#!/usr/bin/python3

# internal modules

from os import system
from os import remove
from os import environ
from shutil import move
from shutil import rmtree
from sys import argv
import random
import string

# custom modules


def clear():
    print('\n' * 100)


def term(cmd):
    _ = system(cmd)


def replace_line(file_name, line_num, text):
    lines = open(file_name, 'r').readlines()
    lines[line_num] = text
    out = open(file_name, 'w')
    out.writelines(lines)
    out.close()


def shmgen():
    shmfoldergen = ''.join(random.SystemRandom().choice(string.ascii_letters + string.digits)
                           for _ in range(random.randint(10, 30)))
    shmentrygen = ''.join(random.SystemRandom().choice(string.ascii_letters + string.digits)
                          for _ in range(random.randint(10, 30)))
    term('mkdir -p /dev/shm/' + shmfoldergen)
    return shmfoldergen, shmentrygen


def encrypt():
    term('gpg -r ' + str(gpgid) + ' -e ' + "'" + '/dev/shm/' + shmfolder + '/' + shmentry + "'")
    move('/dev/shm/' + shmfolder + '/' + shmentry + '.gpg', directory + entryname.replace('/', '', 1) + '.gpg')
    rmtree('/dev/shm/' + shmfolder)


def encryptfolder():
    term('gpg -r ' + str(gpgid) + ' -e ' + "'" + '/dev/shm/' + shmfolder + '/' + shmentry + "'")
    move('/dev/shm/' + shmfolder + '/' + shmentry + '.gpg', directory + folder + '/' + folderentry + '.gpg')
    rmtree('/dev/shm/' + shmfolder)


# argument - filter the argument to usable text

argument = str(argv).replace('[', '', 1).replace(']', '', 1).replace(',', '').replace("'", '') \
    .replace(' ', '', 1).replace('/usr/bin/', '', 1).replace('rpass', '', 1)

# help

if argument == 'help' or argument == '--help' or argument == '-h':
    print('\nrpass  Copyright (C) 2021  Randall Winkhart')
    print("This program comes with ABSOLUTELY NO WARRANTY; for details type `rpass show w'.\nThis is free software, "
          "and you are welcome to redistribute it under certain conditions; type `rpass show c' for details.")
    print('\nUSAGE: rpass [/<entry name>] [OPTION] [FLAG]\n')
    print('Options: ')
    print('help/--help/-h           bring up this menu')
    print('version/-v               display rpass version info')
    print('config                   configure rpass')
    print('add                      add an entry')
    print('gen                      generate a new password')
    print('edit                     edit an existing entry')
    print('copy                     copy details of an entry to your clipboard')
    print('remove/-rm               delete an existing entry')
    print('sync/-s                  manually sync the entry directory via rsync\n')
    print('Flags:')
    print('add:')
    print(' password/-p             add a password entry')
    print(' note/-n                 add a note entry')
    print(' folder/-f               add a new folder for entries')
    print('edit:')
    print(' rename/relocate/-r      rename or relocate an entry')
    print(' username/-u             change the username of an entry')
    print(' password/-p             change the password of an entry')
    print(' note/-n                 change the note attached to an entry')
    print(' url/-l                  change the url attached to an entry')
    print('copy:')
    print(' username/-u             copy the username of an entry to your clipboard')
    print(' password/-p             copy the password of an entry to your clipboard')
    print(' url/-l                  copy the URL of an entry to your clipboard')
    print(' note/-n                 copy the note of an entry to your clipboard')
    print('gen:')
    print(' update/-u               generate a password for an existing entry\n')
    print("Tip: You can quickly read an entry with 'rpass /<entry name>'!")
    print("When specifying entry names, do not include the file extension! '.gpg' is assumed!\n")

# version info

if argument == 'version' or argument == '-v':
    print('\n////////////////////////////////////////////////////////')
    print('/                                                      /')
    print('/     rpass  Copyright (C) 2021  Randall Winkhart      /')
    print('/                                                      /')
    print('/             Version 2021.10.08.mr5                   /')
    print('/          The Housekeeping Update (Pt. 2)             /')
    print('/                                                      /')
    print('////////////////////////////////////////////////////////\n')

