import subprocess
import random
import string
import base64
import sys
import os

#check if input is a file
if os.path.exists(sys.argv[1]) != True:
    print "you typed something wrong, could not find that file"
else:
    file_to_upload = sys.argv[1]
    if "." in sys.argv[1]:
        domain = sys.argv[1][sys.argv[1].index("."):]
    else:
        domain = ".com"

#generate random data for credentials
user_name = ''.join(random.choice(string.ascii_uppercase) for _ in range(10))
email = ''.join(random.choice(string.ascii_uppercase) for _ in range(10)) + "@"\
        + ''.join(random.choice(string.ascii_uppercase) for _ in range(5)) + domain
passphrase = ''.join(random.choice(string.ascii_uppercase) for _ in range(5))

#any key server is good as it will propogate world wide
key_server = "eu.pool.sks-keyservers.net"

#unattended key generation
p = subprocess.Popen('gpg2 --batch --pinentry-mode=loopback --passphrase ' + passphrase +\
                    ' --quick-gen-key "' + user_name + ' ' + email + '" rsa1024',\
                     shell=True, stdout=subprocess.PIPE)
out, err = p.communicate()

#get pub key
p = subprocess.Popen('gpg2 --list-key --with-colons ' + email, shell=True, stdout=subprocess.PIPE)
out, err = p.communicate()
# parse out the key id so we can use it to send keys to the key servers
key = key = [x.replace(':', '').replace('fpr', '') for x in out.split() if "fpr" in x][0] 

#open file in binary and break it up into 1305byte chunks
chunk_list = []
with open(file_to_upload, 'rb') as infile:
    while True:
        chunk = infile.read(1305)
        if not chunk:
            break
        chunk_list.append(chunk)

#encode binary chunks into base64 strings
i = 0
for x in chunk_list:
    new_uid = str(i) + "@" + base64.b64encode(x)
    p = subprocess.Popen("gpg2 --batch --pinentry-mode=loopback --passphrase " + passphrase\
                + " --quick-add-uid "  + email + " " + new_uid, shell=True, stdout=subprocess.PIPE)
    out, err = p.communicate()
    i += 1

#finally send keys to a server
p = subprocess.Popen("gpg2 --keyserver " + key_server + " --send-keys "\
                    + key, shell=True, stdout=subprocess.PIPE)
out, err = p.communicate()

#remove keys when done as they are not needed anymore
p = subprocess.Popen("gpg --batch --yes --delete-secret-keys " + key +\
                "&& gpg --batch --yes --delete-keys " + key, shell=True, stdout=subprocess.PIPE)
out, err = p.communicate()

if not err:
    print "removing temp keys\n"
    print "It can take 3-10mins before your key appears on your chosen server\n"
    print "http://" + key_server + "/pks/lookup?search=" + email + "&op=index"
else:
    print "something went wrong try again"