#!/usr/bin/python

#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

#

# Copyright (c) 2009, 2016, Oracle and/or its affiliates. All rights reserved.

#

import copy

import itertools

import operator

import os

import re

import six

from collections import namedtuple

from six.moves.urllib.parse import unquote

import pkg.actions as actions

import pkg.client.api as api

import pkg.client.api_errors as apx

import pkg.flavor.base as base

import pkg.flavor.elf as elf_dep

import pkg.flavor.hardlink as hardlink

import pkg.flavor.script as script

import pkg.flavor.smf_manifest as smf_manifest

import pkg.fmri as fmri

import pkg.manifest as manifest

import pkg.misc as misc

import pkg.portable as portable

import pkg.variant as variants

from pkg.actions.depend import known_types as dep_types

paths_prefix = "{0}.path".format(base.Dependency.DEPEND_DEBUG_PREFIX)

files_prefix = "{0}.file".format(base.Dependency.DEPEND_DEBUG_PREFIX)

reason_prefix = "{0}.reason".format(base.Dependency.DEPEND_DEBUG_PREFIX)

type_prefix = "{0}.type".format(base.Dependency.DEPEND_DEBUG_PREFIX)

target_prefix = "{0}.target".format(base.Dependency.DEPEND_DEBUG_PREFIX)

bypassed_prefix = "{0}.bypassed".format(base.Dependency.DEPEND_DEBUG_PREFIX)

via_links_prefix = "{0}.via-links".format(base.Dependency.DEPEND_DEBUG_PREFIX)

path_id_prefix = "{0}.path-id".format(base.Dependency.DEPEND_DEBUG_PREFIX)

# A tag used to hold the product of the paths_prefix and files_prefix

# contents, used when bypassing dependency generation

fullpaths_prefix = "{0}.fullpath".format(base.Dependency.DEPEND_DEBUG_PREFIX)

Entries = namedtuple("Entries", ["delivered", "installed"])

# This namedtuple is used to hold two items. The first, delivered, is used to

# hold items which are part of packages being delivered.  The second, installed,

# is used to hold items which are part of packages installed on the system.

LinkInfo = namedtuple("LinkInfo", ["path", "pfmri", "nearest_pfmri",

    "variant_combination", "via_links"])

# This namedtuple is used to hold the information needed when resolving links.

# The 'path' is the path of the file that ultimately resolved the link.  The

# 'pfmri' is the package fmri that delivered 'path.'  The 'nearest_pfmri' is the

# package that delivered the next element in the link chain.  The

# 'variant-combination' is the combination of variants under which the link is

# satisfied.  'via-links' contains the links used to reach 'path.'

class DependencyError(Exception):

        """The parent class for all dependency exceptions."""

        pass

class DropPackageWarning(DependencyError):

        """This exception is used when a package is dropped as it cannot

        satisfy all dependencies."""

        def __init__(self, pkgs, dep):

                self.pkgs = pkgs

                self.dep = dep

        def __str__(self):

                pkg_str = ", ".join(f.get_pkg_stem() for f in self.pkgs)

                return _("WARNING: package {0} was ignored as it cannot "

                    "satisfy all dependencies:\n{1}\n").format(pkg_str,

                    prune_debug_attrs(self.dep))

class UnresolvedDependencyError(DependencyError):

        """This exception is used when no package delivers a file which is

        depended upon."""

        def __init__(self, pth, file_dep, pvars):

                self.path = pth

                self.file_dep = file_dep

                self.pvars = pvars

        def __str__(self):

                dep_str = "\n" + self.file_dep.pretty_print()

                if self.pvars.not_sat_set:

                        return _("{pth} has unresolved dependency '{dep}' "

                            "under the following combinations of "

                            "variants:\n{combo}").format(

                                pth=self.path,

                                dep=dep_str + "\n",

                                combo="\n".join([

                                    " ".join([

                                        ("{0}:{1}".format(name, val))

                                        for name, val in sorted(grp)

                                    ])

                                    for grp in self.pvars.not_sat_set

                            ]))

                else:

                        return _("{pth} has unresolved dependency "

                            "'{dep}'.").format(

                            pth=self.path, dep=dep_str)

        """This exception is used when an action is tagged with a variant or

        variant value which the package is not tagged with."""

        def __init__(self, act_vars, pkg_vars, pth):

                self.act_vars = act_vars

                self.pkg_vars = pkg_vars

                self.path = pth

        def __str__(self):

                    "The action's variants are: {act}\nThe package's "

                    "variants are: {pkg}").format(

                        path=self.path,

                        act=self.act_vars,

                        pkg=self.pkg_vars

                   )