# licensing info

if argument == 'show w':
    clear()
    print('This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;')
    print('without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.')
    print('See the GNU General Public License for more details.')
    print('\nhttps://opensource.org/licenses/GPL-3.0\n')

if argument == 'show c':
    clear()
    print('This program is free software: you can redistribute it and/or modify it under the terms of the GNU General')
    print('Public License as published by the Free Software Foundation, either version 3 of the License,')
    print('or (at your option) any later version.\n')

# config

if argument == 'config':
    try:
        print()
        directorychoice = str(input('Would you like to use the default entry directory (~/.rpass-store/)? (Y/n) '))
        if directorychoice == 'n' or directorychoice == 'N':
            print('\nFormat: /this/path/ends/with/a/slash/\n')
            directory = str(input('Please input a custom directory: '))
            with open("/var/lib/rpass/rpass-dir", 'w') as f:
                f.write(directory + '\n')
        else:
            with open("/var/lib/rpass/rpass-dir", 'w') as f:
                f.write('/home/' + environ.get('USER') + '/.rpass-store/' + '\n')
        print()
        devicetype = str(input('Will this be a client, server, or offline device? (C/s/o) '))
        if devicetype == 's' or devicetype == 'S':
            with open("/var/lib/rpass/rpass-ssh", 'w') as f:
                f.write('\n\n\n\n' + devicetype)
            print('\nMake sure the ssh service is running and properly configured.\n')
            print('Configuration complete!\n')
        else:
            print()
            gpgpresent = str(input(
                'rpass requires the use of a unique gpg key. Do you already have one you are willing to use? (y/N) '))
            if gpgpresent == 'y' or gpgpresent == 'Y':
                print()
                gpgid = str(input('Please input the ID of your gpg key: '))
                with open("/var/lib/rpass/rpass-gpg", 'w') as f:
                    f.write(gpgid + '\n')
            else:
                print()
                gpggen = str(
                    input('Would you like to generate one now? Note that if you choose not to do this, you will have '
                          'to re-run "rpass config" to specify a gpg key when you acquire one (Y/n) '))
                if gpggen == 'n' or gpggen == 'N':
                    exit()
                else:
                    term('gpg --full-generate-key')
                    print()
                    gpgid = str(input('Please input the ID of your gpg key: '))
                    with open("/var/lib/rpass/rpass-gpg", 'w') as f:
                        f.write(gpgid + '\n')
        if devicetype == 'o' or devicetype == 'O':
            with open("/var/lib/rpass/rpass-ssh", 'w') as f:
                f.write('\n\n\n\n' + devicetype)
            print('\nConfiguration complete!\n')
        elif devicetype != 's' and devicetype != 'S':
            devicetype = 'c'
            print('\nMake sure the ssh service on the remote server is running and properly configured.\n')
            sshgen = str(input('rsync support requires a unique ssh key. Would you like to have this '
                               'automatically generated? (Y/n) '))
            if sshgen != 'n' and sshgen != 'N':
                term('ssh-keygen -t ed25519 -f ~/.ssh/rpass')
            print('\nExample input: 10.10.10.10:9999\n')
            ip_port = str(input('What is the IP and ssh port of the remote server? '))
            print()
            ip, sep, port = ip_port.partition(':')
            usernamessh = str(input('What is the username of the remote server? '))
            print('\nDefault directory: ' + '/home/' + usernamessh + '/.rpass-store/\n')
            directoryssh = str(input('Is the remote server using the default password directory? (Y/n) '))
            if directoryssh == 'n' or directoryssh == 'N':
                print('\nFormat: /this/path/ends/with/a/slash/\n')
                directoryssh = str(input('What directory is the server using?: '))
            else:
                directoryssh = ('/home/' + usernamessh + '/.rpass-store/')
            with open("/var/lib/rpass/rpass-ssh", 'w') as f:
                f.write(usernamessh + '\n' + ip + '\n' + port + '\n' + directoryssh + '\n' + devicetype)
            print('\nConfiguration complete!\n')
    except KeyboardInterrupt:
        print('\n\nConfiguration cancelled.\n')
        exit()


