crmux 0.19.1

use std::process::{Command, Stdio};
use std::time::Instant;
use tmux_claude_state::claude_state::{ClaudeState, PermissionMode};
use tmux_claude_state::monitor::{ClaudeSession, MonitorState};

/// Run `git -C <cwd> branch --show-current` and return the branch name.
fn resolve_git_branch(cwd: &str) -> Option<String> {
    let output = Command::new("git")
        .args(["-C", cwd, "branch", "--show-current"])
        .stdin(Stdio::null())
        .stderr(Stdio::null())
        .output()
        .ok()?;
    if output.status.success() {
        let branch = String::from_utf8_lossy(&output.stdout).trim().to_string();
        if branch.is_empty() {
            None
        } else {
            Some(branch)
        }
    } else {
        None
    }
}

/// Parse `git diff --numstat` output and return (`file_count`, insertions, deletions).
fn parse_numstat(output: &[u8]) -> (usize, usize, usize) {
    let text = String::from_utf8_lossy(output);
    let mut files = 0;
    let mut insertions = 0;
    let mut deletions = 0;
    for line in text.lines() {
        let parts: Vec<&str> = line.split('\t').collect();
        if parts.len() >= 2 {
            files += 1;
            // Binary files show "-" instead of numbers
            insertions += parts[0].parse::<usize>().unwrap_or(0);
            deletions += parts[1].parse::<usize>().unwrap_or(0);
        }
    }
    (files, insertions, deletions)
}

/// Run `git diff --numstat` (staged + unstaged) and return a `GitDiffInfo`.
/// Returns `None` if not in a git repo or git is not available.
fn resolve_git_diff(cwd: &str) -> Option<GitDiffInfo> {
    // staged
    let staged_output = Command::new("git")
        .args(["-C", cwd, "diff", "--cached", "--numstat"])
        .stdin(Stdio::null())
        .stderr(Stdio::null())
        .output()
        .ok()?;
    if !staged_output.status.success() {
        return None;
    }
    let (staged_files, staged_ins, staged_del) = parse_numstat(&staged_output.stdout);

    // unstaged
    let unstaged_output = Command::new("git")
        .args(["-C", cwd, "diff", "--numstat"])
        .stdin(Stdio::null())
        .stderr(Stdio::null())
        .output()
        .ok()?;
    let (modified_files, unstaged_ins, unstaged_del) = if unstaged_output.status.success() {
        parse_numstat(&unstaged_output.stdout)
    } else {
        (0, 0, 0)
    };

    let info = GitDiffInfo {
        staged_files,
        modified_files,
        insertions: staged_ins + unstaged_ins,
        deletions: staged_del + unstaged_del,
    };
    Some(info)
}

/// Extract the tmux session name from a `pane_id` (e.g., "dev:%1" → "dev").
/// Returns empty string if `pane_id` has no ':' separator.
fn extract_tmux_session(pane_id: &str) -> String {
    match pane_id.split_once(':') {
        Some((session, _)) => session.to_string(),
        None => String::new(),
    }
}

/// Build the path to a session's JSONL file: `~/.claude/projects/<project_dir>/<session_id>.jsonl`
fn jsonl_path(cwd: &str, session_id: &str) -> Option<String> {
    let home = std::env::var("HOME").ok()?;
    let project_dir = crate::auto_title::cwd_to_project_dir(cwd);
    Some(format!("{home}/.claude/projects/{project_dir}/{session_id}.jsonl"))
}

/// Calculate the number of Shift+Tab presses needed to switch from `current` to `target_mode`.
/// Mode cycle: Ask(0) → Auto Edit(1) → Plan(2) → Ask(0)...
/// Returns 0 if `target_mode` is unknown or already matches current.
pub fn permission_mode_switch_count(current: &PermissionMode, target_mode: &str) -> usize {
    let current_idx = match current {
        PermissionMode::AskBeforeEdits => 0,
        PermissionMode::EditAutomatically => 1,
        PermissionMode::PlanMode => 2,
    };
    let target_idx = match target_mode {
        "accept-edits" => 1,
        "plan-mode" => 2,
        _ => return 0,
    };
    (target_idx + 3 - current_idx) % 3
}

/// Git diff summary for a session's working directory.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GitDiffInfo {
    pub staged_files: usize,
    pub modified_files: usize,
    pub insertions: usize,
    pub deletions: usize,
}


/// A Claude Code session managed by crmux, tracked by PID.
#[derive(Debug, Clone)]
pub struct ManagedSession {
    /// PID of the Claude Code process (stable across pane moves).
    pub pid: u32,
    /// Current tmux pane ID (may change after join-pane).
    pub pane_id: String,
    /// Project name (basename of cwd).
    pub project_name: String,
    /// Current state.
    pub state: ClaudeState,
    /// When the current state was first observed.
    pub state_changed_at: Instant,
    /// Whether this session is marked for filtering and broadcast.
    pub marked: bool,
    /// User-defined title label for this session.
    pub title: Option<String>,
    /// Claude Code `session_id` (from `SessionStart` hook).
    pub session_id: Option<String>,
    /// Model display name (from `statusLine` hook or `SessionStart` hook).
    pub model: Option<String>,
    /// Context window usage percentage (0–100).
    pub context_percent: Option<u8>,
    /// Current working directory.
    pub cwd: String,
    /// Current git branch name (if in a git repo).
    pub git_branch: Option<String>,
    /// Automatically resolved title from Claude Code session metadata.
    pub auto_title: Option<String>,
    /// Current permission mode (Ask / Auto Edit / Plan).
    pub permission_mode: PermissionMode,
    /// Last observed mtime of the session's JSONL file (for change detection).
    pub jsonl_mtime: Option<std::time::SystemTime>,
    /// Whether the session has ever been in the Working state.
    pub has_worked: bool,
    /// Git worktree name (basename of worktree directory), if running in a worktree.
    pub worktree_name: Option<String>,
    /// Git diff summary (staged/modified files and line changes).
    pub git_diff: Option<GitDiffInfo>,
    /// Tmux session name this pane belongs to (extracted from `pane_id`).
    pub tmux_session: String,
}

impl ManagedSession {
    /// Return the display title: manual title takes priority over auto title.
    pub fn display_title(&self) -> Option<&str> {
        self.title
            .as_deref()
            .or(self.auto_title.as_deref())
    }
}

/// Tab for filtering sessions by project.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Tab {
    All,
    Marked,
    Workspace(String),
    Project(String),
}

/// State for tab-based session filtering.
pub struct TabState {
    pub tabs: Vec<Tab>,
    pub selected_tab: usize,
}

impl TabState {
    pub const fn new() -> Self {
        Self {
            tabs: Vec::new(),
            selected_tab: 0,
        }
    }

    /// Rebuild the tab list from sessions' project names.
    /// Maintains the currently selected tab value if it still exists.
    pub fn rebuild_tabs(&mut self, sessions: &[ManagedSession], initial_workspace: Option<&str>) -> bool {
        let mut projects: Vec<String> = sessions
            .iter()
            .map(|s| s.project_name.clone())
            .collect();
        projects.sort();
        projects.dedup();

        let mut new_tabs = vec![Tab::All];

        if sessions.iter().any(|s| s.marked) {
            new_tabs.push(Tab::Marked);
        }

        // Add Workspace tabs when 2+ unique tmux sessions exist
        let mut tmux_sessions: Vec<String> = sessions
            .iter()
            .map(|s| s.tmux_session.clone())
            .collect();
        tmux_sessions.sort();
        tmux_sessions.dedup();
        if tmux_sessions.len() >= 2 {
            for ws in tmux_sessions {
                new_tabs.push(Tab::Workspace(ws));
            }
        }
        for p in projects {
            new_tabs.push(Tab::Project(p));
        }

        // Apply initial_workspace if provided, otherwise maintain selection
        if let Some(ws) = initial_workspace {
            let target = Tab::Workspace(ws.to_string());
            self.tabs = new_tabs;
            if let Some(pos) = self.tabs.iter().position(|t| *t == target) {
                self.selected_tab = pos;
                return true;
            }
            self.selected_tab = 0; // fallback to All
        } else {
            let current = self.current_tab().clone();
            self.tabs = new_tabs;
            if let Some(pos) = self.tabs.iter().position(|t| *t == current) {
                self.selected_tab = pos;
            } else {
                self.selected_tab = 0; // fallback to All
            }
        }
        false
    }

    /// Move to next tab (wraps around).
    pub const fn select_next_tab(&mut self) {
        if !self.tabs.is_empty() {
            self.selected_tab = (self.selected_tab + 1) % self.tabs.len();
        }
    }

    /// Move to previous tab (wraps around).
    pub const fn select_prev_tab(&mut self) {
        if !self.tabs.is_empty() {
            if self.selected_tab == 0 {
                self.selected_tab = self.tabs.len() - 1;
            } else {
                self.selected_tab -= 1;
            }
        }
    }

    /// Get the currently selected tab.
    pub fn current_tab(&self) -> &Tab {
        self.tabs.get(self.selected_tab).unwrap_or(&Tab::All)
    }
}

/// Diff result from syncing with `MonitorState`.
#[derive(Debug, PartialEq, Eq)]
pub struct SyncDiff {
    /// PIDs of newly discovered sessions.
    pub added: Vec<u32>,
    /// PIDs of sessions that disappeared.
    pub removed: Vec<u32>,
    /// PIDs of sessions whose state changed.
    pub state_changed: Vec<u32>,
}

/// A single preview pane entry.
#[derive(Debug, Clone)]
pub struct PreviewEntry {
    /// Project name (basename of cwd).
    pub name: String,
    /// Tmux pane ID.
    pub pane_id: String,
    /// 0-based index in the filtered session list (used for number-key display).
    pub index: usize,
    /// Optional session title label.
    pub title: Option<String>,
    /// Git branch name.
    pub git_branch: Option<String>,
    /// Worktree name.
    pub worktree_name: Option<String>,
    /// Captured pane content.
    pub content: String,
    /// Cursor position (row, col) detected from reverse-video cell in captured content.
    pub cursor_pos: Option<(u16, u16)>,
    /// Git diff summary for display in `title_bottom`.
    pub git_diff: Option<GitDiffInfo>,
    /// Claude session state (for border color).
    pub state: ClaudeState,
    /// Whether the session has ever been in Working state.
    pub has_worked: bool,
    /// When the session state last changed (for pulse timing).
    pub state_changed_at: Instant,
}

/// Maximum number of sessions selectable by number keys (1-9).
pub const MAX_NUMBER_KEYS: usize = 9;

