Index: build/protoc_java.py |
diff --git a/build/protoc_java.py b/build/protoc_java.py |
index 940fd80acf0f7185d757bd91f5314e5deb2c4c95..470667c1f4bff9d7d1b130db2d5435bfda0d26d5 100755 |
--- a/build/protoc_java.py |
+++ b/build/protoc_java.py |
@@ -5,56 +5,64 @@ |
"""Generate java source files from protobuf files. |
-Usage: |
- protoc_java.py {protoc} {proto_path} {java_out} {stamp_file} {proto_files} |
- |
This is a helper file for the genproto_java action in protoc_java.gypi. |
It performs the following steps: |
1. Deletes all old sources (ensures deleted classes are not part of new jars). |
2. Creates source directory. |
-3. Generates Java files using protoc. |
+3. Generates Java files using protoc (output into either --java-out-dir or |
+ --srcjar). |
4. Creates a new stamp file. |
""" |
import os |
+import optparse |
import shutil |
import subprocess |
import sys |
-def main(argv): |
- if len(argv) < 5: |
- usage() |
- return 1 |
- |
- protoc_path, proto_path, java_out, stamp_file = argv[1:5] |
- proto_files = argv[5:] |
+sys.path.append(os.path.join(os.path.dirname(__file__), "android", "gyp")) |
+from util import build_utils |
- # Delete all old sources. |
- if os.path.exists(java_out): |
- shutil.rmtree(java_out) |
- |
- # Create source directory. |
- os.makedirs(java_out) |
+def main(argv): |
+ parser = optparse.OptionParser() |
+ build_utils.AddDepfileOption(parser) |
+ parser.add_option("--protoc", help="Path to protoc binary.") |
+ parser.add_option("--proto-path", help="Path to proto directory.") |
+ parser.add_option("--java-out-dir", |
+ help="Path to output directory for java files.") |
+ parser.add_option("--srcjar", help="Path to output srcjar.") |
+ parser.add_option("--stamp", help="File to touch on success.") |
+ options, args = parser.parse_args(argv) |
- # Specify arguments to the generator. |
- generator_args = ['optional_field_style=reftypes', |
- 'store_unknown_fields=true'] |
- out_arg = '--javanano_out=' + ','.join(generator_args) + ':' + java_out |
+ build_utils.CheckOptions(options, parser, ['protoc', 'proto_path']) |
+ if not options.java_out_dir and not options.srcjar: |
+ print 'One of --java-out-dir or --srcjar must be specified.' |
+ return 1 |
- # Generate Java files using protoc. |
- ret = subprocess.call( |
- [protoc_path, '--proto_path', proto_path, out_arg] + proto_files) |
+ with build_utils.TempDir() as temp_dir: |
+ # Specify arguments to the generator. |
+ generator_args = ['optional_field_style=reftypes', |
+ 'store_unknown_fields=true'] |
+ out_arg = '--javanano_out=' + ','.join(generator_args) + ':' + temp_dir |
+ # Generate Java files using protoc. |
+ build_utils.CheckOutput( |
+ [options.protoc, '--proto_path', options.proto_path, out_arg] |
+ + args) |
- if ret == 0: |
- # Create a new stamp file. |
- with file(stamp_file, 'a'): |
- os.utime(stamp_file, None) |
+ if options.java_out_dir: |
+ build_utils.DeleteDirectory(options.java_out_dir) |
+ shutil.copytree(temp_dir, options.java_out_dir) |
+ else: |
+ build_utils.ZipDir(options.srcjar, temp_dir) |
- return ret |
+ if options.depfile: |
+ build_utils.WriteDepfile( |
+ options.depfile, |
+ args + [options.protoc] + build_utils.GetPythonDependencies()) |
-def usage(): |
- print(__doc__); |
+ if options.stamp: |
+ build_utils.Touch(options.stamp) |
if __name__ == '__main__': |
- sys.exit(main(sys.argv)) |
+ sys.exit(main(sys.argv[1:])) |