#!/usr/bin/env python3
from curses import A_REVERSE, KEY_DOWN, KEY_UP, cbreak, curs_set, endwin, initscr, newwin, noecho, nocbreak
from curses.textpad import rectangle, Textbox
from os import environ, listdir, remove, symlink
from os.path import exists, expanduser, isfile
from pathlib import Path
from random import randint
from re import sub
from shutil import get_terminal_size, which
from sshync import make_profile
from sshyp import copy_id_check, string_gen
from subprocess import PIPE, run
# PORT START UNAME-IMPORT
from os import uname
# PORT END UNAME-IMPORT
home, stdscr, sshyp_data = expanduser("~"), initscr(), []


# creates a radio selection between the provided options
def curses_radio(_options, _pretext):
    curs_set(0)
    _selected = 0
    while True:
        stdscr.clear()
        stdscr.addstr(0, 0, _pretext)
        for _i, _option in enumerate(_options):
            _y = _i + 2
            if _i == _selected:
                stdscr.addstr(_y, 0, "[*] " + _option, A_REVERSE)
            else:
                stdscr.addstr(_y, 0, "[ ] " + _option)
        stdscr.refresh()
        _key = stdscr.getch()
        # update _selected based on user input
        if _key == KEY_UP:
            _selected = (_selected - 1) % len(_options)
        elif _key == KEY_DOWN:
            _selected = (_selected + 1) % len(_options)
        elif _key == ord('\n'):
            break
        stdscr.refresh()
    curs_set(1)
    return _selected


# creates a text-box input
def curses_text(_pretext):
    stdscr.clear()
    stdscr.addstr(0, 0, _pretext)
    _term_columns = get_terminal_size()[0]
    _editwin = newwin(1, _term_columns - 2, 3, 1)
    rectangle(stdscr, 2, 0, 4, _term_columns - 1)
    stdscr.refresh()
    _box = Textbox(_editwin)
    # let the user edit until ctrl+g/enter is struck
    _box.edit()
    # return resulting contents
    return _box.gather().strip()


# cleanly exit curses
def curses_terminate():
    nocbreak()
    endwin()


# device+sync type selection
def install_type():
    _offline_mode = False
    # PORT START TWEAK-DEVTYPE
    _install_type = curses_radio(('server', 'client (ssh-synchronized)', 'client (offline)'),
                                 'device + sync type configuration')
    if _install_type == 0:
        _dev_type = 'server'
        Path(f"{home}/.config/sshyp/deleted").mkdir(mode=0o700, exist_ok=True)
        Path(f"{home}/.config/sshyp/whitelist").mkdir(mode=0o700, exist_ok=True)
        curses_terminate()
        print(f"\nmake sure the ssh service is running and properly configured")
    else:
        _dev_type = 'client'
        if _install_type == 2:
            _offline_mode = True
            if isfile(f"{home}/.config/sshyp/sshyp.sshync"):
                remove(f"{home}/.config/sshyp/sshyp.sshync")
        Path(f"{home}/.local/share/sshyp").mkdir(mode=0o700, parents=True, exist_ok=True)
    # PORT END TWEAK-DEVTYPE
    return _dev_type, _offline_mode


