PBinCLI/pbincli/actions.py

import signal, sys
from urllib.parse import parse_qsl

from pbincli.api import Shortener
from pbincli.format import Paste
from pbincli.utils import PBinCLIError, check_writable, json_encode, uri_validator, validate_url_ending, validate_path_ending


def signal_handler(sig, frame):
    print('Keyboard interrupt received, terminating…')
    sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)


def send(args, api_client, settings=None):
    if settings['short']:
        shortener = Shortener(settings)

    if not args.notext:
        if args.text:
            text = args.text
        elif args.stdin:
            print("Reading text from stdin…")
            text = args.stdin.read()
    elif not args.file:
        PBinCLIError("Nothing to send!")
    else:
        text = ""

    print("Preparing paste…")
    paste = Paste(args.debug)

    if args.verbose: print("Used server: {}".format(api_client.getServer()))

    # get from server supported paste format version and update object
    if args.verbose: print("Getting supported paste format version from server…")
    version = api_client.getVersion()
    paste.setVersion(version)

    if args.verbose: print("Filling paste with data…")
    # set compression type, works only on v2 pastes
    if version == 2:
        paste.setCompression(settings['compression'])

    # add text in paste (if it provided)
    paste.setText(text)

    # If we set PASSWORD variable
    if args.password:
        paste.setPassword(args.password)

    # If we set FILE variable
    if args.file:
        paste.setAttachment(args.file)

    if args.verbose: print("Encrypting paste…")
    paste.encrypt(
        formatter = settings['format'],
        burnafterreading = settings['burn'],
        discussion = settings['discus'],
        expiration = settings['expire'])

    if args.verbose: print("Sending request to server…")
    request = paste.getJSON()

    if args.debug: print("Passphrase:\t{}\nRequest:\t{}".format(paste.getHash(), request))

    # If we use dry option, exit now
    if args.dry: sys.exit(0)

    print("Uploading paste…")
    result = api_client.post(request)

    if args.debug: print("Response:\t{}\n".format(result))

    # Paste was sent. Checking for returned status code
    if not result['status']: # return code is zero
        passphrase = paste.getHash()

        # Paste information
        print("Paste uploaded!\nPasteID:\t{}\nPassword:\t{}\nDelete token:\t{}".format(
            result['id'],
            passphrase,
            result['deletetoken']))

        # Paste link
        print("\nLink:\t\t{}?{}#{}".format(
            settings['server'],
            result['id'],
            passphrase))

        # Paste deletion link
        print("Delete Link:\t{}?pasteid={}&deletetoken={}".format(
            settings['server'],
            result['id'],
            result['deletetoken']))

        # Print links to mirrors if present
        if settings['mirrors']:
            print("\nMirrors:")
            urls = settings['mirrors'].split(',')
            for x in urls:
                print("\t\t{}?{}#{}".format(
                    validate_url_ending(x),
                    result['id'],
                    passphrase))

    elif result['status']: # return code is other then zero
        PBinCLIError("Something went wrong…\nError:\t\t{}".format(result['message']))
    else: # or here no status field in response or it is empty
        PBinCLIError("Something went wrong…\nError: Empty response.")

    if settings['short']:
        print("\nQuerying URL shortening service…")
        shortener.getlink("{}?{}#{}".format(
            settings['server'],
            result['id'],
            passphrase))


def get(args, api_client, settings=None):
    parseduri, isuri = uri_validator(args.pasteinfo)

    if isuri and parseduri.query and parseduri.fragment:
        api_client.server = args.pasteinfo.split("?")[0]
        pasteid = parseduri.query
        passphrase = parseduri.fragment
    elif parseduri.path and parseduri.path != "/" and parseduri.fragment:
        pasteid = parseduri.path
        passphrase = parseduri.fragment
    else:
        PBinCLIError("Provided info hasn't contain valid URL or PasteID#Passphrase string")

    if args.verbose: print("Used server: {}".format(api_client.getServer()))
    if args.debug: print("PasteID:\t{}\nPassphrase:\t{}".format(pasteid, passphrase))

    paste = Paste(args.debug)

    if args.password:
        paste.setPassword(args.password)
        if args.debug: print("Password:\t{}".format(args.password))

    if args.verbose: print("Requesting paste from server…")
    result = api_client.get(pasteid)

    if args.debug: print("Response:\t{}\n".format(result))

    # Paste was received. Checking received status code
    if not result['status']: # return code is zero
        print("Paste received! Decoding…")

        version = result['v'] if 'v' in result else 1
        paste.setVersion(version)

        if version == 2:
            if args.debug: print("Authentication data:\t{}".format(result['adata']))

        paste.setHash(passphrase)
        paste.loadJSON(result)
        paste.decrypt()

        text = paste.getText()

        if args.debug: print("Decoded text size: {}\n".format(len(text)))

        if len(text):
            if args.debug: print("{}\n".format(text.decode()))
            if settings['output']:
                paste_path = validate_path_ending(settings['output']) + "paste-" + pasteid + ".txt"
            else:
                paste_path = "paste-" + pasteid + ".txt"

            print("Found text in paste. Saving it to {}".format(paste_path))

            check_writable(paste_path)
            with open(paste_path, "wb") as f:
                f.write(text)
                f.close()

        attachment, attachment_name = paste.getAttachment()

        if attachment:
            if settings['output']:
                attachment_path = validate_path_ending(settings['output']) + attachment_name
            else:
                attachment_path = attachment_name

            print("Found file, attached to paste. Saving it to {}\n".format(attachment_path))

            check_writable(attachment_path)
            with open(attachment_path, "wb") as f:
                f.write(attachment)
                f.close()

        if version == 1 and 'meta' in result and 'burnafterreading' in result['meta'] and result['meta']['burnafterreading']:
            print("Burn afrer reading flag found. Deleting paste…")
            api_client.delete(json_encode({'pasteid':pasteid,'deletetoken':'burnafterreading'}))

    elif result['status']: # return code is other then zero
        PBinCLIError("Something went wrong…\nError:\t\t{}".format(result['message']))
    else: # or here no status field in response or it is empty
        PBinCLIError("Something went wrong…\nError: Empty response.")


def delete(args, api_client, settings=None):
    parseduri, isuri = uri_validator(args.pasteinfo)

    if isuri:
        api_client.server = args.pasteinfo.split("?")[0]
        query = dict(parse_qsl(parseduri.query))
    else:
        query = dict(parse_qsl(args.pasteinfo))

    if 'pasteid' in query and 'deletetoken' in query:
        pasteid = query['pasteid']
        token = query['deletetoken']
    else:
        PBinCLIError("Provided info hasn't contain required information")

    if args.verbose: print("Used server: {}".format(api_client.getServer()))
    if args.debug: print("PasteID:\t{}\nToken:\t\t{}".format(pasteid, token))

    print("Requesting paste deletion…")
    api_client.delete(json_encode({'pasteid':pasteid,'deletetoken':token}))