wsx_core/

ops.rs

//! Git/project operations projected against the wsx-owned runtime.
// ^ [[wsx Architecture]] Git owns worktree discovery; the daemon owns sessions and panes.

use crate::{
    config::global::GlobalConfig,
    git::{info as git_info, worktree as git_worktree},
    hooks,
    model::workspace::{
        FetchFailReason, GitInfo, PaneInfo, Project, ProjectConfig, SessionInfo, WorkspaceState,
        WorktreeInfo, WorktreeInitialSession,
    },
    runtime::{
        AgentState, Client, ProjectSpec, Request, Response, SessionId, SessionPlacement, Snapshot,
        WorktreeId, WorktreeSpec,
    },
};
use anyhow::{anyhow, bail, Result};
use std::{
    collections::{HashMap, HashSet},
    path::{Path, PathBuf},
};

struct WorktreeState {
    git_info: Option<GitInfo>,
    git_info_fetched_at: Option<std::time::Instant>,
    expanded: bool,
    sessions: Vec<SessionInfo>,
    last_fetched: Option<std::time::Instant>,
    fetch_failed: bool,
    fetch_fail_count: u32,
    fetch_fail_reason: Option<FetchFailReason>,
}

#[derive(Debug, Clone)]
struct DiscoveredProject {
    name: String,
    path: PathBuf,
    worktrees: Vec<git_worktree::WorktreeEntry>,
}

#[derive(Debug, Clone)]
pub struct WorkspaceDiscovery {
    projects: Vec<DiscoveredProject>,
}

impl WorkspaceDiscovery {
    pub fn into_worktrees(self) -> Vec<(PathBuf, Vec<git_worktree::WorktreeEntry>)> {
        self.projects
            .into_iter()
            .map(|project| (project.path, project.worktrees))
            .collect()
    }
}

fn is_git_repo(path: &Path) -> bool {
    path.exists() && path.join(".git").exists()
}

pub fn runtime_snapshot() -> Result<Snapshot> {
    match Client::local().call(&Request::Snapshot)? {
        Response::Snapshot(snapshot) => Ok(snapshot),
        Response::Error(error) => bail!("{}: {}", error.code, error.message),
        _ => bail!("wsx daemon returned an unexpected snapshot response"),
    }
}

fn synchronize(client: &Client, projects: Vec<ProjectSpec>) -> Result<Snapshot> {
    match client.call(&Request::SynchronizeProjects { projects })? {
        Response::Ack { .. } => {}
        Response::Error(error) => bail!("{}: {}", error.code, error.message),
        _ => bail!("wsx daemon returned an unexpected synchronization response"),
    }
    match client.call(&Request::Snapshot)? {
        Response::Snapshot(snapshot) => Ok(snapshot),
        Response::Error(error) => bail!("{}: {}", error.code, error.message),
        _ => bail!("wsx daemon returned an unexpected snapshot response"),
    }
}

pub fn workspace_from_config(config: &GlobalConfig) -> WorkspaceState {
    WorkspaceState {
        projects: config
            .projects
            .iter()
            .filter(|entry| is_git_repo(&entry.path))
            .map(|entry| Project {
                name: entry.name.clone(),
                path: entry.path.clone(),
                default_branch: "main".into(),
                last_agent_active_unix_ms: None,
                last_terminal_active_unix_ms: None,
                worktrees: Vec::new(),
                routines: Vec::new(),
                routine_revision: 0,
                routines_expanded: true,
                config: Some(crate::config::project::load_project_config(&entry.path)),
                expanded: true,
                missing: false,
            })
            .collect(),
    }
}

pub fn discover_workspace(config: &GlobalConfig) -> Result<WorkspaceDiscovery> {
    discover_workspace_with(config, git_worktree::list_worktrees)
}

fn discover_workspace_with<F>(
    config: &GlobalConfig,
    mut list_worktrees: F,
) -> Result<WorkspaceDiscovery>
where
    F: FnMut(&Path) -> Result<Vec<git_worktree::WorktreeEntry>>,
{
    let projects = config
        .projects
        .iter()
        .filter(|entry| is_git_repo(&entry.path))
        .map(|entry| {
            let worktrees = list_worktrees(&entry.path)?;
            Ok(DiscoveredProject {
                name: entry.name.clone(),
                path: entry.path.clone(),
                worktrees,
            })
        })
        .collect::<Result<Vec<_>>>()?;
    Ok(WorkspaceDiscovery { projects })
}

