Tracing V2: Import support for protobuf plugin.

components/tracing/tools/proto_zero_plugin/proto_zero_generator.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "proto_zero_generator.h"
 
 #include <map>
 #include <memory>
 #include <string>
 
@@ skipping 15 matching lines @@
 using google::protobuf::io::ZeroCopyOutputStream;
 
 using google::protobuf::Split;
 using google::protobuf::StripPrefixString;
 using google::protobuf::StripString;
 using google::protobuf::StripSuffixString;
 using google::protobuf::UpperString;
 
 namespace {
 
+inline std::string ProtoStubName(const FileDescriptor* proto) {
+  return StripSuffixString(proto->name(), ".proto") + ".pbzero";
+}
+
 class GeneratorJob {
  public:
   GeneratorJob(const FileDescriptor *file,
                Printer* stub_h_printer,
                Printer* stub_cc_printer)
     : source_(file),
       stub_h_(stub_h_printer),
       stub_cc_(stub_cc_printer) {}
 
   bool GenerateStubs() {
@@ skipping 98 matching lines @@
     std::string guard = package_ + "_" + source_->name() + "_H_";
     UpperString(&guard);
     StripString(&guard, ".-", '_');
 
     stub_h_->Print(
         "$greeting$\n"
         "#ifndef $guard$\n"
         "#define $guard$\n\n"
         "#include <stddef.h>\n"
         "#include <stdint.h>\n\n"
-        "#include \"components/tracing/core/proto_zero_message.h\"\n\n",
+        "#include \"components/tracing/core/proto_zero_message.h\"\n",
         "greeting", greeting,
         "guard", guard);
     stub_cc_->Print(
         "$greeting$\n"
         "// This file intentionally left blank.\n",
         "greeting", greeting);
 
+    // Print dependencies.
+    if (source_->weak_dependency_count() > 0)
+      Abort("Weak imports are not supported.");
+    for (int i = 0; i < source_->dependency_count(); ++i) {
+      const FileDescriptor* dependency = source_->dependency(i);
+      if (dependency->package() != package_)
+        Abort("Imported proto must be in the same package.");

[Primiano Tucci (use gerrit), 2016/08/03 15:36:28]

out of curiosity, why?

[kraynov, 2016/08/04 12:03:54]

It would be complex to deal with namespaces in this case.

+
+      // Dependency name could contatin slashes but importing from upper-level
+      // directories is not possible anyway since build system process each
+      // proto file individually. Hence proto lookup path always equal to the
+      // directory where particular proto file is located and protoc does not
+      // allow reference to upper directory (aka ..) in import path.
+      //
+      // Laconiacally said:

[Primiano Tucci (use gerrit), 2016/08/03 15:36:28]

nit: s/Laconiacally/Laconically/

+      // - source_->name() may never have slashes,
+      // - dependency->name() may have slashes but always reffers to inner path.
+      stub_h_->Print(
+          "#include \"$name$.h\"\n",
+          "name", ProtoStubName(dependency));
+    }
+    stub_h_->Print("\n");
+
     // Print namespaces.
     for (const std::string& ns : namespaces_)
       stub_h_->Print("namespace $ns$ {\n", "ns", ns);
     stub_h_->Print("\n");
     // Print forward declarations.
     for (const Descriptor* message : messages_) {
       stub_h_->Print(
           "class $class$;\n",
           "class", GetCppClassName(message));
     }
@@ skipping 223 matching lines @@
 }
 
 ProtoZeroGenerator::~ProtoZeroGenerator() {
 }
 
 bool ProtoZeroGenerator::Generate(const FileDescriptor* file,
                                   const std::string& options,
                                   GeneratorContext* context,
                                   std::string* error) const {
 
-  const std::string proto_stubs_name =
-      StripSuffixString(file->name(), ".proto") + ".pbzero";
-
   const std::unique_ptr<ZeroCopyOutputStream> stub_h_file_stream(
-      context->Open(proto_stubs_name + ".h"));
+      context->Open(ProtoStubName(file) + ".h"));
   const std::unique_ptr<ZeroCopyOutputStream> stub_cc_file_stream(
-      context->Open(proto_stubs_name + ".cc"));
+      context->Open(ProtoStubName(file) + ".cc"));
 
   // Variables are delimited by $.
   Printer stub_h_printer(stub_h_file_stream.get(), '$');
   Printer stub_cc_printer(stub_cc_file_stream.get(), '$');
 
   GeneratorJob job(file, &stub_h_printer, &stub_cc_printer);
   if (!job.GenerateStubs()) {
     *error = job.GetFirstError();
     return false;
   }
   return true;
 }
 
 }  // namespace proto
 }  // namespace tracing