features_compiler.py 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. #!/usr/bin/env python3
  2. # Copyright 2013 The Chromium Authors. All rights reserved.
  3. # Use of this source code is governed by a BSD-style license that can be
  4. # found in the LICENSE file.
  5. """Generator for C++ features from json files.
  6. Usage example:
  7. features_compiler.py --destdir gen --root /home/Work/src _permissions.json
  8. """
  9. from __future__ import print_function
  10. import optparse
  11. import os
  12. from schema_loader import SchemaLoader
  13. from features_cc_generator import CCGenerator
  14. from features_h_generator import HGenerator
  15. from model import CreateFeature
  16. def _GenerateSchema(filename, root, destdir, namespace):
  17. """Generates C++ features files from the json file |filename|.
  18. """
  19. # Load in the feature permissions from the JSON file.
  20. schema = os.path.normpath(filename)
  21. schema_loader = SchemaLoader(os.path.dirname(os.path.relpath(schema, root)),
  22. os.path.dirname(schema),
  23. [],
  24. None)
  25. schema_filename = os.path.splitext(schema)[0]
  26. feature_defs = schema_loader.LoadSchema(schema)
  27. # Generate a list of the features defined and a list of their models.
  28. feature_list = []
  29. for feature_def, feature in feature_defs.items():
  30. feature_list.append(CreateFeature(feature_def, feature))
  31. source_file_dir, _ = os.path.split(schema)
  32. relpath = os.path.relpath(os.path.normpath(source_file_dir), root)
  33. full_path = os.path.join(relpath, schema)
  34. generators = [
  35. ('%s.cc' % schema_filename, CCGenerator()),
  36. ('%s.h' % schema_filename, HGenerator())
  37. ]
  38. # Generate and output the code for all features.
  39. output_code = []
  40. for filename, generator in generators:
  41. code = generator.Generate(feature_list, full_path, namespace).Render()
  42. if destdir:
  43. with open(os.path.join(destdir, relpath, filename), 'w') as f:
  44. f.write(code)
  45. output_code += [filename, '', code, '']
  46. return '\n'.join(output_code)
  47. if __name__ == '__main__':
  48. parser = optparse.OptionParser(
  49. description='Generates a C++ features model from JSON schema',
  50. usage='usage: %prog [option]... schema')
  51. parser.add_option('-r', '--root', default='.',
  52. help='logical include root directory. Path to schema files from '
  53. 'specified dir will be the include path.')
  54. parser.add_option('-d', '--destdir',
  55. help='root directory to output generated files.')
  56. parser.add_option('-n', '--namespace', default='generated_features',
  57. help='C++ namespace for generated files. e.g extensions::api.')
  58. (opts, filenames) = parser.parse_args()
  59. # Only one file is currently specified.
  60. if len(filenames) != 1:
  61. raise ValueError('One (and only one) file is required (for now).')
  62. result = _GenerateSchema(filenames[0], opts.root, opts.destdir,
  63. opts.namespace)
  64. if not opts.destdir:
  65. print(result)