pub fn synchronize_discovery(discovery: &WorkspaceDiscovery) -> Result<Snapshot> {
    let projects = discovery
        .projects
        .iter()
        .map(|project| ProjectSpec {
            path: project.path.clone(),
            name: project.name.clone(),
            worktrees: project
                .worktrees
                .iter()
                .map(|worktree| WorktreeSpec {
                    path: worktree.path.clone(),
                    branch: worktree.branch.clone(),
                })
                .collect(),
        })
        .collect();
    synchronize(&Client::local(), projects)
}

fn apply_discovery(
    workspace: &mut WorkspaceState,
    config: &GlobalConfig,
    snapshot: &Snapshot,
    discovery: WorkspaceDiscovery,
) -> Result<()> {
    let worktrees = discovery
        .projects
        .into_iter()
        .map(|project| (project.path, project.worktrees))
        .collect();
    refresh_workspace_with_worktrees(workspace, config, snapshot, worktrees)
}

pub fn load_full_workspace(config: &GlobalConfig) -> Result<WorkspaceState> {
    let discovery = discover_workspace(config)?;
    let snapshot = synchronize_discovery(&discovery)?;
    let mut workspace = workspace_from_config(config);
    apply_discovery(&mut workspace, config, &snapshot, discovery)?;
    Ok(workspace)
}

pub fn refresh_workspace_discovery_only(
    workspace: &mut WorkspaceState,
    config: &GlobalConfig,
    worktrees: Vec<(PathBuf, Vec<git_worktree::WorktreeEntry>)>,
) -> Result<()> {
    refresh_workspace_with_discovery(workspace, config, None, worktrees)
}

pub fn refresh_workspace_with_worktrees(
    workspace: &mut WorkspaceState,
    config: &GlobalConfig,
    snapshot: &Snapshot,
    worktrees: Vec<(PathBuf, Vec<git_worktree::WorktreeEntry>)>,
) -> Result<()> {
    refresh_workspace_with_discovery(workspace, config, Some(snapshot), worktrees)
}

fn refresh_workspace_with_discovery(
    workspace: &mut WorkspaceState,
    config: &GlobalConfig,
    snapshot: Option<&Snapshot>,
    worktrees: Vec<(PathBuf, Vec<git_worktree::WorktreeEntry>)>,
) -> Result<()> {
    let mut worktrees_map: HashMap<PathBuf, Vec<git_worktree::WorktreeEntry>> =
        worktrees.into_iter().collect();
    if let Some(snapshot) = snapshot {
        update_project_activity(workspace, snapshot);
    }
    for project in &mut workspace.projects {
        if let Some(default_branch) = worktrees_map
            .get(&project.path)
            .and_then(|entries| entries.iter().find(|entry| entry.is_main))
            .filter(|entry| entry.branch != "HEAD")
            .map(|entry| entry.branch.clone())
        {
            project.default_branch = default_branch;
        }
        let previous: HashMap<PathBuf, WorktreeState> = project
            .worktrees
            .iter()
            .map(|worktree| {
                (
                    worktree.path.clone(),
                    WorktreeState {
                        git_info: worktree.git_info.clone(),
                        git_info_fetched_at: worktree.git_info_fetched_at,
                        expanded: worktree.expanded,
                        sessions: worktree.sessions.clone(),
                        last_fetched: worktree.last_fetched,
                        fetch_failed: worktree.fetch_failed,
                        fetch_fail_count: worktree.fetch_fail_count,
                        fetch_fail_reason: worktree.fetch_fail_reason.clone(),
                    },
                )
            })
            .collect();
        let aliases = config
            .projects
            .iter()
            .find(|entry| entry.path == project.path)
            .map(|entry| &entry.aliases);
        let entries = worktrees_map.remove(&project.path).unwrap_or_default();
        project.worktrees = entries
            .into_iter()
            .filter(|entry| !config.is_worktree_excluded(&entry.path))
            .map(|entry| {
                let old = previous.get(&entry.path);
                Ok(WorktreeInfo {
                    name: entry.name,
                    branch: entry.branch.clone(),
                    path: entry.path.clone(),
                    is_main: entry.is_main,
                    alias: aliases.and_then(|map| map.get(&entry.branch)).cloned(),
                    sessions: match snapshot {
                        Some(snapshot) => sessions_for_worktree(
                            snapshot,
                            &entry.path,
                            old.map(|state| state.sessions.as_slice())
                                .unwrap_or_default(),
                        )?,
                        None => old.map(|state| state.sessions.clone()).unwrap_or_default(),
                    },
                    expanded: old.map(|state| state.expanded).unwrap_or(true),
                    git_info: old.and_then(|state| state.git_info.clone()),
                    fetch_failed: old.map(|state| state.fetch_failed).unwrap_or(false),
                    fetch_fail_count: old.map(|state| state.fetch_fail_count).unwrap_or(0),
                    fetch_fail_reason: old.and_then(|state| state.fetch_fail_reason.clone()),
                    last_fetched: old.and_then(|state| state.last_fetched),
                    git_info_fetched_at: old.and_then(|state| state.git_info_fetched_at),
                })
            })
            .collect::<Result<Vec<_>>>()?;
    }
    workspace.projects.retain(|project| !project.missing);
    for project in &mut workspace.projects {
        project.missing = !is_git_repo(&project.path);
    }
    Ok(())
}