# startup help

if argument == '':
    clear()
    print("Use 'help', '--help', or '-h' to see a list of commands.\n")
    print('rpass is currently early software with very little polish\n')

# import userdata

try:
    sshinfo = open("/var/lib/rpass/rpass-ssh").readlines()
    usernamessh = sshinfo[0].replace('\n', '')
    ip = sshinfo[1].replace('\n', '')
    port = sshinfo[2].replace('\n', '')
    directoryssh = sshinfo[3].replace('\n', '')
    devicetype = sshinfo[4].replace('\n', '')
except (FileNotFoundError, IndexError):
    usernamessh, ip, port, directoryssh, devicetype = (0, 0, 0, 0, 0)
try:
    with open("/var/lib/rpass/rpass-gpg") as f:
        gpgid = f.read().strip()
    with open("/var/lib/rpass/rpass-dir") as f:
        directory = f.read().strip()
    term('mkdir -p ' + directory)
except FileNotFoundError:
    print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
    print("Not all necessary configuration files are present. Please run 'rpass config'!")
    print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
    exit()

# no argument

if argument == '':
    try:
        print()
        term('cd ' + directory + '; ls -1 *')
        print()
        readentry = str(input('Entry to read: '))
        clear()
        term('gpg -d ' + directory + "'" + readentry.replace('/', '', 1) + "'" + '.gpg')
        print()
    except KeyboardInterrupt:
        print('\n\nSelection cancelled.\n')
        exit()

# read shortcut

if argument.startswith('/'):
    if argument.replace('/', '', 1).__contains__('/'):
        folder, sep, folderentry = argument.replace('/', '', 1).partition('/')
        term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
    else:
        term('gpg -d ' + directory + "'" + argument.replace('/', '', 1) + '.gpg' + "'")

# add

if argument == 'add':
    print('\nUSAGE: rpass add [FLAG]')
    print('Flags:')
    print('add:')
    print(' password/-p             add a password entry')
    print(' note/-n                 add a note entry')
    print(' folder/-f               add a new folder for entries\n')

