dnd/deadsands/site_tools/shell.py

from enum import EnumMeta
from inspect import signature

from prompt_toolkit import PromptSession
from prompt_toolkit.completion import FuzzyWordCompleter, NestedCompleter
from rich import print

from site_tools.cli import app


def dmsh():
    session = PromptSession()

    def cmd2dict(cmd):
        sig = signature(cmd)
        if not sig.parameters:
            return None
        cmds = {}
        for k, v in list(sig.parameters.items()):
            print(v, dir(v))
            if v.annotation.__class__ == EnumMeta:
                cmds[k] = FuzzyWordCompleter([e.value for e in v.annotation])
            else:
                cmds[k] = None
        return cmds
    commands = dict(site=dict((c.callback.__name__, cmd2dict(c.callback)) for c in app.registered_commands))
    print(commands)
    completer = NestedCompleter.from_nested_dict(commands)
    text = session.prompt("DM> ", completer=completer, complete_while_typing=True, enable_history_search=False)
    words = text.split()

    print(f"You said {words}")