/// Layout mode for the preview area.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum LayoutMode {
    /// Show a single selected session.
    Single,
    /// Show all filtered sessions in a grid.
    Grid,
    /// Show all filtered sessions in even horizontal split.
    EvenHorizontal,
    /// Show all filtered sessions in even vertical split.
    EvenVertical,
    /// Main pane on the left (60%), remaining stacked vertically on the right.
    MainVertical,
    /// Main pane on top (60%), remaining side by side on the bottom.
    MainHorizontal,
}

impl LayoutMode {
    /// Return the next layout mode in the cycle.
    pub const fn next(self) -> Self {
        match self {
            Self::MainVertical => Self::Single,
            Self::Single => Self::Grid,
            Self::Grid => Self::EvenHorizontal,
            Self::EvenHorizontal => Self::EvenVertical,
            Self::EvenVertical => Self::MainHorizontal,
            Self::MainHorizontal => Self::MainVertical,
        }
    }

    /// Short label for display (e.g. footer).
    pub const fn short_label(self) -> &'static str {
        match self {
            Self::Single => "Single",
            Self::Grid => "Grid",
            Self::EvenHorizontal => "EvenH",
            Self::EvenVertical => "EvenV",
            Self::MainVertical => "MainV",
            Self::MainHorizontal => "MainH",
        }
    }
}

/// Input mode for the sidebar.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum InputMode {
    /// Normal navigation mode.
    Normal,
    /// Text input mode for sending keys to a session.
    Input,
    /// Title input mode for setting session title label.
    Title,
    /// Broadcast mode for sending keys to all marked sessions.
    Broadcast,
    /// Scroll mode for navigating preview with j/k.
    Scroll,
}

/// Application state for the sidebar.
pub struct AppState {
    /// All managed sessions, ordered by discovery time.
    pub sessions: Vec<ManagedSession>,
    /// Index of the currently selected session.
    pub selected_index: usize,
    /// Index of the previously selected session (for `t` key toggle).
    pub prev_selected_index: Option<usize>,
    /// PID of our own sidebar pane's process (excluded from aggregation).
    pub own_pid: Option<u32>,
    /// Preview contents for each visible pane.
    pub preview_contents: Vec<PreviewEntry>,
    /// Current input mode.
    pub input_mode: InputMode,
    /// Buffer for text input in Input mode.
    pub input_buffer: String,
    /// Whether the help popup is currently shown.
    pub show_help: bool,
    /// Help popup scroll offset (in lines from the top).
    pub help_scroll: u16,
    /// Preview scroll offset (0=bottom, positive=scroll up).
    pub preview_scroll: u16,
    /// Preview area height (set during draw loop for scroll amount calculation).
    pub preview_height: u16,
    /// Whether `g` has been pressed once, waiting for the second `g` (vim `gg`).
    pub pending_g: bool,
    /// RPC messages that arrived before the matching session was discovered.
    pub pending_rpc: Vec<crate::rpc::RpcMessage>,
    /// Which mode Esc was pressed in (for Esc Esc cancel forwarding).
    pub esc_source_mode: Option<InputMode>,
    /// Whether the claudeye overlay is visible.
    pub claudeye_visible: bool,
    /// Layout mode for the preview area.
    pub layout_mode: LayoutMode,
    /// Tab state for project-based filtering.
    pub tab_state: TabState,
    /// Accumulated plan info from sessions.
    pub plans: Vec<crate::auto_title::PlanInfo>,
    /// Project dirs already scanned for historical plans.
    pub scanned_project_dirs: std::collections::HashSet<String>,
    /// Latest version available for update (None = unchecked or already latest).
    pub update_available: Option<String>,
    /// Initial workspace to select on first tab rebuild (consumed after use).
    initial_workspace: Option<String>,
    /// Pane ID where crmux itself is running (e.g. "main:0.1").
    pub own_pane_id: Option<String>,
}

impl AppState {
    pub fn new(own_pid: Option<u32>) -> Self {
        Self {
            sessions: Vec::new(),
            selected_index: 0,
            prev_selected_index: None,
            own_pid,
            preview_contents: Vec::new(),
            input_mode: InputMode::Normal,
            input_buffer: String::new(),
            show_help: false,
            help_scroll: 0,
            preview_scroll: 0,
            preview_height: 0,
            pending_g: false,
            pending_rpc: Vec::new(),
            esc_source_mode: None,
            claudeye_visible: false,
            layout_mode: LayoutMode::MainVertical,
            tab_state: TabState::new(),
            plans: Vec::new(),
            scanned_project_dirs: std::collections::HashSet::new(),
            update_available: None,
            initial_workspace: None,
            own_pane_id: None,
        }
    }

    /// Set the initial workspace to select on first tab rebuild.
    pub fn set_initial_workspace(&mut self, workspace: Option<String>) {
        self.initial_workspace = workspace;
    }

    /// Return sessions filtered by the current tab.
    pub fn filtered_sessions(&self) -> Vec<&ManagedSession> {
        match self.tab_state.current_tab() {
            Tab::All => self.sessions.iter().collect(),
            Tab::Workspace(name) => self
                .sessions
                .iter()
                .filter(|s| s.tmux_session == *name)
                .collect(),
            Tab::Marked => self.sessions.iter().filter(|s| s.marked).collect(),
            Tab::Project(name) => self
                .sessions
                .iter()
                .filter(|s| s.project_name == *name)
                .collect(),
        }
    }

    /// Sync with the latest `MonitorState`, returning what changed.
    #[allow(clippy::too_many_lines)]
    pub fn sync_with_monitor(&mut self, monitor: &MonitorState) -> SyncDiff {
        let mut added = Vec::new();
        let mut removed = Vec::new();
        let mut state_changed = Vec::new();

        // Filter out our own sidebar pane
        let incoming: Vec<&ClaudeSession> = monitor
            .sessions
            .iter()
            .filter(|s| self.own_pid != Some(s.pane.pid))
            .collect();

        // Detect removed sessions
        self.sessions.retain(|managed| {
            let still_exists = incoming.iter().any(|s| s.pane.pid == managed.pid);
            if !still_exists {
                removed.push(managed.pid);
            }
            still_exists
        });

        // Detect new and state-changed sessions
        for session in &incoming {
            if let Some(existing) = self
                .sessions
                .iter_mut()
                .find(|m| m.pid == session.pane.pid)
            {
                // Update pane_id in case it changed after join-pane
                existing.pane_id.clone_from(&session.pane.id);
                existing.tmux_session = extract_tmux_session(&session.pane.id);
                existing.permission_mode = session.permission_mode.clone();
                if existing.cwd != session.pane.cwd {
                    existing.cwd.clone_from(&session.pane.cwd);
                    existing.worktree_name.clone_from(&session.pane.worktree_name);
                    existing.git_branch = resolve_git_branch(&session.pane.cwd);
                    existing.git_diff = resolve_git_diff(&session.pane.cwd);
                }
                if existing.state != session.state {
                    state_changed.push(existing.pid);
                    if matches!(session.state, ClaudeState::Working) {
                        existing.has_worked = true;
                    }
                    existing.state = session.state.clone();
                    existing.state_changed_at = session.state_changed_at;
                    existing.git_diff = resolve_git_diff(&existing.cwd);
                }
            } else {
                added.push(session.pane.pid);
                self.sessions.push(ManagedSession {
                    pid: session.pane.pid,
                    pane_id: session.pane.id.clone(),
                    project_name: session.pane.project_name.clone(),
                    state: session.state.clone(),
                    state_changed_at: session.state_changed_at,
                    marked: false,
                    title: None,
                    session_id: None,
                    model: None,
                    context_percent: None,
                    cwd: session.pane.cwd.clone(),
                    git_branch: resolve_git_branch(&session.pane.cwd),
                    auto_title: None,
                    permission_mode: session.permission_mode.clone(),
                    jsonl_mtime: None,
                    has_worked: matches!(session.state, ClaudeState::Working),
                    worktree_name: session.pane.worktree_name.clone(),
                    git_diff: resolve_git_diff(&session.pane.cwd),
                    tmux_session: extract_tmux_session(&session.pane.id),
                });
            }
        }

        // Sort by project_name for stable directory-based ordering
        let selected_pid = self.selected_session().map(|s| s.pid);
        self.sessions.sort_by(|a, b| a.project_name.cmp(&b.project_name));

        // Rebuild tabs from current sessions
        let matched = self.tab_state.rebuild_tabs(&self.sessions, self.initial_workspace.as_deref());
        if matched {
            self.initial_workspace = None;
        }

        // Recalculate selected_index within filtered list
        let filtered_len = self.filtered_sessions().len();
        if let Some(pid) = selected_pid {
            if let Some(pos) = self.filtered_sessions().iter().position(|s| s.pid == pid) {
                self.selected_index = pos;
            } else if filtered_len > 0 {
                self.selected_index = self.selected_index.min(filtered_len - 1);
            } else {
                self.selected_index = 0;
            }
        } else if filtered_len > 0 {
            self.selected_index = self.selected_index.min(filtered_len - 1);
        } else {
            self.selected_index = 0;
        }

        // Clamp prev_selected_index to filtered range
        if let Some(prev) = self.prev_selected_index
            && (filtered_len == 0 || prev >= filtered_len)
        {
            self.prev_selected_index = None;
        }

        // Apply buffered RPC messages to newly added sessions
        if !self.pending_rpc.is_empty() && !added.is_empty() {
            self.pending_rpc.retain(|msg| {
                let Some(pane_id) = msg.params.get("pane_id").and_then(|v| v.as_str()) else {
                    return false;
                };
                self.sessions.iter_mut().find(|s| s.pane_id == pane_id).is_none_or(|session| {
                    Self::apply_rpc_to_session(session, msg);
                    false // applied — remove from buffer
                })
            });
        }

        SyncDiff {
            added,
            removed,
            state_changed,
        }
    }

    /// Move selection down in the filtered list.
    pub fn select_next(&mut self) {
        let len = self.filtered_sessions().len();
        if len > 0 {
            self.prev_selected_index = Some(self.selected_index);
            self.selected_index = (self.selected_index + 1) % len;
        }
        self.preview_scroll = 0;
    }

    /// Move selection up in the filtered list.
    pub fn select_prev(&mut self) {
        let len = self.filtered_sessions().len();
        if len > 0 {
            self.prev_selected_index = Some(self.selected_index);
            if self.selected_index == 0 {
                self.selected_index = len - 1;
            } else {
                self.selected_index -= 1;
            }
        }
        self.preview_scroll = 0;
    }

    /// Switch to the previously selected session (swap selected and prev).
    pub fn select_prev_selected(&mut self) {
        if let Some(prev) = self.prev_selected_index {
            let len = self.filtered_sessions().len();
            if prev < len {
                self.prev_selected_index = Some(self.selected_index);
                self.selected_index = prev;
                self.preview_scroll = 0;
            }
        }
    }

