| 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 """Generate java source files from protobuf files. | 6 """Generate java source files from protobufs |
| 7 | 7 |
| 8 Usage: | 8 Usage: |
| 9 protoc_java.py {protoc} {proto_path} {java_out} {proto_runtime} \ | 9 protoc_java.py {protoc} {proto_path} {java_out} {stamp_file} {proto_files} |
| 10 {stamp_file} {proto_files} | |
| 11 | 10 |
| 12 This is a helper file for the genproto_java action in protoc_java.gypi. | 11 This is a helper file for the genproto_java action in protoc_java.gypi. |
| 13 | 12 |
| 14 It performs the following steps: | 13 It performs the following steps: |
| 15 1. Deletes all old sources (ensures deleted classes are not part of new jars). | 14 1. Deletes all old sources (ensures deleted classes are not part of new jars). |
| 16 2. Creates source directory. | 15 2. Creates source directory. |
| 17 3. Generates Java files using protoc. | 16 3. Generates Java files using protoc. |
| 18 4. Creates a new stamp file. | 17 4. Creates a new stamp file. |
| 19 | |
| 20 proto_runtime must be one of 'nano' and 'lite'. | |
| 21 """ | 18 """ |
| 22 | 19 |
| 23 import os | 20 import os |
| 24 import shutil | 21 import shutil |
| 25 import subprocess | 22 import subprocess |
| 26 import sys | 23 import sys |
| 27 | 24 |
| 28 def main(argv): | 25 def main(argv): |
| 29 if len(argv) < 6: | 26 if len(argv) < 5: |
| 30 usage() | 27 usage() |
| 31 return 1 | 28 return 1 |
| 32 | 29 |
| 33 protoc_path, proto_path, java_out, proto_runtime, stamp_file = argv[1:6] | 30 protoc_path, proto_path, java_out, stamp_file = argv[1:5] |
| 34 proto_files = argv[6:] | 31 proto_files = argv[5:] |
| 35 | 32 |
| 36 # Delete all old sources. | 33 # Delete all old sources |
| 37 if os.path.exists(java_out): | 34 if os.path.exists(java_out): |
| 38 shutil.rmtree(java_out) | 35 shutil.rmtree(java_out) |
| 39 | 36 |
| 40 # Create source directory. | 37 # Create source directory |
| 41 os.makedirs(java_out) | 38 os.makedirs(java_out) |
| 42 | 39 |
| 43 # Figure out which runtime to use. | 40 # Generate Java files using protoc |
| 44 if proto_runtime == 'nano': | |
| 45 out_arg = '--javanano_out=optional_field_style=reftypes,' + \ | |
| 46 'store_unknown_fields=true:' + java_out | |
| 47 elif proto_runtime == 'lite': | |
| 48 out_arg = '--java_out=' + java_out | |
| 49 else: | |
| 50 usage() | |
| 51 return 1 | |
| 52 | |
| 53 # Generate Java files using protoc. | |
| 54 ret = subprocess.call( | 41 ret = subprocess.call( |
| 55 [protoc_path, '--proto_path', proto_path, out_arg] + proto_files) | 42 [protoc_path, '--proto_path', proto_path, '--java_out', java_out] |
| 43 + proto_files) |
| 56 | 44 |
| 57 if ret == 0: | 45 if ret == 0: |
| 58 # Create a new stamp file. | 46 # Create a new stamp file |
| 59 with file(stamp_file, 'a'): | 47 with file(stamp_file, 'a'): |
| 60 os.utime(stamp_file, None) | 48 os.utime(stamp_file, None) |
| 61 | 49 |
| 62 return ret | 50 return ret |
| 63 | 51 |
| 64 def usage(): | 52 def usage(): |
| 65 print(__doc__); | 53 print(__doc__); |
| 66 | 54 |
| 67 if __name__ == '__main__': | 55 if __name__ == '__main__': |
| 68 sys.exit(main(sys.argv)) | 56 sys.exit(main(sys.argv)) |
| OLD | NEW |