| Index: tools/json_schema_compiler/schema_bundle_generator.py
|
| diff --git a/tools/json_schema_compiler/schema_bundle_generator.py b/tools/json_schema_compiler/schema_bundle_generator.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..c17a6748f648fd23cfa2ea8d48a605553cb6ba5c
|
| --- /dev/null
|
| +++ b/tools/json_schema_compiler/schema_bundle_generator.py
|
| @@ -0,0 +1,145 @@
|
| +# Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import code
|
| +import cpp_util
|
| +
|
| +import json
|
| +import os
|
| +
|
| +# TODO(miket/asargent) - parameterize this.
|
| +SOURCE_BASE_PATH = 'chrome/common/extensions/api'
|
| +
|
| +class SchemaBundleGenerator(object):
|
| + """This class contains methods to generate code based on multiple schemas.
|
| + """
|
| +
|
| + def __init__(self, model, api_defs, cpp_type_generator):
|
| + self._model = model
|
| + self._api_defs = api_defs
|
| + self._cpp_type_generator = cpp_type_generator
|
| +
|
| + def GenerateHeader(self, file_base, body_code):
|
| + """Generates a code.Code object for a header file
|
| +
|
| + Parameters:
|
| + - |file_base| - the base of the filename, e.g. 'foo' (for 'foo.h')
|
| + - |body_code| - the code to put in between the multiple inclusion guards"""
|
| + c = code.Code()
|
| + c.Append(cpp_util.CHROMIUM_LICENSE)
|
| + c.Append()
|
| + c.Append(cpp_util.GENERATED_BUNDLE_FILE_MESSAGE % SOURCE_BASE_PATH)
|
| + ifndef_name = cpp_util.GenerateIfndefName(SOURCE_BASE_PATH, file_base)
|
| + c.Append()
|
| + c.Append('#ifndef %s' % ifndef_name)
|
| + c.Append('#define %s' % ifndef_name)
|
| + c.Append('#pragma once')
|
| + c.Append()
|
| + c.Concat(body_code)
|
| + c.Append()
|
| + c.Append('#endif // %s' % ifndef_name)
|
| + c.Append()
|
| + return c
|
| +
|
| + def GenerateAPIHeader(self):
|
| + """Generates the header for API registration / declaration"""
|
| + c = code.Code()
|
| +
|
| + c.Append('#include <string>')
|
| + c.Append()
|
| + c.Append('#include "base/basictypes.h"')
|
| +
|
| + for namespace in self._model.namespaces.values():
|
| + namespace_name = namespace.name.replace(
|
| + "experimental.", "")
|
| + c.Append('#include "chrome/browser/extensions/api/%s/%s_api.h"' % (
|
| + namespace_name, namespace_name))
|
| +
|
| + c.Append()
|
| + c.Append("class ExtensionFunctionRegistry;")
|
| + c.Append()
|
| +
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceStart())
|
| + for namespace in self._model.namespaces.values():
|
| + c.Append("// TODO(miket): emit code for %s" % (namespace.unix_name))
|
| + c.Append()
|
| + c.Concat(self.GenerateFunctionRegistry())
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceEnd())
|
| + c.Append()
|
| + return self.GenerateHeader('generated_api', c)
|
| +
|
| + def GenerateFunctionRegistry(self):
|
| + c = code.Code()
|
| + c.Sblock("class GeneratedFunctionRegistry {")
|
| + c.Append("public:")
|
| + c.Sblock("static void RegisterAll(ExtensionFunctionRegistry* registry) {")
|
| + for namespace in self._model.namespaces.values():
|
| + for function in namespace.functions.values():
|
| + namespace_name = namespace.name.replace(
|
| + "experimental.", "").capitalize()
|
| + function_name = namespace_name + function.name.capitalize()
|
| + c.Append("registry->RegisterFunction<%sFunction>();" % (
|
| + function_name))
|
| + c.Eblock("}")
|
| + c.Eblock("};")
|
| + c.Append()
|
| + return c
|
| +
|
| + def GenerateSchemasHeader(self):
|
| + """Generates a code.Code object for the generated schemas .h file"""
|
| + c = code.Code()
|
| + c.Append('namespace base {')
|
| + c.Append('class ListValue;')
|
| + c.Append('}')
|
| + c.Append()
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceStart())
|
| + c.Append()
|
| + c.Sblock("class GeneratedSchemas {")
|
| + c.Append("public:")
|
| + c.Append("static base::ListValue* Get();")
|
| + c.Eblock("};");
|
| + c.Append()
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceEnd())
|
| + c.Append()
|
| + return self.GenerateHeader('generated_schemas', c)
|
| +
|
| + def GenerateSchemasCC(self):
|
| + """Generates a code.Code object for the generated schemas .cc file"""
|
| + c = code.Code()
|
| + c.Append(cpp_util.CHROMIUM_LICENSE)
|
| + c.Append()
|
| + c.Append('#include "%s"' % (os.path.join(SOURCE_BASE_PATH,
|
| + 'generated_schemas.h')))
|
| + c.Append()
|
| + c.Append('#include "base/json/json_reader.h"')
|
| + c.Append('#include "base/values.h"')
|
| + c.Append()
|
| + c.Append('using base::ListValue;')
|
| + c.Append()
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceStart())
|
| + c.Append()
|
| + c.Sblock('ListValue* GeneratedSchemas::Get() {')
|
| + c.Append('ListValue* list = new ListValue();')
|
| + for api in self._api_defs:
|
| + c.Sblock('{')
|
| + c.Sblock('const char tmp[] =')
|
| + json_content = json.dumps(api, indent=2)
|
| + json_content = json_content.replace('"', '\\"')
|
| + lines = json_content.split('\n')
|
| + for index,line in enumerate(lines):
|
| + line = '"%s"' % line
|
| + if index == len(lines) - 1:
|
| + line += ';'
|
| + c.Append(line)
|
| + c.Eblock()
|
| + c.Append('Value* val = base::JSONReader::Read(std::string(tmp), false);')
|
| + c.Append('list->Append(val);')
|
| + c.Eblock('}')
|
| +
|
| + c.Append('return list;')
|
| + c.Eblock('}')
|
| + c.Append()
|
| + c.Concat(self._cpp_type_generator.GetRootNamespaceEnd())
|
| + c.Append()
|
| + return c
|
|
|