tempo_cli/ui/

timer.rs

use anyhow::Result;
use chrono::{DateTime, Utc, Duration};
use crossterm::event::{self, Event, KeyCode, KeyEventKind};
use ratatui::{
    backend::Backend,
    layout::{Alignment, Constraint, Direction, Layout, Rect},
    style::{Color, Modifier, Style},
    text::{Line, Span},
    widgets::{Block, Borders, Gauge, Paragraph, Wrap},
    Frame, Terminal,
};
use std::time::Duration as StdDuration;

use crate::{
    utils::ipc::IpcClient,
    ui::formatter::Formatter,
};

pub struct InteractiveTimer {
    client: IpcClient,
    start_time: Option<DateTime<Utc>>,
    paused_at: Option<DateTime<Utc>>,
    total_paused: Duration,
    target_duration: i64, // in seconds
    show_milestones: bool,
}

impl InteractiveTimer {
    pub async fn new() -> Result<Self> {
        let socket_path = crate::utils::ipc::get_socket_path()?;
        let client = if socket_path.exists() {
            match IpcClient::connect(&socket_path).await {
                Ok(client) => client,
                Err(_) => IpcClient::new()?,
            }
        } else {
            IpcClient::new()?
        };

        Ok(Self {
            client,
            start_time: None,
            paused_at: None,
            total_paused: Duration::zero(),
            target_duration: 25 * 60, // Default 25 minutes (Pomodoro)
            show_milestones: true,
        })
    }

    pub async fn run<B: Backend>(&mut self, terminal: &mut Terminal<B>) -> Result<()> {
        loop {
            // Update timer state
            self.update_timer_state().await?;

            terminal.draw(|f| {
                self.render_timer(f);
            })?;

            // Handle input
            if event::poll(StdDuration::from_millis(100))? {
                match event::read()? {
                    Event::Key(key) if key.kind == KeyEventKind::Press => {
                        match key.code {
                            KeyCode::Char('q') | KeyCode::Esc => break,
                            KeyCode::Char(' ') => self.toggle_timer().await?,
                            KeyCode::Char('r') => self.reset_timer().await?,
                            KeyCode::Char('s') => self.set_target().await?,
                            KeyCode::Char('m') => self.show_milestones = !self.show_milestones,
                            _ => {}
                        }
                    }
                    _ => {}
                }
            }
        }

        Ok(())
    }

    fn render_timer(&self, f: &mut Frame) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .constraints([
                Constraint::Length(3),  // Title
                Constraint::Length(8),  // Timer display
                Constraint::Length(6),  // Progress bar
                Constraint::Length(6),  // Milestones
                Constraint::Min(0),     // Controls
            ])
            .split(f.size());

        // Title
        let title = Paragraph::new("🕐 Interactive Timer")
            .style(Style::default().fg(Color::Cyan).add_modifier(Modifier::BOLD))
            .alignment(Alignment::Center)
            .block(Block::default().borders(Borders::ALL));
        f.render_widget(title, chunks[0]);

        // Timer display
        self.render_timer_display(f, chunks[1]);
        
        // Progress bar
        self.render_progress_bar(f, chunks[2]);
        
        // Milestones
        if self.show_milestones {
            self.render_milestones(f, chunks[3]);
        }
        
