gitkraft_gui/

state.rs

use std::collections::HashSet;
use std::path::PathBuf;

use gitkraft_core::*;
use iced::{Color, Point, Task};

use crate::message::Message;
use crate::theme::ThemeColors;

// ── Pane resize ───────────────────────────────────────────────────────────────

/// Which vertical divider the user is currently dragging (if any).
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum DragTarget {
    /// The divider between the sidebar and the commit-log panel.
    SidebarRight,
    /// The divider between the commit-log panel and the diff panel.
    CommitLogRight,
    /// The divider between the diff-viewer file list and the diff content
    /// (only visible when a multi-file commit is selected).
    DiffFileListRight,
    /// The divider between the unstaged and staged panels in the staging area.
    StagingUnstagedRight,
    /// The divider between the staged and commit panels in the staging area.
    StagingStagedRight,
    /// Divider between the BRANCH/TAG column and the GRAPH column in the commit log.
    CommitRefColumnRight,
    /// Divider between the GRAPH column and the COMMIT MESSAGE column in the commit log.
    CommitGraphColumnRight,
}

/// Which horizontal divider the user is currently dragging (if any).
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum DragTargetH {
    /// The divider between the middle row and the staging area.
    StagingTop,
    /// The divider between branches and stashes in the sidebar.
    SidebarBranchesBottom,
    /// The divider between stashes and remotes in the sidebar.
    SidebarStashBottom,
}

/// What item was right-clicked to open the context menu.
#[derive(Debug, Clone)]
pub enum ContextMenu {
    /// A local branch.
    Branch {
        name: String,
        is_current: bool,
        /// Index in the filtered local-branch list, used to approximate
        /// the menu's on-screen position.
        local_index: usize,
    },
    /// A remote-tracking branch (e.g. origin/feature-x).
    RemoteBranch { name: String },
    /// A commit in the log.
    Commit { index: usize, oid: String },
    /// A stash entry.
    Stash { index: usize },
    /// An unstaged file in the staging area.
    UnstagedFile { path: String },
    /// A staged file in the staging area.
    StagedFile { path: String },
    /// A file in a commit diff.
    CommitFile { oid: String, file_path: String },
}

// ── Per-repository tab state ──────────────────────────────────────────────────

/// Per-repository state — one instance per open tab.
pub struct RepoTab {
    // ── Repository ────────────────────────────────────────────────────────
    /// Path to the currently opened repository (workdir root).
    pub repo_path: Option<PathBuf>,
    /// High-level information about the opened repository.
    pub repo_info: Option<RepoInfo>,

    // ── Branches ──────────────────────────────────────────────────────────
    /// All branches (local + remote) in the repository.
    pub branches: Vec<BranchInfo>,
    /// Name of the currently checked-out branch.
    pub current_branch: Option<String>,

    // ── Commits ───────────────────────────────────────────────────────────
    /// Commit log (newest first).
    pub commits: Vec<CommitInfo>,
    /// Index into `commits` of the currently selected commit.
    pub selected_commit: Option<usize>,
    /// Anchor commit index for range selection — set on a plain click.
    pub anchor_commit_index: Option<usize>,
    /// Ordered (ascending) list of commit indices in the current range selection.
    pub selected_commits: Vec<usize>,
    /// Per-commit graph layout rows for branch visualisation.
    pub graph_rows: Vec<gitkraft_core::GraphRow>,

    // ── Diff / Staging ────────────────────────────────────────────────────
    /// Unstaged (working-directory) changes.
    pub unstaged_changes: Vec<gitkraft_core::DiffFileEntry>,
    /// Staged (index) changes.
    pub staged_changes: Vec<gitkraft_core::DiffFileEntry>,
    /// Lightweight file list for the currently selected commit (path + status only).
    pub commit_files: Vec<gitkraft_core::DiffFileEntry>,
    /// OID of the currently selected commit (needed for on-demand file diff loading).
    pub selected_commit_oid: Option<String>,
    /// Index of the selected file in `commit_files`.
    pub selected_file_index: Option<usize>,
    /// True while a single-file diff is being loaded.
    pub is_loading_file_diff: bool,
    /// Anchor index for range selection — set on a regular click, kept fixed while
    /// the user extends the selection with Shift+Click.
    pub anchor_file_index: Option<usize>,
    /// Ordered list of file indices currently multi-selected in the commit file list (Shift+Click).
    /// Always stored in ascending index order (lowest index first).
    pub selected_commit_file_indices: Vec<usize>,
    /// Diffs for all multi-selected files (populated when 2+ files are selected).
    pub multi_file_diffs: Vec<gitkraft_core::DiffInfo>,
    /// Combined diff for all commits in the current range selection (populated when
    /// `selected_commits.len() > 1`). Shown in the diff panel instead of single-commit diff.
    pub commit_range_diffs: Vec<gitkraft_core::DiffInfo>,
    /// The diff currently displayed in the diff viewer panel.
    pub selected_diff: Option<DiffInfo>,
    /// Text in the commit-message input.
    pub commit_message: String,

    // ── Stash ─────────────────────────────────────────────────────────────
    /// All stash entries.
    pub stashes: Vec<StashEntry>,

    // ── Remotes ───────────────────────────────────────────────────────────
    /// Configured remotes.
    pub remotes: Vec<RemoteInfo>,

    // ── Per-tab UI state ──────────────────────────────────────────────────
    /// Whether the commit detail pane is visible.
    pub show_commit_detail: bool,
    /// Text in the "new branch name" input.
    pub new_branch_name: String,
    /// Whether the inline branch-creation UI is visible.
    pub show_branch_create: bool,
    /// Whether the Local branches section is expanded.
    pub local_branches_expanded: bool,
    /// Whether the Remote branches section is expanded.
    pub remote_branches_expanded: bool,
    /// Text in the "stash message" input.
    pub stash_message: String,

    /// Set of selected unstaged file paths (for multi-select).
    pub selected_unstaged: std::collections::HashSet<String>,
    /// Set of selected staged file paths (for multi-select).
    pub selected_staged: std::collections::HashSet<String>,
    /// Anchor index for Shift+Click range selection in the unstaged list.
    pub anchor_unstaged_index: Option<usize>,
    /// Anchor index for Shift+Click range selection in the staged list.
    pub anchor_staged_index: Option<usize>,

    /// File path pending discard confirmation (None = no pending discard).
    pub pending_discard: Option<String>,

    // ── Feedback ──────────────────────────────────────────────────────────
    /// Transient status-bar message (e.g. "Branch created").
    pub status_message: Option<String>,
    /// Error message shown in a banner / toast.
    pub error_message: Option<String>,
    /// True while an async operation is in flight.
    pub is_loading: bool,
    /// Cursor position captured at the moment the context menu was opened.
    /// Used to anchor the menu so it doesn't follow the mouse after appearing.
    pub context_menu_pos: (f32, f32),

    /// Currently open context menu, if any.
    pub context_menu: Option<ContextMenu>,
    /// Name of the branch currently being renamed (None = not renaming).
    pub rename_branch_target: Option<String>,
    /// The new name being typed in the rename input.
    pub rename_branch_input: String,

    /// When `Some(oid)`, the tag-creation inline form is visible, targeting that OID.
    pub create_tag_target_oid: Option<String>,
    /// True when creating an annotated tag; false for a lightweight tag.
    pub create_tag_annotated: bool,
    /// The tag name the user is typing.
    pub create_tag_name: String,
    /// The annotated tag message the user is typing (only used when `create_tag_annotated` is true).
    pub create_tag_message: String,
    /// When `Some(oid)`, the inline "create branch at this commit" form is visible.
    pub create_branch_at_oid: Option<String>,

    /// Current scroll offset of the commit log in pixels.
    /// Tracked via `on_scroll` so virtual scrolling can render only the
    /// visible window of rows.
    pub commit_scroll_offset: f32,

    /// Index of the commit row currently being hovered by the mouse.
    /// Used to show ghost branch labels and highlight the row.
    pub hovered_commit: Option<usize>,

    /// Pre-computed branch context labels for each commit row.
    /// For rows with refs, this is empty (the real badges are shown instead).
    /// For rows without refs, this contains the inherited branch name from
    /// the nearest ancestor in the same graph lane.
    pub branch_context: Vec<Option<String>>,

    /// Current scroll offset of the diff viewer in pixels.
    pub diff_scroll_offset: f32,
    /// Pre-computed display strings for each commit:
    /// `(relative_time, truncated_author)`.
    /// Computed once when commits load to avoid per-frame string allocations.
    /// The summary is read directly from `commits[i].summary` to avoid duplication.
    pub commit_display: Vec<(String, String)>,

    /// Whether there are potentially more commits to load beyond those already shown.
    pub has_more_commits: bool,
    /// Guard: true while a background load-more task is in flight (prevents duplicates).
    pub is_loading_more_commits: bool,

    /// When `Some(path)`, the file-history overlay is shown for that repo-relative path.
    pub file_history_path: Option<String>,
    /// Commits loaded for the file-history overlay (newest first).
    pub file_history_commits: Vec<gitkraft_core::CommitInfo>,
    /// Scroll offset of the file-history list in pixels.
    pub file_history_scroll: f32,

    /// When `Some(path)`, the blame overlay is shown for that repo-relative path.
    pub blame_path: Option<String>,
    /// Blame lines loaded for the blame overlay.
    pub blame_lines: Vec<gitkraft_core::BlameLine>,
    /// Scroll offset of the blame view in pixels.
    pub blame_scroll: f32,

    /// When `Some(path)`, a delete-confirmation banner is shown for that file.
    pub pending_delete_file: Option<String>,
}

impl RepoTab {
    /// Create an empty tab (no repo open — shows welcome screen).
    pub fn new_empty() -> Self {
        Self {
            repo_path: None,
            repo_info: None,
            branches: Vec::new(),
            current_branch: None,
            commits: Vec::new(),
            selected_commit: None,
            anchor_commit_index: None,
            selected_commits: Vec::new(),
            graph_rows: Vec::new(),
            unstaged_changes: Vec::new(),
            staged_changes: Vec::new(),
            commit_files: Vec::new(),
            selected_commit_oid: None,
            selected_file_index: None,
            is_loading_file_diff: false,
            anchor_file_index: None,
            selected_commit_file_indices: Vec::new(),
            multi_file_diffs: Vec::new(),
            commit_range_diffs: Vec::new(),
            selected_diff: None,
            commit_message: String::new(),
            stashes: Vec::new(),
            remotes: Vec::new(),
            show_commit_detail: false,
            new_branch_name: String::new(),
            show_branch_create: false,
            local_branches_expanded: true,
            remote_branches_expanded: true,
            stash_message: String::new(),
            selected_unstaged: std::collections::HashSet::new(),
            selected_staged: std::collections::HashSet::new(),
            anchor_unstaged_index: None,
            anchor_staged_index: None,
            pending_discard: None,
            status_message: None,
            error_message: None,
            is_loading: false,
            context_menu: None,
            context_menu_pos: (0.0, 0.0),
            rename_branch_target: None,
            rename_branch_input: String::new(),
            create_tag_target_oid: None,
            create_tag_annotated: false,
            create_tag_name: String::new(),
            create_tag_message: String::new(),
            create_branch_at_oid: None,
            commit_scroll_offset: 0.0,
            hovered_commit: None,
            branch_context: Vec::new(),
            diff_scroll_offset: 0.0,
            commit_display: Vec::new(),
            has_more_commits: true,
            is_loading_more_commits: false,
            file_history_path: None,
            file_history_commits: Vec::new(),
            file_history_scroll: 0.0,
            blame_path: None,
            blame_lines: Vec::new(),
            blame_scroll: 0.0,
            pending_delete_file: None,
        }
    }

    /// Whether a repository is currently open in this tab.
    pub fn has_repo(&self) -> bool {
        self.repo_path.is_some()
    }

    /// Name of the first remote, or "origin" as a fallback.
    pub fn default_remote(&self) -> String {
        self.remotes
            .first()
            .map(|r| r.name.clone())
            .unwrap_or_else(|| "origin".to_string())
    }

    /// Display name for the tab (last path component, or "New Tab").
    pub fn display_name(&self) -> String {
        gitkraft_core::repo_display_name(self.repo_path.as_deref())
    }

