#!/usr/bin/env python
#
#   Copyright 2017 Sandvine
#
#   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 requests
import json
import pprint
import argparse

from dateutil import parser
from dateutil.tz import tzutc
from datetime import datetime
import time

arg_parser = argparse.ArgumentParser(description="Tool to retrieve the latest build from the artifactory server")
arg_parser.add_argument('branch')
arg_parser.add_argument('--user', default='admin')
arg_parser.add_argument('--project', default='osm-stage_3')
arg_parser.add_argument('--url', default='https://artifactory-osm.etsi.org/')
arg_parser.add_argument('--keep', default=5)
arg_parser.add_argument('--password', default='')
args = arg_parser.parse_args()

url = args.url + 'artifactory/api/build/' + args.project + " :: " + args.branch

resp = requests.get(url)
jsonData = json.loads(resp.content)
if 'buildsNumbers' not in jsonData:
    print("Cannot find any valid builds")
    exit(1)

# first entry is the latest build
buildlist = sorted(jsonData['buildsNumbers'], key=lambda x: int(x['uri'][1:]), reverse=True)
print("total builds is {}".format(len(buildlist)))
pprint.pprint(buildlist)

if len(buildlist) < args.keep:
    print("nothing to cleanup")
    exit(0)


def convert_to_ms(datetime):
    # get the millisecond from the date/time
    ms = datetime.split('.')[1].split('+')[0]
    parser_out = parser.parse(datetime)
    timeval = parser_out
    tuple = int(time.mktime(timeval.timetuple()))
    return (tuple * 1000 + int(ms) - (time.timezone * 1000))


def buildPost(dateInMS, buildNumber):
    build = {}
    data = {}
    build['buildName'] = args.project + " :: " + args.branch
    build['buildNumber'] = buildNumber
    build['date'] = str(dateInMS)

    data['buildsCoordinates'] = list()
    data['buildsCoordinates'].append(build)
    return data


delete_url = args.url + 'artifactory/ui/builds/buildsDelete'
headers = {'Content-Type': 'application/json'}

for entry in buildlist[int(args.keep):]:
    ms = convert_to_ms(entry['started'])
    buildNumber = entry['uri'].split('/')[1]
    print("deleting build {} ms {}".format(args.project + " :: " + args.branch + '/' + buildNumber, ms))
    postData = buildPost(ms, entry['uri'].split('/')[1])

    requests.post(delete_url, data=json.dumps(postData), headers=headers, auth=(args.user, args.password))