void EnumGenerator::generateEnum(io::Printer& p) {
p.Print(
"$i$\n"
"$i$var $name$ = {\n",
"name", t_->name(), "i", indent_);
for (int i = 0; i < t_->value_count(); i++) {
p.Print("$i$ $enum_value_name$: $enum_value$,\n", "enum_value_name",
t_->value(i)->name(), "enum_value",
std::to_string(t_->value(i)->number()), "i", indent_);
}
p.Print(
"$i$\n"
"$i$ toString: function(value) {\n"
"$i$ switch(value) {\n",
"i", indent_);
for (int i = 0; i < t_->value_count(); i++) {
p.Print("$i$ case $enum_value$: return '$enum_value_name$';\n",
"enum_value_name", t_->value(i)->name(), "enum_value",
std::to_string(t_->value(i)->number()), "i", indent_);
}
p.Print(
"$i$ }\n"
"$i$ },\n"
"$i$};\n",
"i", indent_);
}
void MessageGenerator::generateMessagePrototype(io::Printer& p) {
p.Print(
"$i$Protobuf.Message.createMessageType($message_name$, "
"_file.descriptor.messageType($message_index$));\n\n",
"message_name", name_, "message_index", std::to_string(t_->index()), "i",
indent_);
}
void RpcGenerator::generateMethodSignature(const MethodDescriptor *method, io::Printer &printer) const
{
string input_type = RemoveExtern(method->input_type()->full_name());
string output_type = RemoveExtern(method->output_type()->full_name());
printer.Print("$method_name$(", "method_name", method->name());
if (input_type != "Void")
printer.Print("const $input_type$* input", "input_type", input_type);
if (input_type != "Void" && output_type != "Void")
printer.Print(", ");
if (output_type != "Void")
printer.Print("$output_type$* output", "output_type", output_type);
if (input_type != "Void" || output_type != "Void")
printer.Print(", ");
printer.Print("::google::protobuf::Closure* done)");
}
void MessageGenerator::generateMessageConstructor(io::Printer& p) {
// For some reason, oneof_decl_count sometimes reports wrong value in the
// constructor. So we initialize it here as a dirty work around.
if (oneof_generators_.empty() && t_->oneof_decl_count() > 0) {
oneof_generators_.reserve(t_->oneof_decl_count());
for (int i = 0; i < t_->oneof_decl_count(); ++i) {
oneof_generators_.emplace_back(t_->oneof_decl(i));
}
}
p.Print(
"$i$var $message_name$ = (function() {\n"
"$i$ var type = function(values) {\n"
"$i$ this._fields = new Array($field_count$);\n"
"$i$ this._oneofs = new Uint32Array($oneof_count$);\n"
"$i$ this._raw = [this._fields, this._oneofs.buffer];\n",
"message_name", name_, "field_count", std::to_string(t_->field_count()),
"oneof_count", std::to_string(t_->oneof_decl_count()), "i", indent_);
for (auto& g : field_generators_) {
g.indent(indent_ + " ");
g.generateInit(p);
}
for (auto& g : oneof_generators_) {
g.indent(indent_ + " ");
g.generateInit(p);
}
p.Print(
// Sealing provides better coding error detection but degrades performance
// by 100% on benchmarks.
// " Object.seal(this);\n"
"$i$ if (values instanceof $message_name$) {\n"
"$i$ this._mergeFromRawArray(values._raw);\n"
"$i$ } else if (values instanceof Array) {\n"
"$i$ this._mergeFromRawArray(values);\n"
"$i$ } else {\n"
"$i$ for (var k in values) {\n"
"$i$ if (this[k] instanceof Function) {\n"
"$i$ this[k](values[k]);\n"
"$i$ } else {\n"
"$i$ this[k] = values[k];\n"
"$i$ }\n"
"$i$ }\n"
"$i$ }\n"
"$i$ };\n\n",
"message_name", name_, "message_index", std::to_string(t_->index()), "i",
indent_);
for (auto& g : enum_generators_) {
g.generateEnum(p);
g.generateNestedAlias(p);
}
for (auto& g : message_generators_) {
g.generateMessage(p);
g.generateNestedAlias(p);
}
p.Print(
"$i$ Protobuf.Message.createMessageType(type, "
"_file.descriptor.messageType($message_index$));\n\n",
"message_name", name_, "message_index", std::to_string(t_->index()), "i",
indent_);
p.Print(
"$i$ type.prototype._mergeFromRawArray = function(rawArray) {\n"
"$i$ if (rawArray && rawArray instanceof Array) {\n"
"$i$ var oneofs = new Uint32Array(rawArray[1]);\n"
"$i$ var field;\n",
"message_name", name_, "field_count", std::to_string(t_->field_count()),
"oneof_count", std::to_string(t_->oneof_decl_count()), "i", indent_);
for (auto& g : field_generators_) {
if (!g.is_oneof()) {
g.indent(indent_ + " ");
g.generateMerge(p, "rawArray");
}
}
for (auto& g : oneof_generators_) {
g.indent(indent_ + " ");
g.generateMerge(p, "rawArray");
}
p.Print(
"$i$ }\n"
"$i$ };\n\n",
"i", indent_);
for (auto& g : oneof_generators_) {
g.indent(indent_ + " ");
g.generate(p);
}
for (auto& g : field_generators_) {
g.indent(indent_ + " ");
g.generateProperty(p);
}
p.Print(
"$i$ return type;\n"
"$i$})();\n\n",
"i", indent_);
}