# 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/.
"""
Apply some defaults and minor modifications to the pgo jobs.
"""

from __future__ import absolute_import, print_function, unicode_literals

from taskgraph.transforms.base import TransformSequence

import logging
logger = logging.getLogger(__name__)

transforms = TransformSequence()


@transforms.add
def run_profile_data(config, jobs):
    for job in jobs:
        build_platform = job['attributes'].get('build_platform')
        instr = 'instrumented-build-{}'.format(job['name'])
        if 'android' in build_platform:
            artifact = 'geckoview-androidTest.apk'
        elif 'macosx64' in build_platform:
            artifact = 'target.dmg'
        elif 'win' in build_platform:
            artifact = 'target.zip'
        else:
            artifact = 'target.tar.bz2'
        job.setdefault('fetches', {})[instr] = [
            artifact,
            'target.crashreporter-symbols.zip',
        ]
        yield job