#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Author: Florian Lambert <florian.lambert@cycloid.io>
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import sys
import argparse
import logging as log
import json
import requests
import yaml

VERSION = '1.0'

PARSER = argparse.ArgumentParser(description='Refresh pipeline Cycloid.io')

PARSER.add_argument("--api-key",
                    required=True,
                    type=str,
                    help='API key')
PARSER.add_argument("-i", "--insecure",
                    action='store_true')
PARSER.add_argument("-t", "--pipeline-template",
                    required=True,
                    type=argparse.FileType('r'),
                    help='Pipeline-template yml file')
PARSER.add_argument("-v", "--pipeline-variables",
                    required=True,
                    type=argparse.FileType('r'),
                    help='Pipeline-variables yml file')
PARSER.add_argument("-o", "--organization",
                    required=True,
                    type=str,
                    help='Organization name')
PARSER.add_argument("-P", "--project",
                    required=True,
                    type=str,
                    help='Project_name')
PARSER.add_argument("-E", "--env",
                    required=True,
                    type=str,
                    help='Environment_name')
PARSER.add_argument("-a", "--api-url",
                    type=str,
                    help='Cycloid api url name',
                    default='https://http-api.cycloid.io')
PARSER.add_argument("--version", action='store_true',
                    help='Print script version')
ARGS = PARSER.parse_args()

def update_pipeline(api_url, token, organization, project, env, pipeline_template, pipeline_variables, insecure=False):
    pipeline_name = "%s-%s" % (project, env)
    try:
        y_pipe=yaml.safe_load(pipeline_template)
        y_vars=yaml.safe_load(pipeline_variables)
    except yaml.YAMLError as exc:
        print(exc)

    data = {
        "passed_config": json.dumps(y_pipe),
        "yaml_vars": json.dumps(y_vars)
    }
    headers = {
        'content-type': 'application/vnd.cycloid.io.v1+json',
        'Authorization':'Bearer %s' % token
    }

    r = requests.put('%s/organizations/%s/projects/%s/pipelines/%s' % (api_url, organization, project, pipeline_name),
                     data=json.dumps(data), headers=headers, verify=insecure)
    log.debug(r.text)
    if r.status_code != 200:
        log.error("Unable pipeline : %s" % r.text)
        exit(1)


if __name__ == "__main__":

    if ARGS.version:
        print("version: %s" % (VERSION))
        sys.exit(0)

    token = login(ARGS.username, ARGS.password, ARGS.api_url, ARGS.organization, ARGS.insecure)
    update_pipeline(api_url=ARGS.api_url,
                    token=ARGS.api_key,
                    organization=ARGS.organization,
                    project=ARGS.project,
                    env=ARGS.env,
                    pipeline_template=ARGS.pipeline_template,
                    pipeline_variables=ARGS.pipeline_variables,
                    insecure=ARGS.insecure)