| #!/usr/bin/env python3 |
| # |
| # Copyright 2018 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import argparse |
| import shutil |
| import sys |
| |
| from util import build_utils |
| import action_helpers # build_utils adds //build to sys.path. |
| import zip_helpers |
| |
| |
| def CreatePathTransform(exclude_globs, include_globs): |
| """Returns a function to strip paths for the given patterns. |
| |
| Args: |
| exclude_globs: List of globs that if matched should be excluded. |
| include_globs: List of globs that if not matched should be excluded. |
| |
| Returns: |
| * None if no filters are needed. |
| * A function "(path) -> path" that returns None when |path| should be |
| stripped, or |path| otherwise. |
| """ |
| if not (exclude_globs or include_globs): |
| return None |
| exclude_globs = list(exclude_globs or []) |
| def path_transform(path): |
| # Exclude filters take precidence over include filters. |
| if build_utils.MatchesGlob(path, exclude_globs): |
| return None |
| if include_globs and not build_utils.MatchesGlob(path, include_globs): |
| return None |
| return path |
| |
| return path_transform |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--input', required=True, |
| help='Input zip file.') |
| parser.add_argument('--output', required=True, |
| help='Output zip file') |
| parser.add_argument('--exclude-globs', |
| help='GN list of exclude globs') |
| parser.add_argument('--include-globs', |
| help='GN list of include globs') |
| argv = build_utils.ExpandFileArgs(sys.argv[1:]) |
| args = parser.parse_args(argv) |
| |
| args.exclude_globs = action_helpers.parse_gn_list(args.exclude_globs) |
| args.include_globs = action_helpers.parse_gn_list(args.include_globs) |
| |
| path_transform = CreatePathTransform(args.exclude_globs, args.include_globs) |
| with action_helpers.atomic_output(args.output) as f: |
| if path_transform: |
| zip_helpers.merge_zips(f.name, [args.input], |
| path_transform=path_transform) |
| else: |
| shutil.copy(args.input, f.name) |
| |
| |
| if __name__ == '__main__': |
| main() |