from textual.app import App, ComposeResult
from textual.containers import Container, Horizontal, Vertical
from textual.widgets import Header, Footer, DirectoryTree, DataTable, Log, Static
from textual.binding import Binding
from textual import on
from pathlib import Path
import os
import subprocess


# --- MOCK DATA GENERATOR (Creates fake files for testing) ---
def create_mock_files():
    root = Path("mock_campaign")
    root.mkdir(exist_ok=True)

    # Create a structure matching your mockup
    dirs = [
        "artwork/scenes/4_belcorras_retreat/lasdas_lament",
        "artwork/scenes/4_belcorras_retreat/affinity",
        "artwork/tokens/creatures",
        "artwork/handouts/letters"
    ]

    for d in dirs:
        full_path = root / d
        full_path.mkdir(parents=True, exist_ok=True)
        # Create fake webp files
        if "lasdas_lament" in str(full_path):
            (full_path / "lasdas_lament.webp").touch()
            (full_path / "lasdas_lament_map_bg.webp").touch()
        if "creatures" in str(full_path):
            (full_path / "nhakazarin.webp").touch()


class CampaignTree(DirectoryTree):
    """A DirectoryTree with Vim-like navigation."""

    BINDINGS = [
        # Map Vim keys to standard Tree actions
        Binding("j", "cursor_down", "Down", show=False),
        Binding("k", "cursor_up", "Up", show=False),
        Binding("h", "cursor_left", "Collapse / Parent", show=True),
        Binding("l", "cursor_right", "Expand / Child", show=True),
    ]


class FileTable(DataTable):
    """A DataTable with custom file selection actions."""

    # Define our constants here for easy access
    ICON_CHECKED = "✔"
    ICON_UNCHECKED = "✘"

    BINDINGS = [
        Binding("space", "toggle_select", "Select/Deselect"),
        Binding("a", "toggle_all", "Select All"),
        Binding("j", "cursor_down", "Down", show=False),
        Binding("k", "cursor_up", "Up", show=False),
        # Category Shortcuts
        Binding("s", "set_category('Scenes')", "Set: Scenes"),
        Binding("t", "set_category('Tokens')", "Set: Tokens"),
        Binding("m", "set_category('Maps')", "Set: Maps"),
        Binding("h", "set_category('Handouts')", "Set: Handouts"),  # 'h' is safe here!
    ]

    def action_toggle_select(self):
        try:
            row_key, _ = self.coordinate_to_cell_key(self.cursor_coordinate)
            current_val = str(self.get_cell(row_key, "Select"))
            new_val = self.ICON_CHECKED if current_val == self.ICON_UNCHECKED else self.ICON_UNCHECKED
            self.update_cell(row_key, "Select", new_val)
            # We can bubble a log message event up if we want, or just print for now
        except Exception:
            pass

    def action_toggle_all(self):
        all_checked = True
        for row_key in self.rows:
            if self.get_cell(row_key, "Select") == self.ICON_UNCHECKED:
                all_checked = False
                break

        target_val = self.ICON_UNCHECKED if all_checked else self.ICON_CHECKED
        for row_key in self.rows:
            self.update_cell(row_key, "Select", target_val)

    def action_set_category(self, category_name: str):
        try:
            row_key, _ = self.coordinate_to_cell_key(self.cursor_coordinate)
            self.update_cell(row_key, "Category", category_name)
        except:
            pass


