#!/usr/bin/env python
"""Script for submitting Galaxy job information to the Galactic radio telescope.

See doc/source/admin/grt.rst for more detailed usage information.
"""
import argparse
import os
import sys
import yaml
import logging
import requests

sample_config = os.path.abspath(os.path.join(os.path.dirname(__file__), 'grt.yml.sample'))
default_config = os.path.abspath(os.path.join(os.path.dirname(__file__), 'grt.yml'))


def main(argv):
    parser = argparse.ArgumentParser()
    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('-r', '--report-directory', help='Directory in which reports are stored',
                        default=os.path.abspath(os.path.join('.', 'reports')))
    parser.add_argument('-c', '--config', help='Path to GRT config file',
                        default=default_config)
    parser.add_argument("-l", "--loglevel", choices=['debug', 'info', 'warning', 'error', 'critical'],
                        help="Set the logging level", default='warning')
    args = parser.parse_args()

    logging.info('Loading GRT configuration...')
    try:
        with open(args.config) as handle:
            config = yaml.safe_load(handle)
    except Exception:
        logging.info('Using default GRT configuration')
        with open(sample_config) as handle:
            config = yaml.safe_load(handle)

    REPORT_DIR = args.report_directory
    GRT_URL = config['grt']['url'].rstrip('/') + '/'
    GRT_INSTANCE_ID = config['grt']['instance_id']
    GRT_API_KEY = config['grt']['api_key']

    # Contact the server and check auth details.
    headers = {
        'AUTHORIZATION': GRT_INSTANCE_ID + ':' + GRT_API_KEY
    }
    r = requests.post(GRT_URL + 'api/whoami', headers=headers)
    data = r.json()
    # we get back some information about which reports had previously been uploaded.
    remote_reports = data['uploaded_reports']
    # so now we can know which to send.
    local_reports = [x.strip('.json') for x in os.listdir(REPORT_DIR) if x.endswith('.json')]
    for report_id in local_reports:
        if report_id not in remote_reports:
            logging.info("Uploading %s", report_id)
            files = {
                'meta': open(os.path.join(REPORT_DIR, report_id + '.json'), 'rb'),
                'data': open(os.path.join(REPORT_DIR, report_id + '.tar.gz'), 'rb')
            }
            data = {
                'identifier': report_id
            }
            r = requests.post(GRT_URL + 'api/v2/upload', files=files, headers=headers, data=data)
            logging.info("Uploaded successfully", report_id)


if __name__ == '__main__':
    main(sys.argv)