if argument == 'add note' or argument == 'add -n':
    try:
        clear()
        entryname = str(input('Name of note: '))
        notes = str(input('Contents of note: '))
    except KeyboardInterrupt:
        entryname, notes = (0, 0)
        print('\n\nEntry add cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):  # check if user input contains slashes
            if entryname.replace('/', '', 1).__contains__('/'):  # create folder and entry if there are two slashes
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines('\n\n\n' + notes + '\n\n')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:  # create entry if there is only one slash
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines('\n\n\n' + notes + '\n\n')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:  # create entry if there are no slashes
            with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                f.writelines('\n\n\n' + notes + '\n\n')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The folder you requested to access does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'add password' or argument == 'add -p':
    try:
        clear()
        entryname = str(input('Name of service: '))
        username = str(input('Username: '))
        password = str(input('Password: '))
        url = str(input('URL: '))
        notes = str(input('Additional notes: '))
    except KeyboardInterrupt:
        entryname, username, password, url, notes = (0, 0, 0, 0, 0)
        print('\n\nEntry add cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines(username + '\n' + password + '\n' + url + '\n' + notes + '\n\n')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines(username + '\n' + password + '\n' + url + '\n' + notes + '\n\n')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                f.writelines(username + '\n' + password + '\n' + url + '\n' + notes + '\n\n')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The folder you requested to access does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'add folder' or argument == 'add -f':
    newfolder = str(input('Name of new folder: '))
    term('mkdir ' + "'" + directory + newfolder + "'")

# edit

if argument == 'edit':
    print('\nUSAGE: rpass edit [FLAG]')
    print('Flags:')
    print('edit:')
    print(' rename/relocate/-r      rename or relocate an entry')
    print(' username/-u             change the username of an entry')
    print(' password/-p             change the password of an entry')
    print(' url/-l                  change the url attached to an entry')
    print(' note/-n                 change the note attached to an entry\n')

if argument == 'edit rename' or argument == 'edit relocate' or argument == 'edit -r':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('What file would you like to edit? '))
        newname = str(input('New Name: '))
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                move(directory + folder + '/' + folderentry + '.gpg', directory + '/' + newname + '.gpg')
            else:
                move(directory + entryname.replace('/', '', 1) + '.gpg',
                     directory + newname.replace('/', '', 1) + '.gpg')
        else:
            move(directory + entryname + '.gpg', directory + newname + '.gpg')
    except KeyboardInterrupt:
        print('\n\nEdit cancelled.\n')
        exit()
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        exit()

if argument == 'edit username' or argument == 'edit -u':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('What file would you like to edit? '))
        username = str(input('New Username: '))
    except KeyboardInterrupt:
        entryname, username = (0, 0)
        print('\n\nEdit cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     folder + '/' + folderentry + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 0, username + '\n')
                remove(directory + folder + '/' + folderentry + '.gpg')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     entryname.replace('/', '', 1) + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 0, username + '\n')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + entryname + '.gpg')
            replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 0, username + '\n')
            remove(directory + entryname + '.gpg')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'edit password' or argument == 'edit -p':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('What file would you like to edit? '))
        password = str(input('New Password: '))
    except KeyboardInterrupt:
        entryname, password = (0, 0)
        print('\n\nEdit cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     folder + '/' + folderentry + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, password + '\n')
                remove(directory + folder + '/' + folderentry + '.gpg')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     entryname.replace('/', '', 1) + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, password + '\n')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + entryname + '.gpg')
            replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, password + '\n')
            remove(directory + entryname + '.gpg')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'edit url' or argument == 'edit -l':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('What file would you like to edit? '))
        url = str(input('New URL: '))
    except KeyboardInterrupt:
        entryname, url = (0, 0)
        print('\n\nEdit cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     folder + '/' + folderentry + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 2, url + '\n')
                remove(directory + folder + '/' + folderentry + '.gpg')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     entryname.replace('/', '', 1) + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 2, url + '\n')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + entryname + '.gpg')
            replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 2, url + '\n')
            remove(directory + entryname + '.gpg')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'edit note' or argument == 'edit -n':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('What file would you like to edit? '))
        notes = str(input('New Note: '))
    except KeyboardInterrupt:
        entryname, notes = (0, 0)
        print('\n\nEdit cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     folder + '/' + folderentry + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 3, notes + '\n')
                remove(directory + folder + '/' + folderentry + '.gpg')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     entryname.replace('/', '', 1) + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 3, notes + '\n')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + entryname + '.gpg')
            replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 3, notes + '\n')
            remove(directory + entryname + '.gpg')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

# copy

if argument == 'copy':
    print('\nUSAGE: rpass copy [FLAG]')
    print('Flags:')
    print('copy:')
    print(' username/-u             copy the username of an entry to your clipboard')
    print(' password/-p             copy the password of an entry to your clipboard')
    print(' url/-l                  copy the URL of an entry to your clipboard')
    print(' note/-n                 copy the note of an entry to your clipboard\n')

