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 |