125 lines
3.1 KiB
Python
125 lines
3.1 KiB
Python
import io
|
|
import logging
|
|
import os
|
|
from pathlib import Path
|
|
from textwrap import dedent
|
|
from typing import Optional
|
|
|
|
import typer
|
|
from dotenv import load_dotenv
|
|
from rich import print
|
|
from rich.logging import RichHandler
|
|
|
|
from ttfrog.path import assets
|
|
|
|
default_data_path = Path("~/.dnd/ttfrog")
|
|
default_host = "127.0.0.1"
|
|
default_port = 2323
|
|
|
|
SETUP_HELP = f"""
|
|
# Please make sure you set the SECRET_KEY in your environment. By default,
|
|
# TableTop Frog will attempt to load these variables from:
|
|
# {default_data_path}/defaults
|
|
#
|
|
# which may contain the following variables as well.
|
|
#
|
|
# See also the --root paramter.
|
|
|
|
DATA_PATH={default_data_path}
|
|
|
|
# Uncomment one or both of these to replace the packaged static assets and templates:
|
|
#
|
|
# STATIC_FILES_PATH={assets()}/public
|
|
# TEMPLATES_PATH={assets()}/templates
|
|
|
|
HOST={default_host}
|
|
PORT={default_port}
|
|
"""
|
|
|
|
db_app = typer.Typer()
|
|
app = typer.Typer()
|
|
app.add_typer(db_app, name="db", help="Manage the database.")
|
|
app_state = dict()
|
|
|
|
|
|
@app.callback()
|
|
@db_app.callback()
|
|
def main(
|
|
context: typer.Context,
|
|
root: Optional[Path] = typer.Option(
|
|
default_data_path,
|
|
help="Path to the TableTop Frog environment",
|
|
),
|
|
):
|
|
app_state["env"] = root.expanduser() / Path("defaults")
|
|
load_dotenv(stream=io.StringIO(SETUP_HELP))
|
|
load_dotenv(app_state["env"])
|
|
debug = os.getenv("DEBUG", None)
|
|
logging.basicConfig(
|
|
format="%(message)s",
|
|
level=logging.DEBUG if debug else logging.INFO,
|
|
handlers=[RichHandler(rich_tracebacks=True, tracebacks_suppress=[typer])],
|
|
)
|
|
|
|
|
|
@app.command()
|
|
def serve(
|
|
context: typer.Context,
|
|
host: str = typer.Argument(
|
|
default_host,
|
|
help="bind address",
|
|
),
|
|
port: int = typer.Argument(
|
|
default_port,
|
|
help="bind port",
|
|
),
|
|
debug: bool = typer.Option(False, help="Enable debugging output"),
|
|
):
|
|
"""
|
|
Start the TableTop Frog server.
|
|
"""
|
|
|
|
# delay loading the app until we have configured our environment
|
|
from ttfrog.db.bootstrap import loader
|
|
from ttfrog.webserver import application
|
|
|
|
print("Starting TableTop Frog server...")
|
|
loader.load()
|
|
application.start(host=host, port=port, debug=debug)
|
|
|
|
|
|
@db_app.command()
|
|
def setup(context: typer.Context):
|
|
"""
|
|
(Re)Initialize TableTop Frog. Idempotent; will preserve any existing configuration.
|
|
"""
|
|
from ttfrog.db.bootstrap import loader
|
|
|
|
if not os.path.exists(app_state["env"]):
|
|
app_state["env"].parent.mkdir(parents=True, exist_ok=True)
|
|
app_state["env"].write_text(dedent(SETUP_HELP))
|
|
print(f"Wrote defaults file {app_state['env']}.")
|
|
loader.load()
|
|
|
|
|
|
@db_app.command()
|
|
def list(context: typer.Context):
|
|
from ttfrog.db.manager import db
|
|
|
|
print("\n".join(sorted(db.tables.keys())))
|
|
|
|
|
|
@db_app.command(context_settings={"allow_extra_args": True, "ignore_unknown_options": True})
|
|
def dump(context: typer.Context):
|
|
"""
|
|
Dump tables (or the entire database) as a JSON blob.
|
|
"""
|
|
from ttfrog.db.manager import db
|
|
|
|
setup(context)
|
|
print(db.dump(context.args))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|