#!/usr/bin/env python3
#
# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Archives a set of files."""

import argparse
import json
import sys
import zipfile

from util import build_utils
import action_helpers  # build_utils adds //build to sys.path.
import zip_helpers


def main(args):
    args = build_utils.ExpandFileArgs(args)
    parser = argparse.ArgumentParser(args)
    parser.add_argument('--input-files', help='GN-list of files to zip.')
    parser.add_argument(
        '--input-files-base-dir',
        help='Paths in the archive will be relative to this directory',
    )
    parser.add_argument('--input-zips', help='GN-list of zips to merge.')
    parser.add_argument(
        '--input-zips-excluded-globs',
        help='GN-list of globs for paths to exclude.',
    )
    parser.add_argument(
        '--output', required=True, help='Path to output archive.'
    )
    compress_group = parser.add_mutually_exclusive_group()
    compress_group.add_argument(
        '--compress', action='store_true', help='Compress entries'
    )
    compress_group.add_argument(
        '--no-compress',
        action='store_false',
        dest='compress',
        help='Do not compress entries',
    )
    parser.add_argument(
        '--comment-json',
        action='append',
        metavar='KEY=VALUE',
        type=lambda x: x.split('=', 1),
        help='Entry to store in JSON-encoded archive comment.',
    )
    action_helpers.add_depfile_arg(parser)
    options = parser.parse_args(args)

    with action_helpers.atomic_output(options.output) as f:
        with zipfile.ZipFile(f.name, 'w') as out_zip:
            depfile_deps = None
            if options.input_files:
                files = action_helpers.parse_gn_list(options.input_files)
                zip_helpers.add_files_to_zip(
                    files,
                    out_zip,
                    base_dir=options.input_files_base_dir,
                    compress=options.compress,
                )

            if options.input_zips:
                files = action_helpers.parse_gn_list(options.input_zips)
                depfile_deps = files
                path_transform = None
                if options.input_zips_excluded_globs:
                    globs = action_helpers.parse_gn_list(
                        options.input_zips_excluded_globs
                    )
                    path_transform = lambda p: (
                        None if build_utils.MatchesGlob(p, globs) else p
                    )
                zip_helpers.merge_zips(
                    out_zip,
                    files,
                    path_transform=path_transform,
                    compress=options.compress,
                )

            if options.comment_json:
                out_zip.comment = json.dumps(
                    dict(options.comment_json), sort_keys=True
                ).encode('utf-8')

    # Depfile used only by dist_jar().
    if options.depfile:
        action_helpers.write_depfile(
            options.depfile, options.output, inputs=depfile_deps
        )


if __name__ == '__main__':
    main(sys.argv[1:])