# gpg configuration
def gpg_config():
    # gpg key selection
    _uid_list = [_item for _item in run(['gpg', '-k', '--with-colons'],
                                        stdout=PIPE, text=True).stdout.splitlines() if _item.startswith('uid')]
    _clean_uid_list = []
    for _uid in _uid_list:
        _clean_uid_list.append(sub(r':+', ':', _uid).split(':')[4])
    _clean_uid_list.append('auto-generate')
    _gpg_id_sel = curses_radio(_clean_uid_list, 'gpg key selection')
    _gpg_id = _clean_uid_list[_gpg_id_sel]
    if _gpg_id == 'auto-generate':
        print('\na unique gpg key is being generated for you...')
        if not isfile(f"{home}/.config/sshyp/gpg-gen"):
            open(f"{home}/.config/sshyp/gpg-gen", 'w').writelines([
                'Key-Type: 1\n', 'Key-Length: 4096\n', 'Key-Usage: sign encrypt\n', 'Name-Real: sshyp\n',
                'Name-Comment: gpg-sshyp\n', 'Name-Email: https://github.com/rwinkhart/sshyp\n',
                'Expire-Date: 0'])
        run(['gpg', '--batch', '--generate-key', f"{home}/.config/sshyp/gpg-gen"])
        remove(f"{home}/.config/sshyp/gpg-gen")
        _gpg_id = run(['gpg', '-k'], stdout=PIPE, text=True).stdout.splitlines()[-3].strip()
    return _gpg_id


# text editor configuration
def editor_config():
    _editor = curses_text('enter the name of your preferred text editor:\n\n\n\n\n'
                          '(ctrl+g/enter to confirm)\n\nexample input: vim')
    return _editor


# ssh+sshync configuration
def ssh_config():
    # ssh+sshync configuration
    _uiport = curses_text('enter the username, ip, and ssh port of your sshyp server:\n\n\n\n\n('
                          'ctrl+g/enter to confirm)\n\nexample inputs:\n\n ipv4: user@10.10.10.'
                          '10:22\n ipv6: user@[2000:2000:2000:2000:2000:2000:2000:2000]:22\n '
                          'domain: user@mydomain.com:22').lstrip('[').replace(']', '')
    _uiport_split = _uiport.split('@')
    _username_ssh = _uiport_split[0]
    _iport = _uiport_split[1].rsplit(':', 1)

    # sshync profile generation
    make_profile(f"{home}/.config/sshyp/sshyp.sshync",
                 f"{home}/.local/share/sshyp/", f"/home/{_username_ssh}/.local/share/sshyp/",
                 f"{home}/.ssh/sshyp", _iport[0], _iport[1], _username_ssh)
    return _iport[1], _username_ssh, _iport[0]


# device id configuration
def dev_id_config(_ip, _username_ssh, _port):
    for _id in listdir(f"{home}/.config/sshyp/devices"):
        remove(f"{home}/.config/sshyp/devices/{_id}")
    _device_id_prefix = curses_text('name this device:\n\n\n\n\n(ctrl+g/enter to confirm)\n\n'
                                    'important:\u001b[0m this id \u001b[4;1mmust\u001b[0m be '
                                    'unique amongst your client devices\n\nthis is used to keep '
                                    'track of database syncing and quick-unlock permissions\n')
    _device_id_suffix = string_gen('f', randint(24, 48))
    _device_id = _device_id_prefix + '-' + _device_id_suffix
    open(f"{home}/.config/sshyp/devices/{_device_id}", 'w')
    # test server connection and attempt to register device id
    copy_id_check(_ip, _username_ssh, _port, _device_id)
    return _device_id


# quick-unlock configuration
def quick_unlock_config():
    _quick_unlock_sel = curses_radio(('yes', 'no'), 'enable quick-unlock?')
    if _quick_unlock_sel == 0:
        _enabled = 'yes'
    else:
        _enabled = 'no'
    return _enabled


# runs secondary configuration menu - clients only
def global_menu(_post_setup):
    # curses initialization
    noecho()
    cbreak()
    stdscr.keypad(True)
    _options, _choice = [], 4

    try:
        if not _post_setup:
            _options.extend(['change device/synchronization types', 'change gpg key', 're-configure ssh',
                             'change device name'])
            _message, _choice = 'all configuration options:', 0
        _options.extend(['[OPTIONAL, RECOMMENDED] set custom text editor',
                         '[OPTIONAL, RECOMMENDED] enable/disable quick-unlock',
                         '[OPTIONAL, NOT IMPLEMENTED] su security mode',
                         '[OPTIONAL, NOT IMPLEMENTED] extensions and updates', 'exit/done'])
        _message = 'required configuration complete - additional configuration options:'
        _choice += curses_radio(_options, _message)

        if _choice == 0:
            install_type()
        elif _choice == 1:
            gpg_config()
        elif _choice == 2:
            ssh_config()
        elif _choice == 3:
            dev_id_config()
        elif _choice == 4:
            editor_config()
        elif _choice == 5:
            quick_unlock_config()
        else:
            pass
        curses_terminate()
    except KeyboardInterrupt:
        curses_terminate()