        // Controls
        self.render_controls(f, chunks[4]);
    }

    fn render_timer_display(&self, f: &mut Frame, area: Rect) {
        let elapsed = self.get_elapsed_time();
        let is_running = self.start_time.is_some() && self.paused_at.is_none();
        
        let time_display = Formatter::format_duration(elapsed);
        let status = if is_running { "🟢 RUNNING" } else if self.start_time.is_some() { "⏸️ PAUSED" } else { "⏹️ STOPPED" };
        let status_color = if is_running { Color::Green } else if self.start_time.is_some() { Color::Yellow } else { Color::Red };

        let timer_text = vec![
            Line::from(Span::styled(
                time_display,
                Style::default().fg(Color::Cyan).add_modifier(Modifier::BOLD)
            )),
            Line::from(Span::raw("")),
            Line::from(vec![
                Span::raw("Status: "),
                Span::styled(status, Style::default().fg(status_color).add_modifier(Modifier::BOLD)),
            ]),
            Line::from(vec![
                Span::raw("Target: "),
                Span::styled(
                    Formatter::format_duration(self.target_duration),
                    Style::default().fg(Color::White)
                ),
            ]),
        ];

        let timer_block = Block::default()
            .borders(Borders::ALL)
            .title("Timer")
            .style(Style::default().fg(Color::White));

        let paragraph = Paragraph::new(timer_text)
            .block(timer_block)
            .alignment(Alignment::Center)
            .wrap(Wrap { trim: true });
        f.render_widget(paragraph, area);
    }

    fn render_progress_bar(&self, f: &mut Frame, area: Rect) {
        let elapsed = self.get_elapsed_time();
        let progress = if self.target_duration > 0 {
            ((elapsed as f64 / self.target_duration as f64) * 100.0).min(100.0)
        } else {
            0.0
        };

        let progress_color = if progress >= 100.0 { Color::Green }
                            else if progress >= 75.0 { Color::Yellow }
                            else { Color::Cyan };

        let progress_bar = Gauge::default()
            .block(Block::default()
                .borders(Borders::ALL)
                .title("Progress to Target")
                .style(Style::default().fg(Color::White)))
            .gauge_style(Style::default().fg(progress_color))
            .percent(progress as u16)
            .label(format!("{:.1}% ({}/{})", 
                progress, 
                Formatter::format_duration(elapsed),
                Formatter::format_duration(self.target_duration)
            ));
        
        f.render_widget(progress_bar, area);
    }

    fn render_milestones(&self, f: &mut Frame, area: Rect) {
        let elapsed = self.get_elapsed_time();
        let milestones = vec![
            (5 * 60, "5 min warm-up"),
            (15 * 60, "15 min focus"),
            (25 * 60, "Pomodoro complete"),
            (45 * 60, "45 min deep work"),
            (60 * 60, "1 hour marathon"),
        ];

        let mut milestone_lines = vec![];
        for (duration, name) in milestones {
            let achieved = elapsed >= duration;
            let icon = if achieved { "✅" } else { "⭕" };
            let style = if achieved { 
                Style::default().fg(Color::Green) 
            } else { 
                Style::default().fg(Color::Gray) 
            };
            
            milestone_lines.push(Line::from(vec![
                Span::styled(format!("{} {}", icon, name), style),
            ]));
        }

        let milestones_block = Block::default()
            .borders(Borders::ALL)
            .title("Milestones")
            .style(Style::default().fg(Color::White));

        let paragraph = Paragraph::new(milestone_lines)
            .block(milestones_block)
            .wrap(Wrap { trim: true });
        f.render_widget(paragraph, area);
    }

    fn render_controls(&self, f: &mut Frame, area: Rect) {
        let controls_text = vec![
            Line::from(Span::styled("Controls:", Style::default().fg(Color::Cyan).add_modifier(Modifier::BOLD))),
            Line::from(Span::raw("Space - Start/Pause timer")),
            Line::from(Span::raw("R - Reset timer")),
            Line::from(Span::raw("S - Set target duration")),
            Line::from(Span::raw("M - Toggle milestones")),
            Line::from(Span::raw("Q/Esc - Quit")),
        ];

        let controls_block = Block::default()
            .borders(Borders::ALL)
            .title("Controls")
            .style(Style::default().fg(Color::White));

        let paragraph = Paragraph::new(controls_text)
            .block(controls_block)
            .wrap(Wrap { trim: true });
        f.render_widget(paragraph, area);
    }

    async fn update_timer_state(&mut self) -> Result<()> {
        // This would sync with the actual session state from the daemon
        // For now, we'll keep local state
        Ok(())
    }

    async fn toggle_timer(&mut self) -> Result<()> {
        if self.start_time.is_none() {
            // Start timer
            self.start_time = Some(Utc::now());
            self.paused_at = None;
        } else if self.paused_at.is_some() {
            // Resume timer
            if let Some(paused_at) = self.paused_at {
                self.total_paused = self.total_paused + (Utc::now() - paused_at);
            }
            self.paused_at = None;
        } else {
            // Pause timer
            self.paused_at = Some(Utc::now());
        }
        Ok(())
    }

    async fn reset_timer(&mut self) -> Result<()> {
        self.start_time = None;
        self.paused_at = None;
        self.total_paused = chrono::Duration::zero();
        Ok(())
    }

    async fn set_target(&mut self) -> Result<()> {
        // In a full implementation, this would show an input dialog
        // For now, cycle through common durations
        self.target_duration = match self.target_duration {
            1500 => 1800,      // 25min -> 30min
            1800 => 2700,      // 30min -> 45min
            2700 => 3600,      // 45min -> 1hour
            3600 => 5400,      // 1hour -> 1.5hour
            5400 => 7200,      // 1.5hour -> 2hour
            _ => 1500,         // Default back to 25min (Pomodoro)
        };
        Ok(())
    }

    fn get_elapsed_time(&self) -> i64 {
        if let Some(start) = self.start_time {
            let end_time = if let Some(paused) = self.paused_at {
                paused
            } else {
                Utc::now()
            };
            
            (end_time - start - self.total_paused).num_seconds().max(0)
        } else {
            0
        }
    }
}