Chromium Code Reviews| Index: third_party/closure_compiler/compile_modules.py |
| diff --git a/third_party/closure_compiler/compile_modules.py b/third_party/closure_compiler/compile_modules.py |
| index 66b8e425cacd2a7b2d86b213b4b28fd5edf0e346..873fb72ad11e9c8501c5a733e5cc2f37abc22776 100755 |
| --- a/third_party/closure_compiler/compile_modules.py |
| +++ b/third_party/closure_compiler/compile_modules.py |
| @@ -8,27 +8,49 @@ import ast |
| from checker import Checker as Checker |
| import os |
| + |
| +def get_rel_path_generator(file_path): |
| + current_dir = os.getcwd() |
| + module_dir = os.path.dirname(file_path) |
| + |
| + if current_dir and module_dir: |
| + here_to_module_dir = os.path.relpath(module_dir, current_dir) |
| + if here_to_module_dir: |
| + return lambda f: os.path.join(here_to_module_dir, f) |
| + |
| + return lambda f: f |
| + |
| + |
| class Module(object): |
| - def __init__(self, name, sources, depends=[], externs=[]): |
| + def __init__(self, name, sources, depends=[], externs=[], |
| + expected_output=None): |
| self.name = name |
| self.sources = sources |
| # TODO(dbeam): support depending on other modules/dependency flattening. |
| self.depends = depends |
| self.externs = externs |
| + self.expected_output = expected_output |
| @staticmethod |
| - def from_dict(d): |
| + def from_dict_and_rel_path_generator(d, rel_path_generator): |
| keys = d.keys() |
| required = ["name", "sources"] |
| assert all(r in keys for r in required), "Module missing name or sources" |
| - allowed = required + ["depends", "externs"] |
| + allowed = required + ["depends", "externs", "expected_output"] |
|
Dan Beam
2014/07/29 18:05:10
this should not be in production code, figure out
Vitaly Pavlenko
2014/07/29 18:53:46
Done.
|
| assert all(k in allowed for k in keys), "Module has unknown key" |
| depends = d["depends"] if "depends" in d else [] |
| externs = d["externs"] if "externs" in d else [] |
| - return Module(d["name"], d["sources"], depends=depends, externs=externs) |
| + |
| + sources = map(rel_path_generator, d["sources"]) |
| + depends = map(rel_path_generator, depends) |
| + externs = map(rel_path_generator, externs) |
| + |
| + expected_output = d["expected_output"] if "expected_output" in d else "" |
| + return Module(d["name"], sources, depends=depends, externs=externs, |
| + expected_output=expected_output) |
| # TODO(dbeam): should ModuleParser be internal to ModuleCompiler or should we |
| @@ -41,9 +63,10 @@ class ModuleParser(object): |
| def parse(self, file_path): |
| if file_path in self._cache: |
| - print "(INFO) Found module file %s in the cache" % file_path |
| + print "(INFO) Found module file %s in the cache" % file_path |
| return self._cache[file_path] |
| + |
| file = open(file_path, "r") |
| file_content = file.read() |
| data = ast.literal_eval(file_content) |
| @@ -52,48 +75,45 @@ class ModuleParser(object): |
| if self._verbose: |
| print "(INFO) Layout: " + os.linesep + file_content + os.linesep |
| - self._cache[file_path] = [Module.from_dict(m) for m in data] |
| + self._cache[file_path] = [Module.from_dict_and_rel_path_generator( |
| + m, get_rel_path_generator(file_path)) for m in data] |
| return self._cache[file_path] |
| class ModuleCompiler(object): |
| - _checker = None |
| - _parser = None |
| - def __init__(self, verbose=False): |
| + def __init__(self, verbose=False, test_expected_output=False): |
| self._verbose = verbose |
| + self._parser = ModuleParser(verbose=self._verbose) |
| + self._checker = Checker(verbose=self._verbose, |
| + return_output=test_expected_output) |
| + self._test_expected_output = test_expected_output |
| def _debug(self, msg, prefix="(INFO) ", suffix=""): |
| if self._verbose: |
| print prefix + msg.strip() + suffix |
| - def compile(self, module_file): |
| - self._debug("MODULE FILE: " + module_file, prefix="") |
| + def compile_source(self, source_path, module): |
| + output = self._checker.check(source_path, |
| + depends=module.depends, |
| + externs=module.externs) |
| - # NOTE: It's possible but unlikely that |_checker| or |_parser|'s verbosity |
| - # isn't the same as |self._verbose| due to this class being called with |
| - # verbose=False then verbose=True in the same program. |
| - self._parser = self._parser or ModuleParser(verbose=self._verbose) |
| - self._checker = self._checker or Checker(verbose=self._verbose) |
| + if self._test_expected_output: |
| + return output |
| - current_dir = os.getcwd() |
| - module_dir = os.path.dirname(module_file) |
| - rel_path = lambda f: f |
| + def get_modules(self, module_file): |
| + return self._parser.parse(module_file) |
| - if current_dir and module_dir: |
| - here_to_module_dir = os.path.relpath(module_dir, current_dir) |
| - if here_to_module_dir: |
| - rel_path = lambda f: os.path.join(here_to_module_dir, f) |
| + def compile_module_file(self, module_file): |
| + self._debug("MODULE FILE: " + module_file, prefix="") |
| - modules = self._parser.parse(module_file) |
| + modules = self.get_modules(module_file) |
| for m in modules: |
| self._debug("MODULE: " + m.name, prefix="", suffix=os.linesep) |
| for s in m.sources: |
| - depends = [rel_path(d) for d in m.depends] |
| - externs = [rel_path(e) for e in m.externs] |
| - self._checker.check(rel_path(s), depends=depends, externs=externs) |
| + self.compile_source(s, m) |
| if s != m.sources[-1]: |
| self._debug(os.linesep, prefix="") |
| @@ -105,7 +125,7 @@ class ModuleCompiler(object): |
| def main(opts): |
| module_compiler = ModuleCompiler(verbose=opts.verbose) |
| for module_file in opts.module_file: |
| - module_compiler.compile(module_file) |
| + module_compiler.compile_module_file(module_file) |
| if __name__ == "__main__": |