dnd-rolltable/rolltable/cli.py

from collections import defaultdict
from enum import Enum
from pathlib import Path
from typing import List

import typer
from rich import print

from rolltable import tables
from rolltable.types import RollTable

app = typer.Typer()
app_state = defaultdict(
    str,
    options=defaultdict(str),
)


class OUTPUT_FORMATS(Enum):
    text = "text"
    yaml = "yaml"
    markdown = "markdown"


@app.callback()
def main(
    frequency: str = typer.Option("default", help="use the specified frequency from the source file"),
    die: int = typer.Option(20, help="The size of the die for which to create a table"),
    hide_rolls: bool = typer.Option(
        False,
        help="If True, do not show the Roll column.",
    ),
    collapsed: bool = typer.Option(True, help="If True, collapse multiple die values with the same option."),
    width: int = typer.Option(120, help="Width of the table."),
    output: OUTPUT_FORMATS = typer.Option(
        "text",
        help="The output format to use.",
    ),
):
    app_state["options"] = {
        "frequency": frequency,
        "die": die,
        "hide_rolls": hide_rolls,
    }
    app_state["collapsed"] = collapsed
    app_state["width"] = width
    app_state["output"] = output


@app.command("custom")
def custom(
    sources: List[Path] = typer.Argument(..., help="Path to one or more yaml-formatted source file."),
):
    """
    Create roll tables from custom sources.
    """

    rt = RollTable([Path(s).read_text() for s in sources], **app_state["options"])
    print_table(rt)


def print_table(table):
    if app_state["output"] == OUTPUT_FORMATS.yaml:
        print(table.as_yaml())
    elif app_state["output"] == OUTPUT_FORMATS.markdown:
        print(table.as_markdown())
    else:
        print(table.as_table(width=app_state["width"], expanded=not app_state["collapsed"]))


def make_callback(roll_table_instance):
    def inner():
        roll_table_instance.frequency = app_state["options"]["frequency"]
        roll_table_instance.die = app_state["options"]["die"]
        print_table(roll_table_instance)

    return inner


# step through all the predfined tables and create a cli for each
for name, table in tables.index.items():
    help_text = name.replace("_", " ").title()
    app.command(name=name, help=f"Create a roll table of {help_text}")(make_callback(table))


if __name__ == "__main__":
    app()