sio-2425/delivery1/client/bin/rep_add_doc

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

from subject import main

sys.path.append(os.path.abspath("../../"))
from lib.symmetric_encryption import *
from lib import digest

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

# session file - document name - file
def addDoc(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('name', nargs='?', default=None)
    parser.add_argument('file', nargs='?', default=None)

    args = parser.parse_args()

    if not args.session or not args.name or not args.file:
        logger.error("Need session file, document's name and file to upload.")
        sys.exit(-1)

    if (not os.path.isfile(os.path.join(BASE_DIR, args.session))):
        logger.error("File '" + args.session + "' not found.")
        sys.exit(-1)

    if (not os.path.isfile(os.path.join(BASE_DIR, args.file))):
        logger.error("File '" + args.file + "' not found")
        sys.exit(-1)

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

    # Get hash of document's name for doc handle
    doc_name = digest.get_hash(bytes(args.name, encoding='utf-8'))


    #Upload document metadata
    doc = {'document_name' : doc_name}
    
    try:
        req = requests.post(f'http://{state['REP_ADDRESS']}/file/upload', json=json.dumps(doc), headers={'Authorization': args.session['token']})
        req.raise_for_status()

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

    #Upload Document content
    #Create salt
    salt = os.urandom(16)

    #Encrypt content
    encrypt_file(salt, BASE_DIR + args.file, BASE_DIR + 'encryptedText')

    with open(BASE_DIR + 'encryptedText', 'rb') as f:
        content = f.read()

    try:
        req = requests.post(f'http://{state['REP_ADDRESS']}/file/upload', json=json.dumps({'dw' : content.hex()}), headers={'Authorization': args.session['token']})
        req.raise_for_status()

    except requests.exceptions.RequestException as errex: 
        logger.error("Failed to obtain response from server.")
        os.remove(BASE_DIR + 'encryptedText')
        sys.exit(1)

    #Delete temporary file
    os.remove(BASE_DIR + 'encryptedText')

    sys.exit(0)

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