cpp_bundle_generator.py revision a02191e04bc25c4935f804f2c080ae28663d096d
12a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Copyright (c) 2012 The Chromium Authors. All rights reserved. 22a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be 32a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# found in the LICENSE file. 42a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import code 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import cpp_util 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from model import Platforms 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from schema_util import CapitalizeFirstLetter 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from schema_util import JsFunctionNameToClassName 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import json 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import os 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import re 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)def _RemoveDescriptions(node): 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Returns a copy of |schema| with "description" fields removed. 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if isinstance(node, dict): 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) result = {} 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for key, value in node.items(): 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Some schemas actually have properties called "description", so only 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # remove descriptions that have string values. 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if key == 'description' and isinstance(value, basestring): 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) result[key] = _RemoveDescriptions(value) 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return result 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if isinstance(node, list): 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return [_RemoveDescriptions(v) for v in node] 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return node 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class CppBundleGenerator(object): 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """This class contains methods to generate code based on multiple schemas. 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 37f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) def __init__(self, 38f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) root, 39f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) model, 40f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) api_defs, 41f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) cpp_type_generator, 42f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) cpp_namespace, 43a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) source_file_dir, 44a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) impl_dir): 4558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) self._root = root 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._model = model 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._api_defs = api_defs 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._cpp_type_generator = cpp_type_generator 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._cpp_namespace = cpp_namespace 50f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) self._source_file_dir = source_file_dir 51a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) self._impl_dir = impl_dir 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.api_cc_generator = _APICCGenerator(self) 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.api_h_generator = _APIHGenerator(self) 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.schemas_cc_generator = _SchemasCCGenerator(self) 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.schemas_h_generator = _SchemasHGenerator(self) 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _GenerateHeader(self, file_base, body_code): 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates a code.Code object for a header file 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Parameters: 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) - |file_base| - the base of the filename, e.g. 'foo' (for 'foo.h') 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) - |body_code| - the code to put in between the multiple inclusion guards""" 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append(cpp_util.CHROMIUM_LICENSE) 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 67f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) c.Append(cpp_util.GENERATED_BUNDLE_FILE_MESSAGE % self._source_file_dir) 68f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ifndef_name = cpp_util.GenerateIfndefName(self._source_file_dir, file_base) 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#ifndef %s' % ifndef_name) 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#define %s' % ifndef_name) 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(body_code) 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#endif // %s' % ifndef_name) 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return c 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _GetPlatformIfdefs(self, model_object): 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates the "defined" conditional for an #if check if |model_object| 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) has platform restrictions. Returns None if there are no restrictions. 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """ 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if model_object.platforms is None: 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return None 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ifdefs = [] 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for platform in model_object.platforms: 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if platform == Platforms.CHROMEOS: 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ifdefs.append('defined(OS_CHROMEOS)') 89f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) elif platform == Platforms.LINUX: 90f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ifdefs.append('defined(OS_LINUX)') 91f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) elif platform == Platforms.MAC: 92f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ifdefs.append('defined(OS_MACOSX)') 93f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) elif platform == Platforms.WIN: 94f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ifdefs.append('defined(OS_WIN)') 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise ValueError("Unsupported platform ifdef: %s" % platform.name) 97f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return ' || '.join(ifdefs) 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _GenerateRegisterFunctions(self, namespace_name, function): 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) function_ifdefs = self._GetPlatformIfdefs(function) 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if function_ifdefs is not None: 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#if %s" % function_ifdefs, indent_level=0) 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) function_name = JsFunctionNameToClassName(namespace_name, function.name) 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("registry->RegisterFunction<%sFunction>();" % ( 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) function_name)) 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if function_ifdefs is not None: 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#endif // %s" % function_ifdefs, indent_level=0) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return c 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _GenerateFunctionRegistryRegisterAll(self): 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('// static') 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Sblock('void GeneratedFunctionRegistry::RegisterAll(' 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'ExtensionFunctionRegistry* registry) {') 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for namespace in self._model.namespaces.values(): 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace_ifdefs = self._GetPlatformIfdefs(namespace) 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if namespace_ifdefs is not None: 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#if %s" % namespace_ifdefs, indent_level=0) 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace_name = CapitalizeFirstLetter(namespace.name.replace( 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "experimental.", "")) 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for function in namespace.functions.values(): 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if function.nocompile: 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(self._GenerateRegisterFunctions(namespace.name, function)) 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for type_ in namespace.types.values(): 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for function in type_.functions.values(): 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if function.nocompile: 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace_types_name = JsFunctionNameToClassName( 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace.name, type_.name) 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(self._GenerateRegisterFunctions(namespace_types_name, 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) function)) 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if namespace_ifdefs is not None: 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#endif // %s" % namespace_ifdefs, indent_level=0) 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Eblock("}") 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return c 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 14458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class _APIHGenerator(object): 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates the header for API registration / declaration""" 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self, cpp_bundle): 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._bundle = cpp_bundle 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Generate(self, namespace): 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include <string>') 1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include "base/basictypes.h"') 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("class ExtensionFunctionRegistry;") 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace)) 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('class GeneratedFunctionRegistry {') 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Sblock(' public:') 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('static void RegisterAll(' 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'ExtensionFunctionRegistry* registry);') 16558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) c.Eblock('};') 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace)) 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return self._bundle._GenerateHeader('generated_api', c) 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 17058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class _APICCGenerator(object): 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates a code.Code object for the generated API .cc file""" 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self, cpp_bundle): 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._bundle = cpp_bundle 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Generate(self, namespace): 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append(cpp_util.CHROMIUM_LICENSE) 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 181f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) c.Append('#include "%s"' % (os.path.join(self._bundle._source_file_dir, 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'generated_api.h'))) 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for namespace in self._bundle._model.namespaces.values(): 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace_name = namespace.unix_name.replace("experimental_", "") 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) implementation_header = namespace.compiler_options.get( 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "implemented_in", 188a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) "%s/%s/%s_api.h" % (self._bundle._impl_dir, 189a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) namespace_name, 190a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) namespace_name)) 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.exists( 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os.path.join(self._bundle._root, 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os.path.normpath(implementation_header))): 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if "implemented_in" in namespace.compiler_options: 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise ValueError('Header file for namespace "%s" specified in ' 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'compiler_options not found: %s' % 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (namespace.unix_name, implementation_header)) 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ifdefs = self._bundle._GetPlatformIfdefs(namespace) 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ifdefs is not None: 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#if %s" % ifdefs, indent_level=0) 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include "%s"' % implementation_header) 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ifdefs is not None: 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append("#endif // %s" % ifdefs, indent_level=0) 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include ' 209a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) '"extensions/browser/extension_function_registry.h"') 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace)) 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(self._bundle._GenerateFunctionRegistryRegisterAll()) 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace)) 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return c 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 21958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class _SchemasHGenerator(object): 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates a code.Code object for the generated schemas .h file""" 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self, cpp_bundle): 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._bundle = cpp_bundle 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Generate(self, namespace): 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include <map>') 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('#include <string>') 22958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) c.Append() 230c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) c.Append('#include "base/strings/string_piece.h"') 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace)) 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append('class GeneratedSchemas {') 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Sblock(' public:') 2367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('// Determines if schema named |name| is generated.') 2377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('static bool IsGenerated(std::string name);') 2387d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 2397d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('// Gets the API schema named |name|.') 2407d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('static base::StringPiece Get(const std::string& name);') 24158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) c.Eblock('};') 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace)) 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return self._bundle._GenerateHeader('generated_schemas', c) 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 24658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 2477d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)def _FormatNameAsConstant(name): 2487d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) """Formats a name to be a C++ constant of the form kConstantName""" 2497d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) name = '%s%s' % (name[0].upper(), name[1:]) 2507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return 'k%s' % re.sub('_[a-z]', 2517d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) lambda m: m.group(0)[1].upper(), 2527d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) name.replace('.', '_')) 2537d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 25458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class _SchemasCCGenerator(object): 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Generates a code.Code object for the generated schemas .cc file""" 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self, cpp_bundle): 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._bundle = cpp_bundle 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Generate(self, namespace): 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c = code.Code() 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append(cpp_util.CHROMIUM_LICENSE) 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 265f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) c.Append('#include "%s"' % (os.path.join(self._bundle._source_file_dir, 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'generated_schemas.h'))) 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2687d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('#include "base/lazy_instance.h"') 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 2707d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('namespace {') 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for api in self._bundle._api_defs: 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) namespace = self._bundle._model.namespaces[api.get('namespace')] 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # JSON parsing code expects lists of schemas, so dump a singleton list. 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) json_content = json.dumps([_RemoveDescriptions(api)], 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) separators=(',', ':')) 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Escape all double-quotes and backslashes. For this to output a valid 277a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch # JSON C string, we need to escape \ and ". Note that some schemas are 278a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch # too large to compile on windows. Split the JSON up into several 279a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch # strings, since apparently that helps. 280a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch max_length = 8192 281a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch segments = [json_content[i:i + max_length].replace('\\', '\\\\') 282a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch .replace('"', '\\"') 283a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch for i in xrange(0, len(json_content), max_length)] 2847d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('const char %s[] = "%s";' % 285a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch (_FormatNameAsConstant(namespace.name), '" "'.join(segments))) 2867d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('}') 2877d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace)) 2887d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 2897d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Sblock('struct Static {') 2907d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Sblock('Static() {') 2917d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) for api in self._bundle._api_defs: 2927d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) namespace = self._bundle._model.namespaces[api.get('namespace')] 2937d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('schemas["%s"] = %s;' % (namespace.name, 2947d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) _FormatNameAsConstant(namespace.name))) 29558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) c.Eblock('}') 2967d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 2977d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('std::map<std::string, const char*> schemas;') 29858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) c.Eblock('};') 2997d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 3007d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('base::LazyInstance<Static> g_lazy_instance;') 3017d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 3027d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('// static') 3037d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Sblock('base::StringPiece GeneratedSchemas::Get(' 3047d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 'const std::string& name) {') 3057d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('return IsGenerated(name) ? ' 3067d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 'g_lazy_instance.Get().schemas[name] : "";') 3077d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Eblock('}') 3087d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append() 3097d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('// static') 3107d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Sblock('bool GeneratedSchemas::IsGenerated(std::string name) {') 3117d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) c.Append('return g_lazy_instance.Get().schemas.count(name) > 0;') 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Eblock('}') 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace)) 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) c.Append() 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return c 317