#!/usr/bin/env python

# -*- coding: utf-8 -*-

"""
This script takes as input the fedmsg messages published under the topic
``buildsys.build.state.change`` and flag the corresponding commit in pagure
if the build was successful.

Authors:    Pierre-Yves Chibon <pingou@pingoured.fr>

"""
from __future__ import unicode_literals

import json
import logging
import os
import sys

import koji
import requests
from requests.packages.urllib3.util import retry


_log = logging.getLogger(__name__)


def _koji_hub_link(msg):
    instance = msg.get('instance', 'primary')
    if instance == 'primary':
        base = "https://koji.fedoraproject.org/kojihub"
    else:
        raise NotImplementedError("Unhandled instance")

    # One last little switch-a-roo for stg
    if '.stg.' in msg['request'][0]:
        base = "https://koji.stg.fedoraproject.org/kojihub"

    return base


def _koji_link(msg):
    instance = msg.get('instance', 'primary')
    if instance == 'primary':
        base = "https://koji.fedoraproject.org/koji"
    else:
        raise NotImplementedError("Unhandled instance")

    # One last little switch-a-roo for stg
    if '.stg.' in msg['request'][0]:
        base = "https://koji.stg.fedoraproject.org/koji"

    return base + "/taskinfo?taskID=%i" \
        % (msg['task_id'])


def main(msg):
    """ Check if the build was successful and if so, flag the commit in
    pagure.
    """

    timeout = (30, 30)
    retries = 3
    requests_session = requests.Session()
    retry_conf = retry.Retry(
        total=retries, connect=retries, read=retries, backoff_factor=1)
    retry_conf.BACKOFF_MAX = 5
    requests_session.mount(
        'http://', requests.adapters.HTTPAdapter(max_retries=retry_conf))
    requests_session.mount(
        'https://', requests.adapters.HTTPAdapter(max_retries=retry_conf))

    request = msg['request'][0]
    instance = msg['instance']
    if instance != 'primary':
        _log.info("Ignoring secondary arch task...")
        return

    # see koji.TASK_STATES for all values
    done_states = {
        1: 'completed',
        3: 'failed',
        4: 'canceled',
    }
    state = msg['new']
    if state not in done_states:
        _log.info('Build is still in progress, let\'s come back later')
        return

    koji_url = _koji_hub_link(msg)
    _log.info('Talking to koji at: %s' % koji_url)
    koji_client = koji.ClientSession(koji_url)

    build_id = msg['build_id']
    build = koji_client.getBuild(build_id)
    git_url = (
        build.get('extra') or {}).get('source', {}).get('original_url')

    if not git_url:
        _log.info(
            'No git url found in the extra information: %s',
            build.get('extra'))
        return 1

    if '#' not in git_url:
        _log.info('No # in the git_url: %s', git_url)
        return 1

    commit = git_url.rsplit('#', 1)[1]

    if state == 1:
        status = 'success'
    elif state == 3:
        status = 'failure'
    else:
        status = 'canceled'

    version = msg['version']
    if msg.get('epoch'):
        version = '%s:%s' % (msg['epoch'], version)

    nvr = '%s-%s-%s' % (
        msg['name'],
        version,
        msg['release']
    )
    data = {
        'username': 'Build %s' % (done_states[state]),
        'status': status,
        'comment': 'Built as %s' % (nvr),
        'url': _koji_link(msg),
    }

    pagure_url = 'https://src.fedoraproject.org'
    env_var = 'API_TOKEN'
    if '.stg.' in request:
        pagure_url = 'https://src.stg.fedoraproject.org'
        env_var = 'API_TOKEN_STG'

    target_url = '/'.join([
        'api',
        '0',
        'rpms',
        msg['name'],
        'c',
        commit,
        'flag'
    ])
    flag_url = pagure_url + '/' + target_url
    _log.info('Flagging commit at: %s' % flag_url)

    headers = {
        "Authorization": "token " + os.environ.get(env_var),
        'User-Agent': 'loopabull@fedora-infr',
    }

    _log.info('payload: %s' % data)

    req = requests_session.request(
        method='POST',
        url=flag_url,
        headers=headers,
        data=data,
    )
    _log.info('Request to %s returned: %s' % (pagure_url, req.status_code))
    _log.debug(req.text)
    if not req.ok:
        _log.info(req.text)
        return 1
    else:
        _log.info('All clear')
        _log.info('User-URL: %s' % pagure_url + '/' + '/'.join([
            'rpms',
            msg['name'],
            'c',
            commit
        ]))


if __name__ == '__main__':
    msg = sys.argv[1]
    msg = json.loads(msg)
    sys.exit(main(msg))