gen_js.cpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445
  1. /**
  2. * Tencent is pleased to support the open source community by making Tars available.
  3. *
  4. * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
  5. *
  6. * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
  7. * in compliance with the License. You may obtain a copy of the License at
  8. *
  9. * https://opensource.org/licenses/BSD-3-Clause
  10. *
  11. * Unless required by applicable law or agreed to in writing, software distributed
  12. * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
  13. * CONDITIONS OF ANY KIND, either express or implied. See the License for the
  14. * specific language governing permissions and limitations under the License.
  15. */
  16. #include "code_generator.h"
  17. string CodeGenerator::generateJS(const EnumPtr &pPtr, const string &sNamespace)
  18. {
  19. ostringstream s;
  20. bool bDependent = false;
  21. if (_bEnumReverseMappings)
  22. {
  23. s << TAB << "(function(" << pPtr->getId() << ") {" << endl;
  24. INC_TAB;
  25. int nenum = -1;
  26. vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
  27. for (size_t i = 0; i < member.size(); i++)
  28. {
  29. bDependent |= isDependent(sNamespace, member[i]->getId());
  30. if (member[i]->hasDefault())
  31. {
  32. nenum = TC_Common::strto<int>(member[i]->def());
  33. }
  34. else
  35. {
  36. nenum++;
  37. }
  38. s << TAB << pPtr->getId() << "[" << pPtr->getId() << "[\"" << member[i]->getId() << "\"] = " << TC_Common::tostr(nenum) << "] = "
  39. << "\"" << member[i]->getId() << "\";" << endl;
  40. }
  41. DEL_TAB;
  42. s << TAB << "}(" << sNamespace << "." << pPtr->getId() << " = {}));" << endl;
  43. }
  44. else
  45. {
  46. s << TAB << sNamespace << "." << pPtr->getId() << " = {" << endl;
  47. INC_TAB;
  48. int nenum = -1;
  49. vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
  50. for (size_t i = 0; i < member.size(); i++)
  51. {
  52. bDependent |= isDependent(sNamespace, member[i]->getId());
  53. if (member[i]->hasDefault())
  54. {
  55. nenum = TC_Common::strto<int>(member[i]->def());
  56. }
  57. else
  58. {
  59. nenum++;
  60. }
  61. s << TAB << "\"" << member[i]->getId() << "\" : " << TC_Common::tostr(nenum) << (i < member.size() - 1 ? "," : "") << endl;
  62. }
  63. DEL_TAB;
  64. s << TAB << "};" << endl;
  65. }
  66. s << TAB << sNamespace << "." << pPtr->getId() << "._classname = \"" << sNamespace << "." << pPtr->getId() << "\";" << endl;
  67. s << TAB << sNamespace << "." << pPtr->getId() << "._write = function(os, tag, val) { return os.writeInt32(tag, val); };" << endl;
  68. s << TAB << sNamespace << "." << pPtr->getId() << "._read = function(is, tag, def) { return is.readInt32(tag, true, def); };" << endl;
  69. if (!_bMinimalMembers || _bEntry || bDependent || isDependent(sNamespace, pPtr->getId())) {
  70. return s.str();
  71. } else {
  72. return "";
  73. }
  74. }
  75. string CodeGenerator::generateJS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream)
  76. {
  77. if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getTypeIdPtr()->getId()))
  78. {
  79. return "";
  80. }
  81. ostringstream s;
  82. if (_bStringBinaryEncoding && GET_CONST_GRAMMAR_PTR(pPtr)->t == CONST_GRAMMAR(STRING))
  83. {
  84. bNeedStream = true;
  85. }
  86. s << TAB << sNamespace << "." << pPtr->getTypeIdPtr()->getId() << " = " << getDefault(pPtr->getTypeIdPtr(), GET_CONST_GRAMMAR_PTR(pPtr)->v, sNamespace, false) << ";" << endl;
  87. return s.str();
  88. }
  89. string CodeGenerator::generateJS(const StructPtr &pPtr, const string &sNamespace, bool &bNeedAssert, bool &bQuickFunc)
  90. {
  91. if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getId()))
  92. {
  93. return "";
  94. }
  95. ostringstream s;
  96. vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
  97. s << TAB << sNamespace << "." << pPtr->getId() << " = function() {"<< endl;
  98. INC_TAB;
  99. for (size_t i = 0; i < member.size(); i++)
  100. {
  101. s << TAB << "this." << member[i]->getId() << " = " << getDefault(member[i], member[i]->def(), sNamespace) << ";" << endl;
  102. }
  103. {
  104. s << TAB << "this._classname = \"" << sNamespace << "." << pPtr->getId() << "\";" << endl;
  105. }
  106. DEL_TAB;
  107. s << TAB << "};" << endl;
  108. s << TAB << sNamespace << "." << pPtr->getId() << "._classname = \"" << sNamespace << "." << pPtr->getId() << "\";" << endl;
  109. string sProto = TC_Common::replace(pPtr->getSid(), "::", ".");
  110. s << TAB << sProto << "._write = function (os, tag, value) { os.writeStruct(tag, value); };" << endl;
  111. s << TAB << sProto << "._read = function (is, tag, def) { return is.readStruct(tag, true, def); };" << endl;
  112. //_readFrom
  113. s << TAB << sNamespace << "." << pPtr->getId() << "._readFrom = function (is) {" << endl;
  114. INC_TAB;
  115. s << TAB << "var tmp = new " << sNamespace << "." << pPtr->getId() << ";" << endl;
  116. for (size_t i = 0; i < member.size(); i++)
  117. {
  118. string sFuncName = toFunctionName(member[i], "read");
  119. s << TAB << "tmp." << member[i]->getId() << " = is." << sFuncName << "(" << member[i]->getTag()
  120. << ", " << (member[i]->isRequire()?"true":"false") << ", ";
  121. if (isSimple(member[i]->getTypePtr()))
  122. {
  123. s << getDefault(member[i], member[i]->def(), sNamespace)
  124. << representArgument(member[i]->getTypePtr());
  125. }
  126. else
  127. {
  128. s << getDataType(member[i]->getTypePtr());
  129. }
  130. s << ");" << endl;
  131. }
  132. s << TAB << "return tmp;" << endl;
  133. DEL_TAB;
  134. s << TAB << "};" << endl;
  135. //_writeTo
  136. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._writeTo = function (os) {" << endl;
  137. INC_TAB;
  138. for (size_t i = 0; i < member.size(); i++)
  139. {
  140. string sFuncName = toFunctionName(member[i], "write");
  141. s << TAB << "os." << sFuncName << "(" << member[i]->getTag() << ", this." << member[i]->getId()
  142. << representArgument(member[i]->getTypePtr()) << ");" << endl;
  143. }
  144. DEL_TAB;
  145. s << TAB << "};" << endl;
  146. /*
  147. * Size Optimize:
  148. * Remove <mutil_map> support.
  149. * Remove toBinBuffer, readFromObject, toObject, new, create members.
  150. */
  151. if (_iOptimizeLevel == Os)
  152. {
  153. return s.str();
  154. }
  155. //_equal
  156. vector<string> key = pPtr->getKey();
  157. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._equal = function (" << (key.size() > 0 ? "anItem" : "") << ") {" << endl;
  158. INC_TAB;
  159. if (key.size() > 0)
  160. {
  161. s << TAB << "return ";
  162. for (size_t i = 0; i < key.size(); i++)
  163. {
  164. for (size_t ii =0; ii < member.size(); ii++)
  165. {
  166. if (key[i] != member[ii]->getId())
  167. {
  168. continue;
  169. }
  170. if (isSimple(member[i]->getTypePtr()))
  171. {
  172. s << (i==0?"":TAB + TAB) << "this." << key[i] << " === " << "anItem." << key[i];
  173. }
  174. else
  175. {
  176. s << (i==0?"":TAB + TAB) << "this._equal(" << "anItem)";
  177. }
  178. }
  179. if (i != key.size() - 1)
  180. {
  181. s << " && " << endl;
  182. }
  183. }
  184. s << ";" << endl;
  185. }
  186. else
  187. {
  188. bNeedAssert = true;
  189. s << TAB << "assert.fail(\"this structure not define key operation\");" << endl;
  190. }
  191. DEL_TAB;
  192. s << TAB << "};" << endl;
  193. //_genKey
  194. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._genKey = function () {" << endl;
  195. INC_TAB;
  196. s << TAB << "if (!this._proto_struct_name_) {" << endl;
  197. INC_TAB;
  198. s << TAB << "this._proto_struct_name_ = \"STRUCT\" + Math.random();" << endl;
  199. DEL_TAB;
  200. s << TAB << "}" << endl;
  201. s << TAB << "return this._proto_struct_name_;" << endl;
  202. DEL_TAB;
  203. s << TAB << "};" << endl;
  204. //toObject
  205. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.toObject = function() { "<< endl;
  206. INC_TAB;
  207. s << TAB << "return {" << endl;
  208. for (size_t i = 0; i < member.size(); i++)
  209. {
  210. INC_TAB;
  211. if (i > 0 && i < member.size()) {
  212. s << "," << endl;
  213. }
  214. if (isSimple(member[i]->getTypePtr())) {
  215. s << TAB << "\"" << member[i]->getId() << "\" : this." << member[i]->getId();
  216. }
  217. else {
  218. s << TAB << "\"" << member[i]->getId() << "\" : this." << member[i]->getId() << ".toObject()";
  219. }
  220. DEL_TAB;
  221. }
  222. s << endl;
  223. s << TAB << "};" << endl;
  224. DEL_TAB;
  225. s << TAB << "};" << endl;
  226. //readFromObject
  227. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.readFromObject = function(json) { "<< endl;
  228. INC_TAB;
  229. for (size_t i = 0; i < member.size(); i++)
  230. {
  231. if (isSimple(member[i]->getTypePtr())) {
  232. s << TAB << "_hasOwnProperty.call(json, \"" << member[i]->getId() << "\") && (this." << member[i]->getId() << " = json." << member[i]->getId() << ");" << endl;
  233. } else {
  234. s << TAB << "_hasOwnProperty.call(json, \"" << member[i]->getId() << "\") && (this." << member[i]->getId() << ".readFromObject(json." << member[i]->getId() << "));" << endl;
  235. }
  236. bQuickFunc = true;
  237. }
  238. s << TAB << "return this;" << endl;
  239. DEL_TAB;
  240. s << TAB << "};" << endl;
  241. //toBinBuffer
  242. s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.toBinBuffer = function () {" << endl;
  243. INC_TAB;
  244. s << TAB << "var os = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "OutputStream();" << endl;
  245. s << TAB << "this._writeTo(os);" << endl;
  246. s << TAB << "return os.getBinBuffer();" << endl;
  247. DEL_TAB;
  248. s << TAB << "};" << endl;
  249. //new
  250. s << TAB << sNamespace << "." << pPtr->getId() << ".new = function () {" << endl;
  251. INC_TAB;
  252. s << TAB << "return new " << sNamespace << "." << pPtr->getId() << "();" << endl;
  253. DEL_TAB;
  254. s << TAB << "};" << endl;
  255. //create
  256. s << TAB << sNamespace << "." << pPtr->getId() << ".create = function (is) {" << endl;
  257. INC_TAB;
  258. s << TAB << "return " << sNamespace << "." << pPtr->getId() << "._readFrom(is);" << endl;
  259. DEL_TAB;
  260. s << TAB << "};" << endl;
  261. return s.str();
  262. }
  263. string CodeGenerator::generateJS(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedAssert, bool &bQuickFunc)
  264. {
  265. ostringstream sstr;
  266. vector<StructPtr> ss(pPtr->getAllStructPtr());
  267. for (size_t last = 0; last != ss.size() && ss.size() != 0;)
  268. {
  269. last = ss.size();
  270. for (vector<StructPtr>::iterator iter=ss.begin(); iter!=ss.end();)
  271. {
  272. string str = generateJS(*iter, pPtr->getId(), bNeedAssert, bQuickFunc);
  273. if (!str.empty()) {
  274. sstr << str << endl;
  275. iter = ss.erase(iter);
  276. } else {
  277. iter++;
  278. }
  279. }
  280. }
  281. ostringstream cstr;
  282. vector<ConstPtr> &cs = pPtr->getAllConstPtr();
  283. for (size_t i = 0; i < cs.size(); i++)
  284. {
  285. string str = generateJS(cs[i], pPtr->getId(), bNeedStream);
  286. if (!str.empty()) {
  287. cstr << str << endl;
  288. }
  289. }
  290. ostringstream estr;
  291. vector<EnumPtr> & es = pPtr->getAllEnumPtr();
  292. for (size_t i = 0; i < es.size(); i++)
  293. {
  294. string str = generateJS(es[i], pPtr->getId());
  295. if (!str.empty()) {
  296. estr << str << endl;
  297. }
  298. }
  299. ostringstream str;
  300. if (!estr.str().empty()) str << estr.str();
  301. if (!cstr.str().empty()) str << cstr.str();
  302. if (!sstr.str().empty())
  303. {
  304. bNeedStream = true;
  305. str << sstr.str();
  306. }
  307. return str.str();
  308. }
  309. bool CodeGenerator::generateJS(const ContextPtr &pPtr)
  310. {
  311. vector<NamespacePtr> namespaces = pPtr->getNamespaces();
  312. ostringstream istr;
  313. set<string> setNamespace;
  314. for(size_t i = 0; i < namespaces.size(); i++)
  315. {
  316. if (setNamespace.count(namespaces[i]->getId()) == 0)
  317. {
  318. istr << TAB << "var " << namespaces[i]->getId() << " = " << namespaces[i]->getId() << " || {};" << endl;
  319. istr << TAB << "module.exports." << namespaces[i]->getId() << " = " << namespaces[i]->getId() << ";" << endl << endl;
  320. setNamespace.insert(namespaces[i]->getId());
  321. }
  322. }
  323. // generate encoders and decoders
  324. ostringstream estr;
  325. bool bNeedAssert = false;
  326. bool bNeedStream = false;
  327. bool bQuickFunc = false;
  328. for(size_t i = 0; i < namespaces.size(); i++)
  329. {
  330. estr << generateJS(namespaces[i], bNeedStream, bNeedAssert, bQuickFunc);
  331. }
  332. if (estr.str().empty())
  333. {
  334. return false;
  335. }
  336. // generate module imports
  337. ostringstream ostr;
  338. for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
  339. {
  340. if (it->second.sModule.empty()) continue;
  341. if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
  342. ostr << "var " << it->second.sModule << " = require(\"" << it->second.sFile << "\");" << endl;
  343. }
  344. // concat generated code
  345. ostringstream sstr;
  346. sstr << printHeaderRemark("Structure");
  347. sstr << DISABLE_ESLINT << endl;
  348. sstr << endl;
  349. sstr << "\"use strict\";" << endl << endl;
  350. if (bNeedAssert)
  351. {
  352. sstr << "var assert = require(\"assert\");" << endl;
  353. }
  354. if (bNeedStream)
  355. {
  356. sstr << "var " << IDL_NAMESPACE_STR << "Stream = require(\"" << _sStreamPath << "\");" << endl;
  357. }
  358. if (bQuickFunc)
  359. {
  360. sstr << endl;
  361. sstr << "var _hasOwnProperty = Object.prototype.hasOwnProperty;" << endl;
  362. }
  363. sstr << ostr.str() << endl;
  364. sstr << istr.str();
  365. sstr << estr.str() << endl;
  366. string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + IDL_TYPE + ".js";
  367. TC_File::makeDirRecursive(_sToPath);
  368. makeUTF8File(sFileName, sstr.str());
  369. return true;
  370. }