ssh-reg/public/userapplication.py

#!/usr/bin/env python3

import argparse
import configparser
import logging
import re
import sqlite3
from os import environ
from os import getcwd
from os import path as ospath

try:
    cwd = environ.get('TILDE_CONF')
    if cwd is None:
        cwd = getcwd()+"/applicationsconfig.ini"
    else:
        if ospath.isfile(cwd) is False:
            cwd = getcwd() + "/applicationsconfig.ini"
    # cwd is now either cwd/applicationsconfig or $TILDE_CONF
    argparser = argparse.ArgumentParser(description='interactive registration formular for tilde platforms')
    argparser.add_argument('-c', '--config', default=cwd, type=str, help='Config file', required=False)
    args = argparser.parse_args()
    CONF_FILE = args.config
    config = configparser.ConfigParser()
    config.read(CONF_FILE)
    logging.basicConfig(format="%(asctime)s: %(message)s", filename=config['DEFAULT']['log_file'],
                        level=int(config['LOG_LEVEL']['log_level']))
    REG_FILE = config['DEFAULT']['applications_db']
except:
    # intended broad, @TODO check them all for errors instead of everything in one
    logging.exception("logging or configparser-Exception: ")
    exit(0)

VALID_SSH = False
VALID_USER = False


def __createTable(cursor, connection):
    try:
        cursor.execute(
            "CREATE TABLE IF NOT EXISTS applications("
            "id INTEGER PRIMARY KEY AUTOINCREMENT,"
            "username TEXT NOT NULL, email TEXT NOT NULL,"
            "name TEXT NOT NULL, pubkey TEXT NOT NULL,"
            "timestamp DATETIME DEFAULT CURRENT_TIMESTAMP CONSTRAINT "
            "timestamp_valid CHECK( timestamp IS strftime('%Y-%m-%d %H:%M:%S', timestamp))"
            ",status INTEGER NOT NULL DEFAULT 0);")
        connection.commit()
    except sqlite3.Error as e:
        logging.exception("Couldn't create needed SQLite Table! Exception: %s" % e)


def addtotable(cursor, connection, username, name, email, pubkey):
    try:
        cursor.execute("INSERT INTO 'applications'(username, name, email, pubkey)VALUES("
                       "?,?,?,?)", [username, name, email, pubkey])
        connection.commit()
    except sqlite3.Error as e:
        logging.exception("Couldn't insert user into the db: %s" % e)


# check if sqlite file does exists or already and has our structure
def __checkSQLite(cursor, connection):
    # SELECT name FROM sqlite_master WHERE type='table' AND name='{table_name}';
    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='applications'")
    res = cursor.fetchall()

    if not res:
        try:
            __createTable(cursor, connection)
        except sqlite3.Error as e:
            logging.exception("couldn't create table on given database. Exception: %s" % e)
    else:
        pass
    return True


def check_username(value):
    global VALID_USER
    if len(value) < 3:
        VALID_USER = False
        return False
    if len(value) > 16:
        VALID_USER = False
        return False
    try:
        from pwd import getpwnam
        getpwnam(value)
        VALID_USER = False
    # everything from pwd throws an KeyError when the given user cannot be found
    except KeyError:
        VALID_USER = True
        return True
    return False


# taken from https://github.com/hashbang/provisor/blob/master/provisor/utils.py, all belongs to them! ;)
def validate_pubkey(value):
    global VALID_SSH
    import base64
    if len(value) > 8192 or len(value) < 80:
        VALID_SSH = False
        return False

    value = value.replace("\"", "").replace("'", "").replace("\\\"", "")
    value = value.split(' ')
    types = ['ecdsa-sha2-nistp256', 'ecdsa-sha2-nistp384',
             'ecdsa-sha2-nistp521', 'ssh-rsa', 'ssh-dss', 'ssh-ed25519']
    if value[0] not in types:
        VALID_SSH = False
        return False

    try:
        base64.decodebytes(bytes(value[1], "utf-8"))
    except TypeError:
        VALID_SSH = False
        return False

    VALID_SSH = True
    return True


def main():
    print("    ▗▀▖      \n▗▖▖ ▐  ▌ ▌▛▀▖\n▘▝▗▖▜▀ ▌ ▌▌ ▌\n  ▝▘▐  ▝▀▘▘ ▘")

    username = input("Welcome to the ~.fun user application form!\n\n"
                     "What is your desired username? [a-z0-9] allowed:\n")
    while (not re.match("[a-z]+[a-z0-9]", username)) or (not check_username(username)):
        username = input("Invalid Username, maybe it exists already?\nValid characters are only a-z and 0-9."
                         "\nMake sure your username starts with a character and not a number"
                         " and is not larger than 16 characters."
                         "\nWhat is your desired username? [a-z0-9] allowed:\n")

    fullname = input("\nPlease enter your full name:\n")
    while not re.match("\w+\s*\w*", username):
        fullname = input("\nThat is not your real name.\nPlease enter your full name:\n")

    email = input("\nPlease enter your email address:\n")
    while not re.match("(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)", email):
        email = input("\nThat is not a valid mail address.\nPlease enter your email address:\n")

    pubkey = input("\nPlease paste your ssh public key:\n")
    while (not re.match("ssh-(\w)+\s(\w+)(\s*)([a-zA-Z0-9@]*)", pubkey)) or (not validate_pubkey(pubkey)):
        pubkey = input("\nPlease enter a valid public key. You can show it with ssh-keygen -f .ssh/id_rsa -y on your "
                       "local machine.\nPlease enter your pubkey:\n")
        validate_pubkey(pubkey)

    print("\nUsername: {0!s}".format(username))
    print("Full Name: {0!s}".format(fullname))
    print("Email: {0!s}".format(email))
    print("Public {0!s}".format(pubkey))

    validation = input("\nIs this information correct? [y/N]")
    while not re.match("[yYnN\n]", validation):
        print("Please answer y for yes or n for no")
        validation = input("Is this information correct? [y/N]")
    if re.match("[yY]", validation):
        print("Thank you for your application! We'll get in touch shortly. 🐧")
        try:
            connection = sqlite3.connect(REG_FILE)
            cursor = connection.cursor()
            __checkSQLite(cursor, connection)
            addtotable(cursor, connection, username, fullname, email, pubkey)
            connection.commit()
            connection.close()
        except sqlite3.Error as e:
            logging.exception("Database {0!s} couldnt be accessed or created. Exception: {0!s}".
                              format(config['DEFAULT']['applications_db'], e))
            if connection:
                connection.close()
            exit(1)

    return 0


if __name__ == "__main__":
    try:
        main()
        exit(0)
    except KeyboardInterrupt:
        pass