#!/usr/bin/python
# -*- coding: UTF-8 -*-

import time
from datetime import datetime, timedelta
import zmq
import json
import numpy as np
import argparse
import base64

REQUEST_TIMEOUT = 5000
RETRIES = 2
SERVER_ENDPOINT = 'tcp://localhost:4444'

filenames = [
    '/Users/jespinoza/Downloads/202016700024003000.png',
    '/Users/jespinoza/Downloads/202016800024003100.png',
    '/Users/jespinoza/Downloads/202016800024003200.png',
]

def send(data):
    '''
    Function to send data to server
    '''
    context = zmq.Context()
    socket = context.socket(zmq.REQ)
    socket.connect(SERVER_ENDPOINT)
    poll = zmq.Poller()
    poll.register(socket, zmq.POLLIN)
    retries = RETRIES
    while True:
        socket.send_string(json.dumps(data))
        socks = dict(poll.poll(REQUEST_TIMEOUT))
        if socks.get(socket) == zmq.POLLIN:
            reply = socket.recv_string()
            if reply == 'ok':
                print(('Server replied (%s)' % reply))
                break
            else:
                print(('Malformed reply from server: %s' % reply))
        else:
            print(('No response from server, retries left {}'.format(retries)))
        socket.setsockopt(zmq.LINGER, 0)
        socket.close()
        poll.unregister(socket)
        retries -= 1
        if retries == 0:
            print('Server seems to be offline...')
            socket = context.socket(zmq.REQ)
            socket.connect(SERVER_ENDPOINT)
            poll.register(socket, zmq.POLLIN)
            break
        # Create new connection
        socket = context.socket(zmq.REQ)
        socket.connect(SERVER_ENDPOINT)
        poll.register(socket, zmq.POLLIN)


def main(realtime, code, date=None, interval=30):
    '''
    Simulate data to be sended to server
    '''
    x = 0
    while True:
        
        if realtime:
            dt = datetime.now()
        else:
            dt = date
        
        print(('Sending {} - {}'.format(code, dt)))

        dato = {
            'time': (datetime.utcnow()-datetime(1970,1,1)).total_seconds(),
            'metadata': {
                'localtime': True,
                'type': 'image',
                },
            'code': code,
            'data': base64.b64encode(bytearray(open(filenames[x], 'rb').read())).decode(),
            'plot': 'Summary'
        }
        
        send(dato)
        
        if realtime:
            time.sleep(interval)
        else:
            dt = dt + timedelta(seconds=interval)
            time.sleep(2)

        x += 1
        if x == 3:
            x = 0

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='''This is a Client for
                                                    realtime app''')
    parser.add_argument('--date', action='store', default=None,
                        help='format: 2018/02/13 12:23:00')
    parser.add_argument('-r', action='store_true', dest='realtime',
                        default=None)
    parser.add_argument('-c', action='store', dest='code', default='300')
    parser.add_argument('-i', action='store', dest='interval', type=int,
                        default=120)

    results = parser.parse_args()
    if not results.realtime:
        try:
            results.date = datetime.strptime(results.date, '%Y/%m/%d %H:%M:%S')
        except:
            raise NameError('''You must specify a date (--date)
                            for non-realtime experiment''')
    main(results.realtime, results.code, results.date, results.interval)