from prompt_toolkit.key_binding import KeyBindings, merge_key_bindings
from prompt_toolkit.key_binding.bindings.basic import load_basic_bindings
from prompt_toolkit.shortcuts import prompt, PromptSession
from rich.table import Table
from rich.console import Console

keybindings = load_basic_bindings()

@keybindings.add('c-q')
def exit_(event):
    """ Pressing 'c-q' will exit the application. """
    exit(0)

@keybindings.add('c-m')
def open_menu(event):
    """ Pressing 'c-m' will open the menu. """
    show_keybindings_menu()
    event.app.exit()

@keybindings.add('enter')
def process_input(event):
    """ Pressing 'enter' will process the input. """
    print(dir(event))
    event.app.exit(result=event.current_buffer.text)

bindings_info = [
    {"Key": "Ctrl-Q", "Action": "Exit the application"},
    {"Key": "Ctrl-M", "Action": "Open the menu"},
]

console = Console()


def show_keybindings_menu():
    table = Table(title="Keybindings")

    table.add_column("Key", justify="right", style="cyan", no_wrap=True)
    table.add_column("Action", style="magenta")

    for binding in bindings_info:
        table.add_row(binding["Key"], binding["Action"])

    console.print(table)


def main_menu():
    session = PromptSession("> ", key_bindings=keybindings)
    while True:
        show_keybindings_menu()

        user_input = session.prompt()
        console.print(f"You entered: {user_input}")


if __name__ == "__main__":
    main_menu()