pub fn refresh_sessions_from_snapshot(
    workspace: &mut WorkspaceState,
    snapshot: &Snapshot,
) -> Result<()> {
    update_project_activity(workspace, snapshot);
    for worktree in workspace
        .projects
        .iter_mut()
        .flat_map(|project| &mut project.worktrees)
    {
        worktree.sessions = sessions_for_worktree(snapshot, &worktree.path, &worktree.sessions)?;
    }
    Ok(())
}

fn update_project_activity(workspace: &mut WorkspaceState, snapshot: &Snapshot) {
    for project in &mut workspace.projects {
        let runtime_project = snapshot
            .projects
            .iter()
            .find(|candidate| candidate.path == project.path)
            .or_else(|| {
                let project_id = snapshot.worktrees.iter().find_map(|runtime_worktree| {
                    project
                        .worktrees
                        .iter()
                        .any(|worktree| worktree.path == runtime_worktree.path)
                        .then_some(runtime_worktree.project_id)
                })?;
                snapshot
                    .projects
                    .iter()
                    .find(|candidate| candidate.id == project_id)
            });
        if let Some(runtime_project) = runtime_project {
            project.last_agent_active_unix_ms = runtime_project.last_agent_active_unix_ms;
            project.last_terminal_active_unix_ms = runtime_project.last_terminal_active_unix_ms;
        }
    }
}

fn sessions_for_worktree(
    snapshot: &Snapshot,
    path: &Path,
    previous: &[SessionInfo],
) -> Result<Vec<SessionInfo>> {
    let Some(worktree) = snapshot
        .worktrees
        .iter()
        .find(|worktree| worktree.path == path)
    else {
        return Ok(Vec::new());
    };
    let previous = previous
        .iter()
        .map(|session| (session.session_id, session))
        .collect::<HashMap<_, _>>();
    let listening_ports = snapshot
        .listening_ports
        .iter()
        .map(|ports| (ports.pane_id, ports.tcp.as_slice()))
        .collect::<HashMap<_, _>>();
    let foreground_jobs = snapshot
        .pane_activity
        .iter()
        .filter_map(|activity| activity.foreground_job.then_some(activity.pane_id))
        .collect::<HashSet<_>>();
    snapshot
        .sessions
        .iter()
        .filter(|session| session.worktree_id == worktree.id)
        .map(|session| {
            let focused = snapshot
                .panes
                .iter()
                .find(|pane| pane.id == session.focused_pane)
                .ok_or_else(|| anyhow!("session {} has no focused pane", session.id))?;
            let focused_agent = focused.agent.as_ref().filter(|agent| agent.attached);
            let old = previous.get(&session.id).copied();
            let panes = session
                .panes
                .iter()
                .map(|pane_id| {
                    let pane = snapshot
                        .panes
                        .iter()
                        .find(|pane| pane.id == *pane_id)
                        .ok_or_else(|| {
                            anyhow!("session {} references missing pane {}", session.id, pane_id)
                        })?;
                    let attached_agent = pane.agent.as_ref().filter(|agent| agent.attached);
                    Ok(PaneInfo {
                        pane_id: pane.id,
                        terminal_id: pane.terminal_id,
                        label: pane.label.clone(),
                        agent: attached_agent.map(|agent| agent.provider.clone()),
                        agent_status: attached_agent
                            .map_or(AgentState::Unknown, |agent| agent.state),
                        revision: pane.revision,
                        exited: pane.exited,
                        listening_ports: listening_ports
                            .get(&pane.id)
                            .copied()
                            .unwrap_or_default()
                            .to_vec(),
                        foreground_job: foreground_jobs.contains(&pane.id),
                        outcome_acknowledged: old
                            .and_then(|session| {
                                session
                                    .panes
                                    .iter()
                                    .find(|previous| previous.terminal_id == pane.terminal_id)
                            })
                            .is_some_and(|previous| {
                                previous.revision == pane.revision && previous.outcome_acknowledged
                            }),
                    })
                })
                .collect::<Result<Vec<_>>>()?;
            let revision = session.revision;
            let outcome_acknowledged = panes
                .iter()
                .find(|pane| pane.pane_id == focused.id)
                .is_some_and(|pane| pane.outcome_acknowledged);
            Ok(SessionInfo {
                session_id: session.id,
                pane_id: focused.id,
                terminal_id: focused.terminal_id,
                agent: focused_agent.map(|agent| agent.provider.clone()),
                display_name: session.label.clone(),
                agent_status: focused_agent.map_or(AgentState::Unknown, |agent| agent.state),
                revision,
                layout: session.layout.clone(),
                panes,
                muted: old.is_some_and(|session| session.muted),
                outcome_acknowledged,
            })
        })
        .collect()
}

