| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """ | 6 """ |
| 7 A simple wrapper for protoc. | 7 A simple wrapper for protoc. |
| 8 Script for //third_party/protobuf/proto_library.gni . | 8 Script for //third_party/protobuf/proto_library.gni . |
| 9 Features: | 9 Features: |
| 10 - Inserts #include for extra header automatically. | 10 - Inserts #include for extra header automatically. |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 help="Output directory for custom generator plugin.") | 82 help="Output directory for custom generator plugin.") |
| 83 | 83 |
| 84 parser.add_argument("--plugin", | 84 parser.add_argument("--plugin", |
| 85 help="Relative path to custom generator plugin.") | 85 help="Relative path to custom generator plugin.") |
| 86 parser.add_argument("--plugin-options", | 86 parser.add_argument("--plugin-options", |
| 87 help="Custom generator plugin options.") | 87 help="Custom generator plugin options.") |
| 88 parser.add_argument("--cc-options", | 88 parser.add_argument("--cc-options", |
| 89 help="Standard C++ generator options.") | 89 help="Standard C++ generator options.") |
| 90 parser.add_argument("--include", | 90 parser.add_argument("--include", |
| 91 help="Name of include to insert into generated headers.") | 91 help="Name of include to insert into generated headers.") |
| 92 | 92 parser.add_argument("--import-dir", action="append", default=[], |
| 93 help="Extra import directory for protos, can be repeated." |
| 94 ) |
| 93 parser.add_argument("protos", nargs="+", | 95 parser.add_argument("protos", nargs="+", |
| 94 help="Input protobuf definition file(s).") | 96 help="Input protobuf definition file(s).") |
| 95 | 97 |
| 96 options = parser.parse_args() | 98 options = parser.parse_args() |
| 97 | 99 |
| 98 proto_dir = os.path.relpath(options.proto_in_dir) | 100 proto_dir = os.path.relpath(options.proto_in_dir) |
| 99 protoc_cmd = [os.path.realpath(options.protoc)] | 101 protoc_cmd = [os.path.realpath(options.protoc)] |
| 100 | 102 |
| 101 protos = options.protos | 103 protos = options.protos |
| 102 headers = [] | 104 headers = [] |
| (...skipping 11 matching lines...) Expand all Loading... |
| 114 headers.append(os.path.join(cc_out_dir, stripped_name + ".pb.h")) | 116 headers.append(os.path.join(cc_out_dir, stripped_name + ".pb.h")) |
| 115 | 117 |
| 116 if options.plugin_out_dir: | 118 if options.plugin_out_dir: |
| 117 plugin_options = FormatGeneratorOptions(options.plugin_options) | 119 plugin_options = FormatGeneratorOptions(options.plugin_options) |
| 118 protoc_cmd += [ | 120 protoc_cmd += [ |
| 119 "--plugin", "protoc-gen-plugin=" + os.path.relpath(options.plugin), | 121 "--plugin", "protoc-gen-plugin=" + os.path.relpath(options.plugin), |
| 120 "--plugin_out", plugin_options + options.plugin_out_dir | 122 "--plugin_out", plugin_options + options.plugin_out_dir |
| 121 ] | 123 ] |
| 122 | 124 |
| 123 protoc_cmd += ["--proto_path", proto_dir] | 125 protoc_cmd += ["--proto_path", proto_dir] |
| 126 for path in options.import_dir: |
| 127 protoc_cmd += ["--proto_path", path] |
| 128 |
| 124 protoc_cmd += [os.path.join(proto_dir, name) for name in protos] | 129 protoc_cmd += [os.path.join(proto_dir, name) for name in protos] |
| 130 |
| 125 ret = subprocess.call(protoc_cmd) | 131 ret = subprocess.call(protoc_cmd) |
| 126 if ret != 0: | 132 if ret != 0: |
| 127 raise RuntimeError("Protoc has returned non-zero status: " | 133 raise RuntimeError("Protoc has returned non-zero status: " |
| 128 "{0} .".format(ret)) | 134 "{0} .".format(ret)) |
| 129 | 135 |
| 130 if options.include: | 136 if options.include: |
| 131 WriteIncludes(headers, options.include) | 137 WriteIncludes(headers, options.include) |
| 132 | 138 |
| 133 | 139 |
| 134 if __name__ == "__main__": | 140 if __name__ == "__main__": |
| 135 try: | 141 try: |
| 136 main(sys.argv) | 142 main(sys.argv) |
| 137 except RuntimeError as e: | 143 except RuntimeError as e: |
| 138 print(e, file=sys.stderr) | 144 print(e, file=sys.stderr) |
| 139 sys.exit(1) | 145 sys.exit(1) |
| OLD | NEW |