forked from forks/microblog.pub
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
"""Basic wizard for setting up microblog.pub configuration files."""
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import bcrypt
|
|
import tomli_w
|
|
from prompt_toolkit import prompt
|
|
from prompt_toolkit.key_binding import KeyBindings
|
|
|
|
from app.key import generate_key
|
|
|
|
_ROOT_DIR = Path().parent.resolve()
|
|
_KEY_PATH = _ROOT_DIR / "data" / "key.pem"
|
|
|
|
|
|
_kb = KeyBindings()
|
|
|
|
|
|
@_kb.add("c-@")
|
|
def _(event):
|
|
"""Save multi-line buffer on CTRL + space"""
|
|
event.current_buffer.validate_and_handle()
|
|
|
|
|
|
def main() -> None:
|
|
print("Welcome to microblog.pub setup wizard\n")
|
|
print("Generating key...")
|
|
if _KEY_PATH.exists():
|
|
yn = ""
|
|
while yn not in ["y", "n"]:
|
|
yn = prompt(
|
|
"WARNING, a key already exists, overwrite it? (y/n): ", default="n"
|
|
).lower()
|
|
if yn == "y":
|
|
generate_key(_KEY_PATH)
|
|
else:
|
|
generate_key(_KEY_PATH)
|
|
|
|
config_file = Path("data/profile.toml")
|
|
|
|
if config_file.exists():
|
|
# Spit out the relative path for the "config artifacts"
|
|
rconfig_file = "data/profile.toml"
|
|
print(
|
|
f"Existing setup detected, please delete {rconfig_file} "
|
|
"before restarting the wizard"
|
|
)
|
|
sys.exit(2)
|
|
|
|
dat: dict[str, Any] = {}
|
|
print("Your identity will be @{username}@{domain}")
|
|
dat["domain"] = prompt("domain: ")
|
|
dat["username"] = prompt("username: ")
|
|
dat["admin_password"] = bcrypt.hashpw(
|
|
prompt("admin password: ", is_password=True).encode(), bcrypt.gensalt()
|
|
).decode()
|
|
dat["name"] = prompt("name (e.g. John Doe): ", default=dat["username"])
|
|
dat["summary"] = prompt(
|
|
(
|
|
"summary (short description, in markdown, "
|
|
"press [CTRL] + [SPACE] to submit):\n"
|
|
),
|
|
key_bindings=_kb,
|
|
multiline=True,
|
|
)
|
|
dat["https"] = True
|
|
proto = "https"
|
|
yn = ""
|
|
while yn not in ["y", "n"]:
|
|
yn = prompt("will the site be served via https? (y/n): ", default="y").lower()
|
|
if yn == "n":
|
|
dat["https"] = False
|
|
proto = "http"
|
|
|
|
dat["id"] = f'{proto}://{dat["domain"]}'
|
|
|
|
print("Note that you can put your icon/avatar in the static/ directory")
|
|
dat["icon_url"] = prompt(
|
|
"icon URL: ", default=f'{proto}://{dat["domain"]}/static/nopic.png'
|
|
)
|
|
dat["secret"] = os.urandom(16).hex()
|
|
|
|
with config_file.open("w") as f:
|
|
f.write(tomli_w.dumps(dat))
|
|
|
|
print("Done")
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print("Aborted")
|
|
sys.exit(1)
|