mod.rs 84.2%

186 of 221 lines run, 34 of 84 functions · run #21 on main · run missed

//! The `spectra` command line (spec §19). Output follows `oxy`: colour and
//! glyphs on a UTF-8 terminal, plain when piped, with `--plain` or with
//! `NO_COLOR`; `--json` prints one document on stdout and nothing else.

pub mod admin;
pub mod agent;
pub mod connect;
pub mod format;
pub mod human;
pub mod identity;
pub mod json;
pub mod marks;
pub mod mcp;
pub mod open;
pub mod project;
pub mod serve;
pub mod ui;

use anyhow::{bail, Context, Result};
use clap::{Parser, Subcommand};
use std::path::PathBuf;

#[derive(Parser)]
#[command(name = "spectra", version = env!("SPECTRA_VERSION_LONG"), about = "A control plane for AI coding agents")]
pub struct Cli {
    /// Plain output: no colour or glyphs (also NO_COLOR, or when not a terminal)
    #[arg(long, global = true)]
    pub plain: bool,
    /// The data directory (default: $XDG_DATA_HOME/spectra, else ~/.local/share/spectra)
    #[arg(long, global = true, env = "SPECTRA_DATA")]
    pub data: Option<PathBuf>,
    #[command(subcommand)]
    pub command: Command,
}