class BadPackageFmri(DependencyError):

        """This exception is used when a manifest's fmri isn't a valid fmri."""

        def __init__(self, path, e):

                self.path = path

                self.exc = e

        def __str__(self):

                return _("The manifest '{path}' has an invalid package "

                    "FMRI:\n\t{exc}").format(path=self.path, exc=self.exc)

class ExtraVariantedDependency(DependencyError):

        """This exception is used when one or more dependency actions have a

        variant set on them which is not in the package's set of variants."""

        def __init__(self, pkg, reason_variants, manual_dep):

                self.pkg = pkg

                assert len(reason_variants) > 0

                self.rvs = reason_variants

                self.manual = manual_dep

        def __str__(self):

                s = ""

                for r, diff in sorted(six.iteritems(self.rvs)):

                        for kind in diff.type_diffs:

                                s += _("\t{r:15} Variant '{kind}' is not "

                                    "declared.\n").format(

                                    r=r, kind=kind)

                        for k, v in diff.value_diffs:

                                s += _("\t{r:15} Variant '{kind}' is not "

                                    "declared to have value '{val}'.\n").format(

                                    r=r, val=v, kind=k)

                if not self.manual:

                        return _("The package '{pkg}' contains actions with "

                            "the\npaths seen below which have variants set on "

                            "them which are not set on the\npackage.  These "

                            "variants must be set on the package or removed "

                            "from the actions.\n\n{rvs}").format(

                            pkg=self.pkg,

                            rvs=s

                       )

                else:

                        return _("The package '{pkg}' contains manually "

                            "specified dependencies\nwhich have variants set "

                            "on them which are not set on the package.  "

                            "These\nvariants must be set on the package or "

                            "removed from the actions.\n\n{rvs}").format(

                            pkg=self.pkg,

                            rvs=s

                       )

class NeedConditionalRequireAny(DependencyError):

        """This exception is used when pkgdepend would need a dependency which

        was both require-any and conditional to properly represent the

        dependency."""

        def __init__(self, conditionals, pkg_vars):

                self.conditionals = conditionals

                self.pkg_vct = pkg_vars

        def __str__(self):

                s = _("""\

pkgdepend has inferred conditional dependencies with different targets but

which share a predicate.  pkg(7) can not represent these dependencies.  This

issue can be resolved by changing the packaging of the links which generated the

conditional dependencies so that they have different predicates or share the

same FMRI.  Each pair of problematic conditional dependencies follows:

""")

                for i, (d1, d2, v) in enumerate(self.conditionals):

                        i += 1

                        v.simplify(self.pkg_vct)

                        if v.is_empty() or not v.sat_set:

                                s += _("Pair {0}\n").format(i)

                        else:

                                s += _("Pair {0} is only problematic in the "

                                    "listed variant combinations:").format(i)

                                s += "\n\t\t" + "\n\t\t".join([

                                    " ".join([

                                        ("{0}:{1}".format(name, val))

                                        for name, val in sorted(grp)

                                    ]) for grp in v.sat_set

                                ])

                                s += "\n"

                        s += d1.pretty_print() + "\n"

                        s += d2.pretty_print() + "\n"

                return s

class __NotSubset(DependencyError):

        def __init__(self, diff):

                self.diff = variants.VCTDifference(tuple(diff.type_diffs),

                    tuple(diff.value_diffs))