    /// Apply a full repo payload to this tab, resetting transient UI state.
    ///
    /// The currently selected commit (if any) is **re-pinned** by OID after the
    /// new commit list arrives, so background auto-refreshes (git-watcher or
    /// staging changes) never clear the user's selection.
    pub fn apply_payload(
        &mut self,
        payload: crate::message::RepoPayload,
        path: std::path::PathBuf,
    ) {
        // ── Save state so we can restore it after the data refresh ────
        let is_refresh = self.repo_info.is_some();
        let prev_oid = self.selected_commit_oid.clone();

        // Save multi-selection OIDs so we can re-map them after the commit
        // list is replaced.  Without this, background auto-refreshes (git
        // watcher, fallback poll) would silently clear a Shift+click range.
        let prev_anchor_oid = self
            .anchor_commit_index
            .and_then(|i| self.commits.get(i).map(|c| c.oid.clone()));
        let prev_selected_oids: Vec<String> = self
            .selected_commits
            .iter()
            .filter_map(|&i| self.commits.get(i).map(|c| c.oid.clone()))
            .collect();

        // Save file-level selection state.  File indices reference into
        // `commit_files` which is NOT replaced during a refresh (the commit
        // content hasn't changed), so these can be restored as-is.
        let prev_anchor_file = self.anchor_file_index;
        let prev_file_indices = self.selected_commit_file_indices.clone();
        let prev_multi_file_diffs = std::mem::take(&mut self.multi_file_diffs);
        let prev_commit_range_diffs = std::mem::take(&mut self.commit_range_diffs);

        // Save staging selection — these are path-based so we can re-validate
        // them against the refreshed unstaged/staged lists.
        let prev_selected_unstaged = std::mem::take(&mut self.selected_unstaged);
        let prev_selected_staged = std::mem::take(&mut self.selected_staged);
        let prev_anchor_unstaged = self.anchor_unstaged_index;
        let prev_anchor_staged = self.anchor_staged_index;

        // Save standalone diff (set from staging click or file preview).
        // When no commit is selected, this is the only diff the user sees.
        let prev_selected_diff = self.selected_diff.take();
        let prev_diff_scroll = self.diff_scroll_offset;

        self.current_branch = payload.info.head_branch.clone();
        self.repo_path = Some(path);
        self.repo_info = Some(payload.info);
        self.branches = payload.branches;

        // Never shrink the commit list during a background refresh — a
        // concurrent pagination (MoreCommitsLoaded) may have extended the
        // list between the refresh start and its completion.  Shrinking
        // would snap the user back and trigger an infinite load/shrink loop.
        if !is_refresh || payload.commits.len() >= self.commits.len() {
            self.commits = payload.commits;
            self.graph_rows = payload.graph_rows;
        }

        self.unstaged_changes = payload.unstaged;
        self.staged_changes = payload.staged;
        self.stashes = payload.stashes;
        self.remotes = payload.remotes;

        // Reset transient UI state.
        // NOTE: selected_commit / commit_files / selected_diff are restored
        // below so they survive background auto-refreshes.
        // NOTE: commit_message is intentionally NOT cleared here — the user
        // may be mid-typing when a background refresh arrives.  It is only
        // cleared on successful commit (CommitCreated handler).
        self.selected_commit = None;
        self.anchor_commit_index = None;
        self.selected_commits.clear();
        self.selected_commit_oid = None;
        self.error_message = None;
        self.status_message = Some("Repository loaded.".into());
        // Scroll position is managed entirely by Iced's scrollable widget;
        // we do NOT reset it here so background refreshes never snap the
        // user back to the top.  Brand-new opens use a scroll_to(0,0) task
        // in handle_repo_loaded instead.
        if !is_refresh {
            self.has_more_commits = true;
        }
        // NOTE: is_loading_more_commits is intentionally NOT reset here.
        // It is owned by the pagination flow (CommitLogScrolled sets it true,
        // handle_more_commits_loaded sets it false).  Resetting it during a
        // background refresh would trick the scroll handler into firing a
        // duplicate pagination, whose identical result then falsely sets
        // has_more_commits = false ("end of history").
        self.anchor_unstaged_index = None;
        self.anchor_staged_index = None;
        self.anchor_file_index = None;
        self.selected_commit_file_indices.clear();

        // ── Restore the previously selected commit by OID ─────────────────
        // If the commit still exists in the refreshed list, re-select it so
        // the diff panel, file list, and selection highlight are all
        // preserved.  This means auto-refreshes (every 5 s fallback, git
        // watcher) never interrupt the user’s view.
        if let Some(oid) = prev_oid {
            if let Some(new_idx) = self.commits.iter().position(|c| c.oid == oid) {
                self.selected_commit = Some(new_idx);
                self.selected_commit_oid = Some(oid);
                // commit_files, selected_file_index, is_loading_file_diff,
                // diff_scroll_offset are intentionally left unchanged —
                // the commit content hasn’t changed.

                // Restore selected_diff (may have been set from commit view,
                // staging click, or file preview).
                self.selected_diff = prev_selected_diff;
                self.diff_scroll_offset = prev_diff_scroll;

                // Restore file-level selection — the commit (and its file
                // list) survived, so these indices are still valid.
                self.anchor_file_index = prev_anchor_file;
                self.selected_commit_file_indices = prev_file_indices;
                self.multi_file_diffs = prev_multi_file_diffs;
                self.commit_range_diffs = prev_commit_range_diffs;
            } else {
                // Commit was rebased / force-pushed away — clear everything.
                self.clear_diff_state();
            }
        } else {
            // No commit was selected. Preserve any standalone diff set from
            // a staging click or file preview so it survives background refreshes.
            if prev_selected_diff.is_some() {
                self.selected_diff = prev_selected_diff;
                self.diff_scroll_offset = prev_diff_scroll;
            } else {
                self.clear_diff_state();
            }
        }

        // ── Restore multi-selection by OID ─────────────────────────────
        // Re-map the saved anchor and range selection from OIDs back to
        // indices in the (possibly reordered) new commit list.
        if let Some(anchor_oid) = prev_anchor_oid {
            if let Some(new_anchor) = self.commits.iter().position(|c| c.oid == anchor_oid) {
                self.anchor_commit_index = Some(new_anchor);
            }
        }
        if !prev_selected_oids.is_empty() {
            let restored: Vec<usize> = prev_selected_oids
                .iter()
                .filter_map(|oid| self.commits.iter().position(|c| &c.oid == oid))
                .collect();
            if !restored.is_empty() {
                self.selected_commits = restored;
                self.selected_commits.sort_unstable();
            }
        }

        // ── Restore staging selection ───────────────────────────────────
        // Keep only paths that still appear in the refreshed file lists.
        self.selected_unstaged = prev_selected_unstaged
            .into_iter()
            .filter(|p| self.unstaged_changes.iter().any(|d| d.display_path() == p))
            .collect();
        self.selected_staged = prev_selected_staged
            .into_iter()
            .filter(|p| self.staged_changes.iter().any(|d| d.display_path() == p))
            .collect();
        // Restore anchors only if they still point to a valid index.
        self.anchor_unstaged_index =
            prev_anchor_unstaged.filter(|&i| i < self.unstaged_changes.len());
        self.anchor_staged_index = prev_anchor_staged.filter(|&i| i < self.staged_changes.len());
    }

    /// Clear all diff-related state (used when the selected commit disappears
    /// or no commit was selected before a refresh).
    fn clear_diff_state(&mut self) {
        self.selected_diff = None;
        self.commit_files.clear();
        self.selected_file_index = None;
        self.is_loading_file_diff = false;
        self.diff_scroll_offset = 0.0;
    }
}

// ── Top-level application state ───────────────────────────────────────────────

/// Top-level application state for the GitKraft GUI.
pub struct GitKraft {
    // ── Tabs ──────────────────────────────────────────────────────────────
    /// All open repository tabs.
    pub tabs: Vec<RepoTab>,
    /// Index of the currently active/visible tab.
    pub active_tab: usize,

    // ── UI state (global, not per-tab) ────────────────────────────────────
    /// Whether the left sidebar is expanded.
    pub sidebar_expanded: bool,

    // ── Pane widths / heights (pixels) ────────────────────────────────────
    /// Width of the left sidebar in pixels.
    pub sidebar_width: f32,
    /// Width of the commit-log panel in pixels.
    pub commit_log_width: f32,
    /// Height of the staging area in pixels.
    pub staging_height: f32,
    /// Width of the diff file-list sidebar in pixels.
    pub diff_file_list_width: f32,
    /// Width of the BRANCH/TAG column in the commit log (pixels).
    pub commit_ref_width: f32,
    /// Width of the GRAPH column in the commit log (pixels).
    pub commit_graph_width: f32,
    /// Fraction of sidebar height allocated to the branches section (0.0–1.0).
    pub sidebar_branches_ratio: f32,
    /// Fraction of sidebar height allocated to the stashes section (0.0–1.0).
    /// Remotes get the remainder: 1.0 - branches_ratio - stash_ratio.
    pub sidebar_stash_ratio: f32,
    /// Fraction of staging area width allocated to the unstaged column (0.0–1.0).
    pub staging_unstaged_ratio: f32,
    /// Fraction of staging area width allocated to the staged column (0.0–1.0).
    /// Commit column gets: 1.0 - unstaged_ratio - staged_ratio.
    pub staging_staged_ratio: f32,

    /// UI scale factor (1.0 = default). Adjusted with Ctrl+/Ctrl- keyboard shortcuts.
    pub ui_scale: f32,

    // ── Drag state ────────────────────────────────────────────────────────
    /// Which vertical divider is being dragged (if any).
    pub dragging: Option<DragTarget>,
    /// Which horizontal divider is being dragged (if any).
    pub dragging_h: Option<DragTargetH>,
    /// Last known mouse X position during a drag (absolute window coords).
    pub drag_start_x: f32,
    /// Last known mouse Y position during a drag (absolute window coords).
    pub drag_start_y: f32,
    /// Whether the first move event has been received for the current vertical drag.
    /// `false` right after `PaneDragStart` — the first `PaneDragMove` sets the
    /// real start position instead of computing a bogus delta from 0.0.
    pub drag_initialized: bool,
    /// Same as `drag_initialized` but for horizontal drags.
    pub drag_initialized_h: bool,

    // ── Cursor ────────────────────────────────────────────────────────────
    /// Last known cursor position in window coordinates.
    /// Updated on every mouse-move event so context menus open at the
    /// exact spot the user right-clicked.
    pub cursor_pos: Point,

    // ── Theme ─────────────────────────────────────────────────────────────
    /// Index into `gitkraft_core::THEME_NAMES` for the currently active theme.
    pub current_theme_index: usize,
    /// Whether the theme follows the OS light/dark setting ("System").
    pub follow_system: bool,

    // ── Persistence ───────────────────────────────────────────────────────
    /// Recently opened repositories (loaded from settings on startup).
    pub recent_repos: Vec<gitkraft_core::RepoHistoryEntry>,

    // ── About ─────────────────────────────────────────────────────────────
    /// Whether the About dialog is visible.
    pub show_about: bool,

    // ── Search ────────────────────────────────────────────────────────────
    /// Whether the search overlay is visible.
    pub search_visible: bool,
    /// Current search query text.
    pub search_query: String,
    /// Search results (commit infos matching the query).
    pub search_results: Vec<gitkraft_core::CommitInfo>,
    /// Index of the selected search result.
    pub search_selected: Option<usize>,

    /// Files changed between the selected search commit and working tree.
    pub search_diff_files: Vec<gitkraft_core::DiffFileEntry>,
    /// Selected file indices in the search diff file list.
    pub search_diff_selected: HashSet<usize>,
    /// The diff content for the currently viewed search diff file(s).
    pub search_diff_content: Vec<gitkraft_core::DiffInfo>,
    /// OID of the commit being diffed against working tree in search.
    pub search_diff_oid: Option<String>,

    /// Configured editor for "Open in editor" actions.
    pub editor: gitkraft_core::Editor,

    /// Current keyboard modifier state (updated via subscription).
    pub keyboard_modifiers: iced::keyboard::Modifiers,

    /// Monotonically-increasing counter incremented on every `AnimationTick`.
    /// Drives the loading-spinner frame selection in all UI widgets.
    pub animation_tick: u64,

    // ── Window geometry ───────────────────────────────────────────────────
    /// Last known window width (updated on WindowResized).
    pub window_width: f32,
    /// Last known window height (updated on WindowResized).
    pub window_height: f32,
    /// Last known window X position (updated on WindowMoved).
    pub window_x: f32,
    /// Last known window Y position (updated on WindowMoved).
    pub window_y: f32,

    /// File preview editor content — shown in the diff pane when a file preview is active.
    pub preview_editor: Option<iced::widget::text_editor::Content>,
    /// Path of the file being previewed (for the header display).
    pub preview_path: Option<String>,
}

impl Default for GitKraft {
    fn default() -> Self {
        Self::new()
    }
}

impl GitKraft {
    /// Build application state from persisted [`AppSettings`].
    ///
    /// Starts with a single empty tab regardless of what was saved — callers
    /// that want to restore the full session should use
    /// [`Self::new_with_session_paths`] instead.
    fn from_settings(settings: gitkraft_core::AppSettings) -> Self {
        let follow_system = settings.follow_system_theme();
        let current_theme_index = settings.theme_index();

        let recent_repos = settings.recent_repos;

        let (
            sidebar_width,
            commit_log_width,
            staging_height,
            diff_file_list_width,
            sidebar_expanded,
            ui_scale,
            commit_ref_width,
            commit_graph_width,
        ) = if let Some(ref layout) = settings.layout {
            (
                layout.sidebar_width.unwrap_or(220.0),
                layout.commit_log_width.unwrap_or(500.0),
                layout.staging_height.unwrap_or(200.0),
                layout.diff_file_list_width.unwrap_or(180.0),
                layout.sidebar_expanded.unwrap_or(true),
                layout.ui_scale.unwrap_or(1.0),
                layout.commit_ref_width.unwrap_or(120.0),
                layout.commit_graph_width.unwrap_or(80.0),
            )
        } else {
            (220.0, 500.0, 200.0, 180.0, true, 1.0, 120.0, 80.0)
        };

        Self {
            tabs: vec![RepoTab::new_empty()],
            active_tab: 0,

            sidebar_expanded,

            sidebar_width,
            commit_log_width,
            staging_height,
            diff_file_list_width,
            commit_ref_width,
            commit_graph_width,
            sidebar_branches_ratio: 0.6,
            sidebar_stash_ratio: 0.25,
            staging_unstaged_ratio: 0.4,
            staging_staged_ratio: 0.3,

            ui_scale,

            dragging: None,
            dragging_h: None,
            drag_start_x: 0.0,
            drag_start_y: 0.0,
            drag_initialized: false,
            drag_initialized_h: false,
            cursor_pos: Point::ORIGIN,

            current_theme_index,
            follow_system,

            recent_repos,

            show_about: false,
            search_visible: false,
            search_query: String::new(),
            search_results: Vec::new(),
            search_selected: None,
            search_diff_files: Vec::new(),
            search_diff_selected: HashSet::new(),
            search_diff_content: Vec::new(),
            search_diff_oid: None,

            keyboard_modifiers: iced::keyboard::Modifiers::default(),
            animation_tick: 0,

            window_width: settings
                .layout
                .as_ref()
                .and_then(|l| l.window_width)
                .unwrap_or(1400.0),
            window_height: settings
                .layout
                .as_ref()
                .and_then(|l| l.window_height)
                .unwrap_or(800.0),
            window_x: settings
                .layout
                .as_ref()
                .and_then(|l| l.window_x)
                .unwrap_or(0.0),
            window_y: settings
                .layout
                .as_ref()
                .and_then(|l| l.window_y)
                .unwrap_or(0.0),

            editor: settings
                .editor_name
                .as_deref()
                .map(|name| {
                    // Try to map persisted name back to Editor variant
                    gitkraft_core::EDITOR_NAMES
                        .iter()
                        .position(|n| n.eq_ignore_ascii_case(name))
                        .map(gitkraft_core::Editor::from_index)
                        .unwrap_or_else(|| {
                            if name.eq_ignore_ascii_case("none") {
                                gitkraft_core::Editor::None
                            } else {
                                gitkraft_core::Editor::Custom(name.to_string())
                            }
                        })
                })
                .unwrap_or_else(detect_system_editor),

            preview_editor: None,
            preview_path: None,
        }
    }

