main.py

"""Main application entry point."""

import sys
import threading
from pathlib import Path

from textual.app import ComposeResult
from textual.containers import Container
from textual.app import App
from textual.binding import Binding

# Add src to path for imports
sys.path.insert(0, str(Path(__file__).parent / "src"))

from config import validate_api_key
from tracker import TimeTracker
from screenshots import ScreenshotThread
from ui import TimeTrackerApp, HeaderBar


class TTApp(App):
    """Main Time Tracking Application."""

    # Global bindings so keys work regardless of widget focus
    BINDINGS = [
        Binding("t", "new_task", "New Task", show=True),
        Binding("s", "start_stop", "Start/Stop", show=True),
        Binding("a", "manual_add", "Manual Add", show=True),
        Binding("d", "delete_session", "Delete Session", show=True),
        Binding("left", "prev_week", "Previous Week", show=True),
        Binding("right", "next_week", "Next Week", show=True),
    ]
    
    TITLE = "Time Tracking Tool"
    CSS = """
    Screen {
        layout: vertical;
    }
    
    HeaderBar {
        height: 1;
        color: white;
        padding: 0 1;
        border-bottom: none;
    }
    
    WeeklyTable {
        height: auto;
        min-height: 3;
        border-bottom: solid $accent;
    }
    
    DailyBreakdown {
        height: auto;
        border-top: solid $accent;
        padding: 1;
    }
    
    TaskListModal {
        align: center middle;
    }
    
    TaskListModal > Vertical {
        width: 50;
        height: auto;
        border: solid $accent;
        background: $panel;
        padding: 1;
    }
    
    NewTaskModal {
        align: center middle;
    }
    
    NewTaskModal > Vertical {
        width: 50;
        height: auto;
        border: solid $accent;
        background: $panel;
        padding: 1;
    }
    
    ManualAddModal {
        align: center middle;
    }
    
    ManualAddModal > Vertical {
        width: 60;
        height: auto;
        border: solid $accent;
        background: $panel;
        padding: 1;
    }
    
    DeleteSessionModal {
        align: center middle;
    }
    
    DeleteSessionModal > Vertical {
        width: 50;
        height: auto;
        border: solid $accent;
        background: $panel;
        padding: 1;
    }
    """
    
    def __init__(self):
        super().__init__()
        # Validate API key first
        validate_api_key()
        
        # Initialize tracker before compose is called
        self.tracker = TimeTracker()
        self.screenshot_thread = None
    
    def on_mount(self):
        """Initialize app on mount."""
        # Tracker already initialized in __init__
        
        # Start auto-save thread
        self.auto_save_thread = threading.Thread(
            target=self._auto_save_loop,
            daemon=True
        )
        self.auto_save_thread.start()
        
        # Start screenshot thread
        self.screenshot_thread = ScreenshotThread(self.tracker)
        self.screenshot_thread.start()
    
    def compose(self) -> ComposeResult:
        """Compose the main app."""
        # Render a persistent header at the top showing keybinds
        yield HeaderBar()
        yield TimeTrackerApp(self.tracker)

    # The following action methods delegate to the main TimeTracker widget
    def _get_main_widget(self):
        try:
            return self.query_one(TimeTrackerApp)
        except Exception:
            return None

    def action_new_task(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_new_task()

    def action_start_stop(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_start_stop()

    def action_manual_add(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_manual_add()

    def action_delete_session(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_delete_session()

    def action_prev_week(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_prev_week()

    def action_next_week(self):
        widget = self._get_main_widget()
        if widget:
            widget.action_next_week()
    
    def _auto_save_loop(self):
        """Periodically save data."""
        import time
        while True:
            time.sleep(60)  # Save every 60 seconds
            self.tracker.store.save()
    
    def on_unmount(self):
        """Cleanup on exit."""
        if self.screenshot_thread:
            self.screenshot_thread.stop()
        self.tracker.store.save()


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