idl_gen_js.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741
  1. /*
  2. * Copyright 2014 Google Inc. All rights reserved.
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. // independent from idl_parser, since this code is not needed for most clients
  17. #include "flatbuffers/flatbuffers.h"
  18. #include "flatbuffers/idl.h"
  19. #include "flatbuffers/util.h"
  20. #include "flatbuffers/code_generators.h"
  21. namespace flatbuffers {
  22. static std::string GeneratedFileName(const std::string &path,
  23. const std::string &file_name) {
  24. return path + file_name + "_generated.js";
  25. }
  26. namespace js {
  27. // Iterate through all definitions we haven't generate code for (enums, structs,
  28. // and tables) and output them to a single file.
  29. class JsGenerator : public BaseGenerator {
  30. public:
  31. JsGenerator(const Parser &parser, const std::string &path,
  32. const std::string &file_name)
  33. : BaseGenerator(parser, path, file_name, "", "."){};
  34. // Iterate through all definitions we haven't generate code for (enums,
  35. // structs, and tables) and output them to a single file.
  36. bool generate() {
  37. if (IsEverythingGenerated()) return true;
  38. std::string enum_code, struct_code, exports_code, code;
  39. generateEnums(&enum_code, &exports_code);
  40. generateStructs(&struct_code, &exports_code);
  41. code = code + "// " + FlatBuffersGeneratedWarning();
  42. // Generate code for all the namespace declarations.
  43. GenNamespaces(&code, &exports_code);
  44. // Output the main declaration code from above.
  45. code += enum_code;
  46. code += struct_code;
  47. if (!exports_code.empty() && !parser_.opts.skip_js_exports) {
  48. code += "// Exports for Node.js and RequireJS\n";
  49. code += exports_code;
  50. }
  51. return SaveFile(GeneratedFileName(path_, file_name_).c_str(), code, false);
  52. }
  53. private:
  54. // Generate code for all enums.
  55. void generateEnums(std::string *enum_code_ptr,
  56. std::string *exports_code_ptr) {
  57. for (auto it = parser_.enums_.vec.begin(); it != parser_.enums_.vec.end();
  58. ++it) {
  59. auto &enum_def = **it;
  60. GenEnum(enum_def, enum_code_ptr, exports_code_ptr);
  61. }
  62. }
  63. // Generate code for all structs.
  64. void generateStructs(std::string *decl_code_ptr,
  65. std::string *exports_code_ptr) {
  66. for (auto it = parser_.structs_.vec.begin();
  67. it != parser_.structs_.vec.end(); ++it) {
  68. auto &struct_def = **it;
  69. GenStruct(parser_, struct_def, decl_code_ptr, exports_code_ptr);
  70. }
  71. }
  72. void GenNamespaces(std::string *code_ptr, std::string *exports_ptr) {
  73. std::set<std::string> namespaces;
  74. for (auto it = parser_.namespaces_.begin();
  75. it != parser_.namespaces_.end(); ++it) {
  76. std::string namespace_so_far;
  77. // Gather all parent namespaces for this namespace
  78. for (auto component = (*it)->components.begin();
  79. component != (*it)->components.end(); ++component) {
  80. if (!namespace_so_far.empty()) {
  81. namespace_so_far += '.';
  82. }
  83. namespace_so_far += *component;
  84. namespaces.insert(namespace_so_far);
  85. }
  86. }
  87. // Make sure parent namespaces come before child namespaces
  88. std::vector<std::string> sorted_namespaces(
  89. namespaces.begin(), namespaces.end());
  90. std::sort(sorted_namespaces.begin(), sorted_namespaces.end());
  91. // Emit namespaces in a form that Closure Compiler can optimize
  92. std::string &code = *code_ptr;
  93. std::string &exports = *exports_ptr;
  94. for (auto it = sorted_namespaces.begin();
  95. it != sorted_namespaces.end(); it++) {
  96. code += "/**\n * @const\n * @namespace\n */\n";
  97. if (it->find('.') == std::string::npos) {
  98. code += "var ";
  99. exports += "this." + *it + " = " + *it + ";\n";
  100. }
  101. code += *it + " = " + *it + " || {};\n\n";
  102. }
  103. }
  104. // Generate a documentation comment, if available.
  105. static void GenDocComment(const std::vector<std::string> &dc,
  106. std::string *code_ptr,
  107. const std::string &extra_lines,
  108. const char *indent = nullptr) {
  109. if (dc.empty() && extra_lines.empty()) {
  110. // Don't output empty comment blocks with 0 lines of comment content.
  111. return;
  112. }
  113. std::string &code = *code_ptr;
  114. if (indent) code += indent;
  115. code += "/**\n";
  116. for (auto it = dc.begin(); it != dc.end(); ++it) {
  117. if (indent) code += indent;
  118. code += " *" + *it + "\n";
  119. }
  120. if (!extra_lines.empty()) {
  121. if (!dc.empty()) {
  122. if (indent) code += indent;
  123. code += " *\n";
  124. }
  125. if (indent) code += indent;
  126. std::string::size_type start = 0;
  127. for (;;) {
  128. auto end = extra_lines.find('\n', start);
  129. if (end != std::string::npos) {
  130. code += " * " + extra_lines.substr(start, end - start) + "\n";
  131. start = end + 1;
  132. } else {
  133. code += " * " + extra_lines.substr(start) + "\n";
  134. break;
  135. }
  136. }
  137. }
  138. if (indent) code += indent;
  139. code += " */\n";
  140. }
  141. static void GenDocComment(std::string *code_ptr,
  142. const std::string &extra_lines) {
  143. GenDocComment(std::vector<std::string>(), code_ptr, extra_lines);
  144. }
  145. // Generate an enum declaration and an enum string lookup table.
  146. void GenEnum(EnumDef &enum_def, std::string *code_ptr,
  147. std::string *exports_ptr) {
  148. if (enum_def.generated) return;
  149. std::string &code = *code_ptr;
  150. std::string &exports = *exports_ptr;
  151. GenDocComment(enum_def.doc_comment, code_ptr, "@enum");
  152. if (enum_def.defined_namespace->components.empty()) {
  153. code += "var ";
  154. exports += "this." + enum_def.name + " = " + enum_def.name + ";\n";
  155. }
  156. code += WrapInNameSpace(enum_def) + " = {\n";
  157. for (auto it = enum_def.vals.vec.begin();
  158. it != enum_def.vals.vec.end(); ++it) {
  159. auto &ev = **it;
  160. if (!ev.doc_comment.empty()) {
  161. if (it != enum_def.vals.vec.begin()) {
  162. code += '\n';
  163. }
  164. GenDocComment(ev.doc_comment, code_ptr, "", " ");
  165. }
  166. code += " " + ev.name + ": " + NumToString(ev.value);
  167. code += (it + 1) != enum_def.vals.vec.end() ? ",\n" : "\n";
  168. }
  169. code += "};\n\n";
  170. }
  171. static std::string GenType(const Type &type) {
  172. switch (type.base_type) {
  173. case BASE_TYPE_BOOL:
  174. case BASE_TYPE_CHAR: return "Int8";
  175. case BASE_TYPE_UTYPE:
  176. case BASE_TYPE_UCHAR: return "Uint8";
  177. case BASE_TYPE_SHORT: return "Int16";
  178. case BASE_TYPE_USHORT: return "Uint16";
  179. case BASE_TYPE_INT: return "Int32";
  180. case BASE_TYPE_UINT: return "Uint32";
  181. case BASE_TYPE_LONG: return "Int64";
  182. case BASE_TYPE_ULONG: return "Uint64";
  183. case BASE_TYPE_FLOAT: return "Float32";
  184. case BASE_TYPE_DOUBLE: return "Float64";
  185. case BASE_TYPE_STRING: return "String";
  186. case BASE_TYPE_VECTOR: return GenType(type.VectorType());
  187. case BASE_TYPE_STRUCT: return type.struct_def->name;
  188. default: return "Table";
  189. }
  190. }
  191. std::string GenGetter(const Type &type, const std::string &arguments) {
  192. switch (type.base_type) {
  193. case BASE_TYPE_STRING: return "this.bb.__string" + arguments;
  194. case BASE_TYPE_STRUCT: return "this.bb.__struct" + arguments;
  195. case BASE_TYPE_UNION: return "this.bb.__union" + arguments;
  196. case BASE_TYPE_VECTOR: return GenGetter(type.VectorType(), arguments);
  197. default: {
  198. auto getter = "this.bb.read" + MakeCamel(GenType(type)) + arguments;
  199. if (type.base_type == BASE_TYPE_BOOL) {
  200. getter = "!!" + getter;
  201. }
  202. if (type.enum_def) {
  203. getter = "/** @type {" + WrapInNameSpace(*type.enum_def) + "} */ (" +
  204. getter + ")";
  205. }
  206. return getter;
  207. }
  208. }
  209. }
  210. std::string GenDefaultValue(const Value &value, const std::string &context) {
  211. if (value.type.enum_def) {
  212. if (auto val = value.type.enum_def->ReverseLookup(
  213. atoi(value.constant.c_str()), false)) {
  214. return WrapInNameSpace(*value.type.enum_def) + "." + val->name;
  215. }
  216. }
  217. switch (value.type.base_type) {
  218. case BASE_TYPE_BOOL:
  219. return value.constant == "0" ? "false" : "true";
  220. case BASE_TYPE_STRING:
  221. return "null";
  222. case BASE_TYPE_LONG:
  223. case BASE_TYPE_ULONG: {
  224. int64_t constant = StringToInt(value.constant.c_str());
  225. return context + ".createLong(" + NumToString((int32_t)constant) +
  226. ", " + NumToString((int32_t)(constant >> 32)) + ")";
  227. }
  228. default:
  229. return value.constant;
  230. }
  231. }
  232. std::string GenTypeName(const Type &type, bool input) {
  233. if (!input) {
  234. if (type.base_type == BASE_TYPE_STRING) {
  235. return "string|Uint8Array";
  236. }
  237. if (type.base_type == BASE_TYPE_STRUCT) {
  238. return WrapInNameSpace(*type.struct_def);
  239. }
  240. }
  241. switch (type.base_type) {
  242. case BASE_TYPE_BOOL: return "boolean";
  243. case BASE_TYPE_LONG:
  244. case BASE_TYPE_ULONG: return "flatbuffers.Long";
  245. default:
  246. if (IsScalar(type.base_type)) {
  247. if (type.enum_def) {
  248. return WrapInNameSpace(*type.enum_def);
  249. }
  250. return "number";
  251. }
  252. return "flatbuffers.Offset";
  253. }
  254. }
  255. // Returns the method name for use with add/put calls.
  256. static std::string GenWriteMethod(const Type &type) {
  257. // Forward to signed versions since unsigned versions don't exist
  258. switch (type.base_type) {
  259. case BASE_TYPE_UTYPE:
  260. case BASE_TYPE_UCHAR: return GenWriteMethod(Type(BASE_TYPE_CHAR));
  261. case BASE_TYPE_USHORT: return GenWriteMethod(Type(BASE_TYPE_SHORT));
  262. case BASE_TYPE_UINT: return GenWriteMethod(Type(BASE_TYPE_INT));
  263. case BASE_TYPE_ULONG: return GenWriteMethod(Type(BASE_TYPE_LONG));
  264. default: break;
  265. }
  266. return IsScalar(type.base_type)
  267. ? MakeCamel(GenType(type))
  268. : (IsStruct(type) ? "Struct" : "Offset");
  269. }
  270. template <typename T>
  271. static std::string MaybeAdd(T value) {
  272. return value != 0 ? " + " + NumToString(value) : "";
  273. }
  274. template <typename T>
  275. static std::string MaybeScale(T value) {
  276. return value != 1 ? " * " + NumToString(value) : "";
  277. }
  278. void GenStructArgs(const StructDef &struct_def,
  279. std::string *annotations,
  280. std::string *arguments,
  281. const std::string &nameprefix) {
  282. for (auto it = struct_def.fields.vec.begin();
  283. it != struct_def.fields.vec.end(); ++it) {
  284. auto &field = **it;
  285. if (IsStruct(field.value.type)) {
  286. // Generate arguments for a struct inside a struct. To ensure names
  287. // don't clash, and to make it obvious these arguments are constructing
  288. // a nested struct, prefix the name with the field name.
  289. GenStructArgs(*field.value.type.struct_def, annotations, arguments,
  290. nameprefix + field.name + "_");
  291. } else {
  292. *annotations += "@param {" + GenTypeName(field.value.type, true);
  293. *annotations += "} " + nameprefix + field.name + "\n";
  294. *arguments += ", " + nameprefix + field.name;
  295. }
  296. }
  297. }
  298. static void GenStructBody(const StructDef &struct_def,
  299. std::string *body,
  300. const std::string &nameprefix) {
  301. *body += " builder.prep(";
  302. *body += NumToString(struct_def.minalign) + ", ";
  303. *body += NumToString(struct_def.bytesize) + ");\n";
  304. for (auto it = struct_def.fields.vec.rbegin();
  305. it != struct_def.fields.vec.rend(); ++it) {
  306. auto &field = **it;
  307. if (field.padding) {
  308. *body += " builder.pad(" + NumToString(field.padding) + ");\n";
  309. }
  310. if (IsStruct(field.value.type)) {
  311. // Generate arguments for a struct inside a struct. To ensure names
  312. // don't clash, and to make it obvious these arguments are constructing
  313. // a nested struct, prefix the name with the field name.
  314. GenStructBody(*field.value.type.struct_def, body,
  315. nameprefix + field.name + "_");
  316. } else {
  317. *body += " builder.write" + GenWriteMethod(field.value.type) + "(";
  318. if (field.value.type.base_type == BASE_TYPE_BOOL) {
  319. *body += "+";
  320. }
  321. *body += nameprefix + field.name + ");\n";
  322. }
  323. }
  324. }
  325. // Generate an accessor struct with constructor for a flatbuffers struct.
  326. void GenStruct(const Parser &parser, StructDef &struct_def, std::string *code_ptr, std::string *exports_ptr) {
  327. if (struct_def.generated) return;
  328. std::string &code = *code_ptr;
  329. std::string &exports = *exports_ptr;
  330. // Emit constructor
  331. bool isStatement = struct_def.defined_namespace->components.empty();
  332. std::string object_name = WrapInNameSpace(struct_def);
  333. GenDocComment(struct_def.doc_comment, code_ptr, "@constructor");
  334. if (isStatement) {
  335. exports += "this." + struct_def.name + " = " + struct_def.name + ";\n";
  336. code += "function " + object_name;
  337. } else {
  338. code += object_name + " = function";
  339. }
  340. code += "() {\n";
  341. code += " /**\n";
  342. code += " * @type {flatbuffers.ByteBuffer}\n";
  343. code += " */\n";
  344. code += " this.bb = null;\n";
  345. code += "\n";
  346. code += " /**\n";
  347. code += " * @type {number}\n";
  348. code += " */\n";
  349. code += " this.bb_pos = 0;\n";
  350. code += isStatement ? "}\n\n" : "};\n\n";
  351. // Generate the __init method that sets the field in a pre-existing
  352. // accessor object. This is to allow object reuse.
  353. code += "/**\n";
  354. code += " * @param {number} i\n";
  355. code += " * @param {flatbuffers.ByteBuffer} bb\n";
  356. code += " * @returns {" + object_name + "}\n";
  357. code += " */\n";
  358. code += object_name + ".prototype.__init = function(i, bb) {\n";
  359. code += " this.bb_pos = i;\n";
  360. code += " this.bb = bb;\n";
  361. code += " return this;\n";
  362. code += "};\n\n";
  363. // Generate a special accessor for the table that when used as the root of a
  364. // FlatBuffer
  365. if (!struct_def.fixed) {
  366. GenDocComment(code_ptr,
  367. "@param {flatbuffers.ByteBuffer} bb\n"
  368. "@param {" + object_name + "=} obj\n"
  369. "@returns {" + object_name + "}");
  370. code += object_name + ".getRootAs" + struct_def.name;
  371. code += " = function(bb, obj) {\n";
  372. code += " return (obj || new " + object_name;
  373. code += ").__init(bb.readInt32(bb.position()) + bb.position(), bb);\n";
  374. code += "};\n\n";
  375. // Generate the identifier check method
  376. if (parser_.root_struct_def_ == &struct_def &&
  377. !parser_.file_identifier_.empty()) {
  378. GenDocComment(code_ptr,
  379. "@param {flatbuffers.ByteBuffer} bb\n"
  380. "@returns {boolean}");
  381. code += object_name + ".bufferHasIdentifier = function(bb) {\n";
  382. code += " return bb.__has_identifier('" + parser_.file_identifier_;
  383. code += "');\n};\n\n";
  384. }
  385. }
  386. // Emit field accessors
  387. for (auto it = struct_def.fields.vec.begin();
  388. it != struct_def.fields.vec.end(); ++it) {
  389. auto &field = **it;
  390. if (field.deprecated) continue;
  391. auto offset_prefix = " var offset = this.bb.__offset(this.bb_pos, " +
  392. NumToString(field.value.offset) + ");\n return offset ? ";
  393. // Emit a scalar field
  394. if (IsScalar(field.value.type.base_type) ||
  395. field.value.type.base_type == BASE_TYPE_STRING) {
  396. GenDocComment(field.doc_comment, code_ptr,
  397. std::string(field.value.type.base_type == BASE_TYPE_STRING ?
  398. "@param {flatbuffers.Encoding=} optionalEncoding\n" : "") +
  399. "@returns {" + GenTypeName(field.value.type, false) + "}");
  400. code += object_name + ".prototype." + MakeCamel(field.name, false);
  401. code += " = function(";
  402. if (field.value.type.base_type == BASE_TYPE_STRING) {
  403. code += "optionalEncoding";
  404. }
  405. code += ") {\n";
  406. if (struct_def.fixed) {
  407. code += " return " + GenGetter(field.value.type, "(this.bb_pos" +
  408. MaybeAdd(field.value.offset) + ")") + ";\n";
  409. } else {
  410. std::string index = "this.bb_pos + offset";
  411. if (field.value.type.base_type == BASE_TYPE_STRING) {
  412. index += ", optionalEncoding";
  413. }
  414. code += offset_prefix + GenGetter(field.value.type,
  415. "(" + index + ")") + " : " + GenDefaultValue(field.value, "this.bb");
  416. code += ";\n";
  417. }
  418. }
  419. // Emit an object field
  420. else {
  421. switch (field.value.type.base_type) {
  422. case BASE_TYPE_STRUCT: {
  423. auto type = WrapInNameSpace(*field.value.type.struct_def);
  424. GenDocComment(field.doc_comment, code_ptr,
  425. "@param {" + type + "=} obj\n@returns {" + type + "}");
  426. code += object_name + ".prototype." + MakeCamel(field.name, false);
  427. code += " = function(obj) {\n";
  428. if (struct_def.fixed) {
  429. code += " return (obj || new " + type;
  430. code += ").__init(this.bb_pos";
  431. code += MaybeAdd(field.value.offset) + ", this.bb);\n";
  432. } else {
  433. code += offset_prefix + "(obj || new " + type + ").__init(";
  434. code += field.value.type.struct_def->fixed
  435. ? "this.bb_pos + offset"
  436. : "this.bb.__indirect(this.bb_pos + offset)";
  437. code += ", this.bb) : null;\n";
  438. }
  439. break;
  440. }
  441. case BASE_TYPE_VECTOR: {
  442. auto vectortype = field.value.type.VectorType();
  443. auto vectortypename = GenTypeName(vectortype, false);
  444. auto inline_size = InlineSize(vectortype);
  445. auto index = "this.bb.__vector(this.bb_pos + offset) + index" +
  446. MaybeScale(inline_size);
  447. std::string args = "@param {number} index\n";
  448. if (vectortype.base_type == BASE_TYPE_STRUCT) {
  449. args += "@param {" + vectortypename + "=} obj\n";
  450. } else if (vectortype.base_type == BASE_TYPE_STRING) {
  451. args += "@param {flatbuffers.Encoding=} optionalEncoding\n";
  452. }
  453. GenDocComment(field.doc_comment, code_ptr, args +
  454. "@returns {" + vectortypename + "}");
  455. code += object_name + ".prototype." + MakeCamel(field.name, false);
  456. code += " = function(index";
  457. if (vectortype.base_type == BASE_TYPE_STRUCT) {
  458. code += ", obj";
  459. } else if (vectortype.base_type == BASE_TYPE_STRING) {
  460. code += ", optionalEncoding";
  461. }
  462. code += ") {\n";
  463. if (vectortype.base_type == BASE_TYPE_STRUCT) {
  464. code += offset_prefix + "(obj || new " + vectortypename;
  465. code += ").__init(";
  466. code += vectortype.struct_def->fixed
  467. ? index
  468. : "this.bb.__indirect(" + index + ")";
  469. code += ", this.bb)";
  470. } else {
  471. if (vectortype.base_type == BASE_TYPE_STRING) {
  472. index += ", optionalEncoding";
  473. }
  474. code += offset_prefix + GenGetter(vectortype, "(" + index + ")");
  475. }
  476. code += " : ";
  477. if (field.value.type.element == BASE_TYPE_BOOL) {
  478. code += "false";
  479. } else if (field.value.type.element == BASE_TYPE_LONG ||
  480. field.value.type.element == BASE_TYPE_ULONG) {
  481. code += "this.bb.createLong(0, 0)";
  482. } else if (IsScalar(field.value.type.element)) {
  483. code += "0";
  484. } else {
  485. code += "null";
  486. }
  487. code += ";\n";
  488. break;
  489. }
  490. case BASE_TYPE_UNION:
  491. GenDocComment(field.doc_comment, code_ptr,
  492. "@param {flatbuffers.Table} obj\n"
  493. "@returns {?flatbuffers.Table}");
  494. code += object_name + ".prototype." + MakeCamel(field.name, false);
  495. code += " = function(obj) {\n";
  496. code += offset_prefix + GenGetter(field.value.type,
  497. "(obj, this.bb_pos + offset)") + " : null;\n";
  498. break;
  499. default:
  500. assert(0);
  501. }
  502. }
  503. code += "};\n\n";
  504. // Adds the mutable scalar value to the output
  505. if (IsScalar(field.value.type.base_type) && parser.opts.mutable_buffer) {
  506. std::string annotations = "@param {" + GenTypeName(field.value.type, true) + "} value\n";
  507. GenDocComment(code_ptr, annotations +
  508. "@returns {boolean}");
  509. code += object_name + ".prototype.mutate_" + field.name + " = function(value) {\n";
  510. code += " var offset = this.bb.__offset(this.bb_pos, " + NumToString(field.value.offset) + ");\n\n";
  511. code += " if (offset === 0) {\n";
  512. code += " return false;\n";
  513. code += " }\n\n";
  514. code += " this.bb.write" + MakeCamel(GenType(field.value.type)) + "(this.bb_pos + offset, value);\n";
  515. code += " return true;\n";
  516. code += "};\n\n";
  517. }
  518. // Emit vector helpers
  519. if (field.value.type.base_type == BASE_TYPE_VECTOR) {
  520. // Emit a length helper
  521. GenDocComment(code_ptr, "@returns {number}");
  522. code += object_name + ".prototype." + MakeCamel(field.name, false);
  523. code += "Length = function() {\n" + offset_prefix;
  524. code += "this.bb.__vector_len(this.bb_pos + offset) : 0;\n};\n\n";
  525. // For scalar types, emit a typed array helper
  526. auto vectorType = field.value.type.VectorType();
  527. if (IsScalar(vectorType.base_type) && !IsLong(vectorType.base_type)) {
  528. GenDocComment(code_ptr, "@returns {" + GenType(vectorType) + "Array}");
  529. code += object_name + ".prototype." + MakeCamel(field.name, false);
  530. code += "Array = function() {\n" + offset_prefix;
  531. code += "new " + GenType(vectorType) + "Array(this.bb.bytes().buffer, "
  532. "this.bb.bytes().byteOffset + this.bb.__vector(this.bb_pos + offset), "
  533. "this.bb.__vector_len(this.bb_pos + offset)) : null;\n};\n\n";
  534. }
  535. }
  536. }
  537. // Emit a factory constructor
  538. if (struct_def.fixed) {
  539. std::string annotations = "@param {flatbuffers.Builder} builder\n";
  540. std::string arguments;
  541. GenStructArgs(struct_def, &annotations, &arguments, "");
  542. GenDocComment(code_ptr, annotations +
  543. "@returns {flatbuffers.Offset}");
  544. code += object_name + ".create" + struct_def.name + " = function(builder";
  545. code += arguments + ") {\n";
  546. GenStructBody(struct_def, &code, "");
  547. code += " return builder.offset();\n};\n\n";
  548. } else {
  549. // Generate a method to start building a new object
  550. GenDocComment(code_ptr,
  551. "@param {flatbuffers.Builder} builder");
  552. code += object_name + ".start" + struct_def.name;
  553. code += " = function(builder) {\n";
  554. code += " builder.startObject(" + NumToString(
  555. struct_def.fields.vec.size()) + ");\n";
  556. code += "};\n\n";
  557. // Generate a set of static methods that allow table construction
  558. for (auto it = struct_def.fields.vec.begin();
  559. it != struct_def.fields.vec.end(); ++it) {
  560. auto &field = **it;
  561. if (field.deprecated) continue;
  562. auto argname = MakeCamel(field.name, false);
  563. if (!IsScalar(field.value.type.base_type)) {
  564. argname += "Offset";
  565. }
  566. // Generate the field insertion method
  567. GenDocComment(code_ptr,
  568. "@param {flatbuffers.Builder} builder\n"
  569. "@param {" + GenTypeName(field.value.type, true) + "} " +
  570. argname);
  571. code += object_name + ".add" + MakeCamel(field.name);
  572. code += " = function(builder, " + argname + ") {\n";
  573. code += " builder.addField" + GenWriteMethod(field.value.type) + "(";
  574. code += NumToString(it - struct_def.fields.vec.begin()) + ", ";
  575. if (field.value.type.base_type == BASE_TYPE_BOOL) {
  576. code += "+";
  577. }
  578. code += argname + ", ";
  579. if (!IsScalar(field.value.type.base_type)) {
  580. code += "0";
  581. } else {
  582. if (field.value.type.base_type == BASE_TYPE_BOOL) {
  583. code += "+";
  584. }
  585. code += GenDefaultValue(field.value, "builder");
  586. }
  587. code += ");\n};\n\n";
  588. if (field.value.type.base_type == BASE_TYPE_VECTOR) {
  589. auto vector_type = field.value.type.VectorType();
  590. auto alignment = InlineAlignment(vector_type);
  591. auto elem_size = InlineSize(vector_type);
  592. // Generate a method to create a vector from a JavaScript array
  593. if (!IsStruct(vector_type)) {
  594. GenDocComment(code_ptr,
  595. "@param {flatbuffers.Builder} builder\n"
  596. "@param {Array.<" + GenTypeName(vector_type, true) +
  597. ">} data\n"
  598. "@returns {flatbuffers.Offset}");
  599. code += object_name + ".create" + MakeCamel(field.name);
  600. code += "Vector = function(builder, data) {\n";
  601. code += " builder.startVector(" + NumToString(elem_size);
  602. code += ", data.length, " + NumToString(alignment) + ");\n";
  603. code += " for (var i = data.length - 1; i >= 0; i--) {\n";
  604. code += " builder.add" + GenWriteMethod(vector_type) + "(";
  605. if (vector_type.base_type == BASE_TYPE_BOOL) {
  606. code += "+";
  607. }
  608. code += "data[i]);\n";
  609. code += " }\n";
  610. code += " return builder.endVector();\n";
  611. code += "};\n\n";
  612. }
  613. // Generate a method to start a vector, data to be added manually after
  614. GenDocComment(code_ptr,
  615. "@param {flatbuffers.Builder} builder\n"
  616. "@param {number} numElems");
  617. code += object_name + ".start" + MakeCamel(field.name);
  618. code += "Vector = function(builder, numElems) {\n";
  619. code += " builder.startVector(" + NumToString(elem_size);
  620. code += ", numElems, " + NumToString(alignment) + ");\n";
  621. code += "};\n\n";
  622. }
  623. }
  624. // Generate a method to stop building a new object
  625. GenDocComment(code_ptr,
  626. "@param {flatbuffers.Builder} builder\n"
  627. "@returns {flatbuffers.Offset}");
  628. code += object_name + ".end" + struct_def.name;
  629. code += " = function(builder) {\n";
  630. code += " var offset = builder.endObject();\n";
  631. for (auto it = struct_def.fields.vec.begin();
  632. it != struct_def.fields.vec.end(); ++it) {
  633. auto &field = **it;
  634. if (!field.deprecated && field.required) {
  635. code += " builder.requiredField(offset, ";
  636. code += NumToString(field.value.offset);
  637. code += "); // " + field.name + "\n";
  638. }
  639. }
  640. code += " return offset;\n";
  641. code += "};\n\n";
  642. // Generate the method to complete buffer construction
  643. if (parser_.root_struct_def_ == &struct_def) {
  644. GenDocComment(code_ptr,
  645. "@param {flatbuffers.Builder} builder\n"
  646. "@param {flatbuffers.Offset} offset");
  647. code += object_name + ".finish" + struct_def.name + "Buffer";
  648. code += " = function(builder, offset) {\n";
  649. code += " builder.finish(offset";
  650. if (!parser_.file_identifier_.empty()) {
  651. code += ", '" + parser_.file_identifier_ + "'";
  652. }
  653. code += ");\n";
  654. code += "};\n\n";
  655. }
  656. }
  657. }
  658. };
  659. } // namespace js
  660. bool GenerateJS(const Parser &parser, const std::string &path,
  661. const std::string &file_name) {
  662. js::JsGenerator generator(parser, path, file_name);
  663. return generator.generate();
  664. }
  665. std::string JSMakeRule(const Parser &parser,
  666. const std::string &path,
  667. const std::string &file_name) {
  668. std::string filebase = flatbuffers::StripPath(
  669. flatbuffers::StripExtension(file_name));
  670. std::string make_rule = GeneratedFileName(path, filebase) + ": ";
  671. auto included_files = parser.GetIncludedFilesRecursive(file_name);
  672. for (auto it = included_files.begin();
  673. it != included_files.end(); ++it) {
  674. make_rule += " " + *it;
  675. }
  676. return make_rule;
  677. }
  678. } // namespace flatbuffers