##############################################################################
# Copyright (c) 2015 EMC and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
"""
"""

from storperf.storperf_master import StorPerfMaster
from threading import Thread
import cPickle
import getopt
import json
import logging
import logging.config
import logging.handlers
import socket
import struct
import sys

import requests


class Usage(Exception):
    pass


def event(event_string):
    logging.getLogger(__name__).info(event_string)


class LogRecordStreamHandler(object):

    def __init__(self):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        self.socket.bind((
            'localhost', logging.handlers.DEFAULT_UDP_LOGGING_PORT))
        self.level = logging.INFO

    def read_logs(self):
        try:
            while True:
                datagram = self.socket.recv(8192)
                chunk = datagram[0:4]
                struct.unpack(">L", chunk)[0]
                chunk = datagram[4:]
                obj = cPickle.loads(chunk)
                record = logging.makeLogRecord(obj)
                if (record.levelno >= self.level):
                    logger = logging.getLogger(record.name)
                    logger.handle(record)

        except Exception as e:
            print "ERROR: " + str(e)
        finally:
            self.socket.close()


def main(argv=None):
    verbose = False
    debug = False
    report = None
    erase = False
    terminate = False
    options = {}

    storperf = StorPerfMaster()

    if argv is None:
        argv = sys.argv
    try:
        try:
            opts, args = getopt.getopt(argv[1:], "t:w:r:f:escvdTh",
                                       ["target=",
                                        "workload=",
                                        "report=",
                                        "configure=",
                                        "erase",
                                        "nossd",
                                        "nowarm",
                                        "verbose",
                                        "debug",
                                        "terminate",
                                        "help",
                                        ])
        except getopt.error, msg:
            raise Usage(msg)

        configuration = None
        options['workload'] = None

        for o, a in opts:
            if o in ("-h", "--help"):
                print __doc__
                return 0
            elif o in ("-t", "--target"):
                options['filename'] = a
            elif o in ("-v", "--verbose"):
                verbose = True
            elif o in ("-d", "--debug"):
                debug = True
            elif o in ("-s", "--nossd"):
                options['nossd'] = a
            elif o in ("-c", "--nowarm"):
                options['nowarm'] = False
            elif o in ("-w", "--workload"):
                options['workload'] = a
            elif o in ("-r", "--report"):
                report = a
            elif o in ("-e", "--erase"):
                erase = True
            elif o in ("-T", "--terminate"):
                terminate = True
            elif o in ("-f", "--configure"):
                configuration = dict(x.split('=') for x in a.split(','))

        if (debug) or (verbose):
            udpserver = LogRecordStreamHandler()

            if (debug):
                udpserver.level = logging.DEBUG

            logging.basicConfig(format="%(asctime)s - %(name)s - " +
                                "%(levelname)s - %(message)s")

            t = Thread(target=udpserver.read_logs, args=())
            t.setDaemon(True)
            t.start()

        if (erase):
            response = requests.delete(
                'http://127.0.0.1:5000/api/v1.0/configure')
            if (response.status_code == 400):
                content = json.loads(response.content)
                raise Usage(content['message'])
            return 0

        if (terminate):
            response = requests.delete(
                'http://127.0.0.1:5000/api/v1.0/job')
            if (response.status_code == 400):
                content = json.loads(response.content)
                raise Usage(content['message'])
            return 0

        if (configuration is not None):
            response = requests.post(
                'http://127.0.0.1:5000/api/v1.0/configure', json=configuration)
            if (response.status_code == 400):
                content = json.loads(response.content)
                raise Usage(content['message'])

        if (report is not None):
            print "Fetching report for %s..." % (report,)
            response = requests.get(
                'http://127.0.0.1:5000/api/v1.0/job?id=%s' % (report,))
            if (response.status_code == 400):
                content = json.loads(response.content)
                raise Usage(content['message'])
            content = json.loads(response.content)
            print content
        else:
            print "Calling start..."
            response = requests.post(
                'http://127.0.0.1:5000/api/v1.0/job', json=options)
            if (response.status_code == 400):
                content = json.loads(response.content)
                raise Usage(content['message'])

            content = json.loads(response.content)
            print "Started job id: " + content['job_id']

    except Usage as e:
        print >> sys.stderr, str(e)
        print >> sys.stderr, "For help use --help"
        return 2

    except Exception as e:
        print >> sys.stderr, str(e)
        return 2


if __name__ == "__main__":
    sys.exit(main())