Plugin support for protobuf compiler (protoc) targets.

tools/protoc_wrapper/protoc_wrapper.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """
 A simple wrapper for protoc.
 
 - Adds includes in generated headers.
 - Handles building with system protobuf as an option.
+- Able to generate C++ target with protoc plugin.
 """
 
 import fnmatch
 import optparse
 import os.path
 import shutil
 import subprocess
 import sys
 import tempfile
 
@@ skipping 73 matching lines @@
                          'along with --protobuf.')
   parser.add_option('--protobuf', dest='generated_header',
                     help='The c++ protobuf header to add the extra header to. '
                          'This must be specified along with --include.')
   parser.add_option('--proto-in-dir',
                     help='The directory containing .proto files.')
   parser.add_option('--proto-in-file', help='Input file to compile.')
   parser.add_option('--use-system-protobuf', type=int, default=0,
                     help='Option to use system-installed protobuf '
                          'instead of bundled one.')
+  parser.add_option('--cc-generator-plugin', default='',
+                    help='Compile C++ target with protoc generator plugin. '
+                         'Expected .pb.h and .pb.cc as output.')
   (options, args) = parser.parse_args(sys.argv)
   if len(args) < 2:
     return 1
 
   if ScanForBadFiles(options.proto_in_dir):
     return 1
 
   proto_path = options.proto_in_dir
   if options.use_system_protobuf == 1:
     proto_path = RewriteProtoFilesForSystemProtobuf(proto_path)
   try:
     # Run what is hopefully protoc.
-    protoc_args = args[1:]
+    protoc = args[1]
+    protoc_args = args[2:]
     protoc_args += ['--proto_path=%s' % proto_path,
                     os.path.join(proto_path, options.proto_in_file)]
-    ret = subprocess.call(protoc_args)
+    if options.cc_generator_plugin != '':
+      args = [x.replace('--cpp_out', '--plugin_out') for x in protoc_args]
+      protoc_args = ['--plugin',
+                     'protoc-gen-plugin=' + options.cc_generator_plugin] + args
+    ret = subprocess.call([protoc] + protoc_args)
     if ret != 0:
       return ret
   finally:
     if options.use_system_protobuf == 1:
       # Remove temporary directory holding re-written files.
       shutil.rmtree(proto_path)
 
   # protoc succeeded, check to see if the generated cpp header needs editing.
   if not options.extra_header or not options.generated_header:
     return 0
   return ModifyHeader(options.generated_header, options.extra_header)
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv))