# 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/.

from __future__ import absolute_import, unicode_literals

import re

from taskgraph.util.scriptworker import generate_beetmover_upstream_artifacts


_ARTIFACT_ID_PER_PLATFORM = {
    'android-aarch64-opt': 'geckoview-default-arm64-v8a',
    'android-api-16-opt': 'geckoview-default-armeabi-v7a',
    'android-x86-opt': 'geckoview-default-x86',
    'android-x86_64-opt': 'geckoview-default-x86_64',
    'android-geckoview-fat-aar-opt': 'geckoview-default',

    'android-aarch64-shippable': 'geckoview{update_channel}-arm64-v8a',
    'android-api-16-shippable': 'geckoview{update_channel}-armeabi-v7a',
    'android-x86-shippable': 'geckoview{update_channel}-x86',
    'android-x86_64-shippable': 'geckoview{update_channel}-x86_64',
    'android-geckoview-fat-aar-shippable': 'geckoview{update_channel}',
}


def get_geckoview_upstream_artifacts(config, job, platform=''):
    if not platform:
        platform = job['attributes']['build_platform']
    upstream_artifacts = generate_beetmover_upstream_artifacts(
        config, job, platform='',
        **get_geckoview_template_vars(config, platform, job['attributes'].get('update-channel'))
    )
    return [{
        key: value for key, value in upstream_artifact.items()
        if key != 'locale'
    } for upstream_artifact in upstream_artifacts]


def get_geckoview_template_vars(config, platform, update_channel):
    version_groups = re.match(r'(\d+).(\d+).*', config.params['version'])
    if version_groups:
        major_version, minor_version = version_groups.groups()

    return {
        'artifact_id': get_geckoview_artifact_id(
            config, platform, update_channel,
        ),
        'build_date': config.params['moz_build_date'],
        'major_version': major_version,
        'minor_version': minor_version,
    }


def get_geckoview_artifact_id(config, platform, update_channel=None):
    if update_channel == 'release':
        update_channel = ''
    elif update_channel is not None:
        update_channel = '-{}'.format(update_channel)
    else:
        # For shippable builds, mozharness defaults to using
        # "nightly-{project}" for the update channel.  For other builds, the
        # update channel is not set, but the value is not substituted.
        update_channel = '-nightly-{}'.format(config.params['project'])
    return _ARTIFACT_ID_PER_PLATFORM[platform].format(update_channel=update_channel)