the-long-night/installation.py

101 lines
2.9 KiB
Python
Executable File

#!/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("\nAttempting to send a test message")
from backends import telegram
telegram.send(config["telegram"], "Test Message from The Long Night")
print("""
Message sent, if you did not recieve a message, check the bot_token and
user_id. Also, ensure that you have started the bot.
""")
# Matrix
print("\nDo you want to enable Matrix messages?\n")
config["matrix"] = {}
matrix_enable = get_boolean("y/N ", False)
config["matrix"]["enabled"] = matrix_enable
if matrix_enable:
print("\nEnter matrix homeserver address\n")
config["matrix"]["server"] = input("> ")
print("""
Use token for authorization? (Reply with no if you want to use
username/password)
""")
if get_boolean("y/N ", False):
print("\nEnter matrix token\n")
config["matrix"]["token"] = input("> ")
else:
print("\nEnter matrix username\n")
config["matrix"]["username"] = input("> ")
print("\nEnter matrix password\n")
config["matrix"]["password"] = input("> ")
print("\nEnter Room ID\n")
config["matrix"]["room_id"] = input("> ")
print("\nAttempting to send a test message")
from backends import matrix
matrix.send(config["matrix"], "Test Message from The Long Night")
print("Message sent, if you did not recieve a message, check the " +
"credentials")
print("\nStoring config.")
with open('config.json', 'w') as f:
json.dump(config, f)
print("Stored config.")