from textual.app import ComposeResult
from textual.binding import Binding
from textual.containers import Vertical, Horizontal
from textual.screen import ModalScreen
from textual.widgets import Label, Input, Button


class NewDiaryModal(ModalScreen[str]):
    BINDINGS = [
        Binding("escape", "cancel", "Cancel"),
        Binding("enter", "create_diary", "Create",priority=True),
    ]
    def __init__(self):
        super().__init__()
        self.name_input = Input(id="NewDiaryModal-NameInput",classes="NewDiaryModal-NameInput") # This ID is fine, it's specific to the input

    def compose(self) -> ComposeResult:

        with Vertical(id="new_diary_dialog",classes="NewDiaryModal-Dialog"):
            yield Label("Create a new diary", classes="NewDiaryModal-Title")
            yield Label("Diary Name:")
            yield self.name_input
            with Horizontal(classes="NewDiaryModal-ButtonsContainer"):
                yield Button("Create", variant="primary", id="create_diary_button",
                             classes="NewDiaryModal-CreateDiaryButton")
                yield Button("Cancel", variant="default", id="cancel_button",
                             classes="NewDiaryModal-CancelButton")

    def on_mount(self):
          self.name_input.focus()

    def on_button_pressed(self, event: Button.Pressed) -> None:
        """Handles button clicks."""
        if event.button.id == "create_diary_button":
            self.action_create_diary()
        elif event.button.id == "cancel_button":
            self.dismiss("")

    def action_cancel(self) -> None:
        """Action to cancel the modal."""
        self.dismiss("")

    def action_create_diary(self) -> None:
        diary_name = self.name_input.value.strip()
        if diary_name:
            self.dismiss(diary_name)
        else:
            self.notify("Diary name cannot be empty.", severity="warning")
            self.name_input.focus()

    def on_input_submitted(self, event: Input.Submitted) -> None:
        if event.input.id == "NewDiaryModal-NameInput":
            self.action_create_diary()