if argument == 'copy username' or argument == 'copy -u':
    try:
        print()
        term('cd ' + directory + '; ls -1 *')
        print()
        readentry = str(input('Entry to copy: '))
    except KeyboardInterrupt:
        readentry = 0
        print('\n\nCopy cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + readentry + '.gpg')
        copyline = open('/dev/shm/' + shmfolder + '/' + shmentry, 'r').readlines()
        if environ.get('WAYLAND_DISPLAY') == 'wayland-0':
            term('wl-copy ' + "'" + copyline[0].replace('\n', '') + "'")
        else:
            term('echo -n ' + "'" + copyline[0].replace('\n', '') + "'" + ' | xclip -sel c')
        rmtree('/dev/shm/' + shmfolder)
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'copy password' or argument == 'copy -p':
    try:
        print()
        term('cd ' + directory + '; ls -1 *')
        print()
        readentry = str(input('Entry to copy: '))
    except KeyboardInterrupt:
        readentry = 0
        print('\n\nCopy cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + readentry + '.gpg')
        copyline = open('/dev/shm/' + shmfolder + '/' + shmentry, 'r').readlines()
        if environ.get('WAYLAND_DISPLAY') == 'wayland-0':
            term('wl-copy ' + "'" + copyline[1].replace('\n', '').replace("'", "'\\''") + "'")
        else:
            term('echo -n ' + "'" + copyline[1].replace('\n', '').replace("'", "'\\''") + "'" + ' | xclip -sel c')
        rmtree('/dev/shm/' + shmfolder)
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'copy url' or argument == 'copy -l':
    try:
        print()
        term('cd ' + directory + '; ls -1 *')
        print()
        readentry = str(input('Entry to copy: '))
    except KeyboardInterrupt:
        readentry = 0
        print('\n\nCopy cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + readentry + '.gpg')
        copyline = open('/dev/shm/' + shmfolder + '/' + shmentry, 'r').readlines()
        if environ.get('WAYLAND_DISPLAY') == 'wayland-0':
            term('wl-copy ' + "'" + copyline[2].replace('\n', '') + "'")
        else:
            term('echo -n ' + "'" + copyline[2].replace('\n', '') + "'" + ' | xclip -sel c')
        rmtree('/dev/shm/' + shmfolder)
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'copy note' or argument == 'copy -n':
    try:
        print()
        term('cd ' + directory + '; ls -1 *')
        print()
        readentry = str(input('Entry to copy: '))
    except KeyboardInterrupt:
        readentry = 0
        print('\n\nCopy cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + readentry + '.gpg')
        copyline = open('/dev/shm/' + shmfolder + '/' + shmentry, 'r').readlines()
        if environ.get('WAYLAND_DISPLAY') == 'wayland-0':
            term('wl-copy ' + "'" + copyline[3].replace('\n', '') + "'")
        else:
            term('echo -n ' + "'" + copyline[3].replace('\n', '') + "'" + ' | xclip -sel c')
        rmtree('/dev/shm/' + shmfolder)
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

# gen

if argument == 'gen':
    try:
        clear()
        entryname = str(input('Name of service: '))
        username = str(input('Username: '))
        passlength = int(input('How many characters should be in the password? '))
        passtype = str(input('Should the password be simple (for compatibility) or complex (for security)? (s/C) '))
        if passtype != 's':
            passtype = string.ascii_letters + string.digits + string.punctuation
        if passtype == 's':
            passtype = string.ascii_letters + string.digits
        passgen = ''.join(random.SystemRandom().choice(passtype) for _ in range(passlength))
        url = str(input('URL: '))
        notes = str(input('Additional notes: '))
    except KeyboardInterrupt:
        entryname, username, passgen, url, notes = (0, 0, 0, 0, 0)
        print('\n\nEntry generation cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines(username + '\n' + passgen + '\n' + url + '\n' + notes + '\n\n')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                    f.writelines(username + '\n' + passgen + '\n' + url + '\n' + notes + '\n\n')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            with open('/dev/shm/' + shmfolder + '/' + shmentry, 'w') as f:
                f.writelines(username + '\n' + passgen + '\n' + url + '\n' + notes + '\n\n')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The folder you requested to access does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

if argument == 'gen update' or argument == 'gen -u':
    try:
        clear()
        term('cd ' + directory + '; ls -1 *')
        print()
        entryname = str(input('Which password would you like to re-generate? '))
        passlength = int(input('How many characters should be in the password? '))
        passtype = str(input('Should the password be simple (for compatibility) or complex (for security)? (s/C) '))
    except KeyboardInterrupt:
        entryname, passlength, passtype = (0, 0, 0)
        print('\n\nEntry generation cancelled.\n')
        exit()
    shmfolder, shmentry = shmgen()
    try:
        if passtype != 's':
            passtype = string.ascii_letters + string.digits + string.punctuation
        if passtype == 's':
            passtype = string.ascii_letters + string.digits
        passgen = ''.join(random.SystemRandom().choice(passtype) for _ in range(passlength))
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                folder, sep, folderentry = entryname.replace('/', '', 1).partition('/')
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     folder + '/' + folderentry + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, passgen + '\n')
                remove(directory + folder + '/' + folderentry.replace('/', '', 1) + '.gpg')
                encryptfolder()
                term('gpg -d ' + directory + folder + '/' + "'" + folderentry + '.gpg' + "'")
            else:
                term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory +
                     entryname.replace('/', '', 1) + '.gpg')
                replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, passgen + '\n')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
                encrypt()
                term('gpg -d ' + directory + "'" + entryname.replace('/', '', 1) + '.gpg' + "'")
        else:
            term('gpg -d --output /dev/shm/' + shmfolder + '/' + shmentry + ' ' + directory + entryname + '.gpg')
            replace_line('/dev/shm/' + shmfolder + '/' + shmentry, 1, passgen + '\n')
            remove(directory + entryname + '.gpg')
            encrypt()
            term('gpg -d ' + directory + "'" + entryname + '.gpg' + "'")
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to edit does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        rmtree('/dev/shm/' + shmfolder)
        exit()

# remove

if argument == 'remove' or argument == '-rm':
    clear()
    term('cd ' + directory + '; ls -1 *')
    print()
    try:
        entryname = str(input('What would you like to delete? '))
        if entryname.startswith('/'):
            if entryname.replace('/', '', 1).__contains__('/'):
                if devicetype == 'c':
                    term('ssh -p ' + port + ' ' + usernamessh + '@' + ip + ' "' + ' rm -rf ' + "'" + directoryssh +
                         entryname.replace('/', '', 1) + '.gpg' + "'" + '"')
                remove(directory + entryname.replace('/', '', 1) + '.gpg')
            else:
                if devicetype == 'c':
                    term('ssh -p ' + port + ' ' + usernamessh + '@' + ip + ' "' + ' rm -rf ' + "'" + directoryssh +
                         entryname.replace('/', '', 1) + "'" + '"')
                rmtree(directory + entryname.replace('/', '', 1))
        else:
            if devicetype == 'c':
                term('ssh -p ' + port + ' ' + usernamessh + '@' + ip + ' "' + ' rm -rf ' + "'" + directoryssh +
                     entryname + '.gpg' + "'" + '"')
            remove(directory + entryname + '.gpg')
    except KeyboardInterrupt:
        print('\n\nDeletion cancelled.\n')
        exit()
    except FileNotFoundError:
        print('\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')
        print('The file you requested to delete does not exist!')
        print('!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n')
        exit()

if argument.__contains__('add') or argument.__contains__('-rm') or argument.__contains__('remove') or argument.\
        __contains__('edit') or argument.__contains__('gen') or argument.__contains__('version') or argument.\
        __contains__('-v') or argument.__contains__('config') or argument.__contains__('copy') or argument.\
        __contains__('show') or argument.__contains__('/') or argument == 'sync' or argument == '-s' or argument == '' \
        or argument == 'help' or argument == '--help' or argument == '-h':
    null = 'null'
else:
    print('\n!!!!!!!!!!!!!!!')
    print('Argument error!')
    print('!!!!!!!!!!!!!!!')
    print("\nUse 'help', '--help', or '-h' to see a list of commands.\n")

# permissions - applied pre-sync to ensure permissions are correct before upload

term('find ' + directory + ' -type d -exec chmod -R 700 {} +')
term('find ' + directory + ' -type f -exec chmod -R 600 {} +')

# rsync - ran at end of program to ensure all files are properly synced
if devicetype == 'c':
    if argument.__contains__('add') or argument.__contains__('-rm') or argument.__contains__('remove') or argument. \
            __contains__('edit') or argument.__contains__('gen') or argument == 'sync' or argument == '-s':
        print('Syncing entries with other device(s)...\n')
        term('rsync -v -H -r -l -t -p -e ' + '"ssh -i ~/.ssh/rpass -p ' + port + '" ' + directory + ' ' +
             usernamessh + '@' + ip + ':' + directoryssh)
        term('rsync -v -H -r -l -t -p --delete -e ' + '"ssh -i ~/.ssh/rpass -p ' + port + '" ' + usernamessh
             + '@' + ip + ':' + directoryssh + ' ' + directory)