# runs initial configuration wizard
def initial_setup():
    # config directory creation
    Path(f"{home}/.config/sshyp/devices").mkdir(mode=0o700, parents=True, exist_ok=True)

    # temporary file symlink creation
    if not exists(f"{home}/.config/sshyp/tmp"):
        # PORT START UNAME-TMP
        if uname()[0] in ('Haiku', 'FreeBSD', 'Darwin'):
            symlink('/tmp', f"{home}/.config/sshyp/tmp")
        elif exists('/data/data/com.termux'):
            symlink('/data/data/com.termux/files/usr/tmp', f"{home}/.config/sshyp/tmp")
        else:
            symlink('/dev/shm', f"{home}/.config/sshyp/tmp")
        # PORT END UNAME-TMP

    # curses initialization
    cbreak()
    stdscr.keypad(True)

    # curses menu tree
    try:
        # device+sync type selection
        _dev_sync_types = install_type()
        sshyp_data.append(_dev_sync_types[0])

        if _dev_sync_types[0] == 'client':

            # gpg configuration
            sshyp_data.append(gpg_config())

            # lock file generation (requires gpg configuration)
            if isfile(f"{home}/.config/sshyp/lock.gpg"):
                remove(f"{home}/.config/sshyp/lock.gpg")
                open(f"{home}/.config/sshyp/lock", 'w')
                run(['gpg', '-qr', str(sshyp_data[1]), '-e', f"{home}/.config/sshyp/lock"])
                remove(f"{home}/.config/sshyp/lock")

            # set default text editor to value of EDITOR environment variable, otherwise default to nano
            if 'EDITOR' in environ:
                sshyp_data.append(environ['EDITOR'])
            else:
                sshyp_data.append('nano')

            # online (synchronized mode) configuration
            if not _dev_sync_types[1]:

                # ssh+sshync configuration
                _ip, _username_ssh, _port = ssh_config()

                # device id configuration
                _device_id = dev_id_config(_ip, _username_ssh, _port)

                # cleanly exit curses
                curses_terminate()

            else:
                # cleanly exit curses
                curses_terminate()

            # PORT START CLIPTOOL
            # check for clipboard tool and display warning if missing
            if uname()[0] in ('Linux', 'FreeBSD'):
                if 'WAYLAND_DISPLAY' in environ:
                    _display_server, _clipboard_tool, _clipboard_package = 'Wayland', 'wl-copy', 'wl-clipboard'
                else:
                    _display_server, _clipboard_tool, _clipboard_package = 'X11', 'xclip', 'xclip'
                if which(_clipboard_tool) is None:
                    print(f'\n\u001b[38;5;9mwarning: you are using {_display_server} and "{_clipboard_tool}" is not '
                          f'present - \ncopying entry fields will not function until '
                          f'"{_clipboard_package}" is installed\u001b[0m')
            # PORT END CLIPTOOL

        # write main config file (sshyp-data)
        with open(f"{home}/.config/sshyp/sshyp-data", 'w') as _config_file:
            _lines = 0
            for _item in sshyp_data:
                _lines += 1
                _config_file.write(str(_item) + '\n')
            while _lines < 4:
                _lines += 1
                _config_file.write('n')

        global_menu(True)

    except KeyboardInterrupt:
        # cleanly exit curses
        curses_terminate()