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

class NewDiaryModal(ModalScreen[str]):
    BINDINGS = [
        ("escape", "cancel", "Cancel"),
    ]
    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":
            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()
        elif event.button.id == "cancel_button":
            self.dismiss("")

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