docker-public-node/starter.py at master · SignRequest/docker-public-node · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import os
import os.path
import urllib.request
import sys
import requests
from pyhocon import ConfigFactory, HOCONConverter
import pywaves as pw
import base58
import string
import random
import math
from tqdm import tqdm
from shutil import copyfile
from pyblake2 import blake2b
from hashlib import sha256

network_names = ['MAINNET', 'TESTNET']

NETWORK = os.environ.get('LTO_NETWORK')


def generate_password(size=12, chars=string.ascii_letters + string.digits):
    return ''.join(random.choice(chars) for i in range(size))


def nested_set(dic, keys, value):
    for key in keys[:-1]:
        dic = dic.setdefault(key, {})
    dic[keys[-1]] = value


def create_configs_dir():
    if not os.path.isdir("/lto/configs"):
        os.mkdir("/lto/configs")
    if not os.path.isdir("/lto/data"):
        os.mkdir("/lto/data")


def parse_env_variables():
    dictionary = dict()
    for env_key in os.environ:
        if "__" in env_key:
            parts = env_key.split('__')
            keys = [x.lower().replace('_', '-') for x in parts]
            nested_set(dictionary, keys, os.environ[env_key])
    return dictionary


def get_wallet_data():
    seed = os.environ.get('LTO_WALLET_SEED', pw.Address().seed)
    seed_base58 = os.environ.get('LTO_WALLET_SEED_BASE58')
    base58_provided = False
    if seed_base58 is not None:
        try:
            base58.b58decode(seed_base58.encode())
            base58_provided = True
        except:
            seed_base58 = base58.b58encode(seed.encode())
    else:
        seed_base58 = base58.b58encode(seed.encode())
    password = os.environ.get('LTO_WALLET_PASSWORD', generate_password())
    if base58_provided is False:
        print('Seed phrase:', seed)
    print('Wallet password:', password)
    return seed_base58, password

def secureHash(message):
    h = blake2b(digest_size=32)
    h.update(message.encode())
    return base58.b58encode(sha256(h.digest()).digest())

if __name__ == "__main__":
    if NETWORK is None or NETWORK not in network_names:
        NETWORK = 'TESTNET'

    create_configs_dir()

    if NETWORK == 'MAINNET':
        copyfile('/lto-node/lto-mainnet.conf', '/lto/configs/lto-config.conf')
    elif NETWORK == 'TESTNET':
        copyfile('/lto-node/lto-testnet.conf', '/lto/configs/lto-config.conf')

    if os.path.isfile('/lto/configs/local.conf'):
        print('config file already created')
        sys.exit(0)

    api_key = os.environ.get('LTO_API_KEY', 'lt1secretapikey!')
    api_key_hash = secureHash(api_key)

    env_dict = parse_env_variables()
    lto_data = get_wallet_data()

    nested_set(env_dict, ['lto', 'directory'], '/lto')
    nested_set(env_dict, ['lto', 'data-directory'], '/lto/data')
    nested_set(env_dict, ['lto', 'wallet', 'seed'], lto_data[0])
    nested_set(env_dict, ['lto', 'wallet', 'password'], lto_data[1])
    nested_set(env_dict, ['lto', 'rest-api', 'api-key-hash'], api_key_hash)

    LTO_NODE_NAME = os.getenv('LTO_NODE_NAME')
    if LTO_NODE_NAME is not None:
        nested_set(env_dict, ['lto', 'network', 'node-name'], LTO_NODE_NAME)

    LTO_DECLARED_ADDRESS = os.getenv('LTO_DECLARED_ADDRESS')
    if LTO_DECLARED_ADDRESS is not None:
        nested_set(env_dict, ['lto', 'network', 'declared-address'], LTO_DECLARED_ADDRESS)

    config = ConfigFactory.from_dict(env_dict)
    local_conf = HOCONConverter.convert(config, 'hocon')
    with open('/lto/configs/local.conf', 'w') as file:
        file.write(local_conf)