#!/bin/python3
import os
import sys
import logging
import requests
import json
import argparse
import datetime

from subject import main

logging.basicConfig(format='%(levelname)s\t- %(message)s')
logger = logging.getLogger()
logger.setLevel(logging.INFO)

state = main(sys.argv)

BASE_DIR = os.path.join(os.path.expanduser('~'), '.sio/')

def validDate(input):
    try:
        date = datetime.datetime.strptime(input,"%d/%m/%Y")
    except ValueError as err:
        logger.error("Date is invalid. Input format: DD/MM/YYYY")
        sys.exit(-1)

    return date

#session file - [-s username] [-d nt/ot/et date]
def list_docs(args):
    parser = argparse.ArgumentParser()

    parser.add_argument("-k", '--key', nargs=1, help="Path to the key file")
    parser.add_argument("-r", '--repo', nargs=1, help="Address:Port of the repository")
    parser.add_argument("-v", '--verbose', help="Increase verbosity", action="store_true")

    parser.add_argument('session', nargs='?', default=None)
    parser.add_argument("-s", '--username', nargs=1, help="Username")
    parser.add_argument("-dnt", '--newerThan', help="Date new than")
    parser.add_argument("-dot", '--olderThan', help="Date older than")
    parser.add_argument("-deq", '--equalTo',   help="Date equal to")

    args = parser.parse_args()

    # Check if session file is valid
    if args.session:
        if (not os.path.isfile(BASE_DIR + args.session)):
            logger.error("File '" + args.session + "' not found.")
            sys.exit(1)
    else:
        logger.error("Need session file.")
        sys.exit(1)

    # Get session file content
    with open(BASE_DIR + args.session, 'r') as f:
        args.session = json.load(f)

    if args.newerThan:
        #Convert date str to datetime in seconds
        args.newerThan = validDate(args.newerThan)
        args.newerThan = int(args.newerThan.timestamp())

        try: 
            subjects = requests.get(f'http://{state['REP_ADDRESS']}/file/list', 
                                    json=json.dumps({'username' : args.username,'datetime' : {'value' : args.newerThan, 'relation' : 'nt'}}),
                                    headers={'Authorization': args.session['token']})
            subjects.raise_for_status() 

        except requests.exceptions.RequestException as errex: 
            logger.error("Failed to obtain response from server.")
            sys.exit(-1)

        subjects = subjects.json()
        
    elif args.equalTo:
        #Convert date str to datetime in seconds
        args.equalTo = validDate(args.equalTo)
        args.equalTo = int(args.equalTo.timestamp())

        try: 
            subjects = requests.get(f'http://{state['REP_ADDRESS']}/file/list', 
                                    json=json.dumps({'username' : args.username,
                                    'datetime' : {'value' : args.equalTo, 'relation' : 'eq'}}),
                                    headers={'Authorization': args.session['token']})
            subjects.raise_for_status() 

        except requests.exceptions.RequestException as errex: 
            logger.error("Failed to obtain response from server.")
            sys.exit(-1)

        subjects = subjects.json()

    elif args.olderThan:
        #Convert date str to datetime in seconds
        args.olderThan = validDate(args.olderThan)
        args.olderThan = int(args.olderThan.timestamp())

        try: 
            subjects = requests.get(f'http://{state['REP_ADDRESS']}/file/list', 
                                    json=json.dumps({'username' : args.username,
                                    'datetime' : {'value' : args.olderThan, 'relation' : 'ot'}}),
                                    headers={'Authorization': args.session['token']})
            subjects.raise_for_status() 

        except requests.exceptions.RequestException as errex: 
            logger.error("Failed to obtain response from server.")
            sys.exit(-1)

        subjects = subjects.json()

    else:
        try: 
            subjects = requests.get(f'http://{state['REP_ADDRESS']}/file/list', json=json.dumps({}), headers={'Authorization': args.session['token']})
            subjects.raise_for_status() 

        except requests.exceptions.RequestException as errex: 
            logger.error("Failed to obtain response from server.")
            sys.exit(-1)

        subjects = subjects.json()

    for s in subjects:
        sys.stdout.write(s['id'] + " - " + s['username'])

    sys.exit(0)

if __name__ == '__main__':
    list_docs(sys.argv[1:])