main_node.py

import os
import sys
import time  # for listener
import platform
import threading
import subprocess
from prompt_toolkit import PromptSession  # pip install prompt_toolkit
from node import ssh_node
from admin import ssh_admin
from peer import ssh_peer
# for housekeeping, separate functions into different files when complete
from DKAPinit import initCheck


def main():
    session = PromptSession()

    print("\nWelcome to the Distributed Key Authority Project.")
    print("Type \'help\' for available commands. Press up for previous commands.\n")

    initCheck()  # check that folders and files required are present

    while True:
        print("> ", end="")
        input = session.prompt()
        print(input[:-1])
        if input == "help":
            print("HELP - COMMANDS AVAILABLE FOR DKAP NODE:")
            print("gen - Generating keypair for this machine. Save private key (ssh-add), commit public key to blockchain.")
            print("del - Delete keypair for this machine. Delete private key (ssh-add -d), commit public key deletion to blockchain.")
        elif input == "exit":
            sys.exit(0)
        elif input == "gen":
            print("Generating keypair for this machine. Save private key (ssh-add), commit public key to blockchain. Maybe on first run only.")
            ssh_node.genKeyPair()
        elif input == "del":
            print(
                "Delete keypair for this machine. Delete private key (ssh-add -d), commit public key deletion to blockchain.")
            ssh_node.delKeyPair()
        else:
            print("Unrecognised command. Type \'help\' for available commands. Press up for previous commands.")


if __name__ == "__main__":
    file_exists = os.path.isfile(".\\node.txt")
    if file_exists:
        node_file = open('node.txt', 'r')
        node_lines = node_file.readlines()
        main_account = node_lines[0].strip()
        private_key = node_lines[1].strip()
        ssh_node = ssh_node(main_account, private_key)
        if platform.system() != "Windows":
            print("DKAP runs on Windows systems only, for now. This application will now exit.")
            sys.exit(1)

        main()
    else:
        print("Please configure your node.txt")