youtube-cli/youtube_tui/screens/home.py

#!/usr/bin/env python3
"""
Home Screen for YouTube TUI
Main dashboard with quick actions
"""

from textual.app import ComposeResult
from textual.containers import Container
from textual.screen import Screen
from textual.widgets import (
    Button,
    Footer,
    Header,
    Static,
)


class HomeScreen(Screen):
    """Main dashboard screen with quick actions"""

    CSS = """
    HomeScreen {
        align: center middle;
    }

    #welcome-container {
        width: 60%;
        height: auto;
        border: double #555555;
        padding: 2 3;
        margin: 2 0;
    }

    #welcome-title {
        width: 100%;
        height: 3;
        content-align: center middle;
        margin-bottom: 1;
    }

    #welcome-info {
        width: 100%;
        height: auto;
        margin-bottom: 2;
        color: $text-muted;
    }

    #quick-actions {
        width: 100%;
        layout: grid;
        grid-gutter: 1;
        grid-columns: 2;
        margin: 1 0;
    }

    Button {
        width: 100%;
        margin: 0 1;
    }

    #status-info {
        width: 100%;
        height: auto;
        margin-top: 2;
        color: $text-muted;
    }

    #status-bar {
        dock: bottom;
        height: 1;
        background: $surface;
        color: $text-muted;
        padding: 0 1;
    }
    """

    BINDINGS = [
        ("s", "open_search", "Search"),
        ("q", "open_queue", "Queue"),
        ("h", "open_history", "History"),
        ("ctrl+h", "show_help", "Help"),
        ("q", "quit", "Quit"),
    ]

    def __init__(self):
        super().__init__()
        self.download_queue = None
        self.download_manager = None

    def compose(self) -> ComposeResult:
        """Compose the home screen"""
        yield Header()
        yield Container(
            Static(
                "[bold cyan]YouTube CLI[/bold cyan] - Browse and download videos",
                id="welcome-title",
            ),
            Static(
                "Use keyboard shortcuts or buttons to navigate",
                id="welcome-info",
            ),
            Container(
                Button("Search Videos", id="search-btn"),
                Button("Download Queue", id="queue-btn"),
                Button("Search History", id="history-btn"),
                Button("Help", id="help-btn"),
                id="quick-actions",
            ),
            Static(id="status-info"),
            id="welcome-container",
        )
        yield Static(id="status-bar")
        yield Footer()

    def on_mount(self) -> None:
        """Called when screen is mounted"""
        if hasattr(self.app, "download_queue"):
            self.download_queue = self.app.download_queue
        if hasattr(self.app, "download_manager"):
            self.download_manager = self.app.download_manager

        self.update_queue_status()
        self.update_status("[green]Welcome to YouTube TUI - Press 's' to search[/green]")

    def update_queue_status(self) -> None:
        """Update queue status info"""
        if self.download_manager:
            status = self.download_manager.get_queue_status()
            status_text = (
                f"Queue: {status['pending_count']} pending, "
                f"{status['downloading_count']} downloading, "
                f"{status['total_count']} total"
            )
        else:
            status_text = "Queue: No active downloads"

        status_info = self.query_one("#status-info", Static)
        status_info.update(f"[dim]{status_text}[/dim]")

    def update_status(self, message: str) -> None:
        """Update the status bar message"""
        status_bar = self.query_one("#status-bar", Static)
        status_bar.update(f"[bold white]{message}[/bold white]")

    def action_open_search(self) -> None:
        """Open search screen"""
        if hasattr(self.app, "push_search_screen"):
            self.app.push_search_screen()
        else:
            from youtube_tui.screens.search import SearchScreen

            self.app.push_screen(SearchScreen())

    def action_open_queue(self) -> None:
        """Open queue screen"""
        if hasattr(self.app, "action_open_queue"):
            self.app.action_open_queue()
        else:
            from youtube_tui.screens.queue import QueueScreen

            self.app.push_screen(QueueScreen())

    def action_open_history(self) -> None:
        """Open search history"""
        from youtube_tui.screens.history import SearchHistoryScreen

        self.app.push_screen(SearchHistoryScreen())

    def action_show_help(self) -> None:
        """Show help screen"""
        from youtube_tui.screens.help import HelpScreen

        self.app.push_screen(HelpScreen())

    def action_quit(self) -> None:
        """Quit the application"""
        self.app.exit()

    def action_cancel(self) -> None:
        """Handle escape key"""
        pass

    def on_button_pressed(self, event: Button.Pressed) -> None:
        """Handle button presses"""
        if event.button.id == "search-btn":
            self.action_open_search()
        elif event.button.id == "queue-btn":
            self.action_open_queue()
        elif event.button.id == "history-btn":
            self.action_open_history()
        elif event.button.id == "help-btn":
            self.action_show_help()