workmux 0.1.167

An opinionated workflow tool that orchestrates git worktrees and tmux
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use anyhow::{Result, anyhow};
use tracing::info;

use crate::config;
use crate::multiplexer::{create_backend, detect_backend};
use crate::state::StateStore;
use crate::workflow::resurrect::{ResurrectAction, plan};
use crate::workflow::{self, SetupOptions, WorkflowContext};

pub fn run(dry_run: bool) -> Result<()> {
    let config = config::Config::load(None)?;
    let mux = create_backend(detect_backend());
    let store = StateStore::new()?;

    let plan = plan(&store, mux.as_ref())?;

    if plan.candidates.is_empty() && plan.unmatched_states == 0 {
        println!("No agent state files found. Nothing to restore.");
        return Ok(());
    }

    // Print plan
    let to_restore: Vec<_> = plan
        .candidates
        .iter()
        .filter(|c| matches!(c.action, ResurrectAction::Restore))
        .collect();

    for candidate in &plan.candidates {
        let status = match &candidate.action {
            ResurrectAction::Restore => "restoring",
            ResurrectAction::SkipAlreadyOpen => "skipping (already open)",
            ResurrectAction::SkipMain => "skipping (main worktree)",
        };
        println!("  {:<20} -> {}", candidate.handle, status);
    }

    if plan.unmatched_states > 0 {
        println!(
            "  ({} unmatched state file(s) ignored)",
            plan.unmatched_states
        );
    }

    if to_restore.is_empty() {
        println!("\nNothing to restore.");
        return Ok(());
    }

    if dry_run {
        println!("\nDry run: would restore {} worktree(s)", to_restore.len());
        return Ok(());
    }

    // Execute restoration
    let context = WorkflowContext::new(config, mux, None)?;
    let mut restored = Vec::new();
    let mut failed = Vec::new();

    for candidate in &plan.candidates {
        if !matches!(candidate.action, ResurrectAction::Restore) {
            continue;
        }

        let options = SetupOptions {
            run_hooks: false,
            run_file_ops: false,
            run_pane_commands: true,
            prompt_file_path: None,
            focus_window: false,
            working_dir: None,
            config_root: None,
            open_if_exists: false,
            mode: candidate.mode,
            continue_session: true,
        };

        info!(
            handle = candidate.handle,
            mode = ?candidate.mode,
            stale_keys = candidate.stale_pane_keys.len(),
            "resurrect:exec opening worktree"
        );

        match workflow::open(&candidate.handle, &context, options, false, false, None) {
            Ok(result) => {
                info!(
                    handle = candidate.handle,
                    resolved = result.resolved_handle,
                    branch = result.branch_name,
                    path = %result.worktree_path.display(),
                    "resurrect:exec restored successfully"
                );
                // Clean up stale state files by specific PaneKey
                for key in &candidate.stale_pane_keys {
                    info!(
                        pane_id = %key.pane_id,
                        "resurrect:exec deleting stale state file"
                    );
                    let _ = store.delete_agent(key);
                }
                restored.push(candidate.handle.clone());
            }
            Err(e) => {
                info!(
                    handle = candidate.handle,
                    error = %e,
                    "resurrect:exec failed to restore"
                );
                eprintln!("  Failed to restore '{}': {}", candidate.handle, e);
                failed.push(candidate.handle.clone());
            }
        }
    }

    // Summary
    if !restored.is_empty() {
        println!(
            "\n✓ Restored {} worktree(s): {}",
            restored.len(),
            restored.join(", ")
        );
    }
    if !failed.is_empty() {
        eprintln!(
            "✗ Failed to restore {} worktree(s): {}",
            failed.len(),
            failed.join(", ")
        );
        return Err(anyhow!("Failed to restore {} worktree(s)", failed.len()));
    }

    Ok(())
}