Watch
0
0
Fork
You've already forked hyperhive
0

Compare commits

...
23 changed files with 226 additions and 83 deletions

View file

@ -138,6 +138,9 @@ Swarm-level:
with **no identity** and a TLS failure that names no cause. with **no identity** and a TLS failure that names no cause.
- **`swarm-queue-client/`** — the one authenticated connect to the swarm - **`swarm-queue-client/`** — the one authenticated connect to the swarm
NATS queue. NATS queue.
- **`swarm-wire/`** — serde-only request and reply contracts services
exchange over the queue (builder, claude login), so naming one links
neither the queue client nor the store client.
- **`swarm-nats-auth/`** — NATS auth-callout responder; without it the - **`swarm-nats-auth/`** — NATS auth-callout responder; without it the
queue is fail-closed and admits nobody. queue is fail-closed and admits nobody.
- **`swarm-agent-verify/`** — checks a `swarm-agent.<agent>.<secret>` - **`swarm-agent-verify/`** — checks a `swarm-agent.<agent>.<secret>`

12
Cargo.lock generated
View file

@ -4811,6 +4811,7 @@ dependencies = [
"serde_json", "serde_json",
"swarm-queue-client", "swarm-queue-client",
"swarm-secret-client", "swarm-secret-client",
"swarm-wire",
"tokio", "tokio",
"tracing", "tracing",
] ]
@ -4846,6 +4847,7 @@ dependencies = [
"serde_json", "serde_json",
"swarm-queue-client", "swarm-queue-client",
"swarm-secret-client", "swarm-secret-client",
"swarm-wire",
"tokio", "tokio",
"tracing", "tracing",
] ]
@ -4889,6 +4891,7 @@ dependencies = [
"swarm-matrix-client", "swarm-matrix-client",
"swarm-queue-client", "swarm-queue-client",
"swarm-secret-client", "swarm-secret-client",
"swarm-wire",
"tempfile", "tempfile",
"tokio", "tokio",
"tower", "tower",
@ -5010,6 +5013,7 @@ dependencies = [
"swarm-agent-verify", "swarm-agent-verify",
"swarm-queue-client", "swarm-queue-client",
"swarm-secret-client", "swarm-secret-client",
"swarm-wire",
"tokio", "tokio",
"tracing", "tracing",
"tracing-subscriber", "tracing-subscriber",
@ -5076,6 +5080,14 @@ dependencies = [
"vaultrs", "vaultrs",
] ]
[[package]]
name = "swarm-wire"
version = "0.1.0"
dependencies = [
"serde",
"serde_json",
]
[[package]] [[package]]
name = "swarmctl" name = "swarmctl"
version = "0.1.0" version = "0.1.0"

View file

@ -43,6 +43,7 @@ members = [
"swarm-logs-auth", "swarm-logs-auth",
"swarm-metrics-auth", "swarm-metrics-auth",
"swarm-secret-client", "swarm-secret-client",
"swarm-wire",
"swarmctl", "swarmctl",
] ]
@ -118,6 +119,7 @@ swarm-controller-api = { path = "swarm-controller-api" }
swarm-matrix-client = { path = "swarm-matrix-client" } swarm-matrix-client = { path = "swarm-matrix-client" }
swarm-queue-client = { path = "swarm-queue-client" } swarm-queue-client = { path = "swarm-queue-client" }
swarm-secret-client = { path = "swarm-secret-client" } swarm-secret-client = { path = "swarm-secret-client" }
swarm-wire = { path = "swarm-wire" }
thiserror = "2" thiserror = "2"
vaultrs = "0.8" vaultrs = "0.8"
# vaultrs's transport crates. Direct dependencies because one endpoint is # vaultrs's transport crates. Direct dependencies because one endpoint is

View file

