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

from bin.lib import encrypt_response_with_iv, decrypt_request_with_iv

sys.path.append(os.path.abspath("../"))
from subject import main

sys.path.append(os.path.abspath('../../'))
from lib import asymmetric_functs

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/')

# Create organization
# organization - username - name - email - public key file
def createOrganization(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('org', nargs='?', default=None)
    parser.add_argument('username', nargs='?', default=None)
    parser.add_argument('name', nargs='?', default=None)
    parser.add_argument('email', nargs='?', default=None)
    parser.add_argument('pubkey', nargs='?', default=None)

    args = parser.parse_args()

    if not args.org or not args.username or not args.name or not args.email or not args.pubkey:
        logger.error("Need organization, username, name, email and key file.")
        sys.exit(1)

    # Validate email
    if not re.match(r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$', args.email):
        logger.error("Need a valid email.")
        sys.exit(1)

    # Validate key file
    if (not os.path.isfile(BASE_DIR + args.pubkey)):
        logger.error("File '" + args.pubkey + "' not found.")
        sys.exit(1)

    # load public key from file
    pubKey = asymmetric_functs.load_public_key(BASE_DIR + args.pubkey)

    payload = encrypt_response_with_iv(json.dumps({'name' : args.org, 'username' : args.username, 'full_name' : args.name, 'email' : args.email, 'public_key' : pubKey}))
    headers = {'Content-Type': 'application/octet-stream'}
    
    try:
        req = requests.post(f'http://{state['REP_ADDRESS']}/org/create', data=payload, headers=headers)
        req.raise_for_status()

    except requests.exceptions.HTTPError:
        logger.error("%d: %s", req.status_code, req.json()['error'])
        sys.exit(-1)
        
    except requests.exceptions.RequestException as errex: 
        logger.error("Failed to obtain response from server.")
        sys.exit(-1)

    if req.status_code == 201:
        try:
            response_data = json.loads(decrypt_request_with_iv(bytes.fromhex(req.text)))
        except Exception as e:
            logger.error("Failed to decrypt the content: %s", e)
            sys.exit(1)
        logger.info(json.dumps(response_data, indent=4))
        sys.exit(0)
    logger.error("Failed to create organization.")
    sys.exit(-1)

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