Add version flag and version info in generated code

javascript/net/grpc/web/generator/grpc_generator.cc

@@ -18,6 +18,7 @@
 
 #include <google/protobuf/compiler/code_generator.h>
 #include <google/protobuf/compiler/plugin.h>
+#include <google/protobuf/compiler/plugin.pb.h>
 #include <google/protobuf/descriptor.h>
 #include <google/protobuf/descriptor.pb.h>
 #include <google/protobuf/io/printer.h>
@@ -40,6 +41,7 @@ using google::protobuf::compiler::CodeGenerator;
 using google::protobuf::compiler::GeneratorContext;
 using google::protobuf::compiler::ParseGeneratorParameter;
 using google::protobuf::compiler::PluginMain;
+using google::protobuf::compiler::Version;
 using google::protobuf::io::Printer;
 using google::protobuf::io::ZeroCopyOutputStream;
 
@@ -77,6 +79,18 @@ const char* kKeyword[] = {
     "volatile",   "while",        "with",
 };
 
+// Edit the version here prior to release
+static const std::string GRPC_WEB_VERSION = "1.3.1";
+
+string GetProtocVersion(GeneratorContext* context) {
+  Version compiler_version;
+  context->GetCompilerVersion(&compiler_version);
+  return std::to_string(compiler_version.major()) + "." +
+         std::to_string(compiler_version.minor()) + "." +
+         std::to_string(compiler_version.patch()) +
+         compiler_version.suffix();
+}
+
 bool IsReserved(const string& ident) {
   for (size_t i = 0; i < sizeof(kKeyword) / sizeof(kKeyword[0]); i++) {
     if (ident == kKeyword[i]) {
@@ -949,7 +963,11 @@ void PrintFileHeader(Printer* printer, const std::map<string, string>& vars) {
       " * @enhanceable\n"
       " * @public\n"
       " */\n\n"
-      "// GENERATED CODE -- DO NOT EDIT!\n\n\n"
+      "// Code generated by protoc-gen-grpc-web. DO NOT EDIT.\n"
+      "// versions:\n"
+      "// \tprotoc-gen-grpc-web v$version$\n"
+      "// \tprotoc              v$protoc_version$\n"
+      "// source: $source_file$\n\n\n"
       "/* eslint-disable */\n"
       "// @ts-nocheck\n\n\n");
 }
@@ -966,7 +984,11 @@ void PrintMethodDescriptorFile(Printer* printer,
   printer->Print(
       " * @public\n"
       " */\n\n"
-      "// GENERATED CODE -- DO NOT EDIT!\n\n\n"
+      "// Code generated by protoc-gen-grpc-web. DO NOT EDIT.\n"
+      "// versions:\n"
+      "// \tprotoc-gen-grpc-web v$version$\n"
+      "// \tprotoc              v$protoc_version$\n"
+      "// source: $source_file$\n\n\n"
       "/* eslint-disable */\n"
       "// @ts-nocheck\n\n\n");
 
@@ -1361,7 +1383,11 @@ void PrintGrpcWebClosureES6File(Printer* printer, const FileDescriptor* file) {
   string package_dot = file->package().empty() ? "" : file->package() + ".";
 
   printer->Print(
-      "// GENERATED CODE -- DO NOT EDIT!\n"
+      "// Code generated by protoc-gen-grpc-web. DO NOT EDIT.\n"
+      "// versions:\n"
+      "// \tprotoc-gen-grpc-web v$version$\n"
+      "// \tprotoc              v$protoc_version$\n"
+      "// source: $source_file$\n"
       "\n"
       "/**\n"
       " * @fileoverview gRPC-Web generated client stub for '$file$'\n"
@@ -1549,6 +1575,10 @@ class GrpcCodeGenerator : public CodeGenerator {
       return true;
     }
 
+    vars["version"]        = GRPC_WEB_VERSION;
+    vars["protoc_version"] = GetProtocVersion(context);
+    vars["source_file"]    = file->name();
+
     string file_name = generator_options.OutputFile(file->name());
     if (generator_options.multiple_files() &&
         ImportStyle::CLOSURE == generator_options.import_style()) {
@@ -1712,6 +1742,11 @@ class GrpcCodeGenerator : public CodeGenerator {
 }  // namespace grpc
 
 int main(int argc, char* argv[]) {
+  if (argc == 2 && std::string(argv[1]) == "--version") {
+    std::cout << argv[0] << " " << grpc::web::GRPC_WEB_VERSION << std::endl;
+    return 0;
+  }
+
   grpc::web::GrpcCodeGenerator generator;
   PluginMain(argc, argv, &generator);
   return 0;