@ -15,10 +15,11 @@ async-nats.workspace = true
futures-util.workspace = true futures-util.workspace = true
hive-log.workspace = true hive-log.workspace = true
serde_json.workspace = true serde_json.workspace = true
# The build subject and the shapes the controller sends and reads back.
swarm-queue-client.workspace = true swarm-queue-client.workspace = true
# The forge token and queue client secret paths, and the store login. # The forge token and queue client secret paths, and the store login.
swarm-secret-client.workspace = true swarm-secret-client.workspace = true
# The build subject and the shapes the controller sends and reads back.
swarm-wire.workspace = true
tokio.workspace = true tokio.workspace = true
tracing.workspace = true tracing.workspace = true

View file

@ -9,7 +9,7 @@ use std::path::PathBuf;
use std::process::Stdio; use std::process::Stdio;
use std::time::Duration; use std::time::Duration;
use swarm_queue_client::builder::ArchetypesReply; use swarm_wire::builder::ArchetypesReply;
/// What every evaluation is run with, fixed at start. /// What every evaluation is run with, fixed at start.
pub struct Evaluator { pub struct Evaluator {
@ -105,7 +105,7 @@ mod tests {
nix: PathBuf::from("nix"), nix: PathBuf::from("nix"),
hyperhive: "/nix/store/h-source".to_owned(), hyperhive: "/nix/store/h-source".to_owned(),
nixpkgs: "/nix/store/n-source".to_owned(), nixpkgs: "/nix/store/n-source".to_owned(),
deadline: swarm_queue_client::builder::ARCHETYPES_DEADLINE, deadline: swarm_wire::builder::ARCHETYPES_DEADLINE,
} }
} }

View file

@ -1,11 +1,11 @@
//! The swarm builder's build service. //! The swarm builder's build service.
//! //!
//! Connects to the swarm queue as its own OIDC client, takes build requests //! Connects to the swarm queue as its own OIDC client, takes build requests
//! from swarm-controller on [`swarm_queue_client::builder::SUBJECT`], builds //! from swarm-controller on [`swarm_wire::builder::SUBJECT`], builds
//! each agent config at its commit, answers on the request's reply subject, //! each agent config at its commit, answers on the request's reply subject,
//! and logs one line per build. It writes nothing to the forge: the //! and logs one line per build. It writes nothing to the forge: the
//! controller turns the reply into a commit status. It also answers //! controller turns the reply into a commit status. It also answers
//! [`swarm_queue_client::builder::ARCHETYPES_SUBJECT`] with the agent //! [`swarm_wire::builder::ARCHETYPES_SUBJECT`] with the agent
//! archetypes its hyperhive source declares, which the controller stores. //! archetypes its hyperhive source declares, which the controller stores.
//! //!
//! The store identity comes from the `BAO_*` environment. The queue client //! The store identity comes from the `BAO_*` environment. The queue client
@ -19,13 +19,13 @@ use std::time::Duration;
use anyhow::{Context, Result, anyhow, bail}; use anyhow::{Context, Result, anyhow, bail};
use futures_util::StreamExt as _; use futures_util::StreamExt as _;
use swarm_queue_client::builder::{
ARCHETYPES_DEADLINE, ARCHETYPES_SUBJECT, BUILD_DEADLINE, BuildAgent, QUEUE_GROUP,
REPLY_DEADLINE, Reply, SUBJECT,
};
use swarm_queue_client::{ClientSecret, QueueConfig}; use swarm_queue_client::{ClientSecret, QueueConfig};
use swarm_secret_client::forge::Credential; use swarm_secret_client::forge::Credential;
use swarm_secret_client::stored_secret::StoredSecret; use swarm_secret_client::stored_secret::StoredSecret;
use swarm_wire::builder::{
ARCHETYPES_DEADLINE, ARCHETYPES_SUBJECT, BUILD_DEADLINE, BuildAgent, QUEUE_GROUP,
REPLY_DEADLINE, Reply, SUBJECT,
};
mod archetypes; mod archetypes;
mod nix_build; mod nix_build;
@ -168,10 +168,10 @@ async fn answer_evaluations(
}; };
let reply = evaluator.eval().await; let reply = evaluator.eval().await;
match &reply { match &reply {
swarm_queue_client::builder::ArchetypesReply::Evaluated { archetypes, .. } => { swarm_wire::builder::ArchetypesReply::Evaluated { archetypes, .. } => {
tracing::info!(count = archetypes.len(), "archetypes evaluated"); tracing::info!(count = archetypes.len(), "archetypes evaluated");
} }
swarm_queue_client::builder::ArchetypesReply::Failed { error_tail } => { swarm_wire::builder::ArchetypesReply::Failed { error_tail } => {
tracing::warn!(%error_tail, "archetype evaluation failed"); tracing::warn!(%error_tail, "archetype evaluation failed");
} }
} }

