keygen.py

import os
import sys
from base64 import b64encode
from base64 import b64decode
import json
import urllib.parse
from Crypto.Hash import SHA256
from Crypto.Random import get_random_bytes
from nacl.signing import SigningKey, VerifyKey
from nacl.public import PrivateKey, PublicKey
from nacl.encoding import Base64Encoder
import validators
import lxml.etree as etree

import uuid
import NessKeys.Prng as prng
from framework.Container import Container
from framework.ARGS import ARGS

from NessKeys.exceptions.UserExist import UserExist

class Keygen:

    def __is_integer(self, n):
        try:
            int(n)
        except ValueError:
            return False
        else:
            return True

    def __manual(self):
        print("*** PrivateNess KEY GENERATOR")
        print("### DESCRIPTION:")
        print("  Generates ciphers for NESS service nodes and NESS service node clients")
        print("  Works on ed25519 for keypairs")
        print("  Adjustable entropy when generating private keys")
        print("### USAGE:")
        print("#### Generate new human")
        print("  ./keygen human <humanname> <Entropy level>")
        print("  Example: $ ./keygen human Neo 5")
        print("#### Generate new user")
        print("  ./keygen user <username> <Entropy level>")
        print("  Example: $ ./keygen user user1 5")
        # print("#### Generate user and add userinfo to bith users (~/.privateness-keys/users.key.json) and user (user.key.json) key")
        # print("  ./keygen userkey <username> <Entropy level>")
        # print("  Example: $ ./keygen userkey user1 5")
        print("#### Generate node Key")
        print("  ./keygen node <Node name or URL> <Tariff> master-user-name \"coma,separated,services\" \"network\"  <Entropy level>")
        print("  Example: $ ./keygen node http://my-node.net 111 master \"prng,files\" inet 5")
        print("#### Generate Faucet Key")
        print("  ./keygen faucet <Faucet URL> <Entropy level>")
        print("  Example: $ ./keygen faucet http://www.faucet.net 5")
        print("#### Generate Backup Key")
        print("  ./keygen backup <Entropy level>")
        print("  Example: $ ./keygen backup 5")
        print("#### Generate seed (phrase)")
        print("  ./keygen seed <length> <Entropy level>")
        print("  Example: $ ./keygen seed 32 5")    
        print("#### Generate seed (words)")
        print("  ./keygen seed <length> <Entropy level>")
        print("  Example: $ ./keygen words 32 5")     
        print("#### Generate seed (bip39)")
        print("  ./keygen bip39 <length (12 or 24)> <Entropy level>")
        print("  Example: $ ./keygen bip39 12 5")    
        print("#### Show this manual")
        print("  $ ./keygen help")
        print("  $ ./keygen -h")

    def process(self):
        if ARGS.args(['user', str, str]):
            username = sys.argv[2]

            if self.__is_integer(ARGS.arg(3)):
                entropy = int(ARGS.arg(3))

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False

            manager = Container.KeyManager()

            try:
                return manager.createUsersKey(username, entropy)            
            except UserExist as e:
                print("User with username '{}' olready exists".format(username))
                print("See users list with $ ./user ls")

            exit(0)
        elif ARGS.args(['human', str, str]):
            humanname = sys.argv[2]

            if self.__is_integer(ARGS.arg(3)):
                entropy = int(ARGS.arg(3))

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False

            manager = Container.KeyManager()

            try:
                return manager.createHumansKey(humanname, entropy)            
            except UserExist as e:
                print("Human with humanname '{}' olready exists".format(humanname))
                print("See humans list with $ ./human ls")

            exit(0)

        elif ARGS.args(['node', str, str, str, str, str, str]):
            url = sys.argv[2]

            if self.__is_integer(sys.argv[3]):
                tariff = int(sys.argv[3])
            else:
                print("<Tariff> must be integer")
                return False

            master_user = sys.argv[4]
            services = sys.argv[5]
            network = sys.argv[6]

            if self.__is_integer(sys.argv[7]):
                entropy = int(sys.argv[7])

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False

            manager = Container.KeyManager()

            return manager.createNodeKey(url, tariff, master_user, services, network, entropy)

        elif ARGS.args(['faucet', str, str]):
            url = sys.argv[2]

            if self.__is_integer(sys.argv[3]):
                entropy = int(sys.argv[3])

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False

            manager = Container.KeyManager()

            return manager.createFaucetkey(url, entropy)

        elif ARGS.args(['words', int, int]):
            words_count = int(sys.argv[2])
            entropy = int(sys.argv[3])

            manager = Container.KeyManager()

            # manager.output = False
            words = manager.generate_word_seed(entropy, words_count)
            print(" ".join(words))
            manager.output = True

        elif ARGS.args(['bip39', int, int]):
            words_count = int(sys.argv[2])
            entropy = int(sys.argv[3])

            manager = Container.KeyManager()

            # manager.output = False
            words = manager.generate_bip39_seed(entropy, words_count)
            print(" ".join(words))
            manager.output = True

        elif ARGS.args(['backup', str]):
            if self.__is_integer(sys.argv[2]):
                entropy = int(sys.argv[2])

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False

            manager = Container.KeyManager()

            bkey = manager.createBackupKey('node', '', entropy)

            print("Write down a seed to paper or password manager:")
            print(bkey.getSeed())

            return True

        elif len(sys.argv) == 4 and sys.argv[1].lower() == 'seed':

            if self.__is_integer(sys.argv[2]):
                length = int(sys.argv[2])

                if length < 16:
                    length = 16
            else:
                print("<length> must be integer")
                return False

            if self.__is_integer(sys.argv[3]):
                entropy = int(sys.argv[3])

                if entropy < 1:
                    entropy = 1
            else:
                print("<Entropy level> must be integer")
                return False
            
            generator = prng.UhePrng()

            for i in range (1, entropy):
                rand = ''
                with open('/dev/random', 'rb') as file:
                    rand = b64encode(file.read(1024)).decode('utf-8')
                    file.close()

                generator.add_entropy(rand, str(uuid.getnode()))

                print('+', end = " ", flush = True)

            print("")
            print (generator.string(length))

        else:
            self.__manual()

keygen = Keygen()
keygen.process()