Chromium Code Reviews| 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..70d70434d35464bd75d36afecf39330a3e832b32 |
| --- /dev/null |
| +++ b/tools/json_schema_compiler/schema_bundle_generator.py |
| @@ -0,0 +1,144 @@ |
| +# 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 |
| + |
| +SOURCE_BASE_PATH = 'chrome/common/extensions/api' |
|
miket_OOO
2012/03/20 21:52:16
I'd like to parameterize this eventually... would
asargent_no_longer_on_chrome
2012/03/20 23:34:07
Done.
|
| + |
| +class SchemaBundleGenerator(object): |
|
miket_OOO
2012/03/20 21:52:16
Style: doc class purpose, please.
asargent_no_longer_on_chrome
2012/03/20 23:34:07
Done.
|
| + |
| + 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 |
| + |
|
miket_OOO
2012/03/20 21:52:16
Stray CR
asargent_no_longer_on_chrome
2012/03/20 23:34:07
Done.
|
| + |
| + 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 |