    /// Create a fresh application state with sensible defaults.
    ///
    /// Loads persisted settings (theme, recent repos) from disk when available.
    /// Always starts with one empty tab — use [`Self::new_with_session_paths`] to
    /// restore the full multi-tab session.
    pub fn new() -> Self {
        Self::from_settings(
            gitkraft_core::features::persistence::ops::load_settings().unwrap_or_default(),
        )
    }

    /// Create state and also return the saved tab paths for startup restore.
    ///
    /// Call this from `main.rs` instead of [`Self::new`]; it sets up loading tabs
    /// for every path in the persisted session and returns those paths so the
    /// caller can spawn parallel `load_repo_at` tasks.
    pub fn new_with_session_paths() -> (Self, Vec<PathBuf>) {
        let settings =
            gitkraft_core::features::persistence::ops::load_settings().unwrap_or_default();
        let open_tabs = settings.open_tabs.clone();
        let active_tab_index = settings.active_tab_index;

        let mut state = Self::from_settings(settings);

        if !open_tabs.is_empty() {
            state.tabs = open_tabs
                .iter()
                .map(|path| {
                    let mut tab = RepoTab::new_empty();
                    // Set the path now so the tab bar shows the right name
                    // while the repo is being loaded in the background.
                    tab.repo_path = Some(path.clone());
                    if path.exists() {
                        tab.is_loading = true;
                        tab.status_message = Some(format!(
                            "Loading {}…",
                            path.file_name().unwrap_or_default().to_string_lossy()
                        ));
                    } else {
                        tab.error_message =
                            Some(format!("Repository not found: {}", path.display()));
                    }
                    tab
                })
                .collect();
            state.active_tab = active_tab_index.min(state.tabs.len().saturating_sub(1));
        }

        (state, open_tabs)
    }

    /// Paths of all tabs where a repository has been fully loaded
    /// (`repo_info` is populated). Used to persist the multi-tab session.
    pub fn open_tab_paths(&self) -> Vec<PathBuf> {
        self.tabs
            .iter()
            .filter(|t| t.repo_info.is_some())
            .filter_map(|t| t.repo_path.clone())
            .collect()
    }

    /// Return the session-persistable (open_tab_paths, active_tab_index) pair.
    ///
    /// `open_tab_paths()` filters out empty tabs, so the raw `self.active_tab`
    /// index does not map 1-to-1.  This helper computes the correct position
    /// of the active tab within the filtered list.
    pub fn session_state(&self) -> (Vec<PathBuf>, usize) {
        let mut paths = Vec::new();
        let mut session_active: usize = 0;
        let mut found_active = false;
        for (i, tab) in self.tabs.iter().enumerate() {
            if tab.repo_info.is_some() {
                if let Some(ref p) = tab.repo_path {
                    if i == self.active_tab {
                        session_active = paths.len();
                        found_active = true;
                    }
                    paths.push(p.clone());
                }
            }
        }
        // If the active tab wasn't in the saved set (e.g. it's an empty tab),
        // default to 0.
        if !found_active {
            session_active = 0;
        }
        (paths, session_active)
    }

    /// Get a reference to the currently active tab.
    pub fn active_tab(&self) -> &RepoTab {
        &self.tabs[self.active_tab]
    }

    /// Get a mutable reference to the currently active tab.
    pub fn active_tab_mut(&mut self) -> &mut RepoTab {
        &mut self.tabs[self.active_tab]
    }

    /// Dismiss any open context menu.
    pub fn dismiss_menu(&mut self) {
        self.active_tab_mut().context_menu = None;
    }

    /// Open a context menu at the current cursor position.
    pub fn open_context_menu(&mut self, menu: ContextMenu) {
        let pos = (self.cursor_pos.x, self.cursor_pos.y);
        let tab = self.active_tab_mut();
        tab.context_menu_pos = pos;
        tab.context_menu = Some(menu);
    }

    /// Whether the active tab has a repository open.
    pub fn has_repo(&self) -> bool {
        self.active_tab().has_repo()
    }

    /// Helper: the display name for the active tab's repo.
    pub fn repo_display_name(&self) -> String {
        self.active_tab().display_name()
    }

    /// Derive the full [`ThemeColors`] from the currently active core theme.
    ///
    /// Call this at the top of view functions:
    /// ```ignore
    /// let c = state.colors();
    /// ```
    pub fn colors(&self) -> ThemeColors {
        ThemeColors::from_core(&gitkraft_core::theme_by_index(self.current_theme_index))
    }

    /// Return a **custom** `iced::Theme` whose `Palette` is derived from the
    /// active core theme.
    ///
    /// This is the key to making every built-in Iced widget (text inputs,
    /// pick-lists, scrollbars, buttons without explicit `.style()`, etc.)
    /// inherit the correct background, text, accent, success and danger
    /// colours.  Without this, Iced falls back to its generic Dark/Light
    /// palette and the UI looks wrong for every non-default theme.
    pub fn iced_theme(&self) -> iced::Theme {
        let core = gitkraft_core::theme_by_index(self.current_theme_index);
        let name = self.current_theme_name().to_string();

        let palette = iced::theme::Palette {
            background: rgb_to_iced(core.background),
            text: rgb_to_iced(core.text_primary),
            primary: rgb_to_iced(core.accent),
            success: rgb_to_iced(core.success),
            warning: rgb_to_iced(core.warning),
            danger: rgb_to_iced(core.error),
        };

        iced::Theme::custom(name, palette)
    }

    /// If a tab with `path` is already open, switch to it.
    /// If the current tab is empty, remove it first.
    /// Returns `true` if an existing tab was found and switched to.
    pub fn switch_to_existing_tab(&mut self, path: &std::path::Path) -> bool {
        let existing_idx = match self
            .tabs
            .iter()
            .position(|t| t.repo_path.as_deref() == Some(path) && t.repo_info.is_some())
        {
            Some(idx) => idx,
            None => return false,
        };
        // Remove the current tab if it's an empty "New Tab" so it
        // doesn't linger in the tab bar after we switch away.
        if !self.active_tab().has_repo() && self.active_tab != existing_idx && self.tabs.len() > 1 {
            self.tabs.remove(self.active_tab);
            let adjusted = if existing_idx > self.active_tab {
                existing_idx - 1
            } else {
                existing_idx
            };
            self.active_tab = adjusted;
        } else {
            self.active_tab = existing_idx;
        }
        true
    }

    /// Name to persist for the current choice (`"System"` while following the OS).
    pub fn persisted_theme_name(&self) -> &'static str {
        if self.follow_system {
            gitkraft_core::SYSTEM_THEME_NAME
        } else {
            self.current_theme_name()
        }
    }

    /// Apply a pick-list choice (`THEME_COUNT` = System, otherwise a preset
    /// index). Returns `false` for out-of-range values.
    pub fn apply_theme_choice(&mut self, choice: usize) -> bool {
        if choice == gitkraft_core::THEME_COUNT {
            self.follow_system = true;
            self.current_theme_index = gitkraft_core::current_system_theme_index();
            true
        } else if choice < gitkraft_core::THEME_COUNT {
            self.follow_system = false;
            self.current_theme_index = choice;
            true
        } else {
            false
        }
    }

    /// Re-read the OS appearance while following the system.
    pub fn refresh_system_theme(&mut self) {
        if self.follow_system {
            self.current_theme_index = gitkraft_core::current_system_theme_index();
        }
    }

    /// The display name of the currently active preset (the resolved preset
    /// while following the system).
    pub fn current_theme_name(&self) -> &'static str {
        gitkraft_core::THEME_NAMES
            .get(self.current_theme_index)
            .copied()
            .unwrap_or("Default")
    }

    /// Refresh all data for the currently active tab's repository.
    ///
    /// Returns [`Task::none()`] if no repository is open in the active tab.
    pub fn refresh_active_tab(&mut self) -> Task<Message> {
        let tab = self.active_tab();
        // Don't spawn a duplicate refresh when one is already in flight
        // (is_loading is also set by the manual RefreshRepo handler).
        if tab.is_loading {
            return Task::none();
        }
        match tab.repo_path.clone() {
            Some(path) => {
                let depth = tab.commits.len();
                // Mark as loading BEFORE spawning the background task to prevent
                // concurrent refresh storms from FileSystemChanged events.  Without
                // this guard, multiple refreshes could overlap — each opening a full
                // git2::Repository with mmapped packfiles, causing GB-scale memory use.
                self.active_tab_mut().is_loading = true;
                crate::features::repo::commands::refresh_repo(path, depth)
            }
            None => Task::none(),
        }
    }

    /// Handle a `Result<(), String>` from a git operation that should trigger
    /// a full repository refresh on success.
    ///
    /// * `Ok(())` — clears `is_loading`, sets `status_message`, refreshes.
    /// * `Err(e)` — clears `is_loading`, sets `error_message`, returns
    ///   [`Task::none()`].
    pub fn on_ok_refresh(
        &mut self,
        result: Result<(), String>,
        ok_msg: &str,
        err_prefix: &str,
    ) -> Task<Message> {
        match result {
            Ok(()) => {
                {
                    let tab = self.active_tab_mut();
                    tab.is_loading = false;
                    tab.status_message = Some(ok_msg.to_string());
                }
                self.refresh_active_tab()
            }
            Err(e) => {
                let tab = self.active_tab_mut();
                tab.is_loading = false;
                tab.error_message = Some(format!("{err_prefix}: {e}"));
                tab.status_message = None;
                Task::none()
            }
        }
    }

    /// Build a [`LayoutSettings`] snapshot from the current pane dimensions.
    pub fn current_layout(&self) -> gitkraft_core::LayoutSettings {
        gitkraft_core::LayoutSettings {
            sidebar_width: Some(self.sidebar_width),
            commit_log_width: Some(self.commit_log_width),
            staging_height: Some(self.staging_height),
            diff_file_list_width: Some(self.diff_file_list_width),
            sidebar_expanded: Some(self.sidebar_expanded),
            ui_scale: Some(self.ui_scale),
            window_width: Some(self.window_width),
            window_height: Some(self.window_height),
            window_x: Some(self.window_x),
            window_y: Some(self.window_y),
            window_maximized: None, // not tracked
            commit_ref_width: Some(self.commit_ref_width),
            commit_graph_width: Some(self.commit_graph_width),
        }
    }
}

/// Convert a core [`gitkraft_core::Rgb`] to an [`iced::Color`].
fn rgb_to_iced(rgb: gitkraft_core::Rgb) -> Color {
    Color::from_rgb8(rgb.r, rgb.g, rgb.b)
}

/// Try to detect the system's preferred editor from environment variables.
fn detect_system_editor() -> gitkraft_core::Editor {
    for var in ["VISUAL", "EDITOR"] {
        if let Ok(val) = std::env::var(var) {
            let bin = val.split('/').next_back().unwrap_or(&val).trim();
            return match bin {
                "nvim" | "neovim" => gitkraft_core::Editor::Neovim,
                "vim" => gitkraft_core::Editor::Vim,
                "hx" | "helix" => gitkraft_core::Editor::Helix,
                "nano" => gitkraft_core::Editor::Nano,
                "micro" => gitkraft_core::Editor::Micro,
                "emacs" => gitkraft_core::Editor::Emacs,
                "code" => gitkraft_core::Editor::VSCode,
                "zed" => gitkraft_core::Editor::Zed,
                "subl" => gitkraft_core::Editor::Sublime,
                _ => gitkraft_core::Editor::Custom(val),
            };
        }
    }
    gitkraft_core::Editor::None
}

