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

from abc import ABC


# TODO(John Sirois): UnionProducts? That seems broken though for ranged version constraints,
# which npm has and are widely used in the community.  For now stay dumb simple (and slow) and
# resolve each node_module individually.
class NodePathsBase(ABC):
    """Maps NpmPackage targets to their resolved NODE_PATH."""

    def __init__(self):
        self._paths_by_target = {}

    def resolved(self, target, node_path):
        """Identifies the given target as resolved to the given path.

        :param target: The target that was resolved to the `node_path` chroot.
        :type target: :class:`pants.contrib.node.targets.npm_package.NpmPackage`
        :param string node_path: The path the given `target` was resolved to.
        """
        self._paths_by_target[target] = node_path

    def node_path(self, target):
        """Returns the path of the resolved root for the given NpmPackage.

        Returns `None` if the target has not been resolved to a chroot.

        :rtype string
        """
        return self._paths_by_target.get(target)

    @property
    def all_node_paths(self):
        """Return all resolved chroots as a list.

        :rtype list string
        """
        return list(self._paths_by_target.values())


class NodePaths(NodePathsBase):
    """Maps Node package targets to their resolved chroot in the pants working dir."""


class NodePathsLocal(NodePathsBase):
    """Maps Node package targets to the directory that the target is defined in.

    This product is intended to be used with the node-install goal which will "install" node targets
    in the directory they are defined rather than in the pants working directory.
    """