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

import os

from pants.base.build_environment import get_buildroot
from pants.fs import archive
from pants.util import dirutil

from pants.contrib.node.tasks.node_task import NodeTask


class NodeBundle(NodeTask):
    """Create an archive bundle of NodeModule targets."""

    @classmethod
    def product_types(cls):
        return ["node_bundles", "deployable_archives"]

    @classmethod
    def prepare(cls, options, round_manager):
        super().prepare(options, round_manager)
        round_manager.require_data("bundleable_js")

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._outdir = self.get_options().pants_distdir

    def execute(self):
        bundleable_js = self.context.products.get_data("bundleable_js")
        bundle_archive_product = self.context.products.get("deployable_archives")
        dirutil.safe_mkdir(self._outdir)  # Make sure dist dir is present.

        for target in self.context.target_roots:
            if self.is_node_bundle(target):
                archiver = archive.create_archiver(target.payload.archive)
                for _, abs_paths in bundleable_js[target.node_module].abs_paths():
                    for abs_path in abs_paths:
                        # build_dir is a symlink.  Since dereference option for tar is set to False, we need to
                        # dereference manually to archive the linked build dir.
                        build_dir = os.path.realpath(abs_path)
                        self.context.log.debug(f"archiving {build_dir}")
                        archivepath = archiver.create(
                            build_dir,
                            self._outdir,
                            target.package_name,
                            prefix=None,
                            dereference=False,
                        )
                        bundle_archive_product.add(target, os.path.dirname(archivepath)).append(
                            os.path.basename(archivepath)
                        )
                        self.context.log.info(
                            f"created {os.path.relpath(archivepath, get_buildroot())}"
                        )