def list_implicit_deps(file_path, proto_dirs, dyn_tok_conv, run_paths,

    remove_internal_deps=True, convert=True, ignore_bypass=False):

        """Given the manifest provided in file_path, use the known dependency

        generators to produce a list of dependencies the files delivered by

        the manifest have.

        'file_path' is the path to the manifest for the package.

        'proto_dirs' is a list of paths to proto areas which hold the files that

        will be delivered by the package.

        'dyn_tok_conv' is the dictionary which maps the dynamic tokens, like

        $PLATFORM, to the values they should be expanded to.

        'run_paths' contains the run paths that are used to find modules.

        'convert' determines whether PublishingDependencies will be transformed

        to DependencyActions prior to being returned.  This is primarily an

        option to facilitate testing and debugging.

        'ignore_bypass' determines whether to bypass generation of dependencies

        against certain files or directories.  This is primarily an option to

        facilitate testing and debugging.

        """

        m, manifest_errs = __make_manifest(file_path, proto_dirs)

        pkg_vars = m.get_all_variants()

        rid_errs = []

        if remove_internal_deps:

                deps, rid_errs = resolve_internal_deps(deps, m, proto_dirs,

                    pkg_vars)

        if convert:

                deps = convert_to_standard_dep_actions(deps)

def convert_to_standard_dep_actions(deps):

        """Convert pkg.base.Dependency objects to

        pkg.actions.dependency.Dependency objects."""

        def norm_attrs(a):

                """Normalize attribute values as lists instead of sets; only

                lists are permitted for multi-value action attributes."""

                for k, v in a.items():

                        if isinstance(v, set):

                                a[k] = list(v)

        res = []

        for d in deps:

                tmp = []

                for c in d.dep_vars.not_sat_set:

                        attrs = d.attrs.copy()

                        attrs.update(c)

                        norm_attrs(attrs)

                        tmp.append(actions.depend.DependencyAction(**attrs))

                if not tmp:

                        attrs = d.attrs.copy()

                        norm_attrs(attrs)

                        tmp.append(actions.depend.DependencyAction(**attrs))

                res.extend(tmp)

        return res

def resolve_internal_deps(deps, mfst, proto_dirs, pkg_vars):

        """Given a list of dependencies, remove those which are satisfied by

        others delivered by the same package.

        'deps' is a list of Dependency objects.

        'mfst' is the Manifest of the package that delivered the dependencies

        found in deps.

        'proto_dir' is the path to the proto area which holds the files that

        will be delivered by the package.

        'pkg_vars' are the variants that this package was published against."""

        res = []

        errs = []

        delivered = {}

        delivered_bn = {}

        files = Entries({}, {})

        links = {}

        # A fake pkg name is used because there is no requirement that a package

        # name itself to generate its dependencies.  Also, the name is entirely

        # a private construction which should not escape to the user.

        add_fmri_path_mapping(files.delivered, links,

            fmri.PkgFmri("INTERNAL@0-0", "0"), mfst)

        for a in mfst.gen_actions_by_type("file"):

                pvars = a.get_variant_template()

                if not pvars:

                        pvars = pkg_vars

                pvc = variants.VariantCombinations(pvars, satisfied=True)

                p = a.attrs["path"]

                bn = os.path.basename(p)

                delivered_bn.setdefault(bn, copy.copy(pvc))

        for d in deps:

                etype, pvars = d.resolve_internal(

                    delivered_files=files.delivered,

                    delivered_base_names=delivered_bn, links=links,

                    resolve_links=resolve_links)

                if etype is None:

                        continue

                pvars.simplify(pkg_vars)

                d.dep_vars = pvars

                res.append(d)

        return res, errs

def no_such_file(action, **kwargs):

        """Function to handle dispatch of files not found on the system."""

        return [], [base.MissingFile(action.attrs["path"])], {}

# Dictionary which maps codes from portable.get_file_type to the functions which

# find dependencies for those types of files.