pub fn expand_path(value: &str) -> PathBuf {
    value
        .strip_prefix("~/")
        .and_then(|tail| dirs::home_dir().map(|home| home.join(tail)))
        .unwrap_or_else(|| PathBuf::from(value))
}
pub fn detect_default_branch(path: &Path) -> String {
    git_info::current_branch(path).unwrap_or_else(|| "main".into())
}

pub fn register_project(path: PathBuf, config: &mut GlobalConfig) -> Result<Project> {
    if path.as_os_str().is_empty() {
        bail!("empty path");
    }
    let path = crate::config::global::normalize_project_path(&path);
    if !path.exists() {
        bail!("path does not exist: {}", path.display());
    }
    if !is_git_repo(&path) {
        bail!("not a git repository: {}", path.display());
    }
    if config.projects.iter().any(|entry| entry.path == path) {
        bail!("project already registered: {}", path.display());
    }
    let name = path
        .file_name()
        .map(|name| name.to_string_lossy().into_owned())
        .unwrap_or_else(|| "unknown".into());
    let project = Project {
        name: name.clone(),
        path: path.clone(),
        default_branch: detect_default_branch(&path),
        last_agent_active_unix_ms: None,
        last_terminal_active_unix_ms: None,
        worktrees: git_worktree::to_worktree_infos(
            git_worktree::list_worktrees(&path).unwrap_or_default(),
            &HashMap::new(),
        ),
        routines: Vec::new(),
        routine_revision: 0,
        routines_expanded: true,
        config: Some(crate::config::project::load_project_config(&path)),
        expanded: true,
        missing: false,
    };
    config.add_project(name, path);
    Ok(project)
}
pub fn unregister_project(path: &PathBuf, config: &mut GlobalConfig) {
    config.remove_project(path);
}

pub fn create_worktree(
    repo_path: &Path,
    default_branch: &str,
    project_config: &ProjectConfig,
    branch: &str,
) -> Result<(PathBuf, Option<String>)> {
    let path = git_worktree::create_worktree(repo_path, branch, default_branch)?;
    let mut warning = hooks::copy_env_files(repo_path, &path, project_config)
        .err()
        .map(|error| format!("Warning: .env copy: {error}"));
    if let Some(command) = &project_config.post_create {
        if let Err(error) = hooks::run_post_create(&path, command) {
            warning = Some(format!("Warning: postCreate: {error}"));
        }
    }
    Ok((path, warning))
}

#[derive(Debug)]
pub struct CreatedWorktree {
    pub path: PathBuf,
    pub warning: Option<String>,
    pub session: Option<(SessionId, String)>,
    pub session_error: Option<String>,
}

pub fn create_configured_worktree(
    config: &GlobalConfig,
    project_name: &str,
    repo_path: &Path,
    default_branch: &str,
    project_config: &ProjectConfig,
    branch: &str,
    fallback_session: WorktreeInitialSession,
) -> Result<CreatedWorktree> {
    let (path, warning) = create_worktree(repo_path, default_branch, project_config, branch)?;
    let session_result = match project_config.initial_session(fallback_session) {
        WorktreeInitialSession::Disabled => None,
        WorktreeInitialSession::Shell => {
            Some(create_session(config, project_name, "", &path, None, None))
        }
        WorktreeInitialSession::Command(command) => Some(create_session(
            config,
            project_name,
            "",
            &path,
            None,
            Some(command),
        )),
    };
    let (session, session_error) = match session_result {
        Some(Ok(session)) => (Some(session), None),
        Some(Err(error)) => (None, Some(error.to_string())),
        None => (None, None),
    };
    Ok(CreatedWorktree {
        path,
        warning,
        session,
        session_error,
    })
}