#[derive(Subcommand)]
pub enum Command {
    /// The version and build of this binary
    Version {
        /// Machine-readable output
        #[arg(long)]
        json: bool,
    },
    /// Make this machine's Spectra: record the owner's identity (read from
    /// their OXY_HOME and proven by signing) and make the instance key
    Init {
        /// The owner's OXY_HOME (default: $OXY_HOME, else ~/.config/oxy)
        #[arg(long)]
        owner_home: Option<PathBuf>,
        /// This instance's name (default: the host name)
        #[arg(long)]
        name: Option<String>,
        /// Set up a server without the owner's secret key: name the owner's
        /// KeyId, and they claim it from their machine with `spectra
        /// connect` (spec §8.6)
        #[arg(long, value_name = "KEYID", conflicts_with = "owner_home")]
        owner: Option<String>,
    },
    /// Use a Spectra on another machine: trust it through your delegation to
    /// its instance key, pin its certificate, and claim it if it is new
    Connect {
        /// The server, https://HOST[:PORT]
        url: String,
        /// The owner's OXY_HOME (default: $OXY_HOME, else ~/.config/oxy)
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// The owner's and the instance's identities: show, refresh
    Identity {
        #[command(subcommand)]
        command: IdentityCmd,
    },
    /// Run the daemon: the ledger, Tuner on the Unix socket and a loopback port
    Serve {
        /// The address and port for browsers and HTTP clients (default:
        /// `[server] bind`, else 127.0.0.1:7350); beyond loopback, other
        /// machines connect over TLS (spec §8.6)
        #[arg(long)]
        bind: Option<String>,
        /// TLS on the TCP port: auto (beyond loopback only), on, or off
        /// (default: `[server] tls`, else auto)
        #[arg(long)]
        tls: Option<String>,
        /// Allow a non-loopback address in plain HTTP
        #[arg(long)]
        insecure: bool,
        /// Open Radar in the browser once serving (signs a link as the owner)
        #[arg(long)]
        open: bool,
    },
    /// Sign in to Radar: a one-time link signed with the owner's key
    Open {
        #[arg(long)]
        owner_home: Option<PathBuf>,
        /// Print the link without opening a browser
        #[arg(long)]
        no_browser: bool,
    },
    /// The stdio MCP server for one agent: signs with its key and forwards to the daemon
    Mcp {
        /// The agent it speaks for
        #[arg(long)]
        agent: String,
        /// The agent's home (default: the one recorded when it was added)
        #[arg(long)]
        home: Option<PathBuf>,
    },
    /// Post a message as yourself
    Say {
        /// What to say (Markdown; @name mentions)
        body: String,
        /// The channel (default: #general)
        #[arg(short, long)]
        channel: Option<String>,
        /// Reply in the thread of this message
        #[arg(long)]
        reply_to: Option<String>,
        #[arg(long)]
        json: bool,
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// Your unread mentions and direct messages; --follow waits for more
    Inbox {
        #[arg(long)]
        follow: bool,
        #[arg(long)]
        json: bool,
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// Search what was said: words, "phrases", prefix*, AND/OR/NOT
    Search {
        query: String,
        #[arg(long)]
        channel: Option<String>,
        #[arg(long)]
        author: Option<String>,
        /// 7d, 3h, 30m
        #[arg(long)]
        since: Option<String>,
        #[arg(long, default_value_t = 20)]
        limit: u32,
        #[arg(long)]
        json: bool,
    },
    /// Any API call, signed as the owner: JSON out
    Api {
        method: String,
        path: String,
        /// The request body (JSON)
        #[arg(short = 'd', long = "body")]
        body: Option<String>,
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// The checkouts agents work in: add, list
    Project {
        #[command(subcommand)]
        command: ProjectCmd,
    },
    /// Agents, each its owner's delegate: add, list, disable
    Agent {
        #[command(subcommand)]
        command: AgentCmd,
    },
    /// The ledger's own checks: verify, check, rebuild
    Admin {
        #[command(subcommand)]
        command: Admin,
    },
}

#[derive(Subcommand)]
pub enum IdentityCmd {
    /// The owner and the instance as the ledger knows them
    Show {
        #[arg(long)]
        json: bool,
    },
    /// Record the owner's current identity after `oxy identity delegate add|remove`
    Refresh {
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
}

#[derive(Subcommand)]
pub enum ProjectCmd {
    /// Register a checkout: its repository id, remote and pinned subrepos
    Add {
        path: PathBuf,
        /// Its name (default: the directory's)
        #[arg(long)]
        name: Option<String>,
        /// Its remote URL (default: the checkout's origin)
        #[arg(long)]
        remote: Option<String>,
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// Every project with what it pins, what pins it, and its agents
    List {
        #[arg(long)]
        json: bool,
    },
}

#[derive(Subcommand)]
pub enum AgentCmd {
    /// Make an agent's key in its own home and record it once you delegate to it
    Add {
        /// Lower-case letters, digits and dashes
        name: String,
        /// claude-code, antigravity, kimi-code, codex or other
        #[arg(long)]
        kind: String,
        /// The project it works in
        #[arg(long)]
        project: String,
        /// The agent's own OXY_HOME, where its key is made and kept
        #[arg(long)]
        home: PathBuf,
        /// Bind the key to this Unix user: a request from another is refused
        #[arg(long)]
        uid: Option<u32>,
        /// Bind the key to the Unix user of its first signed connection
        #[arg(long, conflicts_with = "uid")]
        learn_uid: bool,
        /// How long to wait for your delegation, in seconds
        #[arg(long, default_value_t = 600)]
        wait: u64,
        /// The owner's OXY_HOME
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
    /// Every agent with its owner, key, delegation and state
    List {
        #[arg(long)]
        json: bool,
    },
    /// Refuse an agent's key from now on
    Disable {
        name: String,
        #[arg(long)]
        owner_home: Option<PathBuf>,
    },
}

#[derive(Subcommand)]
pub enum Admin {
    /// Recompute the hash chain and check every checkpoint and proof (exit 1 on any finding)
    Verify {
        /// Machine-readable output
        #[arg(long)]
        json: bool,
    },
    /// Whether the projections equal a rebuild from the ledger
    Check,
    /// Replay the ledger into every projection
    Rebuild,
}

/// What every command needs: where the data lives.
pub struct Ctx {
    pub data: PathBuf,
}

impl Ctx {
    /// The ledger's database, which must exist.
    pub fn db(&self) -> Result<PathBuf> {
        let db = self.data.join("spectra.db");
        if !db.exists() {
            bail!("no Spectra in {} (run `spectra init`, or pass --data)", self.data.display());
        }
        Ok(db)
    }

    /// The ledger, for writing: the instance key signs its checkpoints, and
    /// the lock keeps any other writer out.
    pub async fn writer(&self) -> Result<spectra_core::ledger::Ledger> {
        let db = self.db()?;
        let instance = spectra_core::keys::KeyPair::instance(&self.data, "spectra")?;
        spectra_core::ledger::Ledger::open(&db, Some(instance)).await
    }

    /// The daemon's Unix socket.
    pub fn socket(&self) -> PathBuf {
        self.data.join("spectra.sock")
    }

    /// The Spectra on another machine this one uses, if any (spec §8.6).
    pub fn remote(&self) -> Result<Option<connect::ClientCfg>> {
        connect::load()
    }

    /// Where the daemon is: the configured server, else the local socket.
    pub fn endpoint(&self) -> Result<spectra_mcp::client::Endpoint> {
        Ok(match self.remote()? {
            Some(c) => c.endpoint()?,
            None => spectra_mcp::client::Endpoint::Unix(self.socket()),
        })
    }

    /// A client signing as the owner: to the configured server (spec §8.6),
    /// or to the local daemon when it is serving; none when neither. It
    /// signs with the owner identity the ledger records, since the daemon
    /// may not know a newer one yet.
    pub async fn owner_client(&self, owner_home: &std::path::Path) -> Result<Option<spectra_mcp::client::Client>> {
        use spectra_mcp::client::Client;
        if let Some(cfg) = self.remote()? {
            let ep = cfg.endpoint()?;
            let pair = remote_owner(&ep, owner_home).await?;
            return Ok(Some(Client::at(ep, Some(pair))));
        }
        if !self.data.join("spectra.db").exists() || !Client::alive(&self.socket()).await {
            return Ok(None);
        }
        let ledger = spectra_core::ledger::Ledger::open_read_only(&self.db()?).await?;
        Ok(Some(Client::new(&self.socket(), Some(identity::recorded_owner(&ledger, owner_home).await?))))
    }

    /// Somewhere to append the owner's signed events: the configured server,
    /// the local daemon when it is serving, else straight to the ledger
    /// (spec §8.5, §8.6).
    pub async fn owner_writer(&self, owner_home: &std::path::Path) -> Result<Writer> {
        if let Some(cfg) = self.remote()? {
            let ep = cfg.endpoint()?;
            let pair = remote_owner(&ep, owner_home).await?;
            let recorded = pair.identity.clone();
            return Ok(Writer::Remote { client: Box::new(spectra_mcp::client::Client::at(ep, Some(pair))), recorded });
        }
        let db = self.db()?;
        if spectra_mcp::client::Client::alive(&self.socket()).await {
            let ledger = spectra_core::ledger::Ledger::open_read_only(&db).await?;
            let transport = identity::recorded_owner(&ledger, owner_home).await?;
            return Ok(Writer::Daemon { ledger, client: Box::new(spectra_mcp::client::Client::new(&self.socket(), Some(transport))) });
        }
        Ok(Writer::Direct(self.writer().await?))
    }

    /// Refuse a command that needs the ledger file on this machine.
    pub fn local_only(&self, what: &str) -> Result<()> {
        if self.remote()?.is_some() {
            bail!("{what} runs on the server, which holds the ledger (this machine uses {})", connect::path().display());
        }
        Ok(())
    }

    /// `[owner] home` from spectra.toml.
    pub fn config_owner_home(&self) -> Option<PathBuf> {
        let text = std::fs::read_to_string(self.data.join("spectra.toml")).ok()?;
        let t: toml::Table = text.parse().ok()?;
        t.get("owner")?.get("home")?.as_str().map(PathBuf::from)
    }
}

/// Where the owner's events go.
pub enum Writer {
    Direct(spectra_core::ledger::Ledger),
    Daemon {
        ledger: spectra_core::ledger::Ledger,
        client: Box<spectra_mcp::client::Client>,
    },
    /// A Spectra on another machine; `recorded` is the owner identity it knows.
    Remote {
        client: Box<spectra_mcp::client::Client>,
        recorded: oxy_core::identity::Identity,
    },
}

impl Writer {
    /// The ledger on this machine, for reading, if there is one.
    pub fn ledger(&self) -> Option<&spectra_core::ledger::Ledger> {
        match self {
            Writer::Direct(l) | Writer::Daemon { ledger: l, .. } => Some(l),
            Writer::Remote { .. } => None,
        }
    }

    /// A client of the daemon, when one is serving.
    pub fn client(&self) -> Option<&spectra_mcp::client::Client> {
        match self {
            Writer::Daemon { client, .. } | Writer::Remote { client, .. } => Some(client),
            Writer::Direct(_) => None,
        }
    }

    /// The rotation that brings the recorded owner to the current identity
    /// in `home`, if it has moved.
    pub async fn refresh_event(&self, home: &std::path::Path) -> Result<Option<spectra_core::event::NewEvent>> {
        match self {
            Writer::Direct(l) | Writer::Daemon { ledger: l, .. } => identity::refresh_event(l, home).await,
            Writer::Remote { recorded, .. } => identity::rotation(recorded, home),
        }
    }

    /// Append events signed over their payload, then checkpoint.
    pub async fn append(&self, events: Vec<spectra_core::event::NewEvent>) -> Result<()> {
        match self {
            Writer::Direct(l) => {
                l.append(events).await?;
                l.checkpoint().await?;
            }
            Writer::Daemon { client, .. } | Writer::Remote { client, .. } => {
                let wire: Vec<_> = events.iter().map(spectra_core::daemon::WireEvent::from_new).collect();
                let r = client.send("POST", "/api/v1/ledger/events", &[], serde_json::to_vec(&wire)?).await?;
                if r.status != 200 {
                    let why = r
                        .json()
                        .ok()
                        .and_then(|v| v.get("title").and_then(|t| t.as_str()).map(String::from))
                        .unwrap_or_else(|| String::from_utf8_lossy(&r.body).into());
                    bail!("the daemon refused: {why}");
                }
            }
        }
        Ok(())
    }
}

/// The owner identity a server records, as a key pair from the owner's
/// home: what requests to it are signed with.
async fn remote_owner(ep: &spectra_mcp::client::Endpoint, home: &std::path::Path) -> Result<spectra_core::keys::KeyPair> {
    let r = spectra_mcp::client::Client::at(ep.clone(), None).send("GET", "/api/v1/instance", &[], vec![]).await?;
    let info = r.json()?;
    let key = info.get("owner").and_then(|o| o.as_str()).context("the server has no owner yet: spectra connect claims it")?;
    let keys = home.join("keys");
    spectra_core::keys::KeyPair::load(&keys.join(format!("{key}.secret")), &keys.join(format!("{key}.identity")))
        .with_context(|| format!("the server's owner {} is not an identity in {}", &key[..12.min(key.len())], home.display()))
}

/// An API call; a refusal becomes an error that says why.
pub async fn call(c: &spectra_mcp::client::Client, method: &str, path: &str, body: Option<serde_json::Value>) -> Result<serde_json::Value> {
    let r = c.send(method, path, &[], body.map(|b| serde_json::to_vec(&b)).transpose()?.unwrap_or_default()).await?;
    let v: serde_json::Value = r.json().unwrap_or(serde_json::Value::Null);
    if r.status >= 400 {
        let why = v
            .get("detail")
            .or(v.get("title"))
            .and_then(serde_json::Value::as_str)
            .map(String::from)
            .unwrap_or_else(|| String::from_utf8_lossy(&r.body).into());
        bail!("{why}");
    }
    Ok(v)
}

/// A query string value, percent-encoded.
pub fn encode(s: &str) -> String {
    s.bytes()
        .map(|b| match b {
            b'A'..=b'Z' | b'a'..=b'z' | b'0'..=b'9' | b'-' | b'_' | b'.' | b'~' => (b as char).to_string(),
            _ => format!("%{b:02X}"),
        })
        .collect()
}

fn init_logging() {
    let filter = tracing_subscriber::EnvFilter::try_from_env("SPECTRA_LOG").unwrap_or_else(|_| tracing_subscriber::EnvFilter::new("warn"));
    tracing_subscriber::fmt().with_env_filter(filter).with_writer(std::io::stderr).init();
}

fn default_data() -> PathBuf {
    if let Ok(x) = std::env::var("XDG_DATA_HOME") {
        return PathBuf::from(x).join("spectra");
    }
    let home = std::env::var("HOME").unwrap_or_else(|_| ".".into());
    PathBuf::from(home).join(".local").join("share").join("spectra")
}

fn runtime() -> Result<tokio::runtime::Runtime> {
    Ok(tokio::runtime::Builder::new_multi_thread().enable_all().build()?)
}

pub fn run() -> Result<i32> {
    let cli = Cli::parse();
    ui::init(cli.plain);
    let ctx = Ctx { data: cli.data.unwrap_or_else(default_data) };
    match cli.command {
        Command::Version { json } => version(json),
        Command::Connect { url, owner_home } => runtime()?.block_on(connect::connect(&ctx, &url, owner_home)),
        Command::Init { owner_home, name, owner: Some(key) } => {
            let _ = owner_home;
            runtime()?.block_on(identity::init_for(&ctx, &key, name))
        }
        Command::Init { owner_home, name, owner: None } => runtime()?.block_on(identity::init(&ctx, owner_home, name)),
        Command::Identity { command } => runtime()?.block_on(async {
            match command {
                IdentityCmd::Show { json } => identity::show(&ctx, json).await,
                IdentityCmd::Refresh { owner_home } => identity::refresh(&ctx, owner_home).await,
            }
        }),
        Command::Serve { bind, tls, insecure, open } => {
            init_logging();
            runtime()?.block_on(serve::serve(&ctx, bind, tls, insecure, open))
        }
        Command::Open { owner_home, no_browser } => runtime()?.block_on(open::open(&ctx, owner_home, !no_browser)),
        Command::Mcp { agent, home } => runtime()?.block_on(mcp::run(&ctx, &agent, home)),
        Command::Say { body, channel, reply_to, json, owner_home } => {
            runtime()?.block_on(human::say(&ctx, channel, reply_to, body, owner_home, json))
        }
        Command::Inbox { follow, json, owner_home } => runtime()?.block_on(human::inbox(&ctx, follow, owner_home, json)),
        Command::Search { query, channel, author, since, limit, json } => {
            runtime()?.block_on(human::search(&ctx, &query, channel, author, since, limit, json))
        }
        Command::Api { method, path, body, owner_home } => runtime()?.block_on(human::raw(&ctx, &method, &path, body, owner_home)),
        Command::Project { command } => runtime()?.block_on(async {
            match command {
                ProjectCmd::Add { path, name, remote, owner_home } => project::add(&ctx, &path, name, remote, owner_home).await,
                ProjectCmd::List { json } => project::list(&ctx, json).await,
            }
        }),
        Command::Agent { command } => runtime()?.block_on(async {
            match command {
                AgentCmd::Add { name, kind, project, home, uid, learn_uid, wait, owner_home } => {
                    agent::add(&ctx, agent::Add { name, kind, project, home, uid, learn_uid, wait_secs: wait, owner_home }).await
                }
                AgentCmd::List { json } => agent::list(&ctx, json).await,
                AgentCmd::Disable { name, owner_home } => agent::disable(&ctx, &name, owner_home).await,
            }
        }),
        Command::Admin { command } => runtime()?.block_on(async {
            match command {
                Admin::Verify { json } => admin::verify(&ctx, json).await,
                Admin::Check => admin::check(&ctx).await,
                Admin::Rebuild => admin::rebuild(&ctx).await,
            }
        }),
    }
}

fn version(as_json: bool) -> Result<i32> {
    if as_json {
        println!(
            "{}",
            serde_json::json!({
                "version": env!("CARGO_PKG_VERSION"),
                "build": env!("SPECTRA_BUILD_ID"),
                "mcp_protocol_versions": spectra_mcp::PROTOCOL_VERSIONS,
                "ledger_format": spectra_core::LEDGER_FORMAT,
            })
        );
    } else {
        println!("spectra {}", env!("SPECTRA_VERSION_LONG"));
    }
    Ok(0)
}

`/`	Search
`g` `c`	Code
`g` `h`	History
`g` `b`	Bookmarks
`g` `t`	Trust
`y`	Permalink to this state (files and trees)
`?`	This sheet

1		//! The `spectra` command line (spec §19). Output follows `oxy`: colour and
2		//! glyphs on a UTF-8 terminal, plain when piped, with `--plain` or with
3		//! `NO_COLOR`; `--json` prints one document on stdout and nothing else.
4
5		pub mod admin;
6		pub mod agent;
7		pub mod connect;
8		pub mod format;
9		pub mod human;
10		pub mod identity;
11		pub mod json;
12		pub mod marks;
13		pub mod mcp;
14		pub mod open;
15		pub mod project;
16		pub mod serve;
17		pub mod ui;
18
19		use anyhow::{bail, Context, Result};
20		use clap::{Parser, Subcommand};
21		use std::path::PathBuf;
22
23		#[derive(Parser)]
24		#[command(name = "spectra", version = env!("SPECTRA_VERSION_LONG"), about = "A control plane for AI coding agents")]
25		pub struct Cli {
26		/// Plain output: no colour or glyphs (also NO_COLOR, or when not a terminal)
27		#[arg(long, global = true)]
28		pub plain: bool,
29		/// The data directory (default: $XDG_DATA_HOME/spectra, else ~/.local/share/spectra)
30		#[arg(long, global = true, env = "SPECTRA_DATA")]
31		pub data: Option<PathBuf>,
32		#[command(subcommand)]
33		pub command: Command,
34		}
35
36		#[derive(Subcommand)]
37		pub enum Command {
38		/// The version and build of this binary
39		Version {
40		/// Machine-readable output
41		#[arg(long)]
42		json: bool,
43		},
44		/// Make this machine's Spectra: record the owner's identity (read from
45		/// their OXY_HOME and proven by signing) and make the instance key
46		Init {
47		/// The owner's OXY_HOME (default: $OXY_HOME, else ~/.config/oxy)
48		#[arg(long)]
49		owner_home: Option<PathBuf>,
50		/// This instance's name (default: the host name)
51		#[arg(long)]
52		name: Option<String>,
53		/// Set up a server without the owner's secret key: name the owner's
54		/// KeyId, and they claim it from their machine with `spectra
55		/// connect` (spec §8.6)
56		#[arg(long, value_name = "KEYID", conflicts_with = "owner_home")]
57		owner: Option<String>,
58		},
59		/// Use a Spectra on another machine: trust it through your delegation to
60		/// its instance key, pin its certificate, and claim it if it is new
61		Connect {
62		/// The server, https://HOST[:PORT]
63		url: String,
64		/// The owner's OXY_HOME (default: $OXY_HOME, else ~/.config/oxy)
65		#[arg(long)]
66		owner_home: Option<PathBuf>,
67		},
68		/// The owner's and the instance's identities: show, refresh
69		Identity {
70		#[command(subcommand)]
71		command: IdentityCmd,
72		},
73		/// Run the daemon: the ledger, Tuner on the Unix socket and a loopback port
74		Serve {
75		/// The address and port for browsers and HTTP clients (default:
76		/// `[server] bind`, else 127.0.0.1:7350); beyond loopback, other
77		/// machines connect over TLS (spec §8.6)
78		#[arg(long)]
79		bind: Option<String>,
80		/// TLS on the TCP port: auto (beyond loopback only), on, or off
81		/// (default: `[server] tls`, else auto)
82		#[arg(long)]
83		tls: Option<String>,
84		/// Allow a non-loopback address in plain HTTP
85		#[arg(long)]
86		insecure: bool,
87		/// Open Radar in the browser once serving (signs a link as the owner)
88		#[arg(long)]
89		open: bool,
90		},
91		/// Sign in to Radar: a one-time link signed with the owner's key
92		Open {
93		#[arg(long)]
94		owner_home: Option<PathBuf>,
95		/// Print the link without opening a browser
96		#[arg(long)]
97		no_browser: bool,
98		},
99		/// The stdio MCP server for one agent: signs with its key and forwards to the daemon
100		Mcp {
101		/// The agent it speaks for
102		#[arg(long)]
103		agent: String,
104		/// The agent's home (default: the one recorded when it was added)
105		#[arg(long)]
106		home: Option<PathBuf>,
107		},
108		/// Post a message as yourself
109		Say {
110		/// What to say (Markdown; @name mentions)
111		body: String,
112		/// The channel (default: #general)
113		#[arg(short, long)]
114		channel: Option<String>,
115		/// Reply in the thread of this message
116		#[arg(long)]
117		reply_to: Option<String>,
118		#[arg(long)]
119		json: bool,
120		#[arg(long)]
121		owner_home: Option<PathBuf>,
122		},
123		/// Your unread mentions and direct messages; --follow waits for more
124		Inbox {
125		#[arg(long)]
126		follow: bool,
127		#[arg(long)]
128		json: bool,
129		#[arg(long)]
130		owner_home: Option<PathBuf>,
131		},
132		/// Search what was said: words, "phrases", prefix*, AND/OR/NOT
133		Search {
134		query: String,
135		#[arg(long)]
136		channel: Option<String>,
137		#[arg(long)]
138		author: Option<String>,
139		/// 7d, 3h, 30m
140		#[arg(long)]
141		since: Option<String>,
142		#[arg(long, default_value_t = 20)]
143		limit: u32,
144		#[arg(long)]
145		json: bool,
146		},
147		/// Any API call, signed as the owner: JSON out
148		Api {
149		method: String,
150		path: String,
151		/// The request body (JSON)
152		#[arg(short = 'd', long = "body")]
153		body: Option<String>,
154		#[arg(long)]
155		owner_home: Option<PathBuf>,
156		},
157		/// The checkouts agents work in: add, list
158		Project {
159		#[command(subcommand)]
160		command: ProjectCmd,
161		},
162		/// Agents, each its owner's delegate: add, list, disable
163		Agent {
164		#[command(subcommand)]
165		command: AgentCmd,
166		},
167		/// The ledger's own checks: verify, check, rebuild
168		Admin {
169		#[command(subcommand)]
170		command: Admin,
171		},
172		}
173
174		#[derive(Subcommand)]
175		pub enum IdentityCmd {
176		/// The owner and the instance as the ledger knows them
177		Show {
178		#[arg(long)]
179		json: bool,
180		},
181		/// Record the owner's current identity after `oxy identity delegate add\|remove`
182		Refresh {
183		#[arg(long)]
184		owner_home: Option<PathBuf>,
185		},
186		}
187
188		#[derive(Subcommand)]
189		pub enum ProjectCmd {
190		/// Register a checkout: its repository id, remote and pinned subrepos
191		Add {
192		path: PathBuf,
193		/// Its name (default: the directory's)
194		#[arg(long)]
195		name: Option<String>,
196		/// Its remote URL (default: the checkout's origin)
197		#[arg(long)]
198		remote: Option<String>,
199		#[arg(long)]
200		owner_home: Option<PathBuf>,
201		},
202		/// Every project with what it pins, what pins it, and its agents
203		List {
204		#[arg(long)]
205		json: bool,
206		},
207		}
208
209		#[derive(Subcommand)]
210		pub enum AgentCmd {
211		/// Make an agent's key in its own home and record it once you delegate to it
212		Add {
213		/// Lower-case letters, digits and dashes
214		name: String,
215		/// claude-code, antigravity, kimi-code, codex or other
216		#[arg(long)]
217		kind: String,
218		/// The project it works in
219		#[arg(long)]
220		project: String,
221		/// The agent's own OXY_HOME, where its key is made and kept
222		#[arg(long)]
223		home: PathBuf,
224		/// Bind the key to this Unix user: a request from another is refused
225		#[arg(long)]
226		uid: Option<u32>,
227		/// Bind the key to the Unix user of its first signed connection
228		#[arg(long, conflicts_with = "uid")]
229		learn_uid: bool,
230		/// How long to wait for your delegation, in seconds
231		#[arg(long, default_value_t = 600)]
232		wait: u64,
233		/// The owner's OXY_HOME
234		#[arg(long)]
235		owner_home: Option<PathBuf>,
236		},
237		/// Every agent with its owner, key, delegation and state
238		List {
239		#[arg(long)]
240		json: bool,
241		},
242		/// Refuse an agent's key from now on
243		Disable {
244		name: String,
245		#[arg(long)]
246		owner_home: Option<PathBuf>,
247		},
248		}
249
250		#[derive(Subcommand)]
251		pub enum Admin {
252		/// Recompute the hash chain and check every checkpoint and proof (exit 1 on any finding)
253		Verify {
254		/// Machine-readable output
255		#[arg(long)]
256		json: bool,
257		},
258		/// Whether the projections equal a rebuild from the ledger
259		Check,
260		/// Replay the ledger into every projection
261		Rebuild,
262		}
263
264		/// What every command needs: where the data lives.
265		pub struct Ctx {
266		pub data: PathBuf,
267		}
268
269		impl Ctx {
270		/// The ledger's database, which must exist.
271	149×	pub fn db(&self) -> Result<PathBuf> {
272	149×	let db = self.data.join("spectra.db");
273	149×	if !db.exists() {
274	1×	bail!("no Spectra in {} (run `spectra init`, or pass --data)", self.data.display());
275	148×	}
276	148×	Ok(db)
277	149×	}
278
279		/// The ledger, for writing: the instance key signs its checkpoints, and
280		/// the lock keeps any other writer out.
281	50×	pub async fn writer(&self) -> Result<spectra_core::ledger::Ledger> {
282	50×	let db = self.db()?;
283	50×	let instance = spectra_core::keys::KeyPair::instance(&self.data, "spectra")?;
284	50×	spectra_core::ledger::Ledger::open(&db, Some(instance)).await
285	50×	}
286
287		/// The daemon's Unix socket.
288	103×	pub fn socket(&self) -> PathBuf {
289	103×	self.data.join("spectra.sock")
290	103×	}
291
292		/// The Spectra on another machine this one uses, if any (spec §8.6).
293	122×	pub fn remote(&self) -> Result<Option<connect::ClientCfg>> {
294	122×	connect::load()
295	122×	}
296
297		/// Where the daemon is: the configured server, else the local socket.
298	0×	pub fn endpoint(&self) -> Result<spectra_mcp::client::Endpoint> {
299	0×	Ok(match self.remote()? {
300	0×	Some(c) => c.endpoint()?,
301	0×	None => spectra_mcp::client::Endpoint::Unix(self.socket()),
302		})
303	0×	}
304
305		/// A client signing as the owner: to the configured server (spec §8.6),
306		/// or to the local daemon when it is serving; none when neither. It
307		/// signs with the owner identity the ledger records, since the daemon
308		/// may not know a newer one yet.
309	43×	pub async fn owner_client(&self, owner_home: &std::path::Path) -> Result<Option<spectra_mcp::client::Client>> {
310		use spectra_mcp::client::Client;
311	43×	if let Some(cfg) = self.remote()? {
312	9×	let ep = cfg.endpoint()?;
313	9×	let pair = remote_owner(&ep, owner_home).await?;
314	9×	return Ok(Some(Client::at(ep, Some(pair))));
315	34×	}
316	34×	if !self.data.join("spectra.db").exists() \|\| !Client::alive(&self.socket()).await {
317	30×	return Ok(None);
318	4×	}
319	4×	let ledger = spectra_core::ledger::Ledger::open_read_only(&self.db()?).await?;
320	4×	Ok(Some(Client::new(&self.socket(), Some(identity::recorded_owner(&ledger, owner_home).await?))))
321	43×	}
322
323		/// Somewhere to append the owner's signed events: the configured server,
324		/// the local daemon when it is serving, else straight to the ledger
325		/// (spec §8.5, §8.6).
326	48×	pub async fn owner_writer(&self, owner_home: &std::path::Path) -> Result<Writer> {
327	48×	if let Some(cfg) = self.remote()? {
328	4×	let ep = cfg.endpoint()?;
329	4×	let pair = remote_owner(&ep, owner_home).await?;
330	4×	let recorded = pair.identity.clone();
331	4×	return Ok(Writer::Remote { client: Box::new(spectra_mcp::client::Client::at(ep, Some(pair))), recorded });
332	44×	}
333	44×	let db = self.db()?;
334	44×	if spectra_mcp::client::Client::alive(&self.socket()).await {
335	2×	let ledger = spectra_core::ledger::Ledger::open_read_only(&db).await?;
336	2×	let transport = identity::recorded_owner(&ledger, owner_home).await?;
337	2×	return Ok(Writer::Daemon { ledger, client: Box::new(spectra_mcp::client::Client::new(&self.socket(), Some(transport))) });
338	42×	}
339	42×	Ok(Writer::Direct(self.writer().await?))
340	48×	}
341
342		/// Refuse a command that needs the ledger file on this machine.
343	8×	pub fn local_only(&self, what: &str) -> Result<()> {
344	8×	if self.remote()?.is_some() {
345	1×	bail!("{what} runs on the server, which holds the ledger (this machine uses {})", connect::path().display());
346	7×	}
347	7×	Ok(())
348	8×	}
349
350		/// `[owner] home` from spectra.toml.
351	92×	pub fn config_owner_home(&self) -> Option<PathBuf> {
352	92×	let text = std::fs::read_to_string(self.data.join("spectra.toml")).ok()?;
353	60×	let t: toml::Table = text.parse().ok()?;
354	60×	t.get("owner")?.get("home")?.as_str().map(PathBuf::from)
355	92×	}
356		}
357
358		/// Where the owner's events go.
359		pub enum Writer {
360		Direct(spectra_core::ledger::Ledger),
361		Daemon {
362		ledger: spectra_core::ledger::Ledger,
363		client: Box<spectra_mcp::client::Client>,
364		},
365		/// A Spectra on another machine; `recorded` is the owner identity it knows.
366		Remote {
367		client: Box<spectra_mcp::client::Client>,
368		recorded: oxy_core::identity::Identity,
369		},
370		}
371
372		impl Writer {
373		/// The ledger on this machine, for reading, if there is one.
374	49×	pub fn ledger(&self) -> Option<&spectra_core::ledger::Ledger> {
375	49×	match self {
376	46×	Writer::Direct(l) \| Writer::Daemon { ledger: l, .. } => Some(l),
377	3×	Writer::Remote { .. } => None,
378		}
379	49×	}
380
381		/// A client of the daemon, when one is serving.
382	49×	pub fn client(&self) -> Option<&spectra_mcp::client::Client> {
383	49×	match self {
384	5×	Writer::Daemon { client, .. } \| Writer::Remote { client, .. } => Some(client),
385	44×	Writer::Direct(_) => None,
386		}
387	49×	}
388
389		/// The rotation that brings the recorded owner to the current identity
390		/// in `home`, if it has moved.
391	46×	pub async fn refresh_event(&self, home: &std::path::Path) -> Result<Option<spectra_core::event::NewEvent>> {
392	46×	match self {
393	42×	Writer::Direct(l) \| Writer::Daemon { ledger: l, .. } => identity::refresh_event(l, home).await,
394	4×	Writer::Remote { recorded, .. } => identity::rotation(recorded, home),
395		}
396	46×	}
397
398		/// Append events signed over their payload, then checkpoint.
399	46×	pub async fn append(&self, events: Vec<spectra_core::event::NewEvent>) -> Result<()> {
400	46×	match self {
401	40×	Writer::Direct(l) => {
402	40×	l.append(events).await?;
403	40×	l.checkpoint().await?;
404		}
405	4×	Writer::Daemon { client, .. } \| Writer::Remote { client, .. } => {
406	6×	let wire: Vec<_> = events.iter().map(spectra_core::daemon::WireEvent::from_new).collect();
407	6×	let r = client.send("POST", "/api/v1/ledger/events", &[], serde_json::to_vec(&wire)?).await?;
408	6×	if r.status != 200 {
409	0×	let why = r
410	0×	.json()
411	0×	.ok()
412	0×	.and_then(\|v\| v.get("title").and_then(\|t\| t.as_str()).map(String::from))
413	0×	.unwrap_or_else(\|\| String::from_utf8_lossy(&r.body).into());
414	0×	bail!("the daemon refused: {why}");
415	6×	}
416		}
417		}
418	46×	Ok(())
419	46×	}
420		}
421
422		/// The owner identity a server records, as a key pair from the owner's
423		/// home: what requests to it are signed with.
424	13×	async fn remote_owner(ep: &spectra_mcp::client::Endpoint, home: &std::path::Path) -> Result<spectra_core::keys::KeyPair> {
425	13×	let r = spectra_mcp::client::Client::at(ep.clone(), None).send("GET", "/api/v1/instance", &[], vec![]).await?;
426	13×	let info = r.json()?;
427	13×	let key = info.get("owner").and_then(\|o\| o.as_str()).context("the server has no owner yet: spectra connect claims it")?;
428	13×	let keys = home.join("keys");
429	13×	spectra_core::keys::KeyPair::load(&keys.join(format!("{key}.secret")), &keys.join(format!("{key}.identity")))
430	13×	.with_context(\|\| format!("the server's owner {} is not an identity in {}", &key[..12.min(key.len())], home.display()))
431	13×	}
432
433		/// An API call; a refusal becomes an error that says why.
434	20×	pub async fn call(c: &spectra_mcp::client::Client, method: &str, path: &str, body: Option<serde_json::Value>) -> Result<serde_json::Value> {
435	20×	let r = c.send(method, path, &[], body.map(\|b\| serde_json::to_vec(&b)).transpose()?.unwrap_or_default()).await?;
436	20×	let v: serde_json::Value = r.json().unwrap_or(serde_json::Value::Null);
437	20×	if r.status >= 400 {
438	0×	let why = v
439	0×	.get("detail")
440	0×	.or(v.get("title"))
441	0×	.and_then(serde_json::Value::as_str)
442	0×	.map(String::from)
443	0×	.unwrap_or_else(\|\| String::from_utf8_lossy(&r.body).into());
444	0×	bail!("{why}");
445	20×	}
446	20×	Ok(v)
447	20×	}
448
449		/// A query string value, percent-encoded.
450	3×	pub fn encode(s: &str) -> String {
451	3×	s.bytes()
452	21×	.map(\|b\| match b {
453	21×	b'A'..=b'Z' \| b'a'..=b'z' \| b'0'..=b'9' \| b'-' \| b'_' \| b'.' \| b'~' => (b as char).to_string(),
454	0×	_ => format!("%{b:02X}"),
455	21×	})
456	3×	.collect()
457	3×	}
458
459	19×	fn init_logging() {
460	19×	let filter = tracing_subscriber::EnvFilter::try_from_env("SPECTRA_LOG").unwrap_or_else(\|_\| tracing_subscriber::EnvFilter::new("warn"));
461	19×	tracing_subscriber::fmt().with_env_filter(filter).with_writer(std::io::stderr).init();
462	19×	}
463
464	0×	fn default_data() -> PathBuf {
465	0×	if let Ok(x) = std::env::var("XDG_DATA_HOME") {
466	0×	return PathBuf::from(x).join("spectra");
467	0×	}
468	0×	let home = std::env::var("HOME").unwrap_or_else(\|_\| ".".into());
469	0×	PathBuf::from(home).join(".local").join("share").join("spectra")
470	0×	}
471
472	155×	fn runtime() -> Result<tokio::runtime::Runtime> {
473	155×	Ok(tokio::runtime::Builder::new_multi_thread().enable_all().build()?)
474	155×	}
475
476	156×	pub fn run() -> Result<i32> {
477	156×	let cli = Cli::parse();
478	156×	ui::init(cli.plain);
479	156×	let ctx = Ctx { data: cli.data.unwrap_or_else(default_data) };
480	27×	match cli.command {
481	1×	Command::Version { json } => version(json),
482	7×	Command::Connect { url, owner_home } => runtime()?.block_on(connect::connect(&ctx, &url, owner_home)),
483	5×	Command::Init { owner_home, name, owner: Some(key) } => {
484	5×	let _ = owner_home;
485	5×	runtime()?.block_on(identity::init_for(&ctx, &key, name))
486		}
487	22×	Command::Init { owner_home, name, owner: None } => runtime()?.block_on(identity::init(&ctx, owner_home, name)),
488	2×	Command::Identity { command } => runtime()?.block_on(async {
489	2×	match command {
490	2×	IdentityCmd::Show { json } => identity::show(&ctx, json).await,
491	0×	IdentityCmd::Refresh { owner_home } => identity::refresh(&ctx, owner_home).await,
492		}
493	2×	}),
494	19×	Command::Serve { bind, tls, insecure, open } => {
495	19×	init_logging();
496	19×	runtime()?.block_on(serve::serve(&ctx, bind, tls, insecure, open))
497		}
498	1×	Command::Open { owner_home, no_browser } => runtime()?.block_on(open::open(&ctx, owner_home, !no_browser)),
499	2×	Command::Mcp { agent, home } => runtime()?.block_on(mcp::run(&ctx, &agent, home)),
500	3×	Command::Say { body, channel, reply_to, json, owner_home } => {
501	3×	runtime()?.block_on(human::say(&ctx, channel, reply_to, body, owner_home, json))
502		}
503	2×	Command::Inbox { follow, json, owner_home } => runtime()?.block_on(human::inbox(&ctx, follow, owner_home, json)),
504	2×	Command::Search { query, channel, author, since, limit, json } => {
505	2×	runtime()?.block_on(human::search(&ctx, &query, channel, author, since, limit, json))
506		}
507	2×	Command::Api { method, path, body, owner_home } => runtime()?.block_on(human::raw(&ctx, &method, &path, body, owner_home)),
508	28×	Command::Project { command } => runtime()?.block_on(async {
509	28×	match command {
510	24×	ProjectCmd::Add { path, name, remote, owner_home } => project::add(&ctx, &path, name, remote, owner_home).await,
511	4×	ProjectCmd::List { json } => project::list(&ctx, json).await,
512		}
513	28×	}),
514	32×	Command::Agent { command } => runtime()?.block_on(async {
515	32×	match command {
516	23×	AgentCmd::Add { name, kind, project, home, uid, learn_uid, wait, owner_home } => {
517	23×	agent::add(&ctx, agent::Add { name, kind, project, home, uid, learn_uid, wait_secs: wait, owner_home }).await
518		}
519	6×	AgentCmd::List { json } => agent::list(&ctx, json).await,
520	3×	AgentCmd::Disable { name, owner_home } => agent::disable(&ctx, &name, owner_home).await,
521		}
522	32×	}),
523	28×	Command::Admin { command } => runtime()?.block_on(async {
524	28×	match command {
525	20×	Admin::Verify { json } => admin::verify(&ctx, json).await,
526	6×	Admin::Check => admin::check(&ctx).await,
527	2×	Admin::Rebuild => admin::rebuild(&ctx).await,
528		}
529	28×	}),
530		}
531	156×	}
532
533	1×	fn version(as_json: bool) -> Result<i32> {
534	1×	if as_json {
535	1×	println!(
536	1×	"{}",
537	1×	serde_json::json!({
538	1×	"version": env!("CARGO_PKG_VERSION"),
539	1×	"build": env!("SPECTRA_BUILD_ID"),
540	1×	"mcp_protocol_versions": spectra_mcp::PROTOCOL_VERSIONS,
541	1×	"ledger_format": spectra_core::LEDGER_FORMAT,
542	1×	})
543	1×	);
544	1×	} else {
545	0×	println!("spectra {}", env!("SPECTRA_VERSION_LONG"));
546	0×	}
547	1×	Ok(0)
548	1×	}

Coverage / src/cli/mod.rs 84.2%

Keyboard