View file

@ -16,7 +16,7 @@ use std::path::PathBuf;
use std::process::Stdio; use std::process::Stdio;
use std::time::Duration; use std::time::Duration;
use swarm_queue_client::builder::{BuildAgent, Reply}; use swarm_wire::builder::{BuildAgent, Reply};
use tokio::io::AsyncReadExt as _; use tokio::io::AsyncReadExt as _;
/// The most of `nix`'s stderr a failed build's reply carries, from its end. /// The most of `nix`'s stderr a failed build's reply carries, from its end.
@ -293,7 +293,7 @@ pub(crate) fn tail(stderr: &str, status: std::process::ExitStatus) -> String {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use swarm_queue_client::builder::BUILD_DEADLINE; use swarm_wire::builder::BUILD_DEADLINE;
fn job() -> BuildAgent { fn job() -> BuildAgent {
BuildAgent { BuildAgent {

View file

@ -21,6 +21,8 @@ hive-types.workspace = true
swarm-queue-client.workspace = true swarm-queue-client.workspace = true
# The credentials and queue client secret paths, and the store login. # The credentials and queue client secret paths, and the store login.
swarm-secret-client.workspace = true swarm-secret-client.workspace = true
# The login subject and the requests and replies it carries.
swarm-wire.workspace = true
tokio.workspace = true tokio.workspace = true
tracing.workspace = true tracing.workspace = true

View file

@ -14,8 +14,8 @@ use std::process::Stdio;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use swarm_queue_client::claude_login::Reply;
use swarm_secret_client::claude_login::Credentials; use swarm_secret_client::claude_login::Credentials;
use swarm_wire::claude_login::Reply;
use tokio::io::{AsyncBufReadExt as _, AsyncRead, AsyncWriteExt as _, BufReader}; use tokio::io::{AsyncBufReadExt as _, AsyncRead, AsyncWriteExt as _, BufReader};
use tokio::process::{Child, ChildStdin, Command}; use tokio::process::{Child, ChildStdin, Command};
use tokio::sync::{Mutex, watch}; use tokio::sync::{Mutex, watch};

View file

@ -2,7 +2,7 @@
//! //!
//! Connects to the swarm queue as its own OIDC client, takes an operator's //! Connects to the swarm queue as its own OIDC client, takes an operator's
//! login steps from swarm-controller on //! login steps from swarm-controller on
//! [`swarm_queue_client::claude_login::SUBJECT`], runs `claude auth login` //! [`swarm_wire::claude_login::SUBJECT`], runs `claude auth login`
//! for the agent named ([`login`]), and writes the credentials it produces to //! for the agent named ([`login`]), and writes the credentials it produces to
//! that agent's store leaf, a new version over any stored before. The agent's //! that agent's store leaf, a new version over any stored before. The agent's
//! harness takes a version newer than the one it last installed. //! harness takes a version newer than the one it last installed.
@ -16,10 +16,10 @@ use std::sync::Arc;
use anyhow::{Context, Result, bail}; use anyhow::{Context, Result, bail};
use futures_util::StreamExt as _; use futures_util::StreamExt as _;
use swarm_queue_client::claude_login::{Reply, Request, SESSION_DEADLINE, STEP_DEADLINE, SUBJECT};
use swarm_queue_client::{ClientSecret, QueueConfig}; use swarm_queue_client::{ClientSecret, QueueConfig};
use swarm_secret_client::claude_login::{Credentials, client_secret_path, credentials_path}; use swarm_secret_client::claude_login::{Credentials, client_secret_path, credentials_path};
use swarm_secret_client::stored_secret::StoredSecret; use swarm_secret_client::stored_secret::StoredSecret;
use swarm_wire::claude_login::{Reply, Request, SESSION_DEADLINE, STEP_DEADLINE, SUBJECT};
mod login; mod login;
@ -190,7 +190,7 @@ mod tests {
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use swarm_queue_client::claude_login::{Reply, Request}; use swarm_wire::claude_login::{Reply, Request};
use super::{handle, login}; use super::{handle, login};

View file

@ -118,6 +118,8 @@ swarm-queue-client = { workspace = true, features = [
# name and the object's shape are agreements between the two ends, and a # name and the object's shape are agreements between the two ends, and a
# second spelling here would be a store this hive could not read. # second spelling here would be a store this hive could not read.
swarm-secret-client.workspace = true swarm-secret-client.workspace = true
# The builder and claude login subjects, requests and replies it sends.
swarm-wire.workspace = true
# The appservice calls `matrix_account::agent_token` mints agents' accounts # The appservice calls `matrix_account::agent_token` mints agents' accounts
# with — shared with `swarm-matrix-ctl`, which pins the same device id. # with — shared with `swarm-matrix-ctl`, which pins the same device id.
swarm-matrix-client.workspace = true swarm-matrix-client.workspace = true

View file

@ -16,7 +16,7 @@ use hive_sh4re::permissions::Permission;
use problem_details::ProblemDetails; use problem_details::ProblemDetails;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use swarm_queue_client::archetypes::{ArchetypeMeta, BUILTIN_SOURCE, SourceRecord}; use swarm_queue_client::archetypes::{ArchetypeMeta, BUILTIN_SOURCE, SourceRecord};
use swarm_queue_client::builder::{ use swarm_wire::builder::{
ARCHETYPES_DEADLINE, ARCHETYPES_SUBJECT, ArchetypesReply, EvalArchetypes, ARCHETYPES_DEADLINE, ARCHETYPES_SUBJECT, ArchetypesReply, EvalArchetypes,
}; };
use utoipa::ToSchema; use utoipa::ToSchema;
@ -361,7 +361,7 @@ mod tests {
let mut requests = builder let mut requests = builder
.queue_subscribe( .queue_subscribe(
ARCHETYPES_SUBJECT, ARCHETYPES_SUBJECT,
swarm_queue_client::builder::QUEUE_GROUP.to_owned(), swarm_wire::builder::QUEUE_GROUP.to_owned(),
) )
.await .await
.expect("subscribe"); .expect("subscribe");

View file

@ -16,7 +16,7 @@ use std::future::Future;
use forgejo_api::structs::CommitStatusState; use forgejo_api::structs::CommitStatusState;
use hive_jobq::scheduler::Outcome; use hive_jobq::scheduler::Outcome;
use swarm_queue_client::builder::{BuildAgent, REPLY_DEADLINE, Reply, SUBJECT}; use swarm_wire::builder::{BuildAgent, REPLY_DEADLINE, Reply, SUBJECT};
use crate::agent_renewal::unix_now; use crate::agent_renewal::unix_now;
use crate::forge::{self, CONFIG_ORG}; use crate::forge::{self, CONFIG_ORG};
@ -534,7 +534,7 @@ mod tests {
let builder = nats.client().await; let builder = nats.client().await;
let mut requests = builder let mut requests = builder
.queue_subscribe(SUBJECT, swarm_queue_client::builder::QUEUE_GROUP.to_owned()) .queue_subscribe(SUBJECT, swarm_wire::builder::QUEUE_GROUP.to_owned())
.await .await
.expect("subscribe"); .expect("subscribe");
// A message the builder sends itself and receives back proves the // A message the builder sends itself and receives back proves the

View file

@ -21,7 +21,7 @@ use axum::extract::{Path, State};
use axum::http::{HeaderMap, StatusCode}; use axum::http::{HeaderMap, StatusCode};
use problem_details::ProblemDetails; use problem_details::ProblemDetails;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use swarm_queue_client::claude_login::{REPLY_DEADLINE, Reply, Request, SUBJECT}; use swarm_wire::claude_login::{REPLY_DEADLINE, Reply, Request, SUBJECT};
use utoipa::ToSchema; use utoipa::ToSchema;
use crate::{AppState, error_problem}; use crate::{AppState, error_problem};
@ -243,7 +243,7 @@ mod tests {
use std::sync::Mutex; use std::sync::Mutex;
use axum::http::StatusCode; use axum::http::StatusCode;
use swarm_queue_client::claude_login::{Reply, Request}; use swarm_wire::claude_login::{Reply, Request};
use super::{LoginSink, relay}; use super::{LoginSink, relay};

View file

@ -601,7 +601,7 @@ async fn build_agent(
rev: String, rev: String,
pr: u64, pr: u64,
) -> hive_jobq::scheduler::Outcome { ) -> hive_jobq::scheduler::Outcome {
let job = swarm_queue_client::builder::BuildAgent { agent, rev, pr }; let job = swarm_wire::builder::BuildAgent { agent, rev, pr };
let builds = builder::runs::Builds::Live(deps.queue.clone()); let builds = builder::runs::Builds::Live(deps.queue.clone());
let link_base = std::env::var(PUBLIC_URL_ENV).ok(); let link_base = std::env::var(PUBLIC_URL_ENV).ok();
builder::run( builder::run(

View file

@ -33,6 +33,8 @@ swarm-queue-client = { workspace = true, features = ["notices"] }
# credentials `swarm-agent-verify` checks a presented token against. # credentials `swarm-agent-verify` checks a presented token against.
swarm-secret-client.workspace = true swarm-secret-client.workspace = true
swarm-agent-verify.workspace = true swarm-agent-verify.workspace = true
# The builder and login subjects it grants.
swarm-wire.workspace = true
tokio.workspace = true tokio.workspace = true
tracing.workspace = true tracing.workspace = true
tracing-subscriber.workspace = true tracing-subscriber.workspace = true

View file

@ -281,11 +281,11 @@ impl Policy {
Permissions { Permissions {
publish: vec![Self::inbox(client_id)], publish: vec![Self::inbox(client_id)],
subscribe: Some(vec![ subscribe: Some(vec![
swarm_queue_client::builder::SUBJECT.to_owned(), swarm_wire::builder::SUBJECT.to_owned(),
swarm_queue_client::builder::ARCHETYPES_SUBJECT.to_owned(), swarm_wire::builder::ARCHETYPES_SUBJECT.to_owned(),
Self::inbox(client_id), Self::inbox(client_id),
]), ]),
responses: Some(swarm_queue_client::builder::REPLY_DEADLINE), responses: Some(swarm_wire::builder::REPLY_DEADLINE),
} }
} }
@ -296,10 +296,10 @@ impl Policy {
Permissions { Permissions {
publish: vec![Self::inbox(client_id)], publish: vec![Self::inbox(client_id)],
subscribe: Some(vec![ subscribe: Some(vec![
swarm_queue_client::claude_login::SUBJECT.to_owned(), swarm_wire::claude_login::SUBJECT.to_owned(),
Self::inbox(client_id), Self::inbox(client_id),
]), ]),
responses: Some(swarm_queue_client::claude_login::REPLY_DEADLINE), responses: Some(swarm_wire::claude_login::REPLY_DEADLINE),
} }
} }
@ -661,11 +661,11 @@ impl Policy {
// back on this client's inbox, which it already subscribes to. // back on this client's inbox, which it already subscribes to.
swarm_queue_client::review_classifier::SUBJECT.to_owned(), swarm_queue_client::review_classifier::SUBJECT.to_owned(),
// A build of an agent config, sent as a request the same way. // A build of an agent config, sent as a request the same way.
swarm_queue_client::builder::SUBJECT.to_owned(), swarm_wire::builder::SUBJECT.to_owned(),
// An archetype evaluation, sent as a request the same way. // An archetype evaluation, sent as a request the same way.
swarm_queue_client::builder::ARCHETYPES_SUBJECT.to_owned(), swarm_wire::builder::ARCHETYPES_SUBJECT.to_owned(),
// An operator's claude login step, sent as a request the same way. // An operator's claude login step, sent as a request the same way.
swarm_queue_client::claude_login::SUBJECT.to_owned(), swarm_wire::claude_login::SUBJECT.to_owned(),
// The wanted-state buckets — one per hive, all created and written // The wanted-state buckets — one per hive, all created and written
// by this single client. // by this single client.
// //
@ -1532,8 +1532,8 @@ mod tests {
#[test] #[test]
fn the_builder_hears_its_subject_and_inbox_only_and_answers_by_reply() { fn the_builder_hears_its_subject_and_inbox_only_and_answers_by_reply() {
use swarm_queue_client::builder::{ARCHETYPES_SUBJECT, REPLY_DEADLINE, SUBJECT};
use swarm_queue_client::client_inbox_prefix; use swarm_queue_client::client_inbox_prefix;
use swarm_wire::builder::{ARCHETYPES_SUBJECT, REPLY_DEADLINE, SUBJECT};
let p = policy() let p = policy()
.with_builders(vec![BUILDER.to_owned()]) .with_builders(vec![BUILDER.to_owned()])
@ -1562,7 +1562,7 @@ mod tests {
#[test] #[test]
fn only_the_reader_may_send_a_build() { fn only_the_reader_may_send_a_build() {
use swarm_queue_client::builder::{ARCHETYPES_SUBJECT, SUBJECT}; use swarm_wire::builder::{ARCHETYPES_SUBJECT, SUBJECT};
let p = with_classifier().with_builders(vec![BUILDER.to_owned()]); let p = with_classifier().with_builders(vec![BUILDER.to_owned()]);
let reader = p.permissions("swarm-controller").expect("the reader"); let reader = p.permissions("swarm-controller").expect("the reader");
@ -1589,8 +1589,8 @@ mod tests {
#[test] #[test]
fn the_claude_login_minter_hears_its_subject_and_inbox_only_and_answers_by_reply() { fn the_claude_login_minter_hears_its_subject_and_inbox_only_and_answers_by_reply() {
use swarm_queue_client::claude_login::{REPLY_DEADLINE, SUBJECT};
use swarm_queue_client::client_inbox_prefix; use swarm_queue_client::client_inbox_prefix;
use swarm_wire::claude_login::{REPLY_DEADLINE, SUBJECT};
let p = policy() let p = policy()
.with_claude_logins(vec![CLAUDE_LOGIN.to_owned()]) .with_claude_logins(vec![CLAUDE_LOGIN.to_owned()])
@ -1602,7 +1602,7 @@ mod tests {
for refused in [ for refused in [
">".to_owned(), ">".to_owned(),
"$SWARM.>".to_owned(), "$SWARM.>".to_owned(),
swarm_queue_client::builder::SUBJECT.to_owned(), swarm_wire::builder::SUBJECT.to_owned(),
reply(&client_inbox_prefix("swarm-controller")), reply(&client_inbox_prefix("swarm-controller")),
] { ] {
assert!( assert!(
@ -1618,7 +1618,7 @@ mod tests {
#[test] #[test]
fn only_the_reader_may_send_a_login_step() { fn only_the_reader_may_send_a_login_step() {
use swarm_queue_client::claude_login::SUBJECT; use swarm_wire::claude_login::SUBJECT;
let p = with_classifier() let p = with_classifier()
.with_builders(vec![BUILDER.to_owned()]) .with_builders(vec![BUILDER.to_owned()])

View file

@ -4,7 +4,7 @@
//! //!
//! **swarm-controller is the only writer and the only reader.** The swarm //! **swarm-controller is the only writer and the only reader.** The swarm
//! builder evaluates the source and answers on //! builder evaluates the source and answers on
//! [`crate::builder::ARCHETYPES_SUBJECT`]; the controller checks each //! `swarm_wire::builder::ARCHETYPES_SUBJECT`; the controller checks each
//! archetype's settings the way `POST /api/agents` checks them and writes the //! archetype's settings the way `POST /api/agents` checks them and writes the
//! result here, so the create screen lists archetypes from a value that a //! result here, so the create screen lists archetypes from a value that a
//! create would accept. //! create would accept.

View file

@ -494,13 +494,6 @@ pub mod review_classifier;
mod startup_read; mod startup_read;
pub use startup_read::read_with_startup_retry; pub use startup_read::read_with_startup_retry;
/// The builder's build subject and the request and reply it carries.
pub mod builder;
/// The claude login minter's subject and the requests and replies it
/// carries.
pub mod claude_login;
/// Only the fields this needs; authelia returns several. /// Only the fields this needs; authelia returns several.
#[derive(serde::Deserialize)] #[derive(serde::Deserialize)]
struct TokenResponse { struct TokenResponse {

11
swarm-wire/Cargo.toml Normal file
View file

@ -0,0 +1,11 @@
[package]
name = "swarm-wire"
version.workspace = true
edition.workspace = true
[dependencies]
serde.workspace = true
serde_json.workspace = true
[lints]
workspace = true

View file

@ -82,40 +82,97 @@ pub enum ArchetypesReply {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use serde_json::json;
use super::*; use super::*;
#[test] #[test]
fn the_wire_tags_the_reply_by_state() { fn the_build_request_serializes_as_pinned() {
let json = serde_json::to_value(Reply::Built { let req = BuildAgent {
agent: "atlas".to_owned(),
rev: "abc".to_owned(),
pr: 7,
};
let value = serde_json::to_value(&req).expect("encodes");
assert_eq!(value, json!({"agent": "atlas", "rev": "abc", "pr": 7}));
assert_eq!(
serde_json::from_value::<BuildAgent>(value).expect("decodes"),
req
);
}
#[test]
fn the_reply_serializes_as_pinned() {
for (reply, wire) in [
(
Reply::Built {
out_path: "/nix/store/x".to_owned(), out_path: "/nix/store/x".to_owned(),
}) },
.expect("encodes"); json!({"state": "built", "out_path": "/nix/store/x"}),
assert_eq!(json["state"], "built"); ),
assert_eq!(json["out_path"], "/nix/store/x"); (
let json = serde_json::to_value(Reply::Failed { Reply::Failed {
error_tail: "e".to_owned(), error_tail: "e".to_owned(),
}) },
.expect("encodes"); json!({"state": "failed", "error_tail": "e"}),
assert_eq!(json["state"], "failed"); ),
] {
assert_eq!(serde_json::to_value(&reply).expect("encodes"), wire);
assert_eq!(
serde_json::from_value::<Reply>(wire).expect("decodes"),
reply
);
}
}
#[test]
fn the_archetypes_request_and_reply_serialize_as_pinned() {
let req = EvalArchetypes::default();
let value = serde_json::to_value(&req).expect("encodes");
assert_eq!(value, json!({}));
assert_eq!(
serde_json::from_value::<EvalArchetypes>(value).expect("decodes"),
req
);
let mut archetypes = serde_json::Map::new();
archetypes.insert("coder".to_owned(), json!({"description": "d"}));
for (reply, wire) in [
(
ArchetypesReply::Evaluated {
source: "/nix/store/h".to_owned(),
archetypes,
},
json!({
"state": "evaluated",
"source": "/nix/store/h",
"archetypes": {"coder": {"description": "d"}},
}),
),
(
ArchetypesReply::Failed {
error_tail: "e".to_owned(),
},
json!({"state": "failed", "error_tail": "e"}),
),
] {
assert_eq!(serde_json::to_value(&reply).expect("encodes"), wire);
assert_eq!(
serde_json::from_value::<ArchetypesReply>(wire).expect("decodes"),
reply
);
}
}
#[test]
fn the_subjects_are_pinned() {
assert_eq!(SUBJECT, "$SWARM.builder.build");
assert_eq!(ARCHETYPES_SUBJECT, "$SWARM.builder.archetypes");
assert_eq!(QUEUE_GROUP, "builder");
} }
#[test] #[test]
fn the_reply_deadline_outlasts_the_build() { fn the_reply_deadline_outlasts_the_build() {
assert!(REPLY_DEADLINE > BUILD_DEADLINE); assert!(REPLY_DEADLINE > BUILD_DEADLINE);
} }
#[test]
fn the_archetypes_reply_is_tagged_by_state() {
let json = serde_json::to_value(ArchetypesReply::Evaluated {
source: "/nix/store/h".to_owned(),
archetypes: serde_json::Map::new(),
})
.expect("encodes");
assert_eq!(json["state"], "evaluated");
let json = serde_json::to_value(ArchetypesReply::Failed {
error_tail: "e".to_owned(),
})
.expect("encodes");
assert_eq!(json["state"], "failed");
}
} }

View file

@ -6,9 +6,9 @@
//! each. //! each.
//! //!
//! A login is two requests for one agent: //! A login is two requests for one agent:
//! [`Request::Start`](crate::claude_login::Request::Start) runs //! [`Request::Start`] runs
//! `claude auth login` and answers with the URL it printed, and //! `claude auth login` and answers with the URL it printed, and
//! [`Request::Code`](crate::claude_login::Request::Code) hands it the code the //! [`Request::Code`] hands it the code the
//! operator pasted and answers once the credentials are in the agent's store //! operator pasted and answers once the credentials are in the agent's store
//! leaf. The running login lives in the minter's memory between the two, so //! leaf. The running login lives in the minter's memory between the two, so
//! the swarm runs one minter. //! the swarm runs one minter.
@ -71,19 +71,68 @@ pub enum Reply {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use serde_json::json;
use super::*; use super::*;
#[test] #[test]
fn the_wire_tags_the_request_by_step_and_the_reply_by_state() { fn the_request_serializes_as_pinned() {
let json = serde_json::to_value(Request::Code { for (req, wire) in [
(
Request::Start {
agent: "atlas".to_owned(),
operator: "mara".to_owned(),
},
json!({"step": "start", "agent": "atlas", "operator": "mara"}),
),
(
Request::Code {
agent: "atlas".to_owned(), agent: "atlas".to_owned(),
code: "c".to_owned(), code: "c".to_owned(),
}) },
.expect("encodes"); json!({"step": "code", "agent": "atlas", "code": "c"}),
assert_eq!(json["step"], "code"); ),
assert_eq!(json["agent"], "atlas"); (
let json = serde_json::to_value(Reply::Stored).expect("encodes"); Request::Cancel {
assert_eq!(json["state"], "stored"); agent: "atlas".to_owned(),
},
json!({"step": "cancel", "agent": "atlas"}),
),
] {
assert_eq!(serde_json::to_value(&req).expect("encodes"), wire);
assert!(serde_json::from_value::<Request>(wire).expect("decodes") == req);
}
}
#[test]
fn the_reply_serializes_as_pinned() {
for (reply, wire) in [
(
Reply::Started {
url: "https://u".to_owned(),
},
json!({"state": "started", "url": "https://u"}),
),
(Reply::Stored, json!({"state": "stored"})),
(Reply::Cancelled, json!({"state": "cancelled"})),
(
Reply::Failed {
error: "e".to_owned(),
},
json!({"state": "failed", "error": "e"}),
),
] {
assert_eq!(serde_json::to_value(&reply).expect("encodes"), wire);
assert_eq!(
serde_json::from_value::<Reply>(wire).expect("decodes"),
reply
);
}
}
#[test]
fn the_subject_is_pinned() {
assert_eq!(SUBJECT, "$SWARM.claude-login.request");
} }
#[test] #[test]

9
swarm-wire/src/lib.rs Normal file
View file

@ -0,0 +1,9 @@
//! The request and reply shapes swarm services exchange over the queue: the
//! subjects, their deadlines and the serde types that travel each way.
//!
//! Serde only, with no queue client and no secret-store client. A crate that
//! names a contract (the sender, the answerer, the auth-callout responder that
//! grants the subject) depends on this one without linking either.
pub mod builder;
pub mod claude_login;