pub fn delete_worktree(repo_path: &Path, wt_path: &Path, branch: &str) -> Result<()> {
    let client = Client::local();
    let snapshot = runtime_snapshot()?;
    if let Some(worktree) = snapshot
        .worktrees
        .iter()
        .find(|worktree| worktree.path == wt_path)
    {
        for session in snapshot
            .sessions
            .iter()
            .filter(|session| session.worktree_id == worktree.id)
        {
            expect_ack(client.call(&Request::SessionClose {
                session_id: session.id,
                expected_revision: session.revision,
            })?)?;
        }
    }
    git_worktree::remove_worktree(repo_path, wt_path, branch)
}
fn snapshot_with_worktree<F>(
    snapshot: Snapshot,
    worktree_path: &Path,
    synchronize_once: F,
) -> Result<(Snapshot, WorktreeId)>
where
    F: FnOnce() -> Result<Snapshot>,
{
    if let Some(id) = snapshot
        .worktrees
        .iter()
        .find(|worktree| worktree.path == worktree_path)
        .map(|worktree| worktree.id)
    {
        return Ok((snapshot, id));
    }
    let refreshed = synchronize_once()?;
    let id = refreshed
        .worktrees
        .iter()
        .find(|worktree| worktree.path == worktree_path)
        .map(|worktree| worktree.id)
        .ok_or_else(|| anyhow!("worktree is not synchronized with wsx daemon"))?;
    Ok((refreshed, id))
}

pub fn create_session(
    config: &GlobalConfig,
    project_name: &str,
    _worktree_slug: &str,
    worktree_path: &Path,
    session_label: Option<String>,
    command: Option<String>,
) -> Result<(SessionId, String)> {
    let client = Client::local();
    let (snapshot, worktree_id) =
        snapshot_with_worktree(runtime_snapshot()?, worktree_path, || {
            let discovery = discover_workspace(config)?;
            synchronize_discovery(&discovery)
        })?;
    let base = session_label
        .filter(|label| !label.trim().is_empty())
        .or_else(|| {
            command
                .as_ref()
                .and_then(|command| command.split_whitespace().next().map(str::to_owned))
        })
        .unwrap_or_else(|| project_name.to_owned());
    let used = snapshot
        .sessions
        .iter()
        .filter(|session| session.worktree_id == worktree_id)
        .map(|session| session.label.as_str())
        .collect::<std::collections::HashSet<_>>();
    let mut label = base.clone();
    let mut suffix = 2;
    while used.contains(label.as_str()) {
        label = format!("{base}-{suffix}");
        suffix += 1;
    }
    let response = client.call(&Request::SessionCreate {
        worktree_id,
        label: label.clone(),
        command: Vec::new(),
        initial_input: command,
        rows: 24,
        cols: 80,
    })?;
    let session_id = match response {
        Response::Created { id, .. } => SessionId(id),
        Response::Error(error) => bail!("{}: {}", error.code, error.message),
        _ => bail!("wsx daemon returned an unexpected create response"),
    };
    Ok((session_id, label))
}

pub fn reorder_session(
    session_id: SessionId,
    target_session_id: SessionId,
    placement: SessionPlacement,
    expected_revision: u64,
) -> Result<u64> {
    expect_ack_revision(Client::local().call(&Request::SessionReorder {
        session_id,
        target_session_id,
        placement,
        expected_revision,
    })?)
}

pub fn rename_session(session_id: SessionId, new_label: &str) -> Result<()> {
    let snapshot = runtime_snapshot()?;
    let session = snapshot
        .sessions
        .iter()
        .find(|session| session.id == session_id)
        .ok_or_else(|| anyhow!("session not found"))?;
    expect_ack(Client::local().call(&Request::SessionRename {
        session_id,
        label: new_label.into(),
        expected_revision: session.revision,
    })?)
}
pub fn kill_session(session_id: SessionId) -> Result<()> {
    let snapshot = runtime_snapshot()?;
    let session = snapshot
        .sessions
        .iter()
        .find(|session| session.id == session_id)
        .ok_or_else(|| anyhow!("session not found"))?;
    expect_ack(Client::local().call(&Request::SessionClose {
        session_id,
        expected_revision: session.revision,
    })?)
}