dispatch_dict = {

    portable.ELF: elf_dep.process_elf_dependencies,

    portable.EXEC: script.process_script_deps,

    portable.SMF_MANIFEST: smf_manifest.process_smf_manifest_deps,

    portable.UNFOUND: no_such_file

}

def list_implicit_deps_for_manifest(mfst, proto_dirs, pkg_vars, dyn_tok_conv,

    run_paths, ignore_bypass=False):

        """For a manifest, produce the list of dependencies generated by the

        files it installs.

        'mfst' is the Manifest of the package that delivered the dependencies

        found in deps.

        'proto_dirs' are the paths to the proto areas which hold the files that

        will be delivered by the package.

        'pkg_vars' are the variants that this package was published against.

        'dyn_tok_conv' is the dictionary which maps the dynamic tokens, like

        $PLATFORM, to the values they should be expanded to.

        'run_paths' contains the run paths used to find modules, this is

        overridden by any manifest or action attributes setting

        'pkg.depend.runpath' (portable.PD_RUN_PATH)

        'ignore_bypass' set to True will prevent us from looking up any

        pkg.depend.bypass-generate attributes - this is primarily to aid

        debugging and testing.

        'deps' is a list of dependencies found for the given Manifest.

        'elist' is a list of errors encountered while finding dependencies.

        'missing' is a dictionary mapping a file type that isn't recognized by

        portable.get_file_type to a file which produced that filetype.

        'pkg_attrs' is a dictionary containing metadata that was gathered

        during dependency analysis. Typically these would get turned into

        AttributeActions for that package. """

        deps = []

        elist = []

        missing = {}

        pkg_attrs = {}

        act_list = list(mfst.gen_actions_by_type("file"))

        file_types = portable.get_file_type(act_list)

        if portable.PD_RUN_PATH in mfst:

                # check for multiple values in a set attribute

                run_path_str = mfst[portable.PD_RUN_PATH]

                es = __verify_run_path(run_path_str)

                if es:

                run_paths = run_path_str.split(":")

        mf_bypass = []

        if portable.PD_BYPASS_GENERATE in mfst:

                mf_bypass = __makelist(mfst[portable.PD_BYPASS_GENERATE])

        for i, file_type in enumerate(file_types):

                a = act_list[i]

                a_run_paths = run_paths

                if portable.PD_RUN_PATH in a.attrs:

                        a_run_path_str = a.attrs[portable.PD_RUN_PATH]

                        es = __verify_run_path(a_run_path_str)

                        if es:

                        a_run_paths = a_run_path_str.split(":")

                bypass = __makelist(

                    a.attrs.get(portable.PD_BYPASS_GENERATE, mf_bypass))

                # If we're bypassing all depdendency generation, we can avoid

                # calling our dispatch_dict function altogether.

                if (".*" in bypass or "^.*$" in bypass) and not ignore_bypass:

                        pkg_attrs[bypassed_prefix] = "{0}:.*".format(

                            a.attrs["path"])

                        continue

                try:

                        func = dispatch_dict[file_type]

                except KeyError:

                        if file_type not in missing:

                                missing[file_type] = \

                                    a.attrs[portable.PD_LOCAL_PATH]

                else:

                        try:

                                ds, errs, attrs = func(action=a,

                                    pkg_vars=pkg_vars,

                                    dyn_tok_conv=dyn_tok_conv,

                                    run_paths=a_run_paths)

                                # prune out any dependencies on the files we've

                                # been asked to avoid creating dependencies on

                                if bypass and not ignore_bypass:

                                        ds = \

                                            __bypass_deps(ds, bypass, pkg_attrs)

                                deps.extend(ds)

                                elist.extend(errs)

                                __update_pkg_attrs(pkg_attrs, attrs)

                        except base.DependencyAnalysisError as e:

                                elist.append(e)

        for a in mfst.gen_actions_by_type("hardlink"):

                deps.extend(hardlink.process_hardlink_deps(a, pkg_vars))

def __update_pkg_attrs(pkg_attrs, new_attrs):

        """Update the pkg_attrs dictionary with the contents of new_attrs."""