# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
"""
Add dependencies to release tasks.
"""

from __future__ import absolute_import, print_function, unicode_literals

from taskgraph.transforms.base import TransformSequence

PHASES = ['build', 'promote', 'push', 'ship']

transforms = TransformSequence()


def _get_product(job_or_task):
    # Find the product.
    # XXX Once shipping-product is set for nightly builds as well, we can get
    # rid of this function.
    product = job_or_task.get('shipping-product', job_or_task.get('product'))
    if 'payload' in job_or_task:
        product = product or job_or_task['payload'].get(
            'product',
            job_or_task['payload'].get('properties', {}).get('product')
        )

    if 'run' in job_or_task:
        product = product or job_or_task['run'].get('product')
    if 'extra' in job_or_task:
        product = product or job_or_task['extra'].get('product')
    return product


@transforms.add
def add_dependencies(config, jobs):
    for job in jobs:
        dependencies = {}
        # Add any kind_dependencies_tasks with matching product as dependencies
        product = _get_product(job)
        phase = job.get('shipping-phase')
        if product is None:
            continue

        for dep_task in config.kind_dependencies_tasks:
            # Weed out unwanted tasks.
            # XXX we have run-on-projects which specifies the on-push behavior;
            # we need another attribute that specifies release promotion,
            # possibly which action(s) each task belongs in.
            if product == 'fennec':
                # Don't ship single locale fennec anymore - Bug 1408083
                attr = dep_task.attributes.get
                if attr("locale") or attr("chunk_locales"):
                    continue
                # Skip old-id
                if 'old-id' in dep_task.label:
                    continue
            # We can only depend on tasks in the current or previous phases
            dep_phase = dep_task.attributes.get('shipping_phase')
            if dep_phase and PHASES.index(dep_phase) > PHASES.index(phase):
                continue
            # Add matching product tasks to deps
            if _get_product(dep_task.task) == product or \
                    dep_task.attributes.get('shipping_product') == product:
                dependencies[dep_task.label] = dep_task.label

        job.setdefault('dependencies', {}).update(dependencies)

        yield job