123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173 |
- #!/usr/bin/env python3
- # 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.
- '''Tool to determine inputs and outputs of a grit file.
- '''
- from __future__ import print_function
- import optparse
- import os
- import posixpath
- import sys
- from grit import grd_reader
- from grit import util
- class WrongNumberOfArguments(Exception):
- pass
- def Outputs(filename, defines, ids_file, target_platform=None):
- grd = grd_reader.Parse(
- filename, defines=defines, tags_to_ignore=set(['messages']),
- first_ids_file=ids_file, target_platform=target_platform)
- target = []
- lang_folders = {}
- # Add all explicitly-specified output files
- for output in grd.GetOutputFiles():
- path = output.GetFilename()
- target.append(path)
- if path.endswith('.h'):
- path, filename = os.path.split(path)
- if output.attrs['lang']:
- lang_folders[output.attrs['lang']] = os.path.dirname(path)
- return [t.replace('\\', '/') for t in target]
- def GritSourceFiles():
- files = []
- grit_root_dir = os.path.relpath(os.path.dirname(__file__), os.getcwd())
- for root, dirs, filenames in os.walk(grit_root_dir):
- grit_src = [os.path.join(root, f) for f in filenames
- if f.endswith('.py') and not f.endswith('_unittest.py')]
- files.extend(grit_src)
- return sorted(files)
- def Inputs(filename, defines, ids_file, target_platform=None):
- grd = grd_reader.Parse(
- filename, debug=False, defines=defines, tags_to_ignore=set(['message']),
- first_ids_file=ids_file, target_platform=target_platform)
- files = set()
- for lang, ctx, fallback in grd.GetConfigurations():
- # TODO(tdanderson): Refactor all places which perform the action of setting
- # output attributes on the root. See crbug.com/503637.
- grd.SetOutputLanguage(lang or grd.GetSourceLanguage())
- grd.SetOutputContext(ctx)
- grd.SetFallbackToDefaultLayout(fallback)
- for node in grd.ActiveDescendants():
- with node:
- if (node.name == 'structure' or node.name == 'skeleton' or
- (node.name == 'file' and node.parent and
- node.parent.name == 'translations')):
- path = node.GetInputPath()
- if path is not None:
- files.add(grd.ToRealPath(path))
- # If it's a flattened node, grab inlined resources too.
- if node.name == 'structure' and node.attrs['flattenhtml'] == 'true':
- node.RunPreSubstitutionGatherer()
- files.update(node.GetHtmlResourceFilenames())
- elif node.name == 'grit':
- first_ids_file = node.GetFirstIdsFile()
- if first_ids_file:
- files.add(first_ids_file)
- elif node.name == 'include':
- files.add(grd.ToRealPath(node.GetInputPath()))
- # If it's a flattened node, grab inlined resources too.
- if node.attrs['flattenhtml'] == 'true':
- files.update(node.GetHtmlResourceFilenames())
- elif node.name == 'part':
- files.add(util.normpath(os.path.join(os.path.dirname(filename),
- node.GetInputPath())))
- cwd = os.getcwd()
- return [os.path.relpath(f, cwd) for f in sorted(files)]
- def PrintUsage():
- print('USAGE: ./grit_info.py --inputs [-D foo] [-f resource_ids] <grd-file>')
- print(' ./grit_info.py --outputs [-D foo] [-f resource_ids] ' +
- '<out-prefix> <grd-file>')
- def DoMain(argv):
- os.environ['cwd'] = os.getcwd()
- parser = optparse.OptionParser()
- parser.add_option("--inputs", action="store_true", dest="inputs")
- parser.add_option("--outputs", action="store_true", dest="outputs")
- parser.add_option("-D", action="append", dest="defines", default=[])
- # grit build also supports '-E KEY=VALUE', support that to share command
- # line flags.
- parser.add_option("-E", action="append", dest="build_env", default=[])
- parser.add_option("-p", action="store", dest="predetermined_ids_file")
- parser.add_option("-w", action="append", dest="allowlist_files", default=[])
- parser.add_option("-f", dest="ids_file", default="")
- parser.add_option("-t", dest="target_platform", default=None)
- options, args = parser.parse_args(argv)
- defines = {}
- for define in options.defines:
- name, val = util.ParseDefine(define)
- defines[name] = val
- for env_pair in options.build_env:
- (env_name, env_value) = env_pair.split('=', 1)
- os.environ[env_name] = env_value
- if options.inputs:
- if len(args) > 1:
- raise WrongNumberOfArguments("Expected 0 or 1 arguments for --inputs.")
- inputs = []
- if len(args) == 1:
- filename = args[0]
- inputs = Inputs(filename, defines, options.ids_file,
- options.target_platform)
- # Add in the grit source files. If one of these change, we want to re-run
- # grit.
- inputs.extend(GritSourceFiles())
- inputs = [f.replace('\\', '/') for f in inputs]
- if len(args) == 1:
- # Include grd file as second input (works around gyp expecting it).
- inputs.insert(1, args[0])
- if options.allowlist_files:
- inputs.extend(options.allowlist_files)
- return '\n'.join(inputs)
- elif options.outputs:
- if len(args) != 2:
- raise WrongNumberOfArguments(
- "Expected exactly 2 arguments for --outputs.")
- prefix, filename = args
- outputs = [posixpath.join(prefix, f)
- for f in Outputs(filename, defines,
- options.ids_file, options.target_platform)]
- return '\n'.join(outputs)
- else:
- raise WrongNumberOfArguments("Expected --inputs or --outputs.")
- def main(argv):
- try:
- result = DoMain(argv[1:])
- except WrongNumberOfArguments as e:
- PrintUsage()
- print(e)
- return 1
- print(result)
- return 0
- if __name__ == '__main__':
- sys.exit(main(sys.argv))
|