fn expect_ack(response: Response) -> Result<()> {
    expect_ack_revision(response).map(|_| ())
}

fn expect_ack_revision(response: Response) -> Result<u64> {
    match response {
        Response::Ack { revision } => Ok(revision),
        Response::Error(error) => bail!("{}: {}", error.code, error.message),
        _ => bail!("wsx daemon returned an unexpected mutation response"),
    }
}
pub fn set_alias(config: &mut GlobalConfig, project_path: &PathBuf, branch: &str, alias: &str) {
    config.set_alias(project_path, branch, alias);
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::runtime::{
        self, AgentCapabilities, AgentInfo, AgentInstanceId, Capabilities, Pane, PaneId,
        PaneLayout, Project as RuntimeProject, ProjectId, Session, TerminalId, Worktree,
        WorktreeId,
    };

    #[test]
    fn projection_lists_sessions_directly_under_their_worktree() {
        let snapshot = Snapshot {
            protocol: runtime::PROTOCOL_VERSION,
            epoch: 1,
            revision: 4,
            projects: vec![RuntimeProject {
                id: ProjectId(1),
                path: "/repo".into(),
                name: "repo".into(),
                revision: 1,
                last_agent_active_unix_ms: Some(42),
                last_terminal_active_unix_ms: Some(43),
            }],
            worktrees: vec![Worktree {
                id: WorktreeId(2),
                project_id: ProjectId(1),
                path: "/repo".into(),
                branch: "main".into(),
                revision: 1,
            }],
            sessions: vec![Session {
                id: SessionId(3),
                worktree_id: WorktreeId(2),
                label: "shell".into(),
                primary_pane: PaneId(4),
                focused_pane: PaneId(6),
                panes: vec![PaneId(4), PaneId(6)],
                layout: PaneLayout::Split {
                    axis: runtime::SplitAxis::Vertical,
                    ratio_millis: 500,
                    first: Box::new(PaneLayout::Leaf { pane_id: PaneId(4) }),
                    second: Box::new(PaneLayout::Leaf { pane_id: PaneId(6) }),
                },
                revision: 4,
            }],
            panes: vec![
                Pane {
                    id: PaneId(4),
                    terminal_id: TerminalId(5),
                    session_id: SessionId(3),
                    label: "primary".into(),
                    agent: Some(AgentInfo {
                        id: AgentInstanceId(8),
                        provider: "pi".into(),
                        state: AgentState::Working,
                        attached: true,
                        conversation_id: None,
                        session_ref: None,
                        capabilities: AgentCapabilities::default(),
                        source: "test".into(),
                    }),
                    exited: false,
                    revision: 4,
                },
                Pane {
                    id: PaneId(6),
                    terminal_id: TerminalId(7),
                    session_id: SessionId(3),
                    label: "split".into(),
                    agent: Some(AgentInfo {
                        id: AgentInstanceId(10),
                        provider: "pi".into(),
                        state: AgentState::Done,
                        attached: false,
                        conversation_id: Some("resumable".into()),
                        session_ref: None,
                        capabilities: AgentCapabilities::default(),
                        source: "test".into(),
                    }),
                    exited: false,
                    revision: 9,
                },
            ],
            listening_ports: vec![
                runtime::PanePorts {
                    pane_id: PaneId(4),
                    tcp: vec![5173],
                },
                runtime::PanePorts {
                    pane_id: PaneId(6),
                    tcp: vec![3000, 5173],
                },
            ],
            pane_activity: vec![runtime::PaneActivity {
                pane_id: PaneId(6),
                foreground_job: true,
            }],
            plugin_sidecars: Vec::new(),
            capabilities: Capabilities::default(),
        };
        let sessions = sessions_for_worktree(&snapshot, Path::new("/repo"), &[]).unwrap();
        assert_eq!(sessions.len(), 1);
        assert_eq!(sessions[0].session_id, SessionId(3));
        assert_eq!(sessions[0].display_name, "shell");
        assert_eq!(sessions[0].pane_id, PaneId(6));
        assert_eq!(sessions[0].revision, 4);
        assert_eq!(sessions[0].panes[1].revision, 9);
        assert_eq!(sessions[0].panes.len(), 2);
        assert_eq!(sessions[0].panes[0].label, "primary");
        assert_eq!(sessions[0].panes[0].agent.as_deref(), Some("pi"));
        assert_eq!(sessions[0].panes[0].agent_status, AgentState::Working);
        assert_eq!(sessions[0].panes[1].label, "split");
        assert_eq!(sessions[0].agent, None);
        assert_eq!(sessions[0].agent_status, AgentState::Unknown);
        assert_eq!(sessions[0].panes[1].agent, None);
        assert_eq!(sessions[0].panes[1].agent_status, AgentState::Unknown);
        assert_eq!(sessions[0].listening_ports(), vec![3000, 5173]);
        assert!(sessions[0].has_foreground_job());

        let mut workspace = WorkspaceState {
            projects: vec![Project {
                name: "repo".into(),
                path: "/repo".into(),
                default_branch: "main".into(),
                last_agent_active_unix_ms: None,
                last_terminal_active_unix_ms: None,
                worktrees: Vec::new(),
                routines: Vec::new(),
                routine_revision: 0,
                routines_expanded: true,
                config: None,
                expanded: true,
                missing: false,
            }],
        };
        refresh_sessions_from_snapshot(&mut workspace, &snapshot).unwrap();
        assert_eq!(workspace.projects[0].last_agent_active_unix_ms, Some(42));
        assert_eq!(workspace.projects[0].last_terminal_active_unix_ms, Some(43));
    }

    #[test]
    fn discovery_only_refresh_keeps_worktrees_and_known_activity_without_daemon_state() {
        let root = Path::new(env!("CARGO_MANIFEST_DIR"))
            .join(".work")
            .join(format!("degraded-{}", std::process::id()));
        let _ = std::fs::remove_dir_all(&root);
        std::fs::create_dir_all(root.join(".git")).unwrap();
        let config = GlobalConfig {
            projects: vec![crate::config::global::ProjectEntry {
                name: "degraded".into(),
                path: root.clone(),
                groups: Vec::new(),
                aliases: HashMap::new(),
            }],
            ..GlobalConfig::default()
        };
        let mut workspace = workspace_from_config(&config);
        workspace.projects[0].last_agent_active_unix_ms = Some(11);
        workspace.projects[0].last_terminal_active_unix_ms = Some(12);
        workspace.projects[0].worktrees.push(WorktreeInfo {
            name: "main".into(),
            branch: "main".into(),
            path: root.clone(),
            is_main: true,
            alias: None,
            sessions: vec![SessionInfo {
                session_id: SessionId(31),
                pane_id: PaneId(32),
                terminal_id: TerminalId(33),
                agent: None,
                display_name: "preserved shell".into(),
                agent_status: AgentState::Unknown,
                revision: 7,
                layout: PaneLayout::Leaf {
                    pane_id: PaneId(32),
                },
                panes: vec![PaneInfo {
                    pane_id: PaneId(32),
                    terminal_id: TerminalId(33),
                    label: "terminal".into(),
                    agent: None,
                    agent_status: AgentState::Unknown,
                    revision: 8,
                    exited: false,
                    listening_ports: vec![4173],
                    foreground_job: true,
                    outcome_acknowledged: false,
                }],
                muted: true,
                outcome_acknowledged: false,
            }],
            expanded: false,
            git_info: None,
            fetch_failed: false,
            fetch_fail_count: 0,
            fetch_fail_reason: None,
            last_fetched: None,
            git_info_fetched_at: None,
        });

        refresh_workspace_discovery_only(
            &mut workspace,
            &config,
            vec![(
                root.clone(),
                vec![git_worktree::WorktreeEntry {
                    name: "main".into(),
                    path: root.clone(),
                    branch: "main".into(),
                    is_main: true,
                }],
            )],
        )
        .unwrap();

        assert_eq!(workspace.projects[0].worktrees.len(), 1);
        assert_eq!(workspace.projects[0].last_agent_active_unix_ms, Some(11));
        assert_eq!(workspace.projects[0].last_terminal_active_unix_ms, Some(12));
        let preserved = &workspace.projects[0].worktrees[0];
        assert!(!preserved.expanded);
        assert_eq!(preserved.sessions.len(), 1);
        assert_eq!(preserved.sessions[0].session_id, SessionId(31));
        assert_eq!(preserved.sessions[0].pane_id, PaneId(32));
        assert_eq!(preserved.sessions[0].terminal_id, TerminalId(33));
        assert_eq!(preserved.sessions[0].display_name, "preserved shell");
        assert_eq!(preserved.sessions[0].panes[0].listening_ports, vec![4173]);
        assert!(preserved.sessions[0].panes[0].foreground_job);
        std::fs::remove_dir_all(root).unwrap();
    }

    #[test]
    fn discovery_lists_each_registered_project_once() {
        let root = Path::new(env!("CARGO_MANIFEST_DIR"))
            .join(".work")
            .join(format!("discovery-{}", std::process::id()));
        let _ = std::fs::remove_dir_all(&root);
        let paths = [root.join("one"), root.join("two")];
        for path in &paths {
            std::fs::create_dir_all(path.join(".git")).unwrap();
        }
        let config = GlobalConfig {
            projects: paths
                .iter()
                .map(|path| crate::config::global::ProjectEntry {
                    name: path.file_name().unwrap().to_string_lossy().into_owned(),
                    path: path.clone(),
                    groups: Vec::new(),
                    aliases: HashMap::new(),
                })
                .collect(),
            ..GlobalConfig::default()
        };
        let shell = workspace_from_config(&config);
        assert_eq!(shell.projects.len(), paths.len());
        assert!(shell
            .projects
            .iter()
            .all(|project| project.worktrees.is_empty()));
        let calls = std::cell::Cell::new(0usize);

        let discovery = discover_workspace_with(&config, |path| {
            calls.set(calls.get() + 1);
            Ok(vec![git_worktree::WorktreeEntry {
                name: "main".into(),
                path: path.to_path_buf(),
                branch: "trunk".into(),
                is_main: true,
            }])
        })
        .unwrap();

        assert_eq!(calls.get(), paths.len());
        assert_eq!(discovery.into_worktrees().len(), paths.len());
        let failed =
            discover_workspace_with(&config, |_| Err(anyhow!("worktree discovery failed")));
        assert!(failed.is_err());
        std::fs::remove_dir_all(root).unwrap();
    }

    #[test]
    fn session_worktree_fast_path_skips_synchronization() {
        let (snapshot, worktree_id) = snapshot_with_worktree(
            snapshot_with_worktree_path("/repo"),
            Path::new("/repo"),
            || panic!("synchronized snapshot must not refresh"),
        )
        .unwrap();

        assert_eq!(snapshot.worktrees[0].path, Path::new("/repo"));
        assert_eq!(worktree_id, WorktreeId(2));
    }

    #[test]
    fn missing_session_worktree_retries_synchronization_once() {
        let calls = std::cell::Cell::new(0usize);
        let (snapshot, worktree_id) = snapshot_with_worktree(
            snapshot_with_worktree_path("/other"),
            Path::new("/repo"),
            || {
                calls.set(calls.get() + 1);
                Ok(snapshot_with_worktree_path("/repo"))
            },
        )
        .unwrap();

        assert_eq!(calls.get(), 1);
        assert_eq!(snapshot.worktrees[0].path, Path::new("/repo"));
        assert_eq!(worktree_id, WorktreeId(2));
    }

    #[test]
    fn missing_session_worktree_still_fails_after_one_synchronization() {
        let calls = std::cell::Cell::new(0usize);
        let error = snapshot_with_worktree(
            snapshot_with_worktree_path("/other"),
            Path::new("/repo"),
            || {
                calls.set(calls.get() + 1);
                Ok(snapshot_with_worktree_path("/still-other"))
            },
        )
        .unwrap_err();

        assert_eq!(calls.get(), 1);
        assert_eq!(
            error.to_string(),
            "worktree is not synchronized with wsx daemon"
        );
    }

    fn snapshot_with_worktree_path(path: &str) -> Snapshot {
        Snapshot {
            protocol: runtime::PROTOCOL_VERSION,
            epoch: 1,
            revision: 1,
            projects: vec![RuntimeProject {
                id: ProjectId(1),
                path: "/repo".into(),
                name: "repo".into(),
                revision: 1,
                last_agent_active_unix_ms: None,
                last_terminal_active_unix_ms: None,
            }],
            worktrees: vec![Worktree {
                id: WorktreeId(2),
                project_id: ProjectId(1),
                path: path.into(),
                branch: "main".into(),
                revision: 1,
            }],
            sessions: Vec::new(),
            panes: Vec::new(),
            listening_ports: Vec::new(),
            pane_activity: Vec::new(),
            plugin_sidecars: Vec::new(),
            capabilities: Capabilities::default(),
        }
    }

    #[test]
    fn register_project_rejects_empty_paths() {
        let mut config = GlobalConfig::default();
        assert!(register_project(PathBuf::new(), &mut config).is_err());
    }
}