// ── Tests ─────────────────────────────────────────────────────────────────────

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

    #[test]
    fn new_defaults() {
        let state = GitKraft::new();
        assert!(state.active_tab().repo_path.is_none());
        assert!(!state.has_repo());
        assert_eq!(state.repo_display_name(), "New Tab");
        assert!(state.active_tab().commits.is_empty());
        assert!(state.sidebar_expanded);
        // Default theme index should be valid
        assert!(state.current_theme_index < gitkraft_core::THEME_COUNT);
        // Pane defaults
        assert!(state.sidebar_width > 0.0);
        assert!(state.commit_log_width > 0.0);
        assert!(state.staging_height > 0.0);
        assert!(state.dragging.is_none());
        assert!(state.dragging_h.is_none());
        // Should start with one empty tab
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
    }

    #[test]
    fn repo_display_name_extracts_basename() {
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path = Some(std::path::PathBuf::from("/home/user/my-project"));
        assert_eq!(state.repo_display_name(), "my-project");
    }

    #[test]
    fn colors_returns_theme_colors() {
        let state = GitKraft::new();
        let c = state.colors();
        // The default theme (index 0) is dark, so background should be dark
        assert!(c.bg.r < 0.5);
    }

    #[test]
    fn iced_theme_is_custom_with_correct_palette() {
        let mut state = GitKraft::new();

        // Index 0 = Default (dark) — custom theme with dark background
        state.current_theme_index = 0;
        let iced_t = state.iced_theme();
        let pal = iced_t.palette();
        assert!(pal.background.r < 0.5, "Default theme bg should be dark");
        assert_eq!(iced_t.to_string(), "Default");

        // Solarized Light — custom theme with light background
        let sol_idx = gitkraft_core::theme_index_by_name("Solarized Light");
        state.current_theme_index = sol_idx;
        let iced_t = state.iced_theme();
        let pal = iced_t.palette();
        assert!(pal.background.r > 0.5, "Solarized Light bg should be light");
        assert_eq!(iced_t.to_string(), "Solarized Light");

        // Gruvbox Dark — accent should come from core
        let gruvbox_idx = gitkraft_core::theme_index_by_name("Gruvbox Dark");
        state.current_theme_index = gruvbox_idx;
        let iced_t = state.iced_theme();
        let pal = iced_t.palette();
        let core = gitkraft_core::theme_by_index(gruvbox_idx);
        let expected_accent = rgb_to_iced(core.accent);
        assert!(
            (pal.primary.r - expected_accent.r).abs() < 0.01
                && (pal.primary.g - expected_accent.g).abs() < 0.01
                && (pal.primary.b - expected_accent.b).abs() < 0.01,
            "Gruvbox Dark accent should match core accent"
        );
    }

    #[test]
    fn iced_theme_name_round_trips_through_core() {
        // Ensure the custom theme name matches a core THEME_NAMES entry so
        // that ThemeColors::from_theme() can map it back to the right index.
        for i in 0..gitkraft_core::THEME_COUNT {
            let mut state = GitKraft::new();
            state.current_theme_index = i;
            let iced_t = state.iced_theme();
            let name = iced_t.to_string();
            let resolved = gitkraft_core::theme_index_by_name(&name);
            assert_eq!(
                resolved,
                i,
                "theme index {i} ({}) did not round-trip through iced_theme name",
                gitkraft_core::THEME_NAMES[i]
            );
        }
    }

    #[test]
    fn current_theme_name_round_trips() {
        let mut state = GitKraft::new();
        state.current_theme_index = gitkraft_core::theme_index_by_name("Dracula");
        assert_eq!(state.current_theme_name(), "Dracula");
        state.current_theme_index = 0;
        assert_eq!(state.current_theme_name(), "Default");
    }

    #[test]
    fn repo_tab_new_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.repo_path.is_none());
        assert!(!tab.has_repo());
        assert_eq!(tab.display_name(), "New Tab");
        assert!(tab.commits.is_empty());
        assert!(tab.branches.is_empty());
        assert!(!tab.is_loading);
    }

    #[test]
    fn repo_tab_display_name_with_path() {
        let mut tab = RepoTab::new_empty();
        tab.repo_path = Some(std::path::PathBuf::from("/some/path/cool-repo"));
        assert!(tab.has_repo());
        assert_eq!(tab.display_name(), "cool-repo");
    }

    #[test]
    fn search_defaults() {
        let state = GitKraft::new();
        assert!(!state.search_visible);
        assert!(state.search_query.is_empty());
        assert!(state.search_results.is_empty());
        assert!(state.search_selected.is_none());
    }

    #[test]
    fn context_menu_variants_exist() {
        // Verify all context menu variants can be constructed
        use crate::state::ContextMenu;

        let _branch = ContextMenu::Branch {
            name: "main".to_string(),
            is_current: true,
            local_index: 0,
        };
        let _remote = ContextMenu::RemoteBranch {
            name: "origin/main".to_string(),
        };
        let _commit = ContextMenu::Commit {
            index: 0,
            oid: "abc1234".to_string(),
        };
        let _stash = ContextMenu::Stash { index: 0 };
        let _unstaged = ContextMenu::UnstagedFile {
            path: "src/main.rs".to_string(),
        };
        let _staged = ContextMenu::StagedFile {
            path: "src/lib.rs".to_string(),
        };
    }

    #[test]
    fn repo_tab_context_menu_defaults_to_none() {
        let tab = crate::state::RepoTab::new_empty();
        assert!(tab.context_menu.is_none());
    }

    #[test]
    fn context_menu_variants_constructable() {
        use crate::state::ContextMenu;
        let _ = ContextMenu::Stash { index: 0 };
        let _ = ContextMenu::UnstagedFile {
            path: "a.rs".into(),
        };
        let _ = ContextMenu::StagedFile {
            path: "b.rs".into(),
        };
    }

    #[test]
    fn selected_unstaged_defaults_empty() {
        let tab = crate::state::RepoTab::new_empty();
        assert!(tab.selected_unstaged.is_empty());
        assert!(tab.selected_staged.is_empty());
    }

    #[test]
    fn selected_unstaged_toggle() {
        let mut tab = crate::state::RepoTab::new_empty();
        tab.selected_unstaged.insert("a.rs".to_string());
        tab.selected_unstaged.insert("b.rs".to_string());
        assert_eq!(tab.selected_unstaged.len(), 2);
        assert!(tab.selected_unstaged.contains("a.rs"));
        tab.selected_unstaged.remove("a.rs");
        assert_eq!(tab.selected_unstaged.len(), 1);
        assert!(!tab.selected_unstaged.contains("a.rs"));
    }

    #[test]
    fn detect_system_editor_returns_valid() {
        // Just verify it doesn't panic
        let editor = super::detect_system_editor();
        let _ = editor.display_name();
    }

    // ── Multi-file commit diff selection ──────────────────────────────────

    #[test]
    fn selected_commit_file_indices_defaults_to_empty_vec() {
        let tab = RepoTab::new_empty();
        assert!(tab.selected_commit_file_indices.is_empty());
        // Must be a Vec (ordered), not a HashSet — check it supports indexing
        let v: &Vec<usize> = &tab.selected_commit_file_indices;
        assert_eq!(v.len(), 0);
    }

    #[test]
    fn multi_file_diffs_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.multi_file_diffs.is_empty());
    }

    #[test]
    fn keyboard_modifiers_default_has_no_shift() {
        let state = GitKraft::new();
        assert!(!state.keyboard_modifiers.shift());
    }

    #[test]
    fn selected_commit_file_indices_preserves_insertion_order() {
        let mut tab = RepoTab::new_empty();
        tab.selected_commit_file_indices.push(5);
        tab.selected_commit_file_indices.push(2);
        tab.selected_commit_file_indices.push(8);
        assert_eq!(tab.selected_commit_file_indices, vec![5, 2, 8]);
    }

    #[test]
    fn selected_commit_file_indices_cleared_on_reset() {
        let mut tab = RepoTab::new_empty();
        tab.selected_commit_file_indices.push(0);
        tab.selected_commit_file_indices.push(1);
        tab.selected_commit_file_indices.clear();
        assert!(tab.selected_commit_file_indices.is_empty());
    }

    #[test]
    fn multi_file_diffs_cleared_on_reset() {
        let mut tab = RepoTab::new_empty();
        tab.multi_file_diffs.push(gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "a.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        });
        tab.multi_file_diffs.clear();
        assert!(tab.multi_file_diffs.is_empty());
    }

    #[test]
    fn commit_range_diffs_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.commit_range_diffs.is_empty());
    }

    #[test]
    fn commit_range_diffs_cleared_on_apply_payload() {
        // verify the field is reset — just check it's accessible and clearable
        let mut tab = RepoTab::new_empty();
        tab.commit_range_diffs.push(gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "x.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        });
        tab.commit_range_diffs.clear();
        assert!(tab.commit_range_diffs.is_empty());
    }

    // ── ModifiersChanged update ───────────────────────────────────────────

    #[test]
    fn modifiers_changed_sets_shift_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        assert!(!state.keyboard_modifiers.shift());

        let _ = state.update(Message::ModifiersChanged(iced::keyboard::Modifiers::SHIFT));
        assert!(state.keyboard_modifiers.shift());

        let _ = state.update(Message::ModifiersChanged(
            iced::keyboard::Modifiers::default(),
        ));
        assert!(!state.keyboard_modifiers.shift());
    }

    // ── SelectDiffByIndex update ──────────────────────────────────────────

    fn make_commit_files(names: &[&str]) -> Vec<gitkraft_core::DiffFileEntry> {
        names
            .iter()
            .map(|name| gitkraft_core::DiffFileEntry {
                old_file: String::new(),
                new_file: name.to_string(),
                status: gitkraft_core::FileStatus::Modified,
            })
            .collect()
    }

    #[test]
    fn select_diff_by_index_regular_click_clears_multi_selection() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Provide a repo_path and oid so the update handler can reach the
        // `selected_file_index = Some(index)` assignment (the async task it
        // spawns is dropped without execution — no real repo is needed).
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);
        // Pre-populate a multi-selection
        state.active_tab_mut().selected_commit_file_indices = vec![0, 1];

        // Regular click (no Shift) — should collapse to single-file selection
        let _ = state.update(Message::SelectDiffByIndex(0));

        assert!(state.active_tab().selected_commit_file_indices.is_empty());
        assert_eq!(state.active_tab().selected_file_index, Some(0));
    }

    #[test]
    fn regular_click_preserves_multi_file_diffs_until_load_completes() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);

        // Pre-populate multi_file_diffs as if user had a multi-file selection
        state.active_tab_mut().multi_file_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "a.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];

        // Regular click — should NOT clear multi_file_diffs immediately
        // (deferred to SingleFileDiffLoaded to avoid visual blink).
        let _ = state.update(Message::SelectDiffByIndex(1));

        assert!(
            !state.active_tab().multi_file_diffs.is_empty(),
            "multi_file_diffs must NOT be cleared on click (deferred to diff load)"
        );
        assert_eq!(state.active_tab().selected_file_index, Some(1));
        assert!(state.active_tab().is_loading_file_diff);
    }

    #[test]
    fn regular_click_preserves_diff_scroll_until_load_completes() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs"]);
        state.active_tab_mut().diff_scroll_offset = 150.0;

        // Regular click — should NOT reset diff_scroll_offset immediately.
        let _ = state.update(Message::SelectDiffByIndex(1));

        assert_eq!(
            state.active_tab().diff_scroll_offset,
            150.0,
            "diff_scroll_offset must NOT be reset on click (deferred to diff load)"
        );
    }

    #[test]
    fn single_file_diff_loaded_clears_deferred_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs"]);

        // Pre-populate multi_file_diffs and commit_range_diffs
        state.active_tab_mut().multi_file_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "a.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];
        state.active_tab_mut().commit_range_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "b.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];
        state.active_tab_mut().is_loading_file_diff = true;
        state.active_tab_mut().diff_scroll_offset = 200.0;

        // Simulate SingleFileDiffLoaded arriving with the new diff.
        let new_diff = gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "b.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        };
        let _ = state.update(Message::SingleFileDiffLoaded(Ok(new_diff)));

        // All deferred state should now be cleared.
        assert!(
            state.active_tab().multi_file_diffs.is_empty(),
            "multi_file_diffs must be cleared when new diff arrives"
        );
        assert!(
            state.active_tab().commit_range_diffs.is_empty(),
            "commit_range_diffs must be cleared when new diff arrives"
        );
        assert!(
            state.active_tab().selected_diff.is_some(),
            "selected_diff must be set to the loaded diff"
        );
        assert!(
            !state.active_tab().is_loading_file_diff,
            "is_loading_file_diff must be cleared after load"
        );
        assert_eq!(
            state.active_tab().diff_scroll_offset,
            0.0,
            "diff_scroll_offset must be reset when new diff arrives"
        );
    }

    #[test]
    fn regular_click_preserves_commit_range_diffs_until_load_completes() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs"]);

        // Pre-populate commit_range_diffs
        state.active_tab_mut().commit_range_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "x.rs".to_string(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];

        let _ = state.update(Message::SelectDiffByIndex(0));

        assert!(
            !state.active_tab().commit_range_diffs.is_empty(),
            "commit_range_diffs must NOT be cleared on click (deferred to diff load)"
        );
    }

    #[test]
    fn select_diff_by_index_shift_click_adds_both_files_to_selection() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);
        state.active_tab_mut().selected_file_index = Some(0);

        // Shift+Click on file 1 should anchor 0 and add 1
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectDiffByIndex(1));

        let sel = &state.active_tab().selected_commit_file_indices;
        assert!(sel.contains(&0), "anchor file 0 should be selected");
        assert!(sel.contains(&1), "newly clicked file 1 should be selected");
        assert_eq!(sel.len(), 2);
    }

    #[test]
    fn anchor_file_index_defaults_to_none() {
        let tab = RepoTab::new_empty();
        assert!(tab.anchor_file_index.is_none());
    }

    #[test]
    fn regular_click_sets_anchor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);

        let _ = state.update(Message::SelectDiffByIndex(2));

        assert_eq!(
            state.active_tab().anchor_file_index,
            Some(2),
            "regular click must set anchor to the clicked index"
        );
    }

    #[test]
    fn shift_click_selects_range_downward_from_anchor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files =
            make_commit_files(&["a.rs", "b.rs", "c.rs", "d.rs", "e.rs"]);
        // Anchor at index 1
        state.active_tab_mut().anchor_file_index = Some(1);
        state.active_tab_mut().selected_file_index = Some(1);

        // Shift+Click on index 4 — should select 1, 2, 3, 4
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectDiffByIndex(4));

        let sel = &state.active_tab().selected_commit_file_indices;
        assert_eq!(
            sel,
            &vec![1, 2, 3, 4],
            "range must be contiguous from anchor to click"
        );
    }

    #[test]
    fn shift_click_selects_range_upward_from_anchor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files =
            make_commit_files(&["a.rs", "b.rs", "c.rs", "d.rs", "e.rs"]);
        // Anchor at index 4 (bottom)
        state.active_tab_mut().anchor_file_index = Some(4);
        state.active_tab_mut().selected_file_index = Some(4);

        // Shift+Click on index 1 — should select 1, 2, 3, 4 (ascending)
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectDiffByIndex(1));

        let sel = &state.active_tab().selected_commit_file_indices;
        assert_eq!(
            sel,
            &vec![1, 2, 3, 4],
            "range must be stored ascending regardless of click direction"
        );
    }

    #[test]
    fn shift_click_anchor_fixed_on_subsequent_clicks() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files =
            make_commit_files(&["a.rs", "b.rs", "c.rs", "d.rs", "e.rs"]);
        // Anchor at index 2
        state.active_tab_mut().anchor_file_index = Some(2);
        state.active_tab_mut().selected_file_index = Some(2);
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;

        // First Shift+Click: extend to 4 → range {2, 3, 4}
        let _ = state.update(Message::SelectDiffByIndex(4));
        assert_eq!(
            state.active_tab().selected_commit_file_indices,
            vec![2, 3, 4]
        );

        // Second Shift+Click: shrink back to 3 → range {2, 3} (anchor still 2)
        let _ = state.update(Message::SelectDiffByIndex(3));
        assert_eq!(
            state.active_tab().selected_commit_file_indices,
            vec![2, 3],
            "anchor must stay fixed; second Shift+Click shrinks the range"
        );

        // Third Shift+Click: extend upward → range {0, 1, 2} (anchor still 2)
        let _ = state.update(Message::SelectDiffByIndex(0));
        assert_eq!(
            state.active_tab().selected_commit_file_indices,
            vec![0, 1, 2],
            "anchor must stay fixed; can extend range in either direction"
        );
    }

    #[test]
    fn shift_click_on_anchor_itself_gives_single_item_range() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);
        state.active_tab_mut().anchor_file_index = Some(1);
        state.active_tab_mut().selected_file_index = Some(1);

        // Shift+Click on the anchor itself → single-item range {1}
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectDiffByIndex(1));

        assert_eq!(state.active_tab().selected_commit_file_indices, vec![1]);
        assert!(
            state.active_tab().multi_file_diffs.is_empty(),
            "single-item range must not populate multi_file_diffs"
        );
    }

    #[test]
    fn shift_click_range_is_always_ascending() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().selected_commit_oid = Some("abc123".to_string());
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs", "d.rs"]);
        state.active_tab_mut().anchor_file_index = Some(3);
        state.active_tab_mut().selected_file_index = Some(3);

        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectDiffByIndex(0));

        let sel = &state.active_tab().selected_commit_file_indices;
        let is_sorted = sel.windows(2).all(|w| w[0] < w[1]);
        assert!(
            is_sorted,
            "selection must always be stored in ascending order"
        );
        assert_eq!(sel, &vec![0, 1, 2, 3]);
    }

    #[test]
    fn checkout_file_at_commit_message_variants_exist() {
        use crate::message::Message;
        // Verify the new message variants can be constructed
        let _single =
            Message::CheckoutFileAtCommit("abc123".to_string(), "src/main.rs".to_string());
        let _multi = Message::CheckoutMultiFilesAtCommit(
            "abc123".to_string(),
            vec!["a.rs".to_string(), "b.rs".to_string()],
        );
    }

    #[test]
    fn checkout_file_at_commit_closes_context_menu() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().context_menu = Some(crate::state::ContextMenu::CommitFile {
            oid: "abc123".to_string(),
            file_path: "src/main.rs".to_string(),
        });
        let _ = state.update(Message::CheckoutFileAtCommit(
            "abc123".to_string(),
            "src/main.rs".to_string(),
        ));
        assert!(state.active_tab().context_menu.is_none());
    }

    #[test]
    fn checkout_multi_files_at_commit_closes_context_menu() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().context_menu = Some(crate::state::ContextMenu::CommitFile {
            oid: "abc123".to_string(),
            file_path: "src/main.rs".to_string(),
        });
        let _ = state.update(Message::CheckoutMultiFilesAtCommit(
            "abc123".to_string(),
            vec!["src/main.rs".to_string(), "src/lib.rs".to_string()],
        ));
        assert!(state.active_tab().context_menu.is_none());
    }

    // ── Commit multi-selection ────────────────────────────────────────────

    fn make_test_commits(count: usize) -> Vec<gitkraft_core::CommitInfo> {
        (0..count)
            .map(|i| gitkraft_core::CommitInfo {
                oid: i.to_string(),
                summary: String::new(),
                message: String::new(),
                author_name: String::new(),
                author_email: String::new(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            })
            .collect()
    }

    #[test]
    fn selected_commits_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.selected_commits.is_empty());
        assert!(tab.anchor_commit_index.is_none());
    }

    #[test]
    fn regular_click_commit_sets_anchor_and_clears_range() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path = Some(std::path::PathBuf::from("/tmp/fake"));
        state.active_tab_mut().commits = make_test_commits(3);
        state.active_tab_mut().selected_commits = vec![0, 1, 2];

        let _ = state.update(Message::SelectCommit(1));

        assert_eq!(state.active_tab().anchor_commit_index, Some(1));
        assert!(state.active_tab().selected_commits.is_empty());
        assert_eq!(state.active_tab().selected_commit, Some(1));
    }

    #[test]
    fn shift_click_commit_selects_range_from_anchor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().commits = make_test_commits(5);
        state.active_tab_mut().anchor_commit_index = Some(1);
        state.active_tab_mut().selected_commit = Some(1);

        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectCommit(4));

        assert_eq!(state.active_tab().selected_commits, vec![1, 2, 3, 4]);
    }

    #[test]
    fn shift_click_commit_range_is_ascending_when_clicking_above_anchor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().commits = make_test_commits(5);
        state.active_tab_mut().anchor_commit_index = Some(3);
        state.active_tab_mut().selected_commit = Some(3);

        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        let _ = state.update(Message::SelectCommit(1));

        assert_eq!(state.active_tab().selected_commits, vec![1, 2, 3]);
    }

    // ── ExecuteCommitAction message ───────────────────────────────────────

    #[test]
    fn execute_commit_action_closes_context_menu() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().context_menu = Some(crate::state::ContextMenu::Commit {
            index: 0,
            oid: "abc123".to_string(),
        });

        let _ = state.update(Message::ExecuteCommitAction(
            "abc123".to_string(),
            gitkraft_core::CommitAction::CherryPick,
        ));

        assert!(state.active_tab().context_menu.is_none());
    }

    #[test]
    fn execute_commit_action_sets_loading_when_repo_open() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));

        let _ = state.update(Message::ExecuteCommitAction(
            "abc123".to_string(),
            gitkraft_core::CommitAction::ResetHard,
        ));

        assert!(state.active_tab().is_loading);
    }

    #[test]
    fn execute_commit_action_no_repo_does_not_set_loading() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // No repo_path set

        let _ = state.update(Message::ExecuteCommitAction(
            "abc123".to_string(),
            gitkraft_core::CommitAction::CherryPick,
        ));

        assert!(!state.active_tab().is_loading);
    }

    #[test]
    fn execute_commit_action_sets_status_message_from_action_label() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));

        let _ = state.update(Message::ExecuteCommitAction(
            "abc123".to_string(),
            gitkraft_core::CommitAction::Revert,
        ));

        let status = state.active_tab().status_message.as_deref().unwrap_or("");
        // Status message should contain the action's label
        assert!(
            status.contains("Revert commit"),
            "expected status to contain 'Revert commit', got: {status:?}"
        );
    }

    // ── File history / blame / delete state ──────────────────────────────

    #[test]
    fn file_history_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.file_history_path.is_none());
        assert!(tab.file_history_commits.is_empty());
        assert_eq!(tab.file_history_scroll, 0.0);
    }

    #[test]
    fn blame_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.blame_path.is_none());
        assert!(tab.blame_lines.is_empty());
        assert_eq!(tab.blame_scroll, 0.0);
    }

    #[test]
    fn pending_delete_file_defaults_none() {
        let tab = RepoTab::new_empty();
        assert!(tab.pending_delete_file.is_none());
    }

    #[test]
    fn view_file_history_sets_path_and_clears_blame() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().blame_path = Some("some/file.rs".to_string());

        let _ = state.update(Message::ViewFileHistory("src/main.rs".to_string()));

        assert_eq!(
            state.active_tab().file_history_path.as_deref(),
            Some("src/main.rs")
        );
        // Opening history should close blame
        assert!(state.active_tab().blame_path.is_none());
    }

    #[test]
    fn close_file_history_clears_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().file_history_path = Some("src/lib.rs".to_string());
        state.active_tab_mut().file_history_commits = vec![gitkraft_core::CommitInfo {
            oid: "abc".to_string(),
            summary: "s".to_string(),
            message: "s".to_string(),
            author_name: "a".to_string(),
            author_email: "a@b.com".to_string(),
            time: Default::default(),
            parent_ids: vec![],
            refs: Vec::new(),
        }];

        let _ = state.update(Message::CloseFileHistory);

        assert!(state.active_tab().file_history_path.is_none());
        assert!(state.active_tab().file_history_commits.is_empty());
    }

    #[test]
    fn view_file_blame_sets_path_and_clears_history() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().file_history_path = Some("some/file.rs".to_string());

        let _ = state.update(Message::ViewFileBlame("src/lib.rs".to_string()));

        assert_eq!(state.active_tab().blame_path.as_deref(), Some("src/lib.rs"));
        // Opening blame should close history
        assert!(state.active_tab().file_history_path.is_none());
    }

    #[test]
    fn selecting_new_commit_closes_blame_overlay() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        // Pre-populate a commit list so SelectCommit can find the commit.
        state.active_tab_mut().commits = vec![
            gitkraft_core::CommitInfo {
                oid: "abc1".into(),
                summary: "first".into(),
                message: "first".into(),
                author_name: "A".into(),
                author_email: "a@a.com".into(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            },
            gitkraft_core::CommitInfo {
                oid: "abc2".into(),
                summary: "second".into(),
                message: "second".into(),
                author_name: "A".into(),
                author_email: "a@a.com".into(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            },
        ];
        // Blame is currently open for a file from the first commit.
        state.active_tab_mut().blame_path = Some("src/lib.rs".to_string());
        state.active_tab_mut().blame_lines = vec![gitkraft_core::BlameLine {
            line_number: 1,
            content: "fn main() {}".into(),
            oid: "abc1".into(),
            author_name: "A".into(),
            time: Default::default(),
        }];

        // Click a different commit — blame must close automatically.
        let _ = state.update(Message::SelectCommit(1));

        assert!(
            state.active_tab().blame_path.is_none(),
            "blame_path must be cleared when a new commit is selected"
        );
        assert!(
            state.active_tab().blame_lines.is_empty(),
            "blame_lines must be cleared when a new commit is selected"
        );
    }

    #[test]
    fn close_file_blame_clears_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().blame_path = Some("src/lib.rs".to_string());

        let _ = state.update(Message::CloseFileBlame);

        assert!(state.active_tab().blame_path.is_none());
        assert!(state.active_tab().blame_lines.is_empty());
    }

    #[test]
    fn delete_file_sets_pending() {
        use crate::message::Message;
        let mut state = GitKraft::new();

        let _ = state.update(Message::DeleteFile("src/old.rs".to_string()));

        assert_eq!(
            state.active_tab().pending_delete_file.as_deref(),
            Some("src/old.rs")
        );
        assert!(state.active_tab().context_menu.is_none());
    }

    #[test]
    fn cancel_delete_file_clears_pending() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().pending_delete_file = Some("src/old.rs".to_string());

        let _ = state.update(Message::CancelDeleteFile);

        assert!(state.active_tab().pending_delete_file.is_none());
    }

    #[test]
    fn confirm_delete_file_no_repo_is_noop() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().pending_delete_file = Some("src/old.rs".to_string());
        // No repo_path → should not set is_loading

        let _ = state.update(Message::ConfirmDeleteFile);

        assert!(!state.active_tab().is_loading);
    }

    #[test]
    fn shift_arrow_down_extends_file_list_selection_when_files_loaded() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);
        state.active_tab_mut().selected_file_index = Some(0);
        state.active_tab_mut().anchor_file_index = Some(0);
        // keyboard_modifiers must have SHIFT set for range selection to trigger
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;

        let _ = state.update(Message::ShiftArrowDown);

        assert_eq!(state.active_tab().selected_file_index, Some(1));
        // Range should now include both files
        assert!(state.active_tab().selected_commit_file_indices.contains(&0));
        assert!(state.active_tab().selected_commit_file_indices.contains(&1));
    }

    #[test]
    fn shift_arrow_down_falls_through_to_commit_log_when_no_files() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().commits = make_test_commits(5);
        state.active_tab_mut().selected_commit = Some(1);
        state.active_tab_mut().anchor_commit_index = Some(1);
        state.keyboard_modifiers = iced::keyboard::Modifiers::SHIFT;
        // no commit_files

        let _ = state.update(Message::ShiftArrowDown);

        assert_eq!(state.active_tab().selected_commit, Some(2));
        assert!(state.active_tab().selected_commits.contains(&1));
        assert!(state.active_tab().selected_commits.contains(&2));
    }

    #[test]
    fn file_system_changed_triggers_full_refresh() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));

        // FileSystemChanged calls refresh_active_tab() which now sets
        // is_loading = true synchronously to prevent concurrent refresh storms.
        let _task = state.update(Message::FileSystemChanged);

        // is_loading must be set to prevent overlapping refreshes.
        assert!(
            state.active_tab().is_loading,
            "FileSystemChanged must set is_loading to guard against concurrent refreshes"
        );
        assert!(
            state.active_tab().error_message.is_none(),
            "FileSystemChanged must not set an error message"
        );
    }

    #[test]
    fn refresh_active_tab_skips_when_already_loading() {
        // When is_loading is true (e.g. a manual Ctrl+R refresh is in flight),
        // refresh_active_tab must return Task::none() to avoid spawning a
        // duplicate concurrent refresh.
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path =
            Some(std::path::PathBuf::from("/tmp/fake-repo-for-test"));
        state.active_tab_mut().is_loading = true;

        // With is_loading set, refresh_active_tab should be a no-op.
        let _task = state.refresh_active_tab();
        // The tab must still be loading (unchanged).
        assert!(state.active_tab().is_loading);
    }

    // ── Tab duplication prevention tests ───────────────────────────────────

    /// Helper: build a minimal `RepoPayload` (aka `RepoSnapshot`) for a given path.
    fn fake_payload(workdir: &str) -> crate::message::RepoPayload {
        gitkraft_core::RepoSnapshot {
            info: gitkraft_core::RepoInfo {
                path: std::path::PathBuf::from(format!("{workdir}/.git")),
                workdir: Some(std::path::PathBuf::from(workdir)),
                head_branch: Some("main".into()),
                is_bare: false,
                state: gitkraft_core::RepoState::Clean,
            },
            branches: Vec::new(),
            commits: Vec::new(),
            graph_rows: Vec::new(),
            unstaged: Vec::new(),
            staged: Vec::new(),
            stashes: Vec::new(),
            remotes: Vec::new(),
        }
    }

    /// Helper: set up a tab as if a repo was fully loaded.
    fn setup_loaded_tab(tab: &mut RepoTab, path: &str) {
        tab.repo_path = Some(std::path::PathBuf::from(path));
        tab.repo_info = Some(gitkraft_core::RepoInfo {
            path: std::path::PathBuf::from(format!("{path}/.git")),
            workdir: Some(std::path::PathBuf::from(path)),
            head_branch: Some("main".into()),
            is_bare: false,
            state: gitkraft_core::RepoState::Clean,
        });
    }

    #[test]
    fn open_repo_creates_new_tab_when_repo_already_open() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);

        // Clicking "Open" should create a new tab when the current has a repo.
        let _task = state.update(Message::OpenRepo);

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1);
        // The new tab should be loading (folder picker opening).
        assert!(state.tabs[1].is_loading);
        // The original tab should be untouched.
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    #[test]
    fn open_repo_reuses_empty_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Active tab is empty (no repo loaded).
        assert!(!state.active_tab().has_repo());

        let _task = state.update(Message::OpenRepo);

        // Should NOT create a new tab when the active tab is empty.
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert!(state.tabs[0].is_loading);
    }

    #[test]
    fn repo_selected_deduplicates_already_open_repo() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: fully loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty (simulates the new tab created by OpenRepo)
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        // User selects a folder that matches the already-open repo.
        let _task = state.update(Message::RepoSelected(Some(std::path::PathBuf::from(
            "/home/user/repo-a",
        ))));

        // Should switch to the existing tab and remove the empty one.
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    #[test]
    fn repo_selected_opens_new_repo_in_empty_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: fully loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty (simulates the new tab created by OpenRepo)
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        // User selects a DIFFERENT repo.
        let _task = state.update(Message::RepoSelected(Some(std::path::PathBuf::from(
            "/home/user/repo-b",
        ))));

        // Should keep both tabs; the empty tab is now loading repo-b.
        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1);
        assert!(state.tabs[1]
            .status_message
            .as_deref()
            .unwrap_or("")
            .contains("repo-b"));
    }

    #[test]
    fn repo_selected_cancel_removes_empty_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: fully loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty (created by OpenRepo, waiting for folder picker)
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        // User cancels the folder picker.
        let _task = state.update(Message::RepoSelected(None));

        // The empty tab should be removed; switch back to tab 0.
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    #[test]
    fn repo_selected_cancel_keeps_tab_if_only_one() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Single empty tab — shouldn't be removed on cancel.
        assert_eq!(state.tabs.len(), 1);
        assert!(!state.active_tab().has_repo());

        let _task = state.update(Message::RepoSelected(None));

        assert_eq!(state.tabs.len(), 1);
        assert!(!state.active_tab().is_loading);
    }

    #[test]
    fn open_recent_repo_deduplicates() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: fully loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty tab (simulates user clicking "+")
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        // Opening a recent repo that's already open should switch to it
        // AND remove the empty tab the user was on.
        let _task = state.update(Message::OpenRecentRepo(std::path::PathBuf::from(
            "/home/user/repo-a",
        )));

        assert_eq!(state.tabs.len(), 1, "empty tab must be removed");
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
        );
    }

    #[test]
    fn open_recent_repo_creates_new_tab_when_current_has_repo() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: fully loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        // Opening a different recent repo should create a new tab.
        let _task = state.update(Message::OpenRecentRepo(std::path::PathBuf::from(
            "/home/user/repo-b",
        )));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1);
        assert!(state.tabs[1].is_loading);
    }

    #[test]
    fn open_recent_repo_uses_empty_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Active tab is empty.
        assert!(!state.active_tab().has_repo());

        let _task = state.update(Message::OpenRecentRepo(std::path::PathBuf::from(
            "/home/user/repo-b",
        )));

        // Should NOT create a new tab.
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert!(state.tabs[0].is_loading);
    }

    #[test]
    fn open_recent_repo_dedup_removes_empty_tab_with_multiple_loaded() {
        // Reproduces: user has [A, B*, C], clicks "+" to get [A, B, C, Empty*],
        // then clicks recent repo B → should end up with [A, B*, C] (no empty tab).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        // Simulate clicking "+" to add an empty tab
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 3;

        let _task = state.update(Message::OpenRecentRepo(std::path::PathBuf::from(
            "/home/user/repo-b",
        )));

        assert_eq!(state.tabs.len(), 3, "empty tab must be cleaned up");
        assert_eq!(state.active_tab, 1);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
        );
    }

    // ── Refresh race-condition tests ──────────────────────────────────────

    #[test]
    fn repo_refreshed_targets_correct_tab_after_tab_switch() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a fully loaded
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty tab (user clicked "+")
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1; // user switched to new empty tab

        // Simulate: a RepoRefreshed result arrives for repo-a while tab 1 is active.
        let payload = fake_payload("/home/user/repo-a");
        let _task = state.update(Message::RepoRefreshed(Ok(payload)));

        // The payload must land in tab 0 (which owns repo-a), NOT tab 1.
        assert!(
            state.tabs[0].repo_info.is_some(),
            "tab 0 should still have repo info after refresh"
        );
        assert_eq!(
            state.tabs[0].current_branch.as_deref(),
            Some("main"),
            "tab 0 should have updated branch from payload"
        );
        // Tab 1 must remain empty.
        assert!(
            state.tabs[1].repo_info.is_none(),
            "tab 1 (empty) must NOT receive the refresh payload"
        );
        assert!(
            state.tabs[1].repo_path.is_none(),
            "tab 1 should still have no repo path"
        );
    }

    #[test]
    fn repo_refreshed_targets_active_tab_for_new_open() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Active tab is empty — user just opened a brand-new repo.
        // (RepoSelected set the status but the tab doesn't have repo_path yet
        // matching the payload, so it falls back to active tab.)
        assert_eq!(state.tabs.len(), 1);
        assert!(!state.active_tab().has_repo());

        let payload = fake_payload("/home/user/new-repo");
        let _task = state.update(Message::RepoOpened(Ok(payload)));

        // Should have applied to the active tab (the only tab).
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/new-repo"))
        );
        assert!(state.tabs[0].repo_info.is_some());
    }

    #[test]
    fn repo_refreshed_does_not_duplicate_into_new_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a loaded
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        // User clicks "+" creating an empty tab 1, then switches to it.
        let _task = state.update(Message::NewTab);
        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1);

        // A refresh result for repo-a arrives (was triggered before the tab switch).
        let payload = fake_payload("/home/user/repo-a");
        let _task = state.update(Message::RepoRefreshed(Ok(payload)));

        // Tab 1 must remain empty — the payload should go to tab 0.
        assert!(
            state.tabs[1].repo_path.is_none(),
            "new empty tab must not receive repo-a refresh"
        );
        assert!(
            state.tabs[1].repo_info.is_none(),
            "new empty tab must not have repo_info"
        );
        // Tab 0 must have the refreshed data.
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    #[test]
    fn git_operation_result_targets_correct_tab() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: repo-b
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        // A git operation result arrives for repo-a (e.g. push completed)
        // while user is on tab 1.
        let payload = fake_payload("/home/user/repo-a");
        let _task = state.update(Message::GitOperationResult(Ok(payload)));

        // Payload should land in tab 0 (repo-a), not tab 1.
        assert_eq!(state.tabs[0].current_branch.as_deref(), Some("main"));
        // Tab 1's data should remain unchanged (still repo-b).
        assert_eq!(
            state.tabs[1].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    #[test]
    fn multiple_new_tabs_dont_get_polluted_by_refresh() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a loaded
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        // User creates multiple new tabs.
        let _task = state.update(Message::NewTab);
        let _task = state.update(Message::NewTab);
        assert_eq!(state.tabs.len(), 3);
        assert_eq!(state.active_tab, 2);

        // Refresh arrives for repo-a.
        let payload = fake_payload("/home/user/repo-a");
        let _task = state.update(Message::RepoRefreshed(Ok(payload)));

        // Only tab 0 should be affected.
        assert!(state.tabs[0].repo_info.is_some());
        assert!(state.tabs[1].repo_info.is_none());
        assert!(state.tabs[2].repo_info.is_none());
        assert!(state.tabs[1].repo_path.is_none());
        assert!(state.tabs[2].repo_path.is_none());
    }

    #[test]
    fn repo_selected_dedup_adjusts_index_when_existing_is_after_active() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: empty (newly created by OpenRepo)
        // Tab 1: repo-a loaded
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-a");
        state.active_tab = 0;

        // User selects the same folder as repo-a.
        let _task = state.update(Message::RepoSelected(Some(std::path::PathBuf::from(
            "/home/user/repo-a",
        ))));

        // Empty tab 0 should be removed; we should now be on tab 0 (formerly tab 1).
        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    // ── Multi-selection preservation across refresh ───────────────────────

    #[test]
    fn apply_payload_preserves_multi_selection_by_oid() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(5);
        // Simulate user selecting commits 1..=3 with Shift+click.
        tab.anchor_commit_index = Some(1);
        tab.selected_commits = vec![1, 2, 3];
        tab.selected_commit = Some(3);
        tab.selected_commit_oid = Some(tab.commits[3].oid.clone());

        // Build a payload with the same commits (simulating a background refresh).
        let mut payload = fake_payload("/tmp/repo");
        payload.commits = make_test_commits(5);

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        // The multi-selection must survive the refresh.
        assert_eq!(tab.anchor_commit_index, Some(1));
        assert_eq!(tab.selected_commits, vec![1, 2, 3]);
        assert_eq!(tab.selected_commit, Some(3));
    }

    #[test]
    fn apply_payload_preserves_anchor_even_without_range() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(5);
        tab.anchor_commit_index = Some(2);
        tab.selected_commit = Some(2);
        tab.selected_commit_oid = Some(tab.commits[2].oid.clone());
        // No range selection — just a single click with anchor set.

        let mut payload = fake_payload("/tmp/repo");
        payload.commits = make_test_commits(5);

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        assert_eq!(tab.anchor_commit_index, Some(2));
        assert_eq!(tab.selected_commit, Some(2));
    }

    #[test]
    fn apply_payload_clears_selection_when_commits_disappear() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(5);
        tab.anchor_commit_index = Some(2);
        tab.selected_commits = vec![2, 3, 4];
        tab.selected_commit = Some(4);
        tab.selected_commit_oid = Some(tab.commits[4].oid.clone());

        // Payload has completely different commits (e.g. force-push).
        let mut payload = fake_payload("/tmp/repo");
        payload.commits = (0..3)
            .map(|i| gitkraft_core::CommitInfo {
                oid: format!("new_oid_{i}"),
                summary: format!("new commit {i}"),
                message: String::new(),
                author_name: "Author".into(),
                author_email: "a@b.c".into(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            })
            .collect();

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        // All old OIDs are gone — selection should be cleared.
        assert!(tab.selected_commits.is_empty());
        assert!(tab.anchor_commit_index.is_none());
        assert!(tab.selected_commit.is_none());
    }

    // ── File multi-selection preservation across refresh ─────────────────

    #[test]
    fn apply_payload_preserves_file_selection_when_commit_survives() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(3);
        tab.selected_commit = Some(1);
        tab.selected_commit_oid = Some(tab.commits[1].oid.clone());
        tab.commit_files = make_commit_files(&["a.rs", "b.rs", "c.rs"]);

        // Simulate user Shift+clicking files 0..=2.
        tab.anchor_file_index = Some(0);
        tab.selected_commit_file_indices = vec![0, 1, 2];
        tab.multi_file_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "a.rs".into(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];

        // Background refresh with the same commits.
        let mut payload = fake_payload("/tmp/repo");
        payload.commits = make_test_commits(3);

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        // File selection must survive.
        assert_eq!(tab.anchor_file_index, Some(0));
        assert_eq!(tab.selected_commit_file_indices, vec![0, 1, 2]);
        assert!(
            !tab.multi_file_diffs.is_empty(),
            "multi_file_diffs must survive when commit is preserved"
        );
    }

    #[test]
    fn apply_payload_clears_file_selection_when_commit_disappears() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(3);
        tab.selected_commit = Some(1);
        tab.selected_commit_oid = Some(tab.commits[1].oid.clone());
        tab.commit_files = make_commit_files(&["a.rs", "b.rs"]);
        tab.anchor_file_index = Some(0);
        tab.selected_commit_file_indices = vec![0, 1];
        tab.multi_file_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "a.rs".into(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];

        // Payload with completely different commits (force-push).
        let mut payload = fake_payload("/tmp/repo");
        payload.commits = (0..2)
            .map(|i| gitkraft_core::CommitInfo {
                oid: format!("new_{i}"),
                summary: String::new(),
                message: String::new(),
                author_name: String::new(),
                author_email: String::new(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            })
            .collect();

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        // Commit is gone → file selection must be cleared.
        assert!(tab.anchor_file_index.is_none());
        assert!(tab.selected_commit_file_indices.is_empty());
        assert!(tab.multi_file_diffs.is_empty());
        assert!(tab.commit_files.is_empty());
    }

    #[test]
    fn apply_payload_preserves_commit_range_diffs_when_commit_survives() {
        let mut tab = RepoTab::new_empty();
        tab.commits = make_test_commits(5);
        tab.selected_commit = Some(2);
        tab.selected_commit_oid = Some(tab.commits[2].oid.clone());
        tab.commit_range_diffs = vec![gitkraft_core::DiffInfo {
            old_file: String::new(),
            new_file: "x.rs".into(),
            status: gitkraft_core::FileStatus::Modified,
            hunks: vec![],
        }];

        let mut payload = fake_payload("/tmp/repo");
        payload.commits = make_test_commits(5);

        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        assert!(
            !tab.commit_range_diffs.is_empty(),
            "commit_range_diffs must survive when commit is preserved"
        );
    }

    #[test]
    fn apply_payload_preserves_commit_message() {
        // Regression test: the git watcher triggers refreshes while the user
        // is typing a commit message.  apply_payload must NOT clear it.
        let mut tab = RepoTab::new_empty();
        tab.repo_path = Some(std::path::PathBuf::from("/tmp/repo"));
        tab.commit_message = "WIP: fixing the bug".to_string();

        let payload = fake_payload("/tmp/repo");
        tab.apply_payload(payload, std::path::PathBuf::from("/tmp/repo"));

        assert_eq!(
            tab.commit_message, "WIP: fixing the bug",
            "commit_message must survive background repo refreshes"
        );
    }

    // ── PreviewFile / OpenFiles message tests ─────────────────────────────

    #[test]
    fn preview_file_loaded_sets_preview_editor() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path = Some(std::path::PathBuf::from("/tmp/fake-repo"));

        let content = "line 1\nline 2\nline 3".to_string();
        let _ = state.update(Message::FilePreviewLoaded(Ok((
            "test.rs".to_string(),
            content,
        ))));

        assert!(state.preview_editor.is_some());
        assert_eq!(state.preview_path.as_deref(), Some("test.rs"));
        // The diff view should be cleared so the preview takes over.
        assert!(state.active_tab().selected_diff.is_none());
    }

    #[test]
    fn preview_file_loaded_error_sets_error_message() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.active_tab_mut().repo_path = Some(std::path::PathBuf::from("/tmp/fake-repo"));

        let _ = state.update(Message::FilePreviewLoaded(Err("No such file".to_string())));

        assert!(state.active_tab().error_message.is_some());
        assert!(state
            .active_tab()
            .error_message
            .as_ref()
            .unwrap()
            .contains("No such file"));
    }

    #[test]
    fn close_preview_clears_editor_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        state.preview_editor = Some(iced::widget::text_editor::Content::with_text("hello"));
        state.preview_path = Some("/test/file.txt".to_string());

        let _ = state.update(Message::ClosePreview);

        assert!(state.preview_editor.is_none());
        assert!(state.preview_path.is_none());
    }

    // ── CloseRepo ─────────────────────────────────────────────────────────────

    #[test]
    fn close_repo_single_tab_shows_welcome_screen() {
        // When only one tab is open, CloseRepo should empty the tab (show
        // welcome screen) rather than removing it — there must always be at
        // least one tab.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        assert_eq!(state.tabs.len(), 1);

        let _ = state.update(Message::CloseRepo);

        assert_eq!(state.tabs.len(), 1, "last tab must not be removed");
        assert_eq!(state.active_tab, 0);
        assert!(!state.active_tab().has_repo(), "tab should be empty");
    }

    #[test]
    fn close_repo_with_multiple_tabs_removes_current_tab() {
        // The core of the reported bug: when the currently selected repo is
        // closed and there are other tabs open, the CURRENT tab should be
        // removed and the user should land on the remaining tab — NOT have the
        // current tab emptied in place (which left the other repo displaced).
        //
        // Before fix: CloseRepo replaced active tab with empty, keeping it
        //   selected, so the user saw a blank welcome screen while the other
        //   repo was still in the tab bar but not visible → perceived as
        //   "removes the other one and keeps the current one selected".
        // After fix: active tab is removed, user lands on the other repo.
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a (will be the active/selected tab)
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: repo-b
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        // repo-a (tab 0) is the currently selected tab
        state.active_tab = 0;

        let _ = state.update(Message::CloseRepo);

        // The current tab (repo-a) must be gone; repo-b must be the only
        // remaining tab and must be active.
        assert_eq!(state.tabs.len(), 1, "current tab must be removed");
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
            "repo-b should be the surviving tab"
        );
    }

    #[test]
    fn close_repo_second_tab_active_removes_it_and_keeps_first() {
        // Same scenario but with the SECOND tab as the active (selected) one —
        // the first tab must survive and become the new active tab.
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: repo-b (active)
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1; // repo-b is the currently selected tab

        let _ = state.update(Message::CloseRepo);

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
            "repo-a should be the surviving tab"
        );
    }

    // ── CloseTab ──────────────────────────────────────────────────────────

    #[test]
    fn close_tab_active_tab_first_position() {
        // active_tab=0, close tab 0 → tab 1 (repo-b) becomes active at index 0.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 0;

        let _ = state.update(Message::CloseTab(0));

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    #[test]
    fn close_tab_active_tab_second_position() {
        // active_tab=1, close tab 1 → tab 0 (repo-a) becomes active.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        let _ = state.update(Message::CloseTab(1));

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a"))
        );
    }

    #[test]
    fn close_tab_non_active_tab_before_active() {
        // active_tab=1, close tab 0 → repo-b stays active but shifts to index 0.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        let _ = state.update(Message::CloseTab(0));

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0, "active_tab must shift down by 1");
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
            "repo-b must remain active"
        );
    }

    #[test]
    fn close_tab_non_active_tab_after_active() {
        // active_tab=0, close tab 1 → repo-a stays active at index 0.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 0;

        let _ = state.update(Message::CloseTab(1));

        assert_eq!(state.tabs.len(), 1);
        assert_eq!(state.active_tab, 0, "active_tab must remain unchanged");
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
            "repo-a must remain active"
        );
    }

    #[test]
    fn close_tab_last_remaining_tab_is_noop() {
        // Can't close the last tab — CloseTab must be a no-op.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        assert_eq!(state.tabs.len(), 1);

        let _ = state.update(Message::CloseTab(0));

        assert_eq!(state.tabs.len(), 1, "last tab must not be removed");
        assert!(
            state.active_tab().has_repo(),
            "tab content must be unchanged"
        );
    }

    #[test]
    fn close_tab_out_of_bounds_index_is_noop() {
        // Closing a tab index that doesn't exist must be silently ignored.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        assert_eq!(state.tabs.len(), 2);

        let _ = state.update(Message::CloseTab(99));

        assert_eq!(state.tabs.len(), 2, "no tab should be removed");
    }

    // ── CloseTab — three-tab scenarios ────────────────────────────────────

    #[test]
    fn close_tab_three_tabs_close_first_active() {
        // [repo-a*(0), repo-b(1), repo-c(2)] close 0
        // → [repo-b(0), repo-c(1)], active stays 0 (repo-b).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 0;

        let _ = state.update(Message::CloseTab(0));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    #[test]
    fn close_tab_three_tabs_close_active_middle() {
        // [repo-a(0), repo-b*(1), repo-c(2)] close 1
        // → [repo-a(0), repo-c(1)], active_tab stays 1 (repo-c slides in).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 1;

        let _ = state.update(Message::CloseTab(1));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(
            state.active_tab, 1,
            "active_tab must stay at 1 (repo-c slides in)"
        );
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-c"))
        );
    }

    #[test]
    fn close_tab_three_tabs_close_active_last() {
        // [repo-a(0), repo-b(1), repo-c*(2)] close 2
        // → [repo-a(0), repo-b(1)], active_tab becomes 1 (clamped).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 2;

        let _ = state.update(Message::CloseTab(2));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(
            state.active_tab, 1,
            "active_tab must be clamped to last index"
        );
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    #[test]
    fn close_tab_three_tabs_close_first_when_middle_active() {
        // [repo-a(0), repo-b*(1), repo-c(2)] close 0
        // → [repo-b(0), repo-c(1)], active_tab shifts from 1 to 0.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 1;

        let _ = state.update(Message::CloseTab(0));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(
            state.active_tab, 0,
            "active_tab must decrement when a tab before it is removed"
        );
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
            "repo-b must remain active"
        );
    }

    #[test]
    fn close_tab_three_tabs_close_last_when_middle_active() {
        // [repo-a(0), repo-b*(1), repo-c(2)] close 2
        // → [repo-a(0), repo-b(1)], active_tab stays at 1.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 1;

        let _ = state.update(Message::CloseTab(2));

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(
            state.active_tab, 1,
            "active_tab must not change when a tab after it is removed"
        );
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
            "repo-b must remain active"
        );
    }

    // ── CloseRepo — three-tab scenarios ───────────────────────────────────

    #[test]
    fn close_repo_three_tabs_active_in_middle() {
        // [repo-a(0), repo-b*(1), repo-c(2)] close active
        // → [repo-a(0), repo-c(1)], active_tab stays 1 (repo-c slides in).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 1;

        let _ = state.update(Message::CloseRepo);

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-c"))
        );
    }

    #[test]
    fn close_repo_three_tabs_active_at_end() {
        // [repo-a(0), repo-b(1), repo-c*(2)] close active
        // → [repo-a(0), repo-b(1)], active_tab clamped to 1.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-c");
        state.active_tab = 2;

        let _ = state.update(Message::CloseRepo);

        assert_eq!(state.tabs.len(), 2);
        assert_eq!(state.active_tab, 1, "active_tab clamped to last index");
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    // ── NewTab ────────────────────────────────────────────────────────────

    #[test]
    fn new_tab_appends_empty_tab_and_makes_it_active() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        assert_eq!(state.tabs.len(), 1);

        let _ = state.update(Message::NewTab);

        assert_eq!(state.tabs.len(), 2, "a new empty tab must be appended");
        assert_eq!(state.active_tab, 1, "new tab must become active");
        assert!(!state.active_tab().has_repo(), "new tab must be empty");
        // The original repo tab must be unaffected.
        assert!(state.tabs[0].has_repo());
    }

    #[test]
    fn new_tab_with_multiple_existing_tabs() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 0;

        let _ = state.update(Message::NewTab);

        assert_eq!(state.tabs.len(), 3);
        assert_eq!(state.active_tab, 2);
        assert!(!state.active_tab().has_repo());
    }

    // ── SwitchTab ─────────────────────────────────────────────────────────

    #[test]
    fn switch_tab_to_valid_index() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 0;

        let _ = state.update(Message::SwitchTab(1));

        assert_eq!(state.active_tab, 1);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b"))
        );
    }

    #[test]
    fn switch_tab_to_same_index_is_noop() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        let _ = state.update(Message::SwitchTab(1));

        assert_eq!(state.active_tab, 1, "should remain on tab 1");
    }

    #[test]
    fn switch_tab_out_of_bounds_is_noop() {
        // Firing SwitchTab with a too-large index must not panic or change state.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 0;

        let _ = state.update(Message::SwitchTab(99));

        assert_eq!(state.active_tab, 0, "out-of-bounds switch must be ignored");
    }

    // ── session_state ─────────────────────────────────────────────────────

    #[test]
    fn session_state_matches_open_tab_paths_when_no_empty_tabs() {
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        let (paths, active) = state.session_state();
        assert_eq!(paths.len(), 2);
        assert_eq!(active, 1);
    }

    #[test]
    fn session_state_adjusts_index_when_empty_tab_before_active() {
        // tabs: [Empty(0), repo-a(1), repo-b(2)], active = 2
        // session paths: [repo-a, repo-b] → session active should be 1
        let mut state = GitKraft::new();
        // Tab 0 is already an empty tab from new()
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[2], "/home/user/repo-b");
        state.active_tab = 2;

        let (paths, active) = state.session_state();
        assert_eq!(paths.len(), 2);
        assert_eq!(active, 1, "session index must skip the empty tab at 0");
    }

    #[test]
    fn session_state_defaults_to_zero_when_active_is_empty_tab() {
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.tabs.push(RepoTab::new_empty()); // empty tab at index 1
        state.active_tab = 1; // active is the empty tab

        let (paths, active) = state.session_state();
        assert_eq!(paths.len(), 1);
        assert_eq!(active, 0, "empty active tab should default to 0");
    }

    // ── Race condition: repo_path set before load ─────────────────────────

    #[test]
    fn repo_selected_sets_repo_path_before_load() {
        // After RepoSelected, the empty tab must have repo_path set so that
        // handle_repo_loaded can target it by path even if the user switches
        // tabs during loading.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Simulate OpenRepo creating a new empty tab
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        let _task = state.update(Message::RepoSelected(Some(std::path::PathBuf::from(
            "/home/user/repo-b",
        ))));

        // The active tab should now have repo_path set
        assert_eq!(
            state.tabs[1].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
            "repo_path must be set before async load starts"
        );
    }

    #[test]
    fn open_recent_repo_sets_repo_path_before_load() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Active tab is empty (no repo open)
        assert!(!state.active_tab().has_repo());

        let _task = state.update(Message::OpenRecentRepo(std::path::PathBuf::from(
            "/home/user/repo-a",
        )));

        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
            "repo_path must be set before async load starts"
        );
    }

    #[test]
    fn handle_repo_loaded_targets_correct_tab_after_switch() {
        // Regression test: user opens repo-b in a new tab, then switches
        // back to tab 0 before the load completes.  The loaded data must
        // still go to the correct tab (the one with repo_path = repo-b).
        use crate::message::Message;
        let mut state = GitKraft::new();
        // Tab 0: repo-a loaded
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty, has repo_path set (simulates RepoSelected having set it)
        state.tabs.push(RepoTab::new_empty());
        state.tabs[1].repo_path = Some(std::path::PathBuf::from("/home/user/repo-b"));
        state.tabs[1].is_loading = true;

        // User switches back to tab 0
        state.active_tab = 0;

        // Repo-b load completes
        let payload = fake_payload("/home/user/repo-b");
        let _ = state.update(Message::RepoOpened(Ok(payload)));

        // The payload must have gone to tab 1, NOT tab 0
        assert!(
            state.tabs[1].repo_info.is_some(),
            "repo-b payload must be written to tab 1"
        );
        assert_eq!(
            state.tabs[1].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-b")),
        );
        // Tab 0 must still be repo-a, untouched
        assert_eq!(
            state.tabs[0].repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
        );
    }

    // ── Scroll & pagination fix tests ─────────────────────────────────────

    #[test]
    fn apply_payload_does_not_reset_is_loading_more_commits() {
        // A background refresh must not clear is_loading_more_commits — that
        // flag is owned by the pagination flow.  Clearing it would let the
        // scroll handler fire a duplicate pagination whose identical result
        // falsely sets has_more_commits = false ("end of history").
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        // Simulate pagination in flight
        state.active_tab_mut().is_loading_more_commits = true;

        // Background refresh arrives
        let payload = fake_payload("/home/user/repo-a");
        let _ = state.update(Message::RepoRefreshed(Ok(payload)));

        // is_loading_more_commits must still be true
        assert!(
            state.active_tab().is_loading_more_commits,
            "refresh must not reset is_loading_more_commits"
        );
    }

    #[test]
    fn apply_payload_resets_is_loading_more_commits_for_new_open() {
        // For a brand-new repo open (not a refresh), the flag doesn't matter
        // because no pagination can be in flight.  But is_loading_more_commits
        // is initialized to false in new_empty(), and apply_payload should not
        // set it to true.
        let mut state = GitKraft::new();
        assert!(!state.active_tab().is_loading_more_commits);

        let payload = fake_payload("/home/user/repo-a");
        let _ = state.update(crate::message::Message::RepoOpened(Ok(payload)));

        assert!(!state.active_tab().is_loading_more_commits);
    }

    #[test]
    fn refresh_does_not_shrink_commit_list() {
        // If pagination extended commits to 700 and then a stale refresh
        // arrives with only 500, the commit list must not shrink.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        // Give the tab some initial commits, then extend via simulated pagination
        let tab = state.active_tab_mut();
        tab.commits = make_test_commits(500);
        for i in 0..200 {
            let mut c = tab.commits[0].clone();
            c.oid = format!("extra_{i:04}");
            tab.commits.push(c);
        }
        let extended_len = tab.commits.len();
        tab.has_more_commits = true;

        // Stale refresh arrives with only the original commits (fewer)
        let mut payload = fake_payload("/home/user/repo-a");
        payload.commits = make_test_commits(500);
        payload.graph_rows = gitkraft_core::features::graph::build_graph(&payload.commits);
        let payload_len = payload.commits.len();
        assert!(
            payload_len < extended_len,
            "payload must be shorter for this test"
        );

        let _ = state.update(Message::RepoRefreshed(Ok(payload)));

        assert_eq!(
            state.active_tab().commits.len(),
            extended_len,
            "refresh must not shrink the commit list"
        );
    }

    #[test]
    fn refresh_replaces_commits_when_count_equal_or_greater() {
        // When a refresh loads the same or more commits, they should replace
        // the existing list (to pick up new commits from pushes/rebases).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.active_tab_mut().commits = make_test_commits(5);

        let original_oid = state.active_tab().commits[0].oid.clone();

        // Refresh with same count but different content
        let mut payload = fake_payload("/home/user/repo-a");
        let mut new_commits = make_test_commits(5);
        new_commits[0].oid = "refreshed_oid_0000".to_string();
        payload.commits = new_commits;
        payload.graph_rows = gitkraft_core::features::graph::build_graph(&payload.commits);
        let _ = state.update(Message::RepoRefreshed(Ok(payload)));

        assert_eq!(
            state.active_tab().commits[0].oid,
            "refreshed_oid_0000",
            "refresh with same count must replace commits"
        );
        assert_ne!(state.active_tab().commits[0].oid, original_oid);
    }

    #[test]
    fn refresh_rebuilds_commit_display_when_content_changes_at_same_count() {
        // When a refresh replaces commits with different content at the same
        // count (e.g. after amend/rebase), commit_display must be rebuilt.
        // Previously only a count change triggered the rebuild, leaving stale
        // author names and timestamps in the display.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        let mut initial_commits = make_test_commits(5);
        initial_commits[0].author_name = "Alice".to_string();
        state.active_tab_mut().commits = initial_commits;
        // Simulate a prior load that built commit_display.
        state.active_tab_mut().commit_display = state
            .active_tab()
            .commits
            .iter()
            .map(|c| ("1 day ago".to_string(), c.author_name.clone()))
            .collect();
        assert_eq!(state.active_tab().commit_display[0].1, "Alice");

        // Refresh with same count but different author on the first commit
        let mut payload = fake_payload("/home/user/repo-a");
        let mut new_commits = make_test_commits(5);
        new_commits[0].author_name = "Bob".to_string();
        payload.commits = new_commits;
        payload.graph_rows = gitkraft_core::features::graph::build_graph(&payload.commits);
        let _ = state.update(Message::RepoRefreshed(Ok(payload)));

        // commit_display must reflect the new author, not the stale one.
        assert_eq!(
            state.active_tab().commit_display[0].1,
            "Bob",
            "commit_display must be rebuilt when content changes at same count"
        );
    }

    #[test]
    fn duplicate_more_commits_loaded_sets_has_more_false() {
        // Documents the scenario that caused "end of history" too early:
        // two identical MoreCommitsLoaded results arriving back-to-back.
        // The second sees prev_count == new_total and sets has_more = false.
        // This test verifies the SYMPTOM; the fix is that apply_payload no
        // longer creates the duplicate by resetting is_loading_more_commits.
        use crate::message::{CommitPage, Message};
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        let tab = state.active_tab_mut();
        tab.commits = make_test_commits(10);
        tab.has_more_commits = true;

        // First result: grows from 10 to 15
        let _initial_count = tab.commits.len();
        let mut bigger_commits = tab.commits.clone();
        for i in 0..5 {
            let mut c = bigger_commits[0].clone();
            c.oid = format!("new_{i:04}");
            bigger_commits.push(c);
        }
        let bigger_count = bigger_commits.len();
        let graph_rows = gitkraft_core::features::graph::build_graph(&bigger_commits);
        let _ = state.update(Message::MoreCommitsLoaded(Ok(CommitPage {
            commits: bigger_commits.clone(),
            graph_rows: graph_rows.clone(),
        })));
        assert!(
            state.active_tab().has_more_commits,
            "first result should keep has_more true"
        );
        assert_eq!(state.active_tab().commits.len(), bigger_count);

        // Second identical result: prev_count == new_total → has_more = false
        let _ = state.update(Message::MoreCommitsLoaded(Ok(CommitPage {
            commits: bigger_commits,
            graph_rows,
        })));
        assert!(
            !state.active_tab().has_more_commits,
            "identical second result must set has_more_commits false"
        );
    }

    #[test]
    fn file_system_changed_allowed_during_pagination() {
        // FileSystemChanged is NOT blocked by is_loading_more_commits.
        // The "don't shrink" guard in apply_payload handles the race.
        // Blocking here would permanently freeze auto-refresh if
        // is_loading_more_commits got stuck (e.g. tab switch mid-pagination).
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.active_tab_mut().is_loading_more_commits = true;

        let _task = state.update(Message::FileSystemChanged);

        // is_loading_more_commits stays true (owned by pagination flow)
        assert!(state.active_tab().is_loading_more_commits);
        // The refresh IS allowed — we can't inspect the Task directly,
        // but at minimum the state should not have been corrupted.
    }

    #[test]
    fn has_more_commits_preserved_across_refresh() {
        // A background refresh must not reset has_more_commits.
        use crate::message::Message;
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        state.active_tab_mut().has_more_commits = true;

        let payload = fake_payload("/home/user/repo-a");
        let _ = state.update(Message::RepoRefreshed(Ok(payload)));

        assert!(
            state.active_tab().has_more_commits,
            "refresh must preserve has_more_commits"
        );
    }

    #[test]
    fn switch_to_existing_tab_removes_empty_tab() {
        let mut state = GitKraft::new();
        // Tab 0: loaded repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: empty tab (user clicked "+")
        state.tabs.push(RepoTab::new_empty());
        state.active_tab = 1;

        let found = state.switch_to_existing_tab(std::path::Path::new("/home/user/repo-a"));

        assert!(found, "should find existing tab");
        assert_eq!(state.tabs.len(), 1, "empty tab should be removed");
        assert_eq!(state.active_tab, 0);
        assert_eq!(
            state.active_tab().repo_path.as_deref(),
            Some(std::path::Path::new("/home/user/repo-a")),
        );
    }

    #[test]
    fn switch_to_existing_tab_returns_false_when_not_found() {
        let mut state = GitKraft::new();
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");

        let found = state.switch_to_existing_tab(std::path::Path::new("/home/user/repo-b"));

        assert!(!found, "should not find nonexistent tab");
        assert_eq!(state.tabs.len(), 1);
    }

    #[test]
    fn switch_to_existing_tab_keeps_loaded_current_tab() {
        let mut state = GitKraft::new();
        // Tab 0: repo-a
        setup_loaded_tab(state.active_tab_mut(), "/home/user/repo-a");
        // Tab 1: repo-b (active)
        state.tabs.push(RepoTab::new_empty());
        setup_loaded_tab(&mut state.tabs[1], "/home/user/repo-b");
        state.active_tab = 1;

        let found = state.switch_to_existing_tab(std::path::Path::new("/home/user/repo-a"));

        assert!(found);
        assert_eq!(state.tabs.len(), 2, "loaded tab should not be removed");
        assert_eq!(state.active_tab, 0);
    }

    #[test]
    fn default_remote_returns_first_remote() {
        let mut tab = RepoTab::new_empty();
        tab.remotes = vec![
            gitkraft_core::RemoteInfo {
                name: "upstream".to_string(),
                url: Some("https://example.com/upstream.git".to_string()),
                fetch_refspecs: vec![],
            },
            gitkraft_core::RemoteInfo {
                name: "origin".to_string(),
                url: Some("https://example.com/origin.git".to_string()),
                fetch_refspecs: vec![],
            },
        ];
        assert_eq!(tab.default_remote(), "upstream");
    }

    #[test]
    fn default_remote_fallback_to_origin() {
        let tab = RepoTab::new_empty();
        assert_eq!(tab.default_remote(), "origin");
    }

    #[test]
    fn hovered_commit_defaults_to_none() {
        let tab = RepoTab::new_empty();
        assert_eq!(tab.hovered_commit, None);
    }

    #[test]
    fn hover_commit_message_sets_state() {
        use crate::message::Message;
        let mut state = GitKraft::new();
        let _task = state.update(Message::HoverCommit(Some(5)));
        assert_eq!(state.active_tab().hovered_commit, Some(5));
        let _task = state.update(Message::HoverCommit(None));
        assert_eq!(state.active_tab().hovered_commit, None);
    }

    #[test]
    fn branch_context_defaults_empty() {
        let tab = RepoTab::new_empty();
        assert!(tab.branch_context.is_empty());
    }

    #[test]
    fn ghost_shows_on_selected_commit_without_refs() {
        // When a commit is selected, the ghost branch label should show.
        // This verifies the state fields are set correctly for the view
        // to detect show_ghost = is_hovered || is_selected.
        let mut state = GitKraft::new();
        let tab = state.active_tab_mut();
        tab.repo_path = Some(std::path::PathBuf::from("/tmp/test"));
        tab.commits = vec![
            gitkraft_core::CommitInfo {
                oid: "aaa".into(),
                summary: "tip".into(),
                message: String::new(),
                author_name: "A".into(),
                author_email: String::new(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: vec![gitkraft_core::RefLabel {
                    name: "main".to_string(),
                    kind: gitkraft_core::RefKind::Head,
                }],
            },
            gitkraft_core::CommitInfo {
                oid: "bbb".into(),
                summary: "child".into(),
                message: String::new(),
                author_name: "A".into(),
                author_email: String::new(),
                time: Default::default(),
                parent_ids: Vec::new(),
                refs: Vec::new(),
            },
        ];
        tab.graph_rows = vec![
            gitkraft_core::GraphRow {
                width: 1,
                node_column: 0,
                node_color: 0,
                edges: Vec::new(),
            },
            gitkraft_core::GraphRow {
                width: 1,
                node_column: 0,
                node_color: 0,
                edges: Vec::new(),
            },
        ];
        tab.branch_context = vec![None, Some("main".to_string())];
        tab.selected_commit = Some(1);

        // Verify the state is correct for the view to show ghost.
        let tab = state.active_tab();
        assert_eq!(tab.selected_commit, Some(1));
        assert_eq!(tab.branch_context[1], Some("main".to_string()));
        // Commits with refs don't get a ghost.
        assert_eq!(tab.branch_context[0], None);
    }

    #[test]
    fn theme_choice_system_follows_and_preset_stops() {
        let mut state = GitKraft::new();
        assert!(state.apply_theme_choice(gitkraft_core::THEME_COUNT));
        assert!(state.follow_system);
        assert_eq!(state.persisted_theme_name(), "System");
        assert!(state.current_theme_index < gitkraft_core::THEME_COUNT);
        let nord = gitkraft_core::theme_index_by_name("Nord");
        assert!(state.apply_theme_choice(nord));
        assert!(!state.follow_system);
        assert_eq!(state.persisted_theme_name(), "Nord");
        assert!(!state.apply_theme_choice(gitkraft_core::THEME_COUNT + 5));
        assert_eq!(state.current_theme_index, nord);
    }

    #[test]
    fn refresh_system_theme_only_when_following() {
        let mut state = GitKraft::new();
        state.follow_system = false;
        state.current_theme_index = 9;
        state.refresh_system_theme();
        assert_eq!(state.current_theme_index, 9);
        state.follow_system = true;
        state.refresh_system_theme();
        assert_eq!(
            state.current_theme_index,
            gitkraft_core::current_system_theme_index()
        );
    }
}