import fileinput
import sys, getopt
from pastebin_connection import paste_create, login_user


def add_paste(content, succ, error):
    if content:
        result = paste_create(content)
        if result['success']:
            succ.append(result['data'])
        else:
            error.append(result['data'])
        paste_content = ""

def handle_input(args):
    paste_content = ""
    succ = []
    error = []

    for line in fileinput.input(args):
        if fileinput.isfirstline():
            add_paste(paste_content, succ, error)

        paste_content += line

    add_paste(paste_content, succ, error)
    if succ:
        print("Succesful pastes")
        for url in succ:
            print(url)

        print()

    if error:
        print("Error")
        for url in error:
            print(url)

def print_help():
    print()
    print("Use like cat program to upload to pastebin.")
    print("[-u|--username] [-p|--password] - for username and assword")
    print()
    pass

def main(argv):
    username = ''
    password = ''
    try:
        opts, args = getopt.gnu_getopt(argv, 'hu:p:', ["username=", "password="])
    except getopt.GetoptError:
        print_help()
        sys.exit(2)

    for opt, arg in opts:
        if opt == '-h':
            print_help()
            sys.exit()
        elif opt in ('-u', '--username'):
            username = arg
        elif opt in ('-p', '--password'):
            password = arg

    if username and password:
        res = login_user(username, password)
        if res['success']:
            print('Successful login')
        else:
            print(res['data'])

    handle_input(args)

if __name__ == "__main__":
    main(sys.argv[1:])