| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import os | 5 import os |
| 6 import re | |
| 7 import sys | 6 import sys |
| 8 | 7 |
| 9 import idl_schema | 8 import idl_schema |
| 10 import json_schema | 9 import json_schema |
| 11 from cpp_namespace_environment import CppNamespaceEnvironment | |
| 12 from model import Model, UnixName | |
| 13 | |
| 14 def GenerateFilenames(full_namespace): | |
| 15 # Try to find the file defining the namespace. Eg. for | |
| 16 # nameSpace.sub_name_space.Type' the following heuristics looks for: | |
| 17 # 1. name_space_sub_name_space.json, | |
| 18 # 2. name_space_sub_name_space.idl, | |
| 19 # 3. sub_name_space.json, | |
| 20 # 4. sub_name_space.idl, | |
| 21 # 5. etc. | |
| 22 sub_namespaces = full_namespace.split('.') | |
| 23 filenames = [ ] | |
| 24 basename = None | |
| 25 for namespace in reversed(sub_namespaces): | |
| 26 if basename is not None: | |
| 27 basename = UnixName(namespace + '.' + basename) | |
| 28 else: | |
| 29 basename = UnixName(namespace) | |
| 30 for ext in ['json', 'idl']: | |
| 31 filenames.append('%s.%s' % (basename, ext)) | |
| 32 return filenames | |
| 33 | 10 |
| 34 class SchemaLoader(object): | 11 class SchemaLoader(object): |
| 35 '''Resolves a type name into the namespace the type belongs to. | 12 '''Loads a schema from a provided filename. |
| 36 | 13 |root|: path to the root directory. |
| 37 Properties: | |
| 38 - |root| path to the root directory. | |
| 39 - |path| path to the directory with the API header files, relative to the | |
| 40 root. | |
| 41 - |include_rules| List containing tuples with (path, cpp_namespace_pattern) | |
| 42 used when searching for types. | |
| 43 - |cpp_namespace_pattern| Default namespace pattern | |
| 44 ''' | 14 ''' |
| 45 def __init__(self, | 15 def __init__(self, root): |
| 46 root, | |
| 47 path, | |
| 48 include_rules, | |
| 49 cpp_namespace_pattern): | |
| 50 self._root = root | 16 self._root = root |
| 51 self._include_rules = [(path, cpp_namespace_pattern)] | |
| 52 self._include_rules.extend(include_rules) | |
| 53 | |
| 54 def ResolveNamespace(self, full_namespace): | |
| 55 filenames = GenerateFilenames(full_namespace) | |
| 56 for path, cpp_namespace in self._include_rules: | |
| 57 cpp_namespace_environment = None | |
| 58 if cpp_namespace: | |
| 59 cpp_namespace_environment = CppNamespaceEnvironment(cpp_namespace) | |
| 60 for filename in reversed(filenames): | |
| 61 filepath = os.path.join(path, filename); | |
| 62 if os.path.exists(os.path.join(self._root, filepath)): | |
| 63 return Model().AddNamespace( | |
| 64 self.LoadSchema(filepath)[0], | |
| 65 filepath, | |
| 66 environment=cpp_namespace_environment) | |
| 67 return None | |
| 68 | |
| 69 def ResolveType(self, full_name, default_namespace): | |
| 70 name_parts = full_name.rsplit('.', 1) | |
| 71 if len(name_parts) == 1: | |
| 72 if full_name not in default_namespace.types: | |
| 73 return None | |
| 74 return default_namespace | |
| 75 full_namespace, type_name = full_name.rsplit('.', 1) | |
| 76 namespace = self.ResolveNamespace(full_namespace) | |
| 77 if namespace and type_name in namespace.types: | |
| 78 return namespace | |
| 79 return None | |
| 80 | 17 |
| 81 def LoadSchema(self, schema): | 18 def LoadSchema(self, schema): |
| 82 '''Load a schema definition. The schema parameter must be a file name | 19 '''Load a schema definition. The schema parameter must be a file name |
| 83 with the full path relative to the root.''' | 20 with the full path relative to the root.''' |
| 84 _, schema_extension = os.path.splitext(schema) | 21 _, schema_extension = os.path.splitext(schema) |
| 85 | 22 |
| 86 schema_path = os.path.join(self._root, schema) | 23 schema_path = os.path.join(self._root, schema) |
| 87 if schema_extension == '.json': | 24 if schema_extension == '.json': |
| 88 api_defs = json_schema.Load(schema_path) | 25 api_defs = json_schema.Load(schema_path) |
| 89 elif schema_extension == '.idl': | 26 elif schema_extension == '.idl': |
| 90 api_defs = idl_schema.Load(schema_path) | 27 api_defs = idl_schema.Load(schema_path) |
| 91 else: | 28 else: |
| 92 sys.exit('Did not recognize file extension %s for schema %s' % | 29 sys.exit('Did not recognize file extension %s for schema %s' % |
| 93 (schema_extension, schema)) | 30 (schema_extension, schema)) |
| 94 | 31 |
| 32 # TODO(devlin): This returns a list. Does it need to? Is it ever > 1? |
| 95 return api_defs | 33 return api_defs |
| OLD | NEW |