64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import bcrypt
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
def get_number(prompt):
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
num = int(input(prompt))
|
||
|
|
return num
|
||
|
|
except ValueError:
|
||
|
|
print("Not a number!\n")
|
||
|
|
|
||
|
|
|
||
|
|
def get_boolean(prompt, default):
|
||
|
|
string = input(prompt).lower()
|
||
|
|
if string in ["y", "yes", "true", "t", "1"]:
|
||
|
|
return True
|
||
|
|
elif string in ["n", "no", "false", "f", "0"]:
|
||
|
|
return False
|
||
|
|
return default
|
||
|
|
|
||
|
|
|
||
|
|
config = {}
|
||
|
|
|
||
|
|
# Master password
|
||
|
|
print("""
|
||
|
|
Enter master password. Master password is the password you provide to your bus
|
||
|
|
factor. When the master password is entered, The Long Night will send messages
|
||
|
|
to you via all the configured mediums. If you do not respond within 2 days
|
||
|
|
(configurable), The Long Night will add the supplied key as an authorized key
|
||
|
|
for the user it is running as.
|
||
|
|
""")
|
||
|
|
password = input("> ").encode()
|
||
|
|
config["password"] = bcrypt.hashpw(password, bcrypt.gensalt()).decode()
|
||
|
|
|
||
|
|
print("\nEnter number of days to wait before adding the ssh key\n")
|
||
|
|
config["days"] = get_number("> ")
|
||
|
|
|
||
|
|
|
||
|
|
# SSH key
|
||
|
|
print("\nEnter ssh key to add\n")
|
||
|
|
config["ssh_key"] = input("> ")
|
||
|
|
|
||
|
|
print("\nConfiguring mediums for contact when the master password is entered.")
|
||
|
|
|
||
|
|
# Telegram
|
||
|
|
print("\nDo you want to enable Telegram messages?\n")
|
||
|
|
config["telegram"] = {}
|
||
|
|
tg_enable = get_boolean("Y/n ", True)
|
||
|
|
config["telegram"]["enabled"] = tg_enable
|
||
|
|
|
||
|
|
if tg_enable:
|
||
|
|
print("\nEnter bot token\n")
|
||
|
|
config["telegram"]["bot_token"] = input("> ")
|
||
|
|
print("\nEnter userid\n")
|
||
|
|
config["telegram"]["user_id"] = get_number("> ")
|
||
|
|
|
||
|
|
print("\nStoring config.")
|
||
|
|
with open('config.json', 'w') as f:
|
||
|
|
json.dump(config, f)
|
||
|
|
print("Stored config.")
|