agent-sync 0.3.0

Safely synchronize coding-agent sessions and memories over SSH
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

mod claude;
mod codex;

use crate::core::{OutputFormat, PlanReport, SyncOptions};
use crate::transport::SshTransport;
use anyhow::Result;
use clap::ValueEnum;
use std::fmt;
use std::path::Path;

pub use claude::ClaudePrepared;
pub use codex::CodexPrepared;

#[derive(Clone, Copy, Debug, ValueEnum)]
pub enum AgentKind {
    Codex,
    Claude,
}

impl fmt::Display for AgentKind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(match self {
            Self::Codex => "codex",
            Self::Claude => "claude",
        })
    }
}

pub enum Prepared {
    Claude(ClaudePrepared),
    Codex(CodexPrepared),
}

impl Prepared {
    pub fn report(&self) -> &PlanReport {
        match self {
            Self::Claude(v) => &v.report,
            Self::Codex(v) => &v.report,
        }
    }
    pub fn print(&self, output: OutputFormat, local_root: &Path) -> Result<()> {
        match (self, output) {
            (Self::Claude(value), OutputFormat::Diff) => claude::print_diff(value, local_root),
            (Self::Codex(value), OutputFormat::Diff) => codex::print_diff(value, local_root),
            _ => self.report().print(output),
        }
    }
    pub fn blocked(&self) -> bool {
        !self.report().blockers.is_empty()
    }
}

pub trait Adapter {
    fn doctor(&self, local_root: &Path, remote_root: &str, transport: &SshTransport) -> Result<()>;
    fn prepare(
        &self,
        local_root: &Path,
        remote_root: &str,
        transport: &SshTransport,
        options: &SyncOptions,
    ) -> Result<Prepared>;
    fn resolve_interactive(&self, prepared: &mut Prepared, tty: bool) -> Result<()>;
    fn apply(
        &self,
        prepared: Prepared,
        local_root: &Path,
        remote_root: &str,
        transport: &SshTransport,
        options: &SyncOptions,
    ) -> Result<()>;
}

static CLAUDE: claude::ClaudeAdapter = claude::ClaudeAdapter;
static CODEX: codex::CodexAdapter = codex::CodexAdapter;

pub fn adapter_for(kind: AgentKind) -> &'static dyn Adapter {
    match kind {
        AgentKind::Claude => &CLAUDE,
        AgentKind::Codex => &CODEX,
    }
}