# Copyright 2017 Pants project contributors (see CONTRIBUTORS.md).
# Licensed under the Apache License, Version 2.0 (see LICENSE).

import os
import shutil

from pants.backend.jvm.tasks.nailgun_task import NailgunTask
from pants.util.dirutil import safe_mkdir


class BundleEntries(NailgunTask):
    @classmethod
    def prepare(cls, options, round_manager):
        super().prepare(options, round_manager)
        round_manager.require_data("kythe_entries_files")

    @classmethod
    def register_options(cls, register):
        super().register_options(register)
        register(
            "--archive",
            type=str,
            choices=["none", "uncompressed", "tar", "zip", "gztar", "bztar"],
            default="none",
            fingerprint=True,
            help="Create an archive of this type.",
        )

    def execute(self):
        archive = self.get_options().archive
        if archive == "none":
            return

        for tgt, entries in self.context.products.get_data("kythe_entries_files", dict).items():
            kythe_distdir = os.path.join(self.get_options().pants_distdir, "kythe")
            safe_mkdir(kythe_distdir)
            uncompressed_kythe_distpath = os.path.join(
                kythe_distdir, "{}.entries".format(tgt.address.path_safe_spec)
            )
            if archive == "uncompressed":
                kythe_distpath = uncompressed_kythe_distpath
                shutil.copy(entries, kythe_distpath)
            else:
                kythe_distpath = shutil.make_archive(
                    base_name=uncompressed_kythe_distpath,
                    format=archive,
                    root_dir=os.path.dirname(entries),
                    base_dir=os.path.basename(entries),
                )
            self.context.log.info("Copied entries to {}".format(kythe_distpath))