# --- THE TUI APPLICATION ---
class PublisherApp(App):
    """The Terminal Publishing Tool"""

    CSS = """
    Screen {
        layout: vertical;
        background: #0f1f18;  /* Deep dark green background */
    }

    /* HEADER & FOOTER */
    Header {
        dock: top;
        background: #1a3b2e;
        color: #4caf50;
    }
    Footer {
        dock: bottom;
        background: #1a3b2e;
        color: #4caf50;
    }

    /* MAIN CONTAINER */
    #main_container {
        height: 70%;
        layout: horizontal;
        border-bottom: solid #4caf50;
    }

    /* LEFT PANE (Directory) */
    #left_pane {
        width: 30%;
        height: 100%;
        border-right: solid #4caf50;
        background: #0f1f18;
    }

    /* STYLE CLASS FOR THE STATIC HEADER */
    .pane-header {
        background: #4caf50;
        color: #0f1f18; /* Dark text on green background (Reverse look) */
        text-style: bold;
        padding-left: 1;
    }

    DirectoryTree {
        background: #0f1f18;
        color: #8bc34a; /* Light green text */
        border: hidden; /* We use the container for the border */
    }

    DirectoryTree:focus {
        background: #162d23;
    }

    /* RIGHT PANE (Table) */
    #right_pane {
        width: 70%;
        height: 100%;
        background: #0f1f18;
    }

    DataTable {
        background: #0f1f18;
        color: #a5d6a7;
        scrollbar-gutter: stable;
    }

    DataTable:focus {
        border: solid #4caf50; /* Highlight when active */
    }

    /* Highlight the selected row in the table */
    DataTable > .datatable--cursor {
        background: #2e7d32;
        color: white;
    }

    /* LOG PANE */
    #log_pane {
        height: 30%;
        background: #0a1410;
        color: #81c784;
        padding: 1;
    }
    """

    # KEY BINDINGS
    BINDINGS = [
        ("tab", "toggle_focus", "Switch Pane"),
        ("p", "publish", "Publish Selected"),
        ("r", "refresh", "Refresh Scan"),
        ("q", "quit", "Quit"),
    ]

    def get_git_version(self):
        try:
            # git describe --tags --always --dirty
            # Returns things like: "v1.0.2-4-g9a2b3c" or "v1.0.2"
            return subprocess.check_output(
                ["git", "describe", "--tags", "--always", "--dirty"],
                stderr=subprocess.DEVNULL
            ).decode("utf-8").strip()
        except (subprocess.CalledProcessError, FileNotFoundError):
            return "dev"

    def compose(self) -> ComposeResult:
        """Create child widgets for the app."""
        yield Header(show_clock=True)

        # Main split view
        with Container(id="main_container"):
            # Left Pane: Directory Tree
            with Container(id="left_pane"):
                yield Static(" Directory Browser", classes="pane-header")
                # Start at current directory or mock directory
                start_path = "./mock_campaign" if os.path.exists("./mock_campaign") else "."
                yield CampaignTree(start_path, id="tree_view")

            # Right Pane: Data Table
            with Container(id="right_pane"):
                yield FileTable(id="file_table")

        # Bottom Pane: Log
        log_widget = Log(id="log_pane", highlight=True)
        log_widget.can_focus = False
        yield log_widget

        yield Footer()

    def on_mount(self) -> None:
        """Called when app starts."""
        version_str = self.get_git_version()
        self.title = f"FVTT PUBLISH {version_str}"

        table = self.query_one(FileTable)
        table.cursor_type = "row"

        # DEFINING EXPLICIT KEYS HERE IS THE MAGIC SAUCE
        table.add_column("Select", key="Select")
        table.add_column("Status", key="Status")
        table.add_column("Category", key="Category")
        table.add_column("Filename", key="Filename")
        table.add_column("Location", key="Location")

        self.log_message("System initialized.")

        if not os.path.exists("mock_campaign"):
            create_mock_files()
            try:
                self.query_one(CampaignTree).reload()
            except:
                pass

    # --- ACTIONS & LOGIC ---
    def log_message(self, msg: str):
        """Write to the bottom log window."""
        log = self.query_one(Log)
        log.write_line(f"> {msg}")

    def action_toggle_focus(self):
        """Switch focus between Tree and Table."""
        if self.query_one("#tree_view").has_focus:
            self.query_one("#file_table").focus()
        else:
            self.query_one("#tree_view").focus()

    @on(DirectoryTree.DirectorySelected)
    def handle_directory_click(self, event: DirectoryTree.DirectorySelected):
        """When a directory is selected in the tree, scan it."""
        path = event.path
        self.log_message(f"Scanning directory: {path}")
        self.populate_table(path)

    def populate_table(self, root_path):
        """Clears and refills the table with .webp files from the path."""
        table = self.query_one(FileTable)
        table.clear()

        root = Path(root_path)

        # Simple recursive scan mock
        for path in root.rglob("*.webp"):
            # Simple heuristic mock
            cat = "Scenes"
            if "creatures" in str(path):
                cat = "Tokens"
            if "handouts" in str(path):
                cat = "Handouts"

            try:
                display_location = str(path.parent.relative_to(root))
            except ValueError:
                display_location = str(path.parent)

            # Add row
            table.add_row(
                table.ICON_UNCHECKED,  # Select
                "[NEW]",               # Status
                cat,                   # Category
                path.name,             # Filename
                display_location       # Location
            )


if __name__ == "__main__":
    app = PublisherApp()
    app.run()