148 lines
4.6 KiB
Python
Executable File
148 lines
4.6 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
|
|
|
|
|
|
def get_choice(choices, prompt, error="Invalid Choice", *,
|
|
lowercase_input=False):
|
|
while True:
|
|
inp = input(prompt)
|
|
if lowercase_input:
|
|
inp = inp.lower()
|
|
if inp in choices:
|
|
return inp
|
|
print(error)
|
|
|
|
|
|
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 user ID. (Send /info to @quadnite_bot to find out)\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")
|
|
|
|
# Email
|
|
print("\nDo you want to enable email messages?\n")
|
|
config["email"] = {}
|
|
email_enable = get_boolean("y/N ", False)
|
|
config["email"]["enabled"] = email_enable
|
|
|
|
if email_enable:
|
|
print("\nEnter SMTP root URL\n")
|
|
config["email"]["url"] = input("> ")
|
|
print("\nEnter port\n")
|
|
config["email"]["port"] = get_number("> ")
|
|
print("\nDo you want to enable SSL/TLS/None\n")
|
|
enc = get_choice(["ssl", "tls", "none"], "> ", lowercase_input=True)
|
|
if enc == "ssl":
|
|
config["email"]["ssl"] = True
|
|
config["email"]["tls"] = False
|
|
elif enc == "tls":
|
|
config["email"]["ssl"] = False
|
|
config["email"]["tls"] = True
|
|
else:
|
|
config["email"]["ssl"] = False
|
|
config["email"]["tls"] = False
|
|
print("\nEnter login username\n")
|
|
config["email"]["login"] = input("> ")
|
|
print("\nEnter login password\n")
|
|
config["email"]["password"] = input("> ")
|
|
print("\nEnter from email\n")
|
|
config["email"]["from"] = input("> ")
|
|
print("\nEnter comma seperated list of emails to send the email to\n")
|
|
config["email"]["tos"] = input("> ")
|
|
from backends import mail
|
|
print("\nAttempting to send a test message")
|
|
mail.send(config["email"], "Test Message from The Long Night")
|
|
print("Message sent, if you did not recieve a message, check the " +
|
|
"config")
|
|
|
|
|
|
print("\nStoring config.")
|
|
with open('config.json', 'w') as f:
|
|
json.dump(config, f)
|
|
print("Stored config.")
|