    /// Scroll the preview up by `amount`, clamped to `max_scroll`.
    pub fn scroll_preview_up(&mut self, amount: u16, max_scroll: u16) {
        self.preview_scroll = self.preview_scroll.saturating_add(amount).min(max_scroll);
    }

    /// Scroll the preview down by `amount`, clamped to 0.
    pub const fn scroll_preview_down(&mut self, amount: u16) {
        self.preview_scroll = self.preview_scroll.saturating_sub(amount);
    }

    /// Reset preview scroll to bottom (0).
    pub const fn reset_preview_scroll(&mut self) {
        self.preview_scroll = 0;
    }

    /// Get the currently selected session from the filtered list.
    pub fn selected_session(&self) -> Option<&ManagedSession> {
        let filtered = self.filtered_sessions();
        filtered.get(self.selected_index).copied()
    }

    /// Get a mutable reference to the currently selected session (PID-based lookup).
    pub fn selected_session_mut(&mut self) -> Option<&mut ManagedSession> {
        let pid = self.selected_session().map(|s| s.pid)?;
        self.sessions.iter_mut().find(|s| s.pid == pid)
    }

    /// Get the pane ID of the currently selected session.
    pub fn selected_pane_id(&self) -> Option<&str> {
        self.selected_session().map(|s| s.pane_id.as_str())
    }

    /// Cycle the layout mode: `MainVertical` → Single → Grid → `EvenHorizontal` → `EvenVertical` → `MainHorizontal` → `MainVertical`.
    pub const fn cycle_layout_mode(&mut self) {
        self.layout_mode = self.layout_mode.next();
    }

    /// Toggle the mark on the currently selected session (PID-based).
    pub fn toggle_mark(&mut self) {
        let pid = self.selected_session().map(|s| s.pid);
        if let Some(pid) = pid
            && let Some(session) = self.sessions.iter_mut().find(|s| s.pid == pid)
        {
            session.marked = !session.marked;
        }
    }

    pub fn filtered_pane_ids(&self) -> Vec<String> {
        self.filtered_sessions()
            .iter()
            .map(|s| s.pane_id.clone())
            .collect()
    }

    /// Load historical plans for a given cwd/project. Skips slugs already in `self.plans`.
    pub fn load_historical_plans(&mut self, cwd: &str, project_name: &str) {
        let project_dir = crate::auto_title::cwd_to_project_dir(cwd);
        if !self.scanned_project_dirs.insert(project_dir) {
            return;
        }
        let historical = crate::auto_title::collect_all_plans_for_project(cwd, project_name);
        for plan in historical {
            if !self.plans.iter().any(|p| p.slug == plan.slug) {
                self.plans.push(plan);
            }
        }
    }

    /// Refresh auto titles for sessions that have `session_id` and no manual title.
    /// Also collects plan info from sessions with slugs.
    pub fn refresh_auto_titles(&mut self) {
        // Lazy-scan historical plans for newly discovered project dirs
        let unseen: Vec<(String, String)> = self
            .sessions
            .iter()
            .filter(|s| {
                let pd = crate::auto_title::cwd_to_project_dir(&s.cwd);
                !self.scanned_project_dirs.contains(&pd)
            })
            .map(|s| (s.cwd.clone(), s.project_name.clone()))
            .collect();
        for (cwd, project_name) in unseen {
            self.load_historical_plans(&cwd, &project_name);
        }

        for session in &mut self.sessions {
            if session.title.is_some() {
                continue;
            }
            if let (Some(session_id), cwd) = (&session.session_id, &session.cwd) {
                // Skip if the JSONL file's mtime hasn't changed since last read
                if let Some(path) = jsonl_path(cwd, session_id)
                    && let Ok(meta) = std::fs::metadata(&path)
                    && let Ok(mtime) = meta.modified()
                {
                    if session.jsonl_mtime == Some(mtime) {
                        continue;
                    }
                    session.jsonl_mtime = Some(mtime);
                }

                session.auto_title =
                    crate::auto_title::resolve_auto_title(cwd, session_id);

                if let Some(plan_info) = crate::auto_title::resolve_plan_info(
                    cwd,
                    session_id,
                    &session.project_name,
                ) {
                    if let Some(existing) =
                        self.plans.iter_mut().find(|p| p.slug == plan_info.slug)
                    {
                        *existing = plan_info;
                    } else {
                        self.plans.push(plan_info);
                    }
                }
            }
        }
    }

    /// Find the idle session for a given project name.
    /// If multiple idle sessions exist, return the one that has been idle the longest.
    pub fn find_idle_session_for_project(&self, project: &str) -> Option<&ManagedSession> {
        self.sessions
            .iter()
            .filter(|s| s.project_name == project && s.state == ClaudeState::Idle)
            .min_by_key(|s| s.state_changed_at)
    }

    /// Find a session by `pane_id`.
    pub fn find_session_by_pane_id(&self, pane_id: &str) -> Option<&ManagedSession> {
        self.sessions.iter().find(|s| s.pane_id == pane_id)
    }

    /// Refresh git info (branch names and diff) for all sessions.
    pub fn refresh_git_info(&mut self) {
        for session in &mut self.sessions {
            session.git_branch = resolve_git_branch(&session.cwd);
            session.git_diff = resolve_git_diff(&session.cwd);
        }
    }

    /// Handle an incoming RPC message, updating session metadata.
    /// If the target session is not yet known, the message is buffered in `pending_rpc`.
    pub fn handle_rpc_message(&mut self, msg: &crate::rpc::RpcMessage) {
        if msg.method == "send_text" {
            let Some(text) = msg.params.get("text").and_then(|v| v.as_str()) else {
                return;
            };
            // Resolve target pane: project-targeted or selected pane
            let target_pane = msg
                .params
                .get("project")
                .and_then(|v| v.as_str())
                .map_or_else(
                    || self.selected_pane_id().map(String::from),
                    |project| {
                        self.find_idle_session_for_project(project)
                            .map(|s| s.pane_id.clone())
                    },
                );
            let Some(pane_id) = target_pane else {
                return;
            };
            let no_execute = msg
                .params
                .get("no_execute")
                .and_then(serde_json::Value::as_bool)
                .unwrap_or(false);
            // Switch permission mode if requested
            if let Some(target_mode) = msg.params.get("mode").and_then(|v| v.as_str())
                && let Some(session) = self.find_session_by_pane_id(&pane_id)
            {
                let count =
                    permission_mode_switch_count(&session.permission_mode, target_mode);
                for _ in 0..count {
                    crate::event_handler::run_tmux(&[
                        "send-keys", "-t", &pane_id, "BTab",
                    ]);
                    std::thread::sleep(std::time::Duration::from_millis(100));
                }
                if count > 0 {
                    // Wait for mode switch to settle
                    std::thread::sleep(std::time::Duration::from_millis(200));
                }
            }
            // Use paste-buffer instead of send-keys -l.
            // Claude Code has a bug where send-keys stops working after
            // interrupting multi-line input with Esc,Esc.
            // See: https://github.com/anthropics/claude-code/issues/31739
            crate::event_handler::run_tmux(&[
                "set-buffer", "-b", "crmux-rpc", "--", text,
            ]);
            crate::event_handler::run_tmux(&[
                "paste-buffer", "-b", "crmux-rpc", "-t", &pane_id, "-p",
            ]);
            crate::event_handler::run_tmux(&["delete-buffer", "-b", "crmux-rpc"]);
            if !no_execute {
                std::thread::sleep(std::time::Duration::from_millis(200));
                crate::event_handler::run_tmux(&["send-keys", "-t", &pane_id, "Enter"]);
            }
            return;
        }

        let Some(pane_id) = msg.params.get("pane_id").and_then(|v| v.as_str()) else {
            return;
        };

        if let Some(session) = self.sessions.iter_mut().find(|s| s.pane_id == pane_id) {
            Self::apply_rpc_to_session(session, msg);
        } else {
            // Session not yet discovered — buffer for later
            const MAX_PENDING: usize = 20;
            if self.pending_rpc.len() < MAX_PENDING {
                self.pending_rpc.push(msg.clone());
            }
        }
    }

    /// Apply an RPC message to the matching session.
    fn apply_rpc_to_session(session: &mut ManagedSession, msg: &crate::rpc::RpcMessage) {
        match msg.method.as_str() {
            "session_start" => {
                session.session_id =
                    msg.params.get("session_id").and_then(|v| v.as_str()).map(String::from);
                session.jsonl_mtime = None;
                session.model =
                    msg.params.get("model").and_then(|v| v.as_str()).map(String::from);
            }
            "status_update" => {
                // Extract session_id (fills in if not already set via session_start)
                if session.session_id.is_none() {
                    session.session_id = msg
                        .params
                        .get("session_id")
                        .and_then(|v| v.as_str())
                        .map(String::from);
                    session.jsonl_mtime = None;
                }
                // Extract model.display_name from nested JSON
                if let Some(display_name) = msg
                    .params
                    .get("model")
                    .and_then(|m| m.get("display_name"))
                    .and_then(|v| v.as_str())
                {
                    session.model = Some(display_name.to_string());
                }
                // Extract context window used_percentage
                if let Some(pct) = msg
                    .params
                    .get("context_window")
                    .and_then(|c| c.get("used_percentage"))
                    .and_then(serde_json::Value::as_u64)
                {
                    // Context usage is 0–100, fits in u8.
                    #[allow(clippy::cast_possible_truncation)]
                    let pct = pct as u8;
                    session.context_percent = Some(pct);
                } else {
                    session.context_percent = Some(0);
                }
            }
            _ => {}
        }
    }

    /// Serialize all sessions and visibility state as a JSON value.
    /// If `params` contains a `"project"` key, filter sessions by project name.
    pub fn serialize_sessions(&self, params: &serde_json::Value) -> serde_json::Value {
        let project_filter = params.get("project").and_then(|v| v.as_str());
        let sessions: Vec<serde_json::Value> = self
            .filtered_sessions()
            .into_iter()
            .filter(|s| {
                project_filter.is_none_or(|name| s.project_name == name)
            })
            .map(|s| {
                let state_name = match s.state {
                    ClaudeState::Idle => "Idle",
                    ClaudeState::Working => "Working",
                    ClaudeState::WaitingForApproval => "WaitingForApproval",
                };
                serde_json::json!({
                    "pane_id": s.pane_id,
                    "pid": s.pid,
                    "project_name": s.project_name,
                    "state": state_name,
                    "elapsed_secs": s.state_changed_at.elapsed().as_secs(),
                    "model": s.model,
                    "context_percent": s.context_percent,
                    "title": s.display_title(),
                    "session_id": s.session_id,
                    "git_branch": s.git_branch,
                })
            })
            .collect();
        serde_json::json!({
            "sessions": sessions,
            "visible": self.claudeye_visible,
        })
    }

    /// Serialize all accumulated plans as a JSON value.
    /// If `params` contains a `"project"` key, filter plans by project name.
    pub fn serialize_plans(&self, params: &serde_json::Value) -> serde_json::Value {
        let project_filter = params.get("project").and_then(|v| v.as_str());
        let plans: Vec<serde_json::Value> = self
            .plans
            .iter()
            .filter(|p| {
                project_filter.is_none_or(|name| p.project_name == name)
            })
            .map(|p| {
                serde_json::json!({
                    "slug": p.slug,
                    "title": p.title,
                    "path": p.path,
                    "project_name": p.project_name,
                    "session_id": p.session_id,
                })
            })
            .collect();
        serde_json::json!({ "plans": plans })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use tmux_claude_state::tmux::PaneInfo;

    fn make_session(pid: u32, pane_id: &str, project: &str, state: ClaudeState) -> ClaudeSession {
        ClaudeSession {
            pane: PaneInfo {
                id: pane_id.to_string(),
                pid,
                cwd: format!("/home/user/{project}"),
                project_name: project.to_string(),
                worktree_name: None,
            },
            state,
            permission_mode: PermissionMode::AskBeforeEdits,
            state_changed_at: Instant::now(),
        }
    }

    fn make_monitor(sessions: Vec<ClaudeSession>) -> MonitorState {
        MonitorState {
            sessions,
            any_claude_focused: false,
        }
    }

    // --- New session detection ---

    #[test]
    fn test_sync_detects_new_sessions() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
            make_session(200, "%2", "project-b", ClaudeState::Working),
        ]);

        let diff = app.sync_with_monitor(&monitor);

        assert_eq!(diff.added, vec![100, 200]);
        assert!(diff.removed.is_empty());
        assert!(diff.state_changed.is_empty());
        assert_eq!(app.sessions.len(), 2);
    }

    // --- Session removal ---

    #[test]
    fn test_sync_detects_removed_sessions() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
            make_session(200, "%2", "project-b", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor1);

        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        let diff = app.sync_with_monitor(&monitor2);

        assert!(diff.added.is_empty());
        assert_eq!(diff.removed, vec![200]);
        assert_eq!(app.sessions.len(), 1);
    }

    // --- State change detection ---

    #[test]
    fn test_sync_detects_state_change() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);

        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        let diff = app.sync_with_monitor(&monitor2);

        assert!(diff.added.is_empty());
        assert!(diff.removed.is_empty());
        assert_eq!(diff.state_changed, vec![100]);
        assert_eq!(app.sessions[0].state, ClaudeState::Working);
    }

    // --- Sidebar exclusion ---

    #[test]
    fn test_sync_excludes_own_pid() {
        let mut app = AppState::new(Some(999));
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
            make_session(999, "%9", "sidebar", ClaudeState::Working),
        ]);

        let diff = app.sync_with_monitor(&monitor);

        assert_eq!(diff.added, vec![100]);
        assert_eq!(app.sessions.len(), 1);
        assert_eq!(app.sessions[0].pid, 100);
    }

    // --- cwd/worktree update on sync ---

    #[test]
    fn test_sync_updates_cwd_and_worktree_when_changed() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);

        assert_eq!(app.sessions[0].cwd, "/home/user/project-a");
        assert_eq!(app.sessions[0].worktree_name, None);

        // Simulate worktree move: cwd and worktree_name change
        let monitor2 = make_monitor(vec![ClaudeSession {
            pane: PaneInfo {
                id: "%1".to_string(),
                pid: 100,
                cwd: "/home/user/project-a-worktree".to_string(),
                project_name: "project-a".to_string(),
                worktree_name: Some("feature-branch".to_string()),
            },
            state: ClaudeState::Idle,
            permission_mode: PermissionMode::AskBeforeEdits,
            state_changed_at: Instant::now(),
        }]);
        app.sync_with_monitor(&monitor2);

        assert_eq!(app.sessions[0].cwd, "/home/user/project-a-worktree");
        assert_eq!(
            app.sessions[0].worktree_name,
            Some("feature-branch".to_string())
        );
    }

    // --- Selection operations ---

    #[test]
    fn test_select_next_wraps_around() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.selected_index, 0);
        app.select_next();
        assert_eq!(app.selected_index, 1);
        app.select_next();
        assert_eq!(app.selected_index, 2);
        app.select_next();
        assert_eq!(app.selected_index, 0); // wraps
    }

    #[test]
    fn test_select_prev_wraps_around() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.selected_index, 0);
        app.select_prev();
        assert_eq!(app.selected_index, 1); // wraps to end
        app.select_prev();
        assert_eq!(app.selected_index, 0);
    }

    #[test]
    fn test_select_on_empty() {
        let mut app = AppState::new(None);
        app.select_next();
        assert_eq!(app.selected_index, 0);
        app.select_prev();
        assert_eq!(app.selected_index, 0);
    }

    #[test]
    fn test_selected_session() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.selected_session().unwrap().pid, 100);
        app.select_next();
        assert_eq!(app.selected_session().unwrap().pid, 200);
    }

    #[test]
    fn test_selected_pane_id() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.selected_pane_id(), Some("%1"));
    }

    #[test]
    fn test_selected_pane_id_empty() {
        let app = AppState::new(None);
        assert_eq!(app.selected_pane_id(), None);
    }

    // --- selected_index fix on removal ---

    #[test]
    fn test_selected_index_fixed_on_removal() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.selected_index = 2; // select last

        // Remove all but one
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);

        assert_eq!(app.selected_index, 0);
    }

    // --- Pane ID update after join-pane ---

    // --- PreviewEntry ---

    #[test]
    fn test_preview_entry_with_title() {
        let entry = PreviewEntry {
            name: "crmux".to_string(),
            pane_id: "%1".to_string(),
            index: 0,
            title: Some("development".to_string()),
            git_branch: None,
            worktree_name: None,
            content: "hello".to_string(),
            cursor_pos: None,
            git_diff: None,
            state: ClaudeState::Idle,
            has_worked: false,
            state_changed_at: Instant::now(),
        };
        assert_eq!(entry.name, "crmux");
        assert_eq!(entry.title, Some("development".to_string()));
    }

    #[test]
    fn test_preview_entry_without_title() {
        let entry = PreviewEntry {
            name: "crmux".to_string(),
            pane_id: "%1".to_string(),
            index: 0,
            title: None,
            git_branch: None,
            worktree_name: None,
            content: "hello".to_string(),
            cursor_pos: None,
            git_diff: None,
            state: ClaudeState::Idle,
            has_worked: false,
            state_changed_at: Instant::now(),
        };
        assert_eq!(entry.title, None);
    }

    // --- Preview content ---

    #[test]
    fn test_preview_contents_default_empty() {
        let app = AppState::new(None);
        assert!(app.preview_contents.is_empty());
    }

    // --- Input mode ---

    #[test]
    fn test_initial_input_mode_is_normal() {
        let app = AppState::new(None);
        assert_eq!(app.input_mode, InputMode::Normal);
    }

    #[test]
    fn test_initial_input_buffer_is_empty() {
        let app = AppState::new(None);
        assert!(app.input_buffer.is_empty());
    }

    #[test]
    fn test_initial_show_help_is_false() {
        let app = AppState::new(None);
        assert!(!app.show_help);
    }

    // --- Mark operations ---

    #[test]
    fn test_new_session_is_not_marked() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert!(!app.sessions[0].marked);
    }

    #[test]
    fn test_toggle_mark_marks_selected() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.toggle_mark();
        assert!(app.sessions[0].marked);
        assert!(!app.sessions[1].marked);
    }

    #[test]
    fn test_toggle_mark_unmarks_marked() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.toggle_mark();
        assert!(app.sessions[0].marked);
        app.toggle_mark();
        assert!(!app.sessions[0].marked);
    }

    #[test]
    fn test_toggle_mark_on_empty_does_nothing() {
        let mut app = AppState::new(None);
        app.toggle_mark(); // should not panic
    }

    #[test]
    fn test_marked_sessions_returns_marked_only() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.selected_index = 0;
        app.toggle_mark();
        app.selected_index = 2;
        app.toggle_mark();

        let marked: Vec<_> = app.sessions.iter().filter(|s| s.marked).collect();
        assert_eq!(marked.len(), 2);
        assert_eq!(marked[0].pid, 100);
        assert_eq!(marked[1].pid, 300);
    }

    #[test]
    fn test_marked_sessions_empty_when_none_marked() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert!(app.sessions.iter().all(|s| !s.marked));
    }

    #[test]
    fn test_mark_preserved_on_sync() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.toggle_mark(); // mark session 100

        // Re-sync with same sessions (state change)
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Working),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);

        assert!(app.sessions[0].marked); // mark preserved
        assert!(!app.sessions[1].marked);
    }

    // --- Title field ---

    #[test]
    fn test_new_session_has_no_title() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].title, None);
    }

    #[test]
    fn test_title_preserved_on_sync() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.sessions[0].title = Some("refactoring auth".to_string());

        // Re-sync with state change
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor2);

        assert_eq!(app.sessions[0].title, Some("refactoring auth".to_string()));
    }

    // --- selected_session_mut ---

    #[test]
    fn test_selected_session_mut() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.selected_session_mut().unwrap().title = Some("testing".to_string());
        assert_eq!(app.sessions[0].title, Some("testing".to_string()));
        assert_eq!(app.sessions[1].title, None);
    }

    #[test]
    fn test_selected_session_mut_empty() {
        let mut app = AppState::new(None);
        assert!(app.selected_session_mut().is_none());
    }

    // --- Preview scroll tests ---

    #[test]
    fn test_initial_preview_scroll_is_zero() {
        let app = AppState::new(None);
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_initial_preview_height_is_zero() {
        let app = AppState::new(None);
        assert_eq!(app.preview_height, 0);
    }

    #[test]
    fn test_initial_pending_g_is_false() {
        let app = AppState::new(None);
        assert!(!app.pending_g);
    }

    #[test]
    fn test_scroll_preview_up() {
        let mut app = AppState::new(None);
        app.scroll_preview_up(10, 90);
        assert_eq!(app.preview_scroll, 10);
    }

    #[test]
    fn test_scroll_preview_up_clamps_to_max() {
        let mut app = AppState::new(None);
        app.scroll_preview_up(100, 90);
        assert_eq!(app.preview_scroll, 90);
    }

    #[test]
    fn test_scroll_preview_up_saturating() {
        let mut app = AppState::new(None);
        app.preview_scroll = 80;
        app.scroll_preview_up(20, 90);
        assert_eq!(app.preview_scroll, 90);
    }

    #[test]
    fn test_scroll_preview_down() {
        let mut app = AppState::new(None);
        app.preview_scroll = 20;
        app.scroll_preview_down(10);
        assert_eq!(app.preview_scroll, 10);
    }

    #[test]
    fn test_scroll_preview_down_clamps_to_zero() {
        let mut app = AppState::new(None);
        app.preview_scroll = 5;
        app.scroll_preview_down(10);
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_reset_preview_scroll() {
        let mut app = AppState::new(None);
        app.preview_scroll = 42;
        app.reset_preview_scroll();
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_select_next_resets_scroll() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.preview_scroll = 20;
        app.select_next();
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_select_prev_resets_scroll() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.selected_index = 1;
        app.preview_scroll = 15;
        app.select_prev();
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_pane_id_updated_on_sync() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        assert_eq!(app.sessions[0].pane_id, "%1");

        // Same PID, different pane_id (after join-pane)
        let monitor2 = make_monitor(vec![
            make_session(100, "%5", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);
        assert_eq!(app.sessions[0].pane_id, "%5");
    }

    // --- RPC message handling ---

    #[test]
    fn test_new_session_has_no_rpc_fields() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].session_id, None);
        assert_eq!(app.sessions[0].model, None);
    }

    #[test]
    fn test_handle_rpc_session_start() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-abc",
                "model": "claude-sonnet-4-6",
            }),
        });

        assert_eq!(app.sessions[0].session_id, Some("sess-abc".to_string()));
        assert_eq!(app.sessions[0].model, Some("claude-sonnet-4-6".to_string()));
    }

    #[test]
    fn test_handle_rpc_session_start_unknown_pane() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%99",
                "session_id": "sess-xyz",
            }),
        });

        // Should not crash, and existing session should be unchanged
        assert_eq!(app.sessions[0].session_id, None);
        // Unknown pane should be buffered for later
        assert_eq!(app.pending_rpc.len(), 1);
    }

    #[test]
    fn test_handle_rpc_missing_pane_id() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "session_id": "sess-abc",
            }),
        });

        // Without pane_id, nothing should be updated
        assert_eq!(app.sessions[0].session_id, None);
    }

    #[test]
    fn test_handle_rpc_unknown_method() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "unknown_method".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
            }),
        });

        // Unknown method should not change anything
        assert_eq!(app.sessions[0].session_id, None);
    }

    #[test]
    fn test_rpc_fields_preserved_on_sync() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-abc",
                "model": "opus",
            }),
        });

        // Re-sync with state change
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor2);

        assert_eq!(app.sessions[0].session_id, Some("sess-abc".to_string()));
        assert_eq!(app.sessions[0].model, Some("opus".to_string()));
    }

    // --- display_title ---

    #[test]
    fn test_display_title_manual_over_auto() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].title = Some("manual".to_string());
        app.sessions[0].auto_title = Some("auto".to_string());
        assert_eq!(app.sessions[0].display_title(), Some("manual"));
    }

    #[test]
    fn test_display_title_auto_when_no_manual() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].auto_title = Some("auto".to_string());
        assert_eq!(app.sessions[0].display_title(), Some("auto"));
    }

    #[test]
    fn test_display_title_none_when_both_empty() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].display_title(), None);
    }

    // --- cwd and git_branch ---

    #[test]
    fn test_new_session_has_cwd() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].cwd, "/home/user/project-a");
    }

    #[test]
    fn test_new_session_has_no_git_branch() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].git_branch, None);
    }

    #[test]
    fn test_git_branch_preserved_on_sync() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.sessions[0].git_branch = Some("feature-branch".to_string());

        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor2);

        assert_eq!(app.sessions[0].git_branch, Some("feature-branch".to_string()));
    }

    #[test]
    fn test_refresh_git_info_sets_branch_for_valid_repo() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        // Use this repo's own cwd for a known git repo
        app.sessions[0].cwd = env!("CARGO_MANIFEST_DIR").to_string();
        app.refresh_git_info();
        assert!(app.sessions[0].git_branch.is_some());
    }

    #[test]
    fn test_refresh_git_info_none_for_non_repo() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "tmp", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].cwd = "/tmp".to_string();
        app.refresh_git_info();
        assert_eq!(app.sessions[0].git_branch, None);
    }

    // --- Pending RPC buffer ---

    #[test]
    fn test_rpc_before_session_is_buffered() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-early",
                "model": "opus",
            }),
        });

        // Should be buffered in pending_rpc
        assert_eq!(app.pending_rpc.len(), 1);
        assert_eq!(app.pending_rpc[0].params["pane_id"], "%1");
    }

    #[test]
    fn test_pending_rpc_applied_after_sync() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);

        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-early",
                "model": "opus",
            }),
        });

        // Now session appears via monitor
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Pending RPC should have been applied and removed
        assert!(app.pending_rpc.is_empty());
        assert_eq!(app.sessions[0].session_id, Some("sess-early".to_string()));
        assert_eq!(app.sessions[0].model, Some("opus".to_string()));
    }

    #[test]
    fn test_pending_rpc_unmatched_retained() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);

        // Two RPCs for different panes
        for (pane, sid) in [("%1", "sess-1"), ("%2", "sess-2")] {
            app.handle_rpc_message(&RpcMessage {
                method: "session_start".to_string(),
                params: serde_json::json!({
                    "pane_id": pane,
                    "session_id": sid,
                }),
            });
        }
        assert_eq!(app.pending_rpc.len(), 2);

        // Only %1 session appears
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // %1 applied and removed, %2 still pending
        assert_eq!(app.pending_rpc.len(), 1);
        assert_eq!(app.pending_rpc[0].params["pane_id"], "%2");
        assert_eq!(app.sessions[0].session_id, Some("sess-1".to_string()));
    }

    // --- status_update RPC ---

    #[test]
    fn test_handle_rpc_status_update_sets_model_display_name() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "model": { "display_name": "Opus" },
            }),
        });

        assert_eq!(app.sessions[0].model, Some("Opus".to_string()));
    }

    #[test]
    fn test_status_update_overwrites_session_start_model() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // session_start sets model to ID format
        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-abc",
                "model": "claude-opus-4-6",
            }),
        });
        assert_eq!(app.sessions[0].model, Some("claude-opus-4-6".to_string()));

        // status_update overwrites with display_name
        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "model": { "display_name": "Opus" },
            }),
        });
        assert_eq!(app.sessions[0].model, Some("Opus".to_string()));
    }

    #[test]
    fn test_status_update_without_model_is_noop() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].model = Some("existing".to_string());

        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
            }),
        });

        // Model should remain unchanged
        assert_eq!(app.sessions[0].model, Some("existing".to_string()));
    }

    // --- context_percent ---

    #[test]
    fn test_status_update_sets_context_percent() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "model": { "display_name": "Opus" },
                "context_window": {
                    "used_percentage": 50,
                },
            }),
        });

        assert_eq!(app.sessions[0].context_percent, Some(50));
    }

    #[test]
    fn test_status_update_context_percent_zero_when_no_used_percentage() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "model": { "display_name": "Opus" },
                "context_window": {},
            }),
        });

        assert_eq!(app.sessions[0].context_percent, Some(0));
    }

    #[test]
    fn test_status_update_sets_session_id() {
        use crate::rpc::RpcMessage;
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].session_id, None);

        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-from-statusline",
                "model": { "display_name": "Opus" },
            }),
        });

        assert_eq!(app.sessions[0].session_id, Some("sess-from-statusline".to_string()));
    }

    #[test]
    fn test_status_update_does_not_overwrite_existing_session_id() {
        use crate::rpc::RpcMessage;
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        // session_start sets session_id first
        app.handle_rpc_message(&RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-original",
                "model": "opus",
            }),
        });
        assert_eq!(app.sessions[0].session_id, Some("sess-original".to_string()));

        // status_update should NOT overwrite existing session_id
        app.handle_rpc_message(&RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "sess-new",
                "model": { "display_name": "Opus" },
            }),
        });
        assert_eq!(app.sessions[0].session_id, Some("sess-original".to_string()));
    }

    #[test]
    fn test_new_session_has_no_context_percent() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].context_percent, None);
    }

    // --- claudeye_visible ---

    #[test]
    fn test_claudeye_visible_default_false() {
        let app = AppState::new(None);
        assert!(!app.claudeye_visible);
    }

    #[test]
    fn test_claudeye_visible_toggle() {
        let mut app = AppState::new(None);
        app.claudeye_visible = false;
        assert!(!app.claudeye_visible);
        app.claudeye_visible = true;
        assert!(app.claudeye_visible);
    }

    // --- serialize_sessions ---

    #[test]
    fn test_serialize_sessions_empty() {
        let app = AppState::new(None);
        let result = app.serialize_sessions(&serde_json::json!({}));
        assert_eq!(result["sessions"], serde_json::json!([]));
        assert_eq!(result["visible"], false);
    }

    #[test]
    fn test_serialize_sessions_one() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].model = Some("Opus".to_string());
        app.sessions[0].context_percent = Some(23);
        app.sessions[0].title = Some("implementing feature X".to_string());
        app.sessions[0].session_id = Some("abc-123".to_string());
        app.sessions[0].git_branch = Some("main".to_string());

        let result = app.serialize_sessions(&serde_json::json!({}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions.len(), 1);

        let s = &sessions[0];
        assert_eq!(s["pane_id"], "%1");
        assert_eq!(s["pid"], 100);
        assert_eq!(s["project_name"], "crmux");
        assert_eq!(s["state"], "Working");
        assert_eq!(s["model"], "Opus");
        assert_eq!(s["context_percent"], 23);
        assert_eq!(s["title"], "implementing feature X");
        assert_eq!(s["session_id"], "abc-123");
        assert_eq!(s["git_branch"], "main");
        // elapsed_secs should be a non-negative number
        assert!(s["elapsed_secs"].as_u64().is_some());
        assert_eq!(result["visible"], false);
    }

    #[test]
    fn test_serialize_sessions_waiting_for_approval_state() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::WaitingForApproval),
        ]);
        app.sync_with_monitor(&monitor);

        let result = app.serialize_sessions(&serde_json::json!({}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions[0]["state"], "WaitingForApproval");
    }

    #[test]
    fn test_serialize_sessions_visible_true() {
        let mut app = AppState::new(None);
        app.claudeye_visible = true;
        let result = app.serialize_sessions(&serde_json::json!({}));
        assert_eq!(result["visible"], true);
    }

    #[test]
    fn test_serialize_sessions_filtered_by_tab() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
            make_session(200, "%2", "aegis", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // All tab returns all sessions
        let result = app.serialize_sessions(&serde_json::json!({}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions.len(), 3);

        // Select "crmux" project tab
        app.tab_state.selected_tab = 2; // Tab::Project("crmux")
        let result = app.serialize_sessions(&serde_json::json!({}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions.len(), 2);
        assert!(sessions.iter().all(|s| s["project_name"] == "crmux"));
    }

    #[test]
    fn test_serialize_sessions_filter_by_project_param() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
            make_session(200, "%2", "aegis", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        let result = app.serialize_sessions(&serde_json::json!({"project": "crmux"}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions.len(), 2);
        assert!(sessions.iter().all(|s| s["project_name"] == "crmux"));
    }

    #[test]
    fn test_serialize_sessions_filter_by_project_no_match() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        let result = app.serialize_sessions(&serde_json::json!({"project": "nonexistent"}));
        let sessions = result["sessions"].as_array().unwrap();
        assert_eq!(sessions.len(), 0);
    }

    // --- TabState tests ---

    #[test]
    fn test_tab_state_initial() {
        let ts = TabState::new();
        assert!(ts.tabs.is_empty());
        assert_eq!(ts.selected_tab, 0);
        assert_eq!(*ts.current_tab(), Tab::All);
    }

    #[test]
    fn test_tab_state_rebuild_from_sessions() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
            make_session(200, "%2", "aegis", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.tab_state.tabs, vec![
            Tab::All,
            Tab::Project("aegis".to_string()),
            Tab::Project("crmux".to_string()),
        ]);
    }

    #[test]
    fn test_tab_state_alphabetical_sort() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "zebra", ClaudeState::Idle),
            make_session(200, "%2", "alpha", ClaudeState::Idle),
            make_session(300, "%3", "middle", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.tab_state.tabs, vec![
            Tab::All,
            Tab::Project("alpha".to_string()),
            Tab::Project("middle".to_string()),
            Tab::Project("zebra".to_string()),
        ]);
    }

    #[test]
    fn test_tab_state_dedup() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.tab_state.tabs, vec![
            Tab::All,
            Tab::Project("crmux".to_string()),
        ]);
    }

    #[test]
    fn test_tab_state_selection_maintained_on_rebuild() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.tab_state.selected_tab = 2; // crmux tab

        // Re-sync with an additional project
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Working),
            make_session(300, "%3", "zeta", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);

        // crmux tab should still be selected
        assert_eq!(*app.tab_state.current_tab(), Tab::Project("crmux".to_string()));
    }

    #[test]
    fn test_tab_state_fallback_on_project_disappear() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        app.tab_state.selected_tab = 2; // crmux tab

        // crmux session disappears
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);

        // Should fallback to All
        assert_eq!(*app.tab_state.current_tab(), Tab::All);
    }

    #[test]
    fn test_tab_state_next_prev_wrap() {
        let mut ts = TabState::new();
        ts.tabs = vec![Tab::All, Tab::Project("a".into()), Tab::Project("b".into())];
        ts.selected_tab = 0;

        ts.select_next_tab();
        assert_eq!(ts.selected_tab, 1);
        ts.select_next_tab();
        assert_eq!(ts.selected_tab, 2);
        ts.select_next_tab();
        assert_eq!(ts.selected_tab, 0); // wraps

        ts.select_prev_tab();
        assert_eq!(ts.selected_tab, 2); // wraps back
        ts.select_prev_tab();
        assert_eq!(ts.selected_tab, 1);
    }

    // --- Filtering tests ---

    #[test]
    fn test_filtered_sessions_all_tab() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        let filtered = app.filtered_sessions();
        assert_eq!(filtered.len(), 2);
    }

    #[test]
    fn test_filtered_sessions_project_tab() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);
        app.tab_state.selected_tab = 2; // crmux tab

        let filtered = app.filtered_sessions();
        assert_eq!(filtered.len(), 2);
        assert!(filtered.iter().all(|s| s.project_name == "crmux"));
    }

    #[test]
    fn test_selected_index_adjusted_on_filter() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);
        app.selected_index = 2; // third item in All tab

        // Switch to aegis tab (only 1 session)
        app.tab_state.selected_tab = 1; // aegis
        let filtered = app.filtered_sessions();
        // selected_index should not exceed filtered len
        let adjusted = app.selected_index.min(filtered.len().saturating_sub(1));
        assert!(adjusted < filtered.len());
    }

    #[test]
    fn test_select_next_on_filtered() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        // Switch to crmux tab
        app.tab_state.selected_tab = 2;
        app.selected_index = 0;

        app.select_next();
        assert_eq!(app.selected_index, 1);
        app.select_next();
        assert_eq!(app.selected_index, 0); // wraps at 2, not 3
    }

    // --- permission_mode_switch_count ---

    #[test]
    fn test_mode_switch_ask_to_plan() {
        let count = permission_mode_switch_count(&PermissionMode::AskBeforeEdits, "plan-mode");
        assert_eq!(count, 2);
    }

    #[test]
    fn test_mode_switch_ask_to_auto_edit() {
        let count = permission_mode_switch_count(&PermissionMode::AskBeforeEdits, "accept-edits");
        assert_eq!(count, 1);
    }

    #[test]
    fn test_mode_switch_plan_to_ask() {
        let count = permission_mode_switch_count(&PermissionMode::PlanMode, "accept-edits");
        assert_eq!(count, 2);
    }

    #[test]
    fn test_mode_switch_same_mode_returns_zero() {
        let count = permission_mode_switch_count(&PermissionMode::PlanMode, "plan-mode");
        assert_eq!(count, 0);
    }

    #[test]
    fn test_mode_switch_unknown_target_returns_zero() {
        let count = permission_mode_switch_count(&PermissionMode::AskBeforeEdits, "unknown");
        assert_eq!(count, 0);
    }

    #[test]
    fn test_mode_switch_auto_edit_to_plan() {
        let count = permission_mode_switch_count(&PermissionMode::EditAutomatically, "plan-mode");
        assert_eq!(count, 1);
    }

    // --- send_text RPC ---

    #[test]
    fn test_send_text_missing_text_does_not_panic() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({}),
        });
        // Should not panic
    }

    #[test]
    fn test_send_text_no_selected_session_does_not_panic() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        // No sessions — selected_pane_id() returns None

        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({ "text": "hello" }),
        });
        // Should not panic
    }

    #[test]
    fn test_send_text_does_not_enter_pending_rpc() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        // No sessions — would normally be buffered in pending_rpc

        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({ "text": "hello" }),
        });

        assert!(app.pending_rpc.is_empty());
    }

    // --- find_idle_session_for_project ---

    #[test]
    fn test_find_idle_session_for_project_returns_idle_session() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        let session = app.find_idle_session_for_project("crmux");
        assert!(session.is_some());
        assert_eq!(session.unwrap().pid, 100);
    }

    #[test]
    fn test_find_idle_session_for_project_no_idle() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        assert!(app.find_idle_session_for_project("crmux").is_none());
    }

    #[test]
    fn test_find_idle_session_for_project_no_matching_project() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert!(app.find_idle_session_for_project("torudo").is_none());
    }

    #[test]
    fn test_find_idle_session_for_project_picks_longest_idle() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        // Make session 100 have an older state_changed_at (longer idle)
        app.sessions[0].state_changed_at = Instant::now() - std::time::Duration::from_secs(60);
        app.sessions[1].state_changed_at = Instant::now() - std::time::Duration::from_secs(10);

        let session = app.find_idle_session_for_project("crmux");
        assert_eq!(session.unwrap().pid, 100);
    }

    // --- send_text with project ---

    #[test]
    fn test_send_text_with_project_no_idle_session_does_not_panic() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({ "text": "hello", "project": "crmux" }),
        });
        // Should not panic
    }

    #[test]
    fn test_send_text_no_execute_works_without_project() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.selected_index = 0;

        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({
                "text": "hello",
                "no_execute": true
            }),
        });
        // Should not panic — no_execute works on selected pane too
    }

    #[test]
    fn test_send_text_with_project_does_not_enter_pending_rpc() {
        use crate::rpc::RpcMessage;

        let mut app = AppState::new(None);
        app.handle_rpc_message(&RpcMessage {
            method: "send_text".to_string(),
            params: serde_json::json!({ "text": "hello", "project": "nonexistent" }),
        });

        assert!(app.pending_rpc.is_empty());
    }

    #[test]
    fn test_sync_rebuilds_tabs() {
        let mut app = AppState::new(None);
        let monitor1 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor1);
        assert_eq!(app.tab_state.tabs.len(), 2); // All + aegis

        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);
        assert_eq!(app.tab_state.tabs.len(), 3); // All + aegis + crmux
    }

    // --- plans ---

    #[test]
    fn test_plans_initially_empty() {
        let app = AppState::new(None);
        assert!(app.plans.is_empty());
    }

    #[test]
    fn test_serialize_plans_empty() {
        let app = AppState::new(None);
        let result = app.serialize_plans(&serde_json::json!({}));
        assert_eq!(result["plans"], serde_json::json!([]));
    }

    #[test]
    fn test_serialize_plans_with_entries() {
        let mut app = AppState::new(None);
        app.plans.push(crate::auto_title::PlanInfo {
            slug: "my-plan".to_string(),
            title: "My Plan Title".to_string(),
            path: "/home/user/.claude/plans/my-plan.md".to_string(),
            project_name: "crmux".to_string(),
            session_id: "sess-001".to_string(),
        });

        let result = app.serialize_plans(&serde_json::json!({}));
        let plans = result["plans"].as_array().unwrap();
        assert_eq!(plans.len(), 1);
        assert_eq!(plans[0]["slug"], "my-plan");
        assert_eq!(plans[0]["title"], "My Plan Title");
        assert_eq!(plans[0]["path"], "/home/user/.claude/plans/my-plan.md");
        assert_eq!(plans[0]["project_name"], "crmux");
        assert_eq!(plans[0]["session_id"], "sess-001");
    }

    #[test]
    fn test_serialize_plans_filter_by_project() {
        let mut app = AppState::new(None);
        app.plans.push(crate::auto_title::PlanInfo {
            slug: "plan-a".to_string(),
            title: "Plan A".to_string(),
            path: "/path/plan-a.md".to_string(),
            project_name: "crmux".to_string(),
            session_id: "sess-001".to_string(),
        });
        app.plans.push(crate::auto_title::PlanInfo {
            slug: "plan-b".to_string(),
            title: "Plan B".to_string(),
            path: "/path/plan-b.md".to_string(),
            project_name: "other-project".to_string(),
            session_id: "sess-002".to_string(),
        });

        let result = app.serialize_plans(&serde_json::json!({"project": "crmux"}));
        let plans = result["plans"].as_array().unwrap();
        assert_eq!(plans.len(), 1);
        assert_eq!(plans[0]["slug"], "plan-a");
        assert_eq!(plans[0]["project_name"], "crmux");
    }

    #[test]
    fn test_serialize_plans_filter_no_match() {
        let mut app = AppState::new(None);
        app.plans.push(crate::auto_title::PlanInfo {
            slug: "plan-a".to_string(),
            title: "Plan A".to_string(),
            path: "/path/plan-a.md".to_string(),
            project_name: "crmux".to_string(),
            session_id: "sess-001".to_string(),
        });

        let result = app.serialize_plans(&serde_json::json!({"project": "nonexistent"}));
        let plans = result["plans"].as_array().unwrap();
        assert_eq!(plans.len(), 0);
    }

    #[test]
    fn test_plans_dedup_by_slug() {
        let mut app = AppState::new(None);
        let plan1 = crate::auto_title::PlanInfo {
            slug: "my-plan".to_string(),
            title: "Old Title".to_string(),
            path: "/path/my-plan.md".to_string(),
            project_name: "proj".to_string(),
            session_id: "sess-001".to_string(),
        };
        let plan2 = crate::auto_title::PlanInfo {
            slug: "my-plan".to_string(),
            title: "New Title".to_string(),
            path: "/path/my-plan.md".to_string(),
            project_name: "proj".to_string(),
            session_id: "sess-002".to_string(),
        };

        // Simulate what refresh_auto_titles does
        app.plans.push(plan1);
        if let Some(existing) = app.plans.iter_mut().find(|p| p.slug == plan2.slug) {
            *existing = plan2;
        }

        assert_eq!(app.plans.len(), 1);
        assert_eq!(app.plans[0].title, "New Title");
        assert_eq!(app.plans[0].session_id, "sess-002");
    }

    #[test]
    fn test_load_historical_plans_no_overwrite() {
        let mut app = AppState::new(None);
        // Pre-populate with an existing plan
        app.plans.push(crate::auto_title::PlanInfo {
            slug: "existing-plan".to_string(),
            title: "Current Title".to_string(),
            path: "/path/existing-plan.md".to_string(),
            project_name: "myproject".to_string(),
            session_id: "sess-active".to_string(),
        });

        // Simulate what load_historical_plans does internally:
        // historical plans with same slug should not overwrite existing
        let historical = vec![crate::auto_title::PlanInfo {
            slug: "existing-plan".to_string(),
            title: "Historical Title".to_string(),
            path: "/path/existing-plan.md".to_string(),
            project_name: "myproject".to_string(),
            session_id: "sess-old".to_string(),
        }];
        for plan in historical {
            if !app.plans.iter().any(|p| p.slug == plan.slug) {
                app.plans.push(plan);
            }
        }

        assert_eq!(app.plans.len(), 1);
        assert_eq!(app.plans[0].title, "Current Title"); // not overwritten
        assert_eq!(app.plans[0].session_id, "sess-active"); // not overwritten
    }

    #[test]
    fn test_scanned_project_dirs_dedup() {
        let mut app = AppState::new(None);
        let project_dir = crate::auto_title::cwd_to_project_dir("/work/myproject");

        // First insert succeeds
        assert!(app.scanned_project_dirs.insert(project_dir.clone()));
        // Second insert returns false (already present)
        assert!(!app.scanned_project_dirs.insert(project_dir));
        assert_eq!(app.scanned_project_dirs.len(), 1);
    }

    #[test]
    fn test_load_historical_plans_skips_scanned_dir() {
        let mut app = AppState::new(None);
        // Mark project dir as already scanned
        let project_dir = crate::auto_title::cwd_to_project_dir("/work/myproject");
        app.scanned_project_dirs.insert(project_dir);

        // load_historical_plans should early-return without adding anything
        // (even though the dir doesn't exist, the point is it won't even try)
        app.load_historical_plans("/work/myproject", "myproject");
        assert!(app.plans.is_empty());
    }

    // --- mtime-based change detection ---

    #[test]
    fn test_refresh_auto_titles_skips_when_mtime_unchanged() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].session_id = Some("test-session".to_string());

        // Create a temp JSONL file so mtime can be read
        let tmpdir = tempfile::tempdir().unwrap();
        let cwd = tmpdir.path().to_str().unwrap().to_string();
        app.sessions[0].cwd = cwd.clone();
        let project_dir = crate::auto_title::cwd_to_project_dir(&cwd);
        app.scanned_project_dirs.insert(project_dir.clone());

        let home = std::env::var("HOME").unwrap();
        let jsonl_dir = format!("{home}/.claude/projects/{project_dir}");
        std::fs::create_dir_all(&jsonl_dir).unwrap();
        let jsonl_file = format!("{jsonl_dir}/test-session.jsonl");
        std::fs::write(&jsonl_file, "{}\n").unwrap();

        // First call: should read and set mtime
        app.refresh_auto_titles();
        assert!(app.sessions[0].jsonl_mtime.is_some());
        let first_mtime = app.sessions[0].jsonl_mtime;

        // Set a sentinel auto_title to verify it's NOT re-resolved
        app.sessions[0].auto_title = Some("sentinel".to_string());

        // Second call: mtime unchanged, should skip (auto_title stays as sentinel)
        app.refresh_auto_titles();
        assert_eq!(app.sessions[0].auto_title, Some("sentinel".to_string()));
        assert_eq!(app.sessions[0].jsonl_mtime, first_mtime);

        // Cleanup
        let _ = std::fs::remove_file(&jsonl_file);
        let _ = std::fs::remove_dir(&jsonl_dir);
    }

    #[test]
    fn test_refresh_auto_titles_reads_when_mtime_is_none() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].session_id = Some("test-session-none".to_string());

        let tmpdir = tempfile::tempdir().unwrap();
        let cwd = tmpdir.path().to_str().unwrap().to_string();
        app.sessions[0].cwd = cwd.clone();
        let project_dir = crate::auto_title::cwd_to_project_dir(&cwd);
        app.scanned_project_dirs.insert(project_dir.clone());

        let home = std::env::var("HOME").unwrap();
        let jsonl_dir = format!("{home}/.claude/projects/{project_dir}");
        std::fs::create_dir_all(&jsonl_dir).unwrap();
        let jsonl_file = format!("{jsonl_dir}/test-session-none.jsonl");
        std::fs::write(&jsonl_file, "{}\n").unwrap();

        // jsonl_mtime is None initially
        assert!(app.sessions[0].jsonl_mtime.is_none());

        // Should read the file and set mtime
        app.refresh_auto_titles();
        assert!(app.sessions[0].jsonl_mtime.is_some());

        // Cleanup
        let _ = std::fs::remove_file(&jsonl_file);
        let _ = std::fs::remove_dir(&jsonl_dir);
    }

    #[test]
    fn test_refresh_auto_titles_re_reads_after_file_change() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].session_id = Some("test-session-change".to_string());

        let tmpdir = tempfile::tempdir().unwrap();
        let cwd = tmpdir.path().to_str().unwrap().to_string();
        app.sessions[0].cwd = cwd.clone();
        let project_dir = crate::auto_title::cwd_to_project_dir(&cwd);
        app.scanned_project_dirs.insert(project_dir.clone());

        let home = std::env::var("HOME").unwrap();
        let jsonl_dir = format!("{home}/.claude/projects/{project_dir}");
        std::fs::create_dir_all(&jsonl_dir).unwrap();
        let jsonl_file = format!("{jsonl_dir}/test-session-change.jsonl");
        std::fs::write(&jsonl_file, "{}\n").unwrap();

        // First call
        app.refresh_auto_titles();
        let first_mtime = app.sessions[0].jsonl_mtime;
        app.sessions[0].auto_title = Some("sentinel".to_string());

        // Modify the file (sleep briefly to ensure mtime changes)
        std::thread::sleep(std::time::Duration::from_millis(50));
        std::fs::write(&jsonl_file, "{}\n{}\n").unwrap();

        // Second call: mtime changed, should re-read (auto_title will be re-resolved)
        app.refresh_auto_titles();
        assert_ne!(app.sessions[0].jsonl_mtime, first_mtime);
        // auto_title should have been re-resolved (overwritten from sentinel)
        assert_ne!(app.sessions[0].auto_title, Some("sentinel".to_string()));

        // Cleanup
        let _ = std::fs::remove_file(&jsonl_file);
        let _ = std::fs::remove_dir(&jsonl_dir);
    }

    #[test]
    fn test_session_start_resets_jsonl_mtime() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Set a non-None mtime
        app.sessions[0].jsonl_mtime = Some(std::time::SystemTime::UNIX_EPOCH);

        let msg = crate::rpc::RpcMessage {
            method: "session_start".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "new-session-id",
                "model": "claude-3"
            }),
        };
        app.handle_rpc_message(&msg);

        assert_eq!(app.sessions[0].session_id, Some("new-session-id".to_string()));
        assert_eq!(app.sessions[0].jsonl_mtime, None);
    }

    #[test]
    fn test_status_update_first_session_id_resets_jsonl_mtime() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // session_id is None, set a non-None mtime
        app.sessions[0].jsonl_mtime = Some(std::time::SystemTime::UNIX_EPOCH);

        let msg = crate::rpc::RpcMessage {
            method: "status_update".to_string(),
            params: serde_json::json!({
                "pane_id": "%1",
                "session_id": "first-session-id",
                "context_window": { "used_percentage": 42 }
            }),
        };
        app.handle_rpc_message(&msg);

        assert_eq!(app.sessions[0].session_id, Some("first-session-id".to_string()));
        assert_eq!(app.sessions[0].jsonl_mtime, None);
    }

    #[test]
    fn test_parse_numstat_normal() {
        let output = b"10\t5\tsrc/main.rs\n3\t0\tsrc/lib.rs\n";
        let (files, ins, del) = parse_numstat(output);
        assert_eq!(files, 2);
        assert_eq!(ins, 13);
        assert_eq!(del, 5);
    }

    #[test]
    fn test_parse_numstat_binary() {
        let output = b"-\t-\timage.png\n2\t1\tsrc/main.rs\n";
        let (files, ins, del) = parse_numstat(output);
        assert_eq!(files, 2);
        assert_eq!(ins, 2);
        assert_eq!(del, 1);
    }

    #[test]
    fn test_parse_numstat_empty() {
        let (files, ins, del) = parse_numstat(b"");
        assert_eq!(files, 0);
        assert_eq!(ins, 0);
        assert_eq!(del, 0);
    }

    // --- Marked tab ---

    #[test]
    fn test_marked_tab_appears_when_sessions_marked() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // No marked sessions → no Marked tab
        assert!(!app.tab_state.tabs.contains(&Tab::Marked));

        // Mark a session
        app.sessions[0].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);

        // Now Marked tab should appear
        assert!(app.tab_state.tabs.contains(&Tab::Marked));
    }

    #[test]
    fn test_marked_tab_position_after_all() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        app.sessions[0].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);

        // Marked tab should be at index 1 (right after All)
        assert_eq!(app.tab_state.tabs[0], Tab::All);
        assert_eq!(app.tab_state.tabs[1], Tab::Marked);
    }

    #[test]
    fn test_filtered_sessions_marked_tab() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
            make_session(300, "%3", "crmux", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        // Mark first and third sessions
        app.sessions[0].marked = true;
        app.sessions[2].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);

        // Switch to Marked tab
        let marked_pos = app.tab_state.tabs.iter().position(|t| *t == Tab::Marked).unwrap();
        app.tab_state.selected_tab = marked_pos;

        let filtered = app.filtered_sessions();
        assert_eq!(filtered.len(), 2);
        assert_eq!(filtered[0].pid, 100);
        assert_eq!(filtered[1].pid, 300);
    }

    #[test]
    fn test_marked_tab_disappears_when_no_marked_sessions() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Mark then unmark
        app.sessions[0].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);
        assert!(app.tab_state.tabs.contains(&Tab::Marked));

        app.sessions[0].marked = false;
        app.tab_state.rebuild_tabs(&app.sessions, None);
        assert!(!app.tab_state.tabs.contains(&Tab::Marked));
    }

    #[test]
    fn test_marked_tab_fallback_to_all_when_empty() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "aegis", ClaudeState::Idle),
            make_session(200, "%2", "crmux", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Mark a session and select Marked tab
        app.sessions[0].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);
        let marked_pos = app.tab_state.tabs.iter().position(|t| *t == Tab::Marked).unwrap();
        app.tab_state.selected_tab = marked_pos;
        assert_eq!(*app.tab_state.current_tab(), Tab::Marked);

        // Unmark all → Marked tab disappears → fallback to All
        app.sessions[0].marked = false;
        app.tab_state.rebuild_tabs(&app.sessions, None);
        assert_eq!(*app.tab_state.current_tab(), Tab::All);
    }

    // --- LayoutMode tests ---

    #[test]
    fn test_layout_mode_default_main_vertical() {
        let state = AppState::new(None);
        assert_eq!(state.layout_mode, LayoutMode::MainVertical);
    }

    #[test]
    fn test_cycle_layout_mode() {
        let mut state = AppState::new(None);
        assert_eq!(state.layout_mode, LayoutMode::MainVertical);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::Single);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::Grid);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::EvenHorizontal);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::EvenVertical);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::MainHorizontal);
        state.cycle_layout_mode();
        assert_eq!(state.layout_mode, LayoutMode::MainVertical);
    }

    // --- Workspace tab tests ---

    #[test]
    fn test_tmux_session_extracted_from_pane_id() {
        let mut app = AppState::new(None);
        // pane_id format: "session_name:window.pane" e.g. "dev:%1"
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].tmux_session, "dev");
    }

    #[test]
    fn test_tmux_session_updated_on_pane_id_change() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        assert_eq!(app.sessions[0].tmux_session, "dev");

        // Simulate pane move to different session
        let monitor2 = make_monitor(vec![
            make_session(100, "staging:%3", "project-a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);
        assert_eq!(app.sessions[0].tmux_session, "staging");
    }

    #[test]
    fn test_workspace_tabs_appear_with_two_sessions() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert!(app.tab_state.tabs.iter().any(|t| matches!(t, Tab::Workspace(_))));
        // Should have Workspace(dev) and Workspace(staging)
        let ws_tabs: Vec<_> = app.tab_state.tabs.iter()
            .filter_map(|t| if let Tab::Workspace(name) = t { Some(name.as_str()) } else { None })
            .collect();
        assert!(ws_tabs.contains(&"dev"));
        assert!(ws_tabs.contains(&"staging"));
    }

    #[test]
    fn test_workspace_tabs_not_shown_with_one_session() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "dev:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Both in same tmux session "dev" → only 1 unique session → no Workspace tabs
        assert!(!app.tab_state.tabs.iter().any(|t| matches!(t, Tab::Workspace(_))));
    }

    #[test]
    fn test_workspace_filter_returns_matching_sessions() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
            make_session(300, "dev:%3", "project-c", ClaudeState::Working),
        ]);
        app.sync_with_monitor(&monitor);

        // Select the Workspace(dev) tab
        let ws_pos = app.tab_state.tabs.iter().position(|t| *t == Tab::Workspace("dev".to_string())).unwrap();
        app.tab_state.selected_tab = ws_pos;

        let filtered = app.filtered_sessions();
        assert_eq!(filtered.len(), 2);
        assert!(filtered.iter().all(|s| s.tmux_session == "dev"));
    }

    #[test]
    fn test_workspace_tab_order() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);
        // Mark a session to get Marked tab
        app.sessions[0].marked = true;
        app.tab_state.rebuild_tabs(&app.sessions, None);

        // Order: All → Marked → Workspace(dev) → Workspace(staging) → Project(...)
        let tab_names: Vec<String> = app.tab_state.tabs.iter().map(|t| match t {
            Tab::All => "All".to_string(),
            Tab::Marked => "Marked".to_string(),
            Tab::Workspace(w) => format!("Workspace({w})"),
            Tab::Project(p) => format!("Project({p})"),
        }).collect();

        let all_pos = tab_names.iter().position(|t| t == "All").unwrap();
        let marked_pos = tab_names.iter().position(|t| t == "Marked").unwrap();
        let ws_pos = tab_names.iter().position(|t| t.starts_with("Workspace")).unwrap();
        let proj_pos = tab_names.iter().position(|t| t.starts_with("Project")).unwrap();

        assert!(all_pos < marked_pos, "All should come before Marked");
        assert!(marked_pos < ws_pos, "Marked should come before Workspace");
        assert!(ws_pos < proj_pos, "Workspace should come before Project");
    }

    #[test]
    fn test_workspace_tab_selection_maintained() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Select workspace tab
        let ws_pos = app.tab_state.tabs.iter().position(|t| *t == Tab::Workspace("dev".to_string())).unwrap();
        app.tab_state.selected_tab = ws_pos;

        // Re-sync — selection should be maintained
        app.sync_with_monitor(&monitor);
        assert_eq!(*app.tab_state.current_tab(), Tab::Workspace("dev".to_string()));
    }

    #[test]
    fn test_initial_workspace_selects_workspace_tab() {
        let mut app = AppState::new(None);
        app.set_initial_workspace(Some("staging".to_string()));
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(*app.tab_state.current_tab(), Tab::Workspace("staging".to_string()));
    }

    #[test]
    fn test_initial_workspace_fallback_to_all_when_not_found() {
        let mut app = AppState::new(None);
        app.set_initial_workspace(Some("nonexistent".to_string()));
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(*app.tab_state.current_tab(), Tab::All);
    }

    #[test]
    fn test_initial_workspace_consumed_after_first_rebuild() {
        let mut app = AppState::new(None);
        app.set_initial_workspace(Some("staging".to_string()));
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // First rebuild selects staging
        assert_eq!(*app.tab_state.current_tab(), Tab::Workspace("staging".to_string()));

        // Manually switch to All
        app.tab_state.selected_tab = 0;

        // Second sync should NOT override back to staging (initial_workspace consumed)
        app.sync_with_monitor(&monitor);
        assert_eq!(*app.tab_state.current_tab(), Tab::All);
    }

    #[test]
    fn test_initial_workspace_deferred_until_sessions_arrive() {
        let mut app = AppState::new(None);
        app.set_initial_workspace(Some("staging".to_string()));

        // First sync: no sessions yet (empty monitor)
        let empty_monitor = make_monitor(vec![]);
        app.sync_with_monitor(&empty_monitor);

        // Should still be All (no workspace tabs exist)
        assert_eq!(*app.tab_state.current_tab(), Tab::All);

        // Second sync: sessions arrive
        let monitor = make_monitor(vec![
            make_session(100, "dev:%1", "project-a", ClaudeState::Idle),
            make_session(200, "staging:%2", "project-b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        // Now staging workspace should be selected
        assert_eq!(*app.tab_state.current_tab(), Tab::Workspace("staging".to_string()));
    }

    // --- prev_selected_index (t key) ---

    #[test]
    fn test_select_next_saves_prev_selected() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.prev_selected_index, None);
        app.select_next(); // 0 -> 1
        assert_eq!(app.prev_selected_index, Some(0));
        app.select_next(); // 1 -> 2
        assert_eq!(app.prev_selected_index, Some(1));
    }

    #[test]
    fn test_select_prev_saves_prev_selected() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.select_prev(); // 0 -> 2 (wrap)
        assert_eq!(app.prev_selected_index, Some(0));
    }

    #[test]
    fn test_select_prev_selected_swaps_indices() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.select_next(); // 0 -> 1, prev=0
        app.select_next(); // 1 -> 2, prev=1
        app.select_prev_selected(); // 2 -> 1, prev=2
        assert_eq!(app.selected_index, 1);
        assert_eq!(app.prev_selected_index, Some(2));
    }

    #[test]
    fn test_select_prev_selected_toggles() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.select_next(); // 0 -> 1, prev=0
        app.select_next(); // 1 -> 2, prev=1

        app.select_prev_selected(); // 2 -> 1, prev=2
        assert_eq!(app.selected_index, 1);

        app.select_prev_selected(); // 1 -> 2, prev=1
        assert_eq!(app.selected_index, 2);
        assert_eq!(app.prev_selected_index, Some(1));
    }

    #[test]
    fn test_select_prev_selected_noop_when_none() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        assert_eq!(app.prev_selected_index, None);
        app.select_prev_selected();
        assert_eq!(app.selected_index, 0); // unchanged
        assert_eq!(app.prev_selected_index, None);
    }

    #[test]
    fn test_select_prev_selected_resets_preview_scroll() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.select_next(); // 0 -> 1, prev=0
        app.preview_scroll = 10;
        app.select_prev_selected();
        assert_eq!(app.preview_scroll, 0);
    }

    #[test]
    fn test_prev_selected_index_clamped_on_session_removal() {
        let mut app = AppState::new(None);
        let monitor = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
            make_session(200, "%2", "b", ClaudeState::Idle),
            make_session(300, "%3", "c", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor);

        app.select_next();
        app.select_next(); // now at 2, prev=1
        app.prev_selected_index = Some(2); // point to last item

        // Remove sessions so only 1 remains
        let monitor2 = make_monitor(vec![
            make_session(100, "%1", "a", ClaudeState::Idle),
        ]);
        app.sync_with_monitor(&monitor2);

        // prev_selected_index should be None (was 2, but only 1 session now)
        assert_eq!(app.prev_selected_index, None);
    }

    // --- own_pane_id ---

    #[test]
    fn test_own_pane_id_default_none() {
        let app = AppState::new(None);
        assert_eq!(app.own_pane_id, None);
    }

    #[test]
    fn test_own_pane_id_can_be_set() {
        let mut app = AppState::new(None);
        app.own_pane_id = Some("main:0.1".to_string());
        assert_eq!(app.own_pane_id.as_deref(), Some("main:0.1"));
    }

}