refactor(#2581): carve per-agent mcp.sock protocol into hive-agent-sock crate
This commit is contained in:
parent
f64ab47de0
commit
d0beec8a40
26 changed files with 603 additions and 542 deletions
14
Cargo.lock
generated
14
Cargo.lock
generated
|
|
@ -1523,6 +1523,7 @@ dependencies = [
|
||||||
"clap",
|
"clap",
|
||||||
"forgejo-api",
|
"forgejo-api",
|
||||||
"futures-util",
|
"futures-util",
|
||||||
|
"hive-agent-sock",
|
||||||
"hive-claude",
|
"hive-claude",
|
||||||
"hive-sh4re",
|
"hive-sh4re",
|
||||||
"http-body-util",
|
"http-body-util",
|
||||||
|
|
@ -1551,6 +1552,7 @@ dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"axum",
|
"axum",
|
||||||
"clap",
|
"clap",
|
||||||
|
"hive-agent-sock",
|
||||||
"hive-sh4re",
|
"hive-sh4re",
|
||||||
"rmcp",
|
"rmcp",
|
||||||
"serde",
|
"serde",
|
||||||
|
|
@ -1560,13 +1562,21 @@ dependencies = [
|
||||||
"tracing-subscriber",
|
"tracing-subscriber",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "hive-agent-sock"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"hive-sh4re",
|
||||||
|
"serde",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "hive-agent-wake"
|
name = "hive-agent-wake"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"clap",
|
"clap",
|
||||||
"hive-sh4re",
|
"hive-agent-sock",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
|
@ -1579,6 +1589,7 @@ name = "hive-bash-mcp"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
|
"hive-agent-sock",
|
||||||
"hive-sh4re",
|
"hive-sh4re",
|
||||||
"libc",
|
"libc",
|
||||||
"rmcp",
|
"rmcp",
|
||||||
|
|
@ -1604,6 +1615,7 @@ dependencies = [
|
||||||
"clap-markdown",
|
"clap-markdown",
|
||||||
"clap_complete",
|
"clap_complete",
|
||||||
"forgejo-api",
|
"forgejo-api",
|
||||||
|
"hive-agent-sock",
|
||||||
"hive-host-sock",
|
"hive-host-sock",
|
||||||
"hive-priv-sock",
|
"hive-priv-sock",
|
||||||
"hive-sh4re",
|
"hive-sh4re",
|
||||||
|
|
|
||||||
|
|
@ -3,6 +3,7 @@ resolver = "3"
|
||||||
members = [
|
members = [
|
||||||
"hive-agent",
|
"hive-agent",
|
||||||
"hive-agent-mcp",
|
"hive-agent-mcp",
|
||||||
|
"hive-agent-sock",
|
||||||
"hive-agent-wake",
|
"hive-agent-wake",
|
||||||
"hive-bash-mcp",
|
"hive-bash-mcp",
|
||||||
"hive-c0re",
|
"hive-c0re",
|
||||||
|
|
@ -48,6 +49,7 @@ clap = { version = "4", features = ["derive"] }
|
||||||
clap_complete = "4"
|
clap_complete = "4"
|
||||||
indicatif = "0.18"
|
indicatif = "0.18"
|
||||||
hive-sh4re = { path = "hive-sh4re" }
|
hive-sh4re = { path = "hive-sh4re" }
|
||||||
|
hive-agent-sock = { path = "hive-agent-sock" }
|
||||||
hive-claude = { path = "hive-claude" }
|
hive-claude = { path = "hive-claude" }
|
||||||
hive-host-sock = { path = "hive-host-sock" }
|
hive-host-sock = { path = "hive-host-sock" }
|
||||||
hive-priv-sock = { path = "hive-priv-sock" }
|
hive-priv-sock = { path = "hive-priv-sock" }
|
||||||
|
|
|
||||||
|
|
@ -14,6 +14,7 @@ workspace = true
|
||||||
anyhow.workspace = true
|
anyhow.workspace = true
|
||||||
axum.workspace = true
|
axum.workspace = true
|
||||||
clap.workspace = true
|
clap.workspace = true
|
||||||
|
hive-agent-sock.workspace = true
|
||||||
hive-sh4re.workspace = true
|
hive-sh4re.workspace = true
|
||||||
rmcp.workspace = true
|
rmcp.workspace = true
|
||||||
serde.workspace = true
|
serde.workspace = true
|
||||||
|
|
|
||||||
|
|
@ -123,9 +123,9 @@ impl AgentServer {
|
||||||
/// (hive-sh4re type aliases), so this single method covers both sockets.
|
/// (hive-sh4re type aliases), so this single method covers both sockets.
|
||||||
async fn dispatch(
|
async fn dispatch(
|
||||||
&self,
|
&self,
|
||||||
req: hive_sh4re::Request,
|
req: hive_agent_sock::Request,
|
||||||
) -> (Result<hive_sh4re::Response, anyhow::Error>, u32) {
|
) -> (Result<hive_agent_sock::Response, anyhow::Error>, u32) {
|
||||||
match client::request_retried::<_, hive_sh4re::Response>(&self.socket, &req).await {
|
match client::request_retried::<_, hive_agent_sock::Response>(&self.socket, &req).await {
|
||||||
Ok((r, n)) => (Ok(r), n),
|
Ok((r, n)) => (Ok(r), n),
|
||||||
Err(e) => (Err(e), 0),
|
Err(e) => (Err(e), 0),
|
||||||
}
|
}
|
||||||
|
|
@ -152,7 +152,7 @@ impl AgentServer {
|
||||||
}
|
}
|
||||||
run_tool_envelope("send", log, async move {
|
run_tool_envelope("send", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Send {
|
.dispatch(hive_agent_sock::Request::Send {
|
||||||
to: args.to,
|
to: args.to,
|
||||||
body: args.body,
|
body: args.body,
|
||||||
in_reply_to: args.in_reply_to,
|
in_reply_to: args.in_reply_to,
|
||||||
|
|
@ -183,7 +183,7 @@ impl AgentServer {
|
||||||
let log = format!("{args:?}");
|
let log = format!("{args:?}");
|
||||||
run_tool_envelope("ask", log, async move {
|
run_tool_envelope("ask", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Ask {
|
.dispatch(hive_agent_sock::Request::Ask {
|
||||||
question: args.question,
|
question: args.question,
|
||||||
options: args.options,
|
options: args.options,
|
||||||
multi: args.multi,
|
multi: args.multi,
|
||||||
|
|
@ -192,7 +192,7 @@ impl AgentServer {
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
let s = match resp {
|
let s = match resp {
|
||||||
Ok(hive_sh4re::Response::QuestionQueued { id }) => format!(
|
Ok(hive_agent_sock::Response::QuestionQueued { id }) => format!(
|
||||||
"question queued (id={id}); answer will arrive as a system \
|
"question queued (id={id}); answer will arrive as a system \
|
||||||
`question_answered` event in your inbox"
|
`question_answered` event in your inbox"
|
||||||
),
|
),
|
||||||
|
|
@ -217,7 +217,7 @@ impl AgentServer {
|
||||||
let id = args.id;
|
let id = args.id;
|
||||||
run_tool_envelope("answer", log, async move {
|
run_tool_envelope("answer", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Answer {
|
.dispatch(hive_agent_sock::Request::Answer {
|
||||||
id,
|
id,
|
||||||
answer: args.answer,
|
answer: args.answer,
|
||||||
})
|
})
|
||||||
|
|
@ -255,7 +255,7 @@ impl AgentServer {
|
||||||
run_tool_envelope("recv", log, async move {
|
run_tool_envelope("recv", log, async move {
|
||||||
let waited = args.wait_seconds.is_some_and(|w| w > 0);
|
let waited = args.wait_seconds.is_some_and(|w| w > 0);
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Recv {
|
.dispatch(hive_agent_sock::Request::Recv {
|
||||||
wait_seconds: args.wait_seconds,
|
wait_seconds: args.wait_seconds,
|
||||||
max: args.max,
|
max: args.max,
|
||||||
})
|
})
|
||||||
|
|
@ -280,10 +280,10 @@ impl AgentServer {
|
||||||
let log = format!("{args:?}");
|
let log = format!("{args:?}");
|
||||||
run_tool_envelope("ack_until", log, async move {
|
run_tool_envelope("ack_until", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::AckUntil { up_to: args.up_to })
|
.dispatch(hive_agent_sock::Request::AckUntil { up_to: args.up_to })
|
||||||
.await;
|
.await;
|
||||||
let rendered = match resp {
|
let rendered = match resp {
|
||||||
Ok(hive_sh4re::Response::Acked { count }) => {
|
Ok(hive_agent_sock::Response::Acked { count }) => {
|
||||||
format!("acked {count} message(s) up to id {}", args.up_to)
|
format!("acked {count} message(s) up to id {}", args.up_to)
|
||||||
}
|
}
|
||||||
other => reply_err(other, "ack_until"),
|
other => reply_err(other, "ack_until"),
|
||||||
|
|
@ -312,11 +312,11 @@ impl AgentServer {
|
||||||
run_tool_envelope("get_loose_ends", String::new(), async move {
|
run_tool_envelope("get_loose_ends", String::new(), async move {
|
||||||
let is_self_query = args.agent.is_none();
|
let is_self_query = args.agent.is_none();
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::GetLooseEnds { agent: args.agent })
|
.dispatch(hive_agent_sock::Request::GetLooseEnds { agent: args.agent })
|
||||||
.await;
|
.await;
|
||||||
// Extract the vec so we can augment before rendering.
|
// Extract the vec so we can augment before rendering.
|
||||||
let mut loose_ends = match resp {
|
let mut loose_ends = match resp {
|
||||||
Ok(hive_sh4re::Response::LooseEnds { loose_ends }) => loose_ends,
|
Ok(hive_agent_sock::Response::LooseEnds { loose_ends }) => loose_ends,
|
||||||
other => return annotate_retries(reply_err(other, "get_loose_ends"), retries),
|
other => return annotate_retries(reply_err(other, "get_loose_ends"), retries),
|
||||||
};
|
};
|
||||||
// Prepend matrix unread entry for self-queries only (can't
|
// Prepend matrix unread entry for self-queries only (can't
|
||||||
|
|
@ -365,7 +365,7 @@ impl AgentServer {
|
||||||
return e;
|
return e;
|
||||||
}
|
}
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::SetStatus { text: args.text })
|
.dispatch(hive_agent_sock::Request::SetStatus { text: args.text })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(resp, "set_status", "status updated".to_owned()),
|
format_ack(resp, "set_status", "status updated".to_owned()),
|
||||||
|
|
@ -397,7 +397,7 @@ impl AgentServer {
|
||||||
let log = args.name.clone().unwrap_or_else(|| "<self>".to_owned());
|
let log = args.name.clone().unwrap_or_else(|| "<self>".to_owned());
|
||||||
run_tool_envelope("get_agent_meta", log, async move {
|
run_tool_envelope("get_agent_meta", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::GetAgentMeta { name: args.name })
|
.dispatch(hive_agent_sock::Request::GetAgentMeta { name: args.name })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(format_agent_meta(resp), retries)
|
annotate_retries(format_agent_meta(resp), retries)
|
||||||
})
|
})
|
||||||
|
|
@ -425,7 +425,7 @@ impl AgentServer {
|
||||||
};
|
};
|
||||||
let kind_label = loose_end_kind_label(kind);
|
let kind_label = loose_end_kind_label(kind);
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::CancelLooseEnd { kind, id })
|
.dispatch(hive_agent_sock::Request::CancelLooseEnd { kind, id })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(
|
format_ack(
|
||||||
|
|
@ -452,10 +452,10 @@ impl AgentServer {
|
||||||
let log = format!("{args:?}");
|
let log = format!("{args:?}");
|
||||||
run_tool_envelope("create_repo", log, async move {
|
run_tool_envelope("create_repo", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::CreateRepo { repo: args.repo })
|
.dispatch(hive_agent_sock::Request::CreateRepo { repo: args.repo })
|
||||||
.await;
|
.await;
|
||||||
let s = match resp {
|
let s = match resp {
|
||||||
Ok(hive_sh4re::Response::RepoCreated {
|
Ok(hive_agent_sock::Response::RepoCreated {
|
||||||
full_name,
|
full_name,
|
||||||
clone_url,
|
clone_url,
|
||||||
}) => format!("created repo {full_name} — clone: {clone_url}"),
|
}) => format!("created repo {full_name} — clone: {clone_url}"),
|
||||||
|
|
@ -495,7 +495,7 @@ impl AgentServer {
|
||||||
(None, Some(t)) => hive_sh4re::ReminderTiming::At { unix_timestamp: t },
|
(None, Some(t)) => hive_sh4re::ReminderTiming::At { unix_timestamp: t },
|
||||||
};
|
};
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Remind {
|
.dispatch(hive_agent_sock::Request::Remind {
|
||||||
message: args.message,
|
message: args.message,
|
||||||
timing,
|
timing,
|
||||||
file_path: args.file_path,
|
file_path: args.file_path,
|
||||||
|
|
@ -549,7 +549,7 @@ impl AgentServer {
|
||||||
let name = args.name.clone();
|
let name = args.name.clone();
|
||||||
run_tool_envelope("restart", log, async move {
|
run_tool_envelope("restart", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Restart { name: args.name })
|
.dispatch(hive_agent_sock::Request::Restart { name: args.name })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(resp, "restart", format!("restarted {name}")),
|
format_ack(resp, "restart", format!("restarted {name}")),
|
||||||
|
|
@ -571,7 +571,7 @@ impl AgentServer {
|
||||||
let name = args.name.clone();
|
let name = args.name.clone();
|
||||||
run_tool_envelope("kill", log, async move {
|
run_tool_envelope("kill", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Kill { name: args.name })
|
.dispatch(hive_agent_sock::Request::Kill { name: args.name })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(format_ack(resp, "kill", format!("killed {name}")), retries)
|
annotate_retries(format_ack(resp, "kill", format!("killed {name}")), retries)
|
||||||
})
|
})
|
||||||
|
|
@ -592,7 +592,7 @@ impl AgentServer {
|
||||||
let name = args.name.clone();
|
let name = args.name.clone();
|
||||||
run_tool_envelope("update", log, async move {
|
run_tool_envelope("update", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Update { name: args.name })
|
.dispatch(hive_agent_sock::Request::Update { name: args.name })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(resp, "update", format!("updated {name}")),
|
format_ack(resp, "update", format!("updated {name}")),
|
||||||
|
|
@ -614,9 +614,11 @@ impl AgentServer {
|
||||||
)]
|
)]
|
||||||
async fn list_containers(&self) -> String {
|
async fn list_containers(&self) -> String {
|
||||||
run_tool_envelope("list_containers", String::new(), async move {
|
run_tool_envelope("list_containers", String::new(), async move {
|
||||||
let (resp, retries) = self.dispatch(hive_sh4re::Request::ListDescendants).await;
|
let (resp, retries) = self
|
||||||
|
.dispatch(hive_agent_sock::Request::ListDescendants)
|
||||||
|
.await;
|
||||||
let body = match resp {
|
let body = match resp {
|
||||||
Ok(hive_sh4re::Response::Containers { containers }) => {
|
Ok(hive_agent_sock::Response::Containers { containers }) => {
|
||||||
if containers.is_empty() {
|
if containers.is_empty() {
|
||||||
"no descendant containers".to_owned()
|
"no descendant containers".to_owned()
|
||||||
} else {
|
} else {
|
||||||
|
|
@ -659,7 +661,7 @@ impl AgentServer {
|
||||||
let log = format!("{args:?}");
|
let log = format!("{args:?}");
|
||||||
run_tool_envelope("get_host_journal", log, async move {
|
run_tool_envelope("get_host_journal", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::GetHostJournal {
|
.dispatch(hive_agent_sock::Request::GetHostJournal {
|
||||||
unit: args.unit,
|
unit: args.unit,
|
||||||
container: args.container,
|
container: args.container,
|
||||||
lines: args.lines,
|
lines: args.lines,
|
||||||
|
|
@ -670,7 +672,7 @@ impl AgentServer {
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
let result = match resp {
|
let result = match resp {
|
||||||
Ok(hive_sh4re::Response::HostJournal { content }) => content,
|
Ok(hive_agent_sock::Response::HostJournal { content }) => content,
|
||||||
other => reply_err(other, "get_host_journal"),
|
other => reply_err(other, "get_host_journal"),
|
||||||
};
|
};
|
||||||
annotate_retries(result, retries)
|
annotate_retries(result, retries)
|
||||||
|
|
@ -699,7 +701,7 @@ impl AgentServer {
|
||||||
let name = args.name.clone();
|
let name = args.name.clone();
|
||||||
run_tool_envelope("request_init_config", log, async move {
|
run_tool_envelope("request_init_config", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::RequestInitConfig {
|
.dispatch(hive_agent_sock::Request::RequestInitConfig {
|
||||||
name: args.name,
|
name: args.name,
|
||||||
description: args.description,
|
description: args.description,
|
||||||
})
|
})
|
||||||
|
|
@ -727,7 +729,7 @@ impl AgentServer {
|
||||||
let name = args.name.clone();
|
let name = args.name.clone();
|
||||||
run_tool_envelope("start", log, async move {
|
run_tool_envelope("start", log, async move {
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::Start { name: args.name })
|
.dispatch(hive_agent_sock::Request::Start { name: args.name })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(resp, "start", format!("started {name}")),
|
format_ack(resp, "start", format!("started {name}")),
|
||||||
|
|
@ -750,13 +752,13 @@ impl AgentServer {
|
||||||
run_tool_envelope("get_logs", log, async move {
|
run_tool_envelope("get_logs", log, async move {
|
||||||
let lines = args.lines.map(|n| n.min(500));
|
let lines = args.lines.map(|n| n.min(500));
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::GetLogs {
|
.dispatch(hive_agent_sock::Request::GetLogs {
|
||||||
agent: agent.clone(),
|
agent: agent.clone(),
|
||||||
lines,
|
lines,
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
let s = match resp {
|
let s = match resp {
|
||||||
Ok(hive_sh4re::Response::Logs { content }) => {
|
Ok(hive_agent_sock::Response::Logs { content }) => {
|
||||||
if content.is_empty() {
|
if content.is_empty() {
|
||||||
format!("(no journal output for {agent})")
|
format!("(no journal output for {agent})")
|
||||||
} else {
|
} else {
|
||||||
|
|
@ -790,7 +792,7 @@ impl AgentServer {
|
||||||
args.inputs.join(", ")
|
args.inputs.join(", ")
|
||||||
};
|
};
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::RequestUpdateMetaInputs {
|
.dispatch(hive_agent_sock::Request::RequestUpdateMetaInputs {
|
||||||
inputs: args.inputs,
|
inputs: args.inputs,
|
||||||
description: args.description,
|
description: args.description,
|
||||||
})
|
})
|
||||||
|
|
@ -829,7 +831,7 @@ impl AgentServer {
|
||||||
run_tool_envelope("request_schedule_prompt", log, async move {
|
run_tool_envelope("request_schedule_prompt", log, async move {
|
||||||
let target_count = args.targets.len();
|
let target_count = args.targets.len();
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::RequestSchedulePrompt(
|
.dispatch(hive_agent_sock::Request::RequestSchedulePrompt(
|
||||||
hive_sh4re::SchedulePromptPayload {
|
hive_sh4re::SchedulePromptPayload {
|
||||||
targets: args.targets,
|
targets: args.targets,
|
||||||
body: args.body,
|
body: args.body,
|
||||||
|
|
@ -865,7 +867,7 @@ impl AgentServer {
|
||||||
run_tool_envelope("fire_schedule_now", log, async move {
|
run_tool_envelope("fire_schedule_now", log, async move {
|
||||||
let id = args.id;
|
let id = args.id;
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::FireScheduleNow { id })
|
.dispatch(hive_agent_sock::Request::FireScheduleNow { id })
|
||||||
.await;
|
.await;
|
||||||
annotate_retries(
|
annotate_retries(
|
||||||
format_ack(resp, "fire_schedule_now", format!("fired #{id} now")),
|
format_ack(resp, "fire_schedule_now", format!("fired #{id} now")),
|
||||||
|
|
@ -888,7 +890,7 @@ impl AgentServer {
|
||||||
run_tool_envelope("cancel_schedule", log, async move {
|
run_tool_envelope("cancel_schedule", log, async move {
|
||||||
let id = args.id;
|
let id = args.id;
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::CancelSchedule {
|
.dispatch(hive_agent_sock::Request::CancelSchedule {
|
||||||
id: args.id,
|
id: args.id,
|
||||||
targets: args.targets,
|
targets: args.targets,
|
||||||
})
|
})
|
||||||
|
|
@ -920,7 +922,7 @@ impl AgentServer {
|
||||||
run_tool_envelope("edit_schedule", log, async move {
|
run_tool_envelope("edit_schedule", log, async move {
|
||||||
let id = args.id;
|
let id = args.id;
|
||||||
let (resp, retries) = self
|
let (resp, retries) = self
|
||||||
.dispatch(hive_sh4re::Request::EditSchedule {
|
.dispatch(hive_agent_sock::Request::EditSchedule {
|
||||||
id: args.id,
|
id: args.id,
|
||||||
body: args.body,
|
body: args.body,
|
||||||
description: args.description.map(Some),
|
description: args.description.map(Some),
|
||||||
|
|
@ -947,9 +949,9 @@ impl AgentServer {
|
||||||
)]
|
)]
|
||||||
async fn list_schedules(&self) -> String {
|
async fn list_schedules(&self) -> String {
|
||||||
run_tool_envelope("list_schedules", String::new(), async move {
|
run_tool_envelope("list_schedules", String::new(), async move {
|
||||||
let (resp, retries) = self.dispatch(hive_sh4re::Request::ListSchedules).await;
|
let (resp, retries) = self.dispatch(hive_agent_sock::Request::ListSchedules).await;
|
||||||
let body = match resp {
|
let body = match resp {
|
||||||
Ok(hive_sh4re::Response::Schedules { schedules }) => {
|
Ok(hive_agent_sock::Response::Schedules { schedules }) => {
|
||||||
serde_json::to_string(&schedules)
|
serde_json::to_string(&schedules)
|
||||||
.unwrap_or_else(|e| format!("list_schedules: serialise: {e:#}"))
|
.unwrap_or_else(|e| format!("list_schedules: serialise: {e:#}"))
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -10,9 +10,12 @@
|
||||||
/// transport error: …"`. Handlers match their own happy-path variant and route
|
/// transport error: …"`. Handlers match their own happy-path variant and route
|
||||||
/// everything else here via a catch-all arm (`other => reply_err(other, tool)`),
|
/// everything else here via a catch-all arm (`other => reply_err(other, tool)`),
|
||||||
/// so the triplet lives in exactly one place.
|
/// so the triplet lives in exactly one place.
|
||||||
pub(super) fn reply_err(resp: Result<hive_sh4re::Response, anyhow::Error>, tool: &str) -> String {
|
pub(super) fn reply_err(
|
||||||
|
resp: Result<hive_agent_sock::Response, anyhow::Error>,
|
||||||
|
tool: &str,
|
||||||
|
) -> String {
|
||||||
match resp {
|
match resp {
|
||||||
Ok(hive_sh4re::Response::Err { message }) => format!("{tool} failed: {message}"),
|
Ok(hive_agent_sock::Response::Err { message }) => format!("{tool} failed: {message}"),
|
||||||
Ok(other) => format!("{tool} unexpected response: {other:?}"),
|
Ok(other) => format!("{tool} unexpected response: {other:?}"),
|
||||||
Err(e) => format!("{tool} transport error: {e:#}"),
|
Err(e) => format!("{tool} transport error: {e:#}"),
|
||||||
}
|
}
|
||||||
|
|
@ -23,12 +26,12 @@ pub(super) fn reply_err(resp: Result<hive_sh4re::Response, anyhow::Error>, tool:
|
||||||
/// behavior.
|
/// behavior.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn format_ack(
|
pub fn format_ack(
|
||||||
resp: Result<hive_sh4re::Response, anyhow::Error>,
|
resp: Result<hive_agent_sock::Response, anyhow::Error>,
|
||||||
tool: &str,
|
tool: &str,
|
||||||
ok_msg: String,
|
ok_msg: String,
|
||||||
) -> String {
|
) -> String {
|
||||||
match resp {
|
match resp {
|
||||||
Ok(hive_sh4re::Response::Ok) => ok_msg,
|
Ok(hive_agent_sock::Response::Ok) => ok_msg,
|
||||||
other => reply_err(other, tool),
|
other => reply_err(other, tool),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -44,9 +47,9 @@ pub fn format_ack(
|
||||||
/// so the model can tell where one ends and the next begins;
|
/// so the model can tell where one ends and the next begins;
|
||||||
/// per-message redelivery banners included.
|
/// per-message redelivery banners included.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn format_recv(resp: Result<hive_sh4re::Response, anyhow::Error>, waited: bool) -> String {
|
pub fn format_recv(resp: Result<hive_agent_sock::Response, anyhow::Error>, waited: bool) -> String {
|
||||||
match resp {
|
match resp {
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages,
|
messages,
|
||||||
remaining,
|
remaining,
|
||||||
}) => render_recv_messages(&messages, remaining, waited),
|
}) => render_recv_messages(&messages, remaining, waited),
|
||||||
|
|
@ -57,7 +60,7 @@ pub fn format_recv(resp: Result<hive_sh4re::Response, anyhow::Error>, waited: bo
|
||||||
// stop unmissably tells claude to flush + end. `remaining` is forced
|
// stop unmissably tells claude to flush + end. `remaining` is forced
|
||||||
// to 0 — the inbox is fenced, so a "N more pending" hint would be
|
// to 0 — the inbox is fenced, so a "N more pending" hint would be
|
||||||
// misleading.
|
// misleading.
|
||||||
Ok(hive_sh4re::Response::GracefulStop) => {
|
Ok(hive_agent_sock::Response::GracefulStop) => {
|
||||||
render_recv_messages(&[graceful_stop_message()], 0, waited)
|
render_recv_messages(&[graceful_stop_message()], 0, waited)
|
||||||
}
|
}
|
||||||
other => reply_err(other, "recv"),
|
other => reply_err(other, "recv"),
|
||||||
|
|
@ -354,9 +357,9 @@ pub(super) fn loose_end_kind_label(kind: hive_sh4re::CancelLooseEndKind) -> &'st
|
||||||
/// `running: no` line tells the caller WHY. See
|
/// `running: no` line tells the caller WHY. See
|
||||||
/// `docs/turn-loop/mcp.md::Core tools` (`get_agent_meta`).
|
/// `docs/turn-loop/mcp.md::Core tools` (`get_agent_meta`).
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn format_agent_meta(resp: Result<hive_sh4re::Response, anyhow::Error>) -> String {
|
pub fn format_agent_meta(resp: Result<hive_agent_sock::Response, anyhow::Error>) -> String {
|
||||||
match resp {
|
match resp {
|
||||||
Ok(hive_sh4re::Response::AgentMeta {
|
Ok(hive_agent_sock::Response::AgentMeta {
|
||||||
name,
|
name,
|
||||||
running,
|
running,
|
||||||
hyperhive_rev,
|
hyperhive_rev,
|
||||||
|
|
@ -477,7 +480,7 @@ mod tests {
|
||||||
#[test]
|
#[test]
|
||||||
fn empty_recv_after_wait_appends_idle_hint() {
|
fn empty_recv_after_wait_appends_idle_hint() {
|
||||||
let out = format_recv(
|
let out = format_recv(
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages: vec![],
|
messages: vec![],
|
||||||
remaining: 0,
|
remaining: 0,
|
||||||
}),
|
}),
|
||||||
|
|
@ -490,7 +493,7 @@ mod tests {
|
||||||
#[test]
|
#[test]
|
||||||
fn empty_recv_without_wait_has_no_hint() {
|
fn empty_recv_without_wait_has_no_hint() {
|
||||||
let out = format_recv(
|
let out = format_recv(
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages: vec![],
|
messages: vec![],
|
||||||
remaining: 0,
|
remaining: 0,
|
||||||
}),
|
}),
|
||||||
|
|
@ -502,7 +505,7 @@ mod tests {
|
||||||
#[test]
|
#[test]
|
||||||
fn single_recv_with_remaining_appends_pending_hint() {
|
fn single_recv_with_remaining_appends_pending_hint() {
|
||||||
let out = format_recv(
|
let out = format_recv(
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages: vec![msg(7, "alice", "hi")],
|
messages: vec![msg(7, "alice", "hi")],
|
||||||
remaining: 3,
|
remaining: 3,
|
||||||
}),
|
}),
|
||||||
|
|
@ -516,7 +519,7 @@ mod tests {
|
||||||
#[test]
|
#[test]
|
||||||
fn single_recv_no_remaining_has_no_pending_hint() {
|
fn single_recv_no_remaining_has_no_pending_hint() {
|
||||||
let out = format_recv(
|
let out = format_recv(
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages: vec![msg(7, "alice", "hi")],
|
messages: vec![msg(7, "alice", "hi")],
|
||||||
remaining: 0,
|
remaining: 0,
|
||||||
}),
|
}),
|
||||||
|
|
@ -528,7 +531,7 @@ mod tests {
|
||||||
#[test]
|
#[test]
|
||||||
fn batch_recv_with_remaining_appends_pending_hint_once() {
|
fn batch_recv_with_remaining_appends_pending_hint_once() {
|
||||||
let out = format_recv(
|
let out = format_recv(
|
||||||
Ok(hive_sh4re::Response::Messages {
|
Ok(hive_agent_sock::Response::Messages {
|
||||||
messages: vec![msg(7, "alice", "hi"), msg(8, "bob", "yo")],
|
messages: vec![msg(7, "alice", "hi"), msg(8, "bob", "yo")],
|
||||||
remaining: 9,
|
remaining: 9,
|
||||||
}),
|
}),
|
||||||
|
|
|
||||||
11
hive-agent-sock/Cargo.toml
Normal file
11
hive-agent-sock/Cargo.toml
Normal file
|
|
@ -0,0 +1,11 @@
|
||||||
|
[package]
|
||||||
|
name = "hive-agent-sock"
|
||||||
|
edition.workspace = true
|
||||||
|
version.workspace = true
|
||||||
|
|
||||||
|
[lints]
|
||||||
|
workspace = true
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
hive-sh4re.workspace = true
|
||||||
|
serde.workspace = true
|
||||||
414
hive-agent-sock/src/lib.rs
Normal file
414
hive-agent-sock/src/lib.rs
Normal file
|
|
@ -0,0 +1,414 @@
|
||||||
|
//! Per-agent + manager socket wire types (`/run/hive/mcp.sock`).
|
||||||
|
//!
|
||||||
|
//! The unified `Request` / `Response` protocol spoken between an agent's
|
||||||
|
//! in-container harness (and the manager) and the `hive-c0re` daemon over the
|
||||||
|
//! per-agent mcp socket. Re-homed out of `hive-sh4re` so this one socket owns
|
||||||
|
//! its own protocol crate, mirroring `hive-host-sock` / `hive-priv-sock`. The
|
||||||
|
//! shared payload types it references (`Message`, `LooseEnd`, `Approval`, …)
|
||||||
|
//! stay in `hive-sh4re`, which this crate depends on.
|
||||||
|
|
||||||
|
use hive_sh4re::{
|
||||||
|
CancelLooseEndKind, ContainerInfo, DeliveredMessage, InboxRow, JournalPriority, LooseEnd,
|
||||||
|
MatrixIdentity, ReminderStats, ReminderTiming, SchedulePromptPayload, WireSchedule,
|
||||||
|
};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
/// serde `default` helper for `Response::AgentMeta::running` (absent = true).
|
||||||
|
fn default_true() -> bool {
|
||||||
|
true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unified request enum for both agent and manager sockets. The agent's
|
||||||
|
/// identity is the socket it arrived on. Privileged variants are marked
|
||||||
|
/// `*(privileged)*` — an agent socket returns `Err` for them server-side.
|
||||||
|
///
|
||||||
|
/// `AgentRequest` and `ManagerRequest` are type aliases for this enum;
|
||||||
|
/// existing callers continue to compile unchanged.
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
#[serde(tag = "cmd", rename_all = "snake_case")]
|
||||||
|
pub enum Request {
|
||||||
|
/// Send a message to another agent.
|
||||||
|
Send {
|
||||||
|
to: String,
|
||||||
|
body: String,
|
||||||
|
/// Optional id of the message being replied to. Stored in the
|
||||||
|
/// broker DB and returned on `Recv` so the dashboard can render
|
||||||
|
/// threads. Ignored if the id is unknown or out of retention.
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
in_reply_to: Option<i64>,
|
||||||
|
},
|
||||||
|
/// Pop pending messages from this agent's inbox.
|
||||||
|
/// Delivery + ack cycle: see
|
||||||
|
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
||||||
|
Recv {
|
||||||
|
#[serde(default)]
|
||||||
|
wait_seconds: Option<u64>,
|
||||||
|
#[serde(default)]
|
||||||
|
max: Option<u32>,
|
||||||
|
},
|
||||||
|
/// Non-mutating: how many pending messages are addressed to me?
|
||||||
|
/// Used by the harness to render a status line after each tool call.
|
||||||
|
Status,
|
||||||
|
/// Operator-injected message TO this agent (from this agent's own web
|
||||||
|
/// UI). Recipient is implicit — `from` is `"operator"`. Effectively the
|
||||||
|
/// per-agent equivalent of the old dashboard T4LK form, but scoped to
|
||||||
|
/// the agent whose page the operator is on.
|
||||||
|
OperatorMsg { body: String },
|
||||||
|
/// Wake-up event injected from inside the container. Recipient is
|
||||||
|
/// implicit (this agent); `from` is caller-chosen. See
|
||||||
|
/// `docs/conventions.md::Wake injection` for the trust model and
|
||||||
|
/// typical callers. The wake is persisted in the sqlite broker
|
||||||
|
/// like any other message — the agent can ack it via `AckUntil`.
|
||||||
|
Wake { from: String, body: String },
|
||||||
|
/// Last `limit` messages addressed to this agent, newest-first.
|
||||||
|
/// Non-mutating — pulls from the broker without delivering. The
|
||||||
|
/// per-agent web UI uses this to render its own inbox section.
|
||||||
|
Recent { limit: u64 },
|
||||||
|
/// Surface a question to either the operator or another agent.
|
||||||
|
/// Routing + shape: see
|
||||||
|
/// `docs/conventions.md::Question routing (Ask / Answer)`.
|
||||||
|
Ask {
|
||||||
|
question: String,
|
||||||
|
#[serde(default)]
|
||||||
|
options: Vec<String>,
|
||||||
|
#[serde(default)]
|
||||||
|
multi: bool,
|
||||||
|
#[serde(default)]
|
||||||
|
ttl_seconds: Option<u64>,
|
||||||
|
#[serde(default)]
|
||||||
|
to: Option<String>,
|
||||||
|
},
|
||||||
|
/// Answer a question previously routed to this agent via
|
||||||
|
/// `HelperEvent::QuestionAsked`. Authorised callers + threading
|
||||||
|
/// back via `HelperEvent::QuestionAnswered`: see
|
||||||
|
/// `docs/conventions.md::Question routing (Ask / Answer)`.
|
||||||
|
Answer { id: i64, answer: String },
|
||||||
|
/// Schedule a reminder message to be delivered to this agent at a
|
||||||
|
/// future time. The reminder lands in the agent's inbox as an auto-sent
|
||||||
|
/// message from `"reminder"`. Use for agent follow-ups (e.g. check task
|
||||||
|
/// status, retry failed operation). Message length is limited; pass
|
||||||
|
/// `file_path` to store in a file and get a path-reference message
|
||||||
|
/// instead.
|
||||||
|
Remind {
|
||||||
|
message: String,
|
||||||
|
timing: ReminderTiming,
|
||||||
|
#[serde(default)]
|
||||||
|
file_path: Option<String>,
|
||||||
|
},
|
||||||
|
/// Loose-ends view. On the agent socket: `None` = self; direct
|
||||||
|
/// children are always accessible; non-children require the
|
||||||
|
/// `query_agent_state` capability — rejected with an error otherwise;
|
||||||
|
/// `"*"` is always rejected (use the manager socket). On the manager
|
||||||
|
/// socket: `None` = manager self, `"*"` = hive-wide, any name =
|
||||||
|
/// that agent. See `docs/conventions.md::Loose-ends wire shape`.
|
||||||
|
GetLooseEnds {
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
agent: Option<String>,
|
||||||
|
},
|
||||||
|
/// Upsert a *todo* (loose-ends v2) from an in-container
|
||||||
|
/// subsystem (matrix / forge / bash). `subsystem` is the producer
|
||||||
|
/// marker; `key` is the optional subsystem-specific dedup key (a
|
||||||
|
/// matrix room id, a bash task id). Re-pushing an identical keyed
|
||||||
|
/// todo is a no-op; a new-or-changed one coalesces a wake to the
|
||||||
|
/// agent. Keyless todos always insert as one-offs.
|
||||||
|
UpsertTodo {
|
||||||
|
subsystem: String,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
key: Option<String>,
|
||||||
|
summary: String,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
source: Option<String>,
|
||||||
|
},
|
||||||
|
/// Clear producer-resolved todo(s) by `(subsystem, key)`. `key =
|
||||||
|
/// Some(k)` clears the one keyed row; `key = None` clears **all** of
|
||||||
|
/// the subsystem's keyless todos (rows with no key can't be told
|
||||||
|
/// apart — clear a specific one via `MarkTodoDone` by id). `all =
|
||||||
|
/// true` wipes the producer's whole set (cancel-and-recreate on
|
||||||
|
/// daemon restart).
|
||||||
|
ClearTodo {
|
||||||
|
subsystem: String,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
key: Option<String>,
|
||||||
|
#[serde(default)]
|
||||||
|
all: bool,
|
||||||
|
},
|
||||||
|
/// List todos, optionally filtered to one `subsystem` (a producer
|
||||||
|
/// enumerating its own set). `None` = all.
|
||||||
|
ListTodos {
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
subsystem: Option<String>,
|
||||||
|
},
|
||||||
|
/// The agent marks one of its own todos done, by id.
|
||||||
|
MarkTodoDone { id: i64 },
|
||||||
|
/// Count of pending (un-delivered) reminders. On the agent socket:
|
||||||
|
/// same target rules as `GetLooseEnds` (self/children free;
|
||||||
|
/// non-children require `query_agent_state`; `"*"` rejected).
|
||||||
|
/// On the manager socket: `None` = self, any name = that agent.
|
||||||
|
/// Used by the harness's per-turn stats sink.
|
||||||
|
CountPendingReminders {
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
agent: Option<String>,
|
||||||
|
},
|
||||||
|
/// Reminder statistics: counts of scheduled, delivered, and pending
|
||||||
|
/// reminders over a time window. `since_secs` filters to reminders
|
||||||
|
/// created in the last N seconds (0 = all). On the agent socket:
|
||||||
|
/// same target rules as `GetLooseEnds` (self/children free;
|
||||||
|
/// non-children require `query_agent_state`; `"*"` rejected).
|
||||||
|
/// On the manager socket: `None` = self, any name = that agent.
|
||||||
|
ReminderRollup {
|
||||||
|
/// Only count reminders created in the last N seconds from now.
|
||||||
|
/// Pass 0 to include all reminders.
|
||||||
|
#[serde(default)]
|
||||||
|
since_secs: u64,
|
||||||
|
/// Whose reminders to roll up. `None` = the caller's own.
|
||||||
|
/// `Some("<name>")` = that agent's (requires `query_agent_state`
|
||||||
|
/// capability on the agent socket; always available on the manager
|
||||||
|
/// socket).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
agent: Option<String>,
|
||||||
|
},
|
||||||
|
/// Set a free-text status string visible on the dashboard. The harness
|
||||||
|
/// writes `{state_dir}/hyperhive-status` locally before sending this
|
||||||
|
/// request; hive-c0re just triggers a dashboard rescan on receipt.
|
||||||
|
/// Pass an empty string to clear the status.
|
||||||
|
SetStatus { text: String },
|
||||||
|
/// Fetch identity + status for an agent. `name = None` =
|
||||||
|
/// self-introspection; `Some(<agent>)` = target query. See
|
||||||
|
/// `docs/conventions.md::Agent metadata`.
|
||||||
|
GetAgentMeta {
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
name: Option<String>,
|
||||||
|
},
|
||||||
|
/// Cancel an open thread the agent owns. Authorisation +
|
||||||
|
/// per-kind semantics in
|
||||||
|
/// `docs/conventions.md::Loose-ends wire shape`.
|
||||||
|
CancelLooseEnd { kind: CancelLooseEndKind, id: i64 },
|
||||||
|
/// Create a git repo *through hive-c0re*. Agents can't create
|
||||||
|
/// repos with their own forge token (`max_repo_creation = 0`); this is
|
||||||
|
/// the sanctioned path. hive-c0re creates `repo` in the c0re-owned
|
||||||
|
/// `agents` org, adds the calling agent as a write collaborator (not
|
||||||
|
/// owner), and applies operator-team branch protection so the author
|
||||||
|
/// can't merge its own PRs. Returns the new repo's full name.
|
||||||
|
CreateRepo { repo: String },
|
||||||
|
/// Mark every message popped since the last `AckTurn` as handled.
|
||||||
|
/// Harness↔broker pairing fired after `TurnOutcome::Ok`. See
|
||||||
|
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
||||||
|
AckTurn,
|
||||||
|
/// Mark every inbox message with broker row id `<= up_to` as
|
||||||
|
/// handled (`acked_at` set), whether still pending or already
|
||||||
|
/// delivered. The agent-facing bulk-triage escape hatch for a
|
||||||
|
/// redelivered / accumulated backlog: instead of popping and
|
||||||
|
/// re-reading dozens of already-handled messages one turn at a
|
||||||
|
/// time, the agent acks everything up to the id it has seen.
|
||||||
|
/// Recipient-scoped — an agent can only ack its own rows. See
|
||||||
|
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
||||||
|
AckUntil { up_to: i64 },
|
||||||
|
/// Requeue every popped-but-unacked message back into the inbox.
|
||||||
|
/// Harness fires this once at boot to recover from
|
||||||
|
/// crashed-mid-turn sessions. See
|
||||||
|
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
||||||
|
RequeueInflight,
|
||||||
|
/// Harness → c0re: "I saw the `GracefulStop` signal, ran my
|
||||||
|
/// stop-checkpoint turn (durable `/state` flushed) and am exiting my
|
||||||
|
/// serve loop now." Lets the `GracefulStop` orchestration stop the
|
||||||
|
/// container immediately instead of waiting out its timeout fallback.
|
||||||
|
GracefulStopComplete,
|
||||||
|
/// *(capability-gated: `read_host_journal`)* Fetch recent lines
|
||||||
|
/// from the host journal. Filters are all optional; omitting all
|
||||||
|
/// returns the last `lines` entries from the global journal.
|
||||||
|
GetHostJournal {
|
||||||
|
/// Filter to a specific systemd unit (e.g. `hive-c0re.service`).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
unit: Option<String>,
|
||||||
|
/// Machine name to pass to journalctl `-M` verbatim (e.g. `h-iris`).
|
||||||
|
/// The caller is responsible for the correct nspawn machine name.
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
container: Option<String>,
|
||||||
|
/// Number of journal lines to return (default 30, max 100).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
lines: Option<u32>,
|
||||||
|
/// Minimum syslog priority level.
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
priority: Option<JournalPriority>,
|
||||||
|
/// Regex to match against log message fields (journalctl `--grep`).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
grep: Option<String>,
|
||||||
|
/// Show entries on or newer than this timestamp (journalctl `--since`).
|
||||||
|
/// ISO 8601 or journalctl-accepted relative strings (e.g. `"-1h"`).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
since: Option<String>,
|
||||||
|
/// Show entries on or older than this timestamp (journalctl `--until`).
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
until: Option<String>,
|
||||||
|
},
|
||||||
|
|
||||||
|
// ---- privileged (manager socket only for now) ---------------------------
|
||||||
|
/// *(privileged)* Initialise a brand-new agent's proposed config repo
|
||||||
|
/// and queue an approval for the operator to review.
|
||||||
|
RequestInitConfig {
|
||||||
|
name: String,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
description: Option<String>,
|
||||||
|
},
|
||||||
|
/// *(privileged)* Stop a sub-agent (graceful).
|
||||||
|
Kill { name: String },
|
||||||
|
/// *(privileged)* Start a previously-stopped sub-agent container.
|
||||||
|
Start { name: String },
|
||||||
|
/// *(privileged)* Restart a sub-agent container (stop + start).
|
||||||
|
Restart { name: String },
|
||||||
|
/// *(privileged)* Rebuild a sub-agent against the current hyperhive
|
||||||
|
/// flake + agent.nix. No approval required.
|
||||||
|
Update { name: String },
|
||||||
|
/// *(privileged)* Fetch recent journal lines for a sub-agent container.
|
||||||
|
GetLogs {
|
||||||
|
agent: String,
|
||||||
|
#[serde(default)]
|
||||||
|
lines: Option<u32>,
|
||||||
|
},
|
||||||
|
/// *(privileged)* Queue an approval to run `nix flake update [inputs...]`.
|
||||||
|
RequestUpdateMetaInputs {
|
||||||
|
#[serde(default)]
|
||||||
|
inputs: Vec<String>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
description: Option<String>,
|
||||||
|
},
|
||||||
|
/// *(privileged)* Queue an approval to add a scheduled prompt.
|
||||||
|
RequestSchedulePrompt(SchedulePromptPayload),
|
||||||
|
/// *(privileged)* Cancel a scheduled prompt.
|
||||||
|
CancelSchedule {
|
||||||
|
id: i64,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
targets: Option<Vec<String>>,
|
||||||
|
},
|
||||||
|
/// *(privileged)* List every schedule in the queue.
|
||||||
|
ListSchedules,
|
||||||
|
/// List all containers that are topological descendants of the calling
|
||||||
|
/// agent (direct children + their subtrees). Scoped to the caller's
|
||||||
|
/// subtree; gated by the `lifecycle` tool group. The result includes all
|
||||||
|
/// known descendants regardless of whether the container is currently
|
||||||
|
/// running — use `running` to distinguish.
|
||||||
|
ListDescendants,
|
||||||
|
/// *(privileged)* Fire a scheduled prompt out of band immediately.
|
||||||
|
FireScheduleNow { id: i64 },
|
||||||
|
/// *(privileged)* Edit an existing schedule's mutable fields.
|
||||||
|
EditSchedule {
|
||||||
|
id: i64,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
body: Option<String>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
description: Option<Option<String>>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
interval_seconds: Option<Option<u64>>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
next_fire_at_unix: Option<i64>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
targets_add: Option<Vec<String>>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
targets_remove: Option<Vec<String>>,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Backwards-compatible aliases. Both sockets now speak the unified `Request`
|
||||||
|
/// / `Response` wire; the server-side privilege gate rejects privileged
|
||||||
|
/// variants on agent sockets with `Err { message: "privileged variant..." }`.
|
||||||
|
pub type AgentRequest = Request;
|
||||||
|
pub type ManagerRequest = Request;
|
||||||
|
|
||||||
|
/// Unified response enum for both agent and manager sockets. Privileged
|
||||||
|
/// variants (`Logs`, `Schedules`) are never returned on agent sockets.
|
||||||
|
///
|
||||||
|
/// `AgentResponse` and `ManagerResponse` are type aliases for this enum.
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
#[serde(tag = "kind", rename_all = "snake_case")]
|
||||||
|
pub enum Response {
|
||||||
|
/// `Send` succeeded.
|
||||||
|
Ok,
|
||||||
|
/// Either `Send` failed or `Recv` errored.
|
||||||
|
Err { message: String },
|
||||||
|
/// `Recv` result: zero or more messages, FIFO-ordered, never
|
||||||
|
/// longer than the `max` the caller passed. Empty vec = nothing
|
||||||
|
/// pending (the "(empty)" path for the formatter). Per-row `id` +
|
||||||
|
/// `redelivered` carry the broker's row id (tracked by the harness
|
||||||
|
/// for `AckTurn`, and surfaced to claude as a `[msg #<id>]` marker
|
||||||
|
/// so `AckUntil` has something to reference) and the "previously
|
||||||
|
/// popped, not acked" flag — see `DeliveredMessage` for details.
|
||||||
|
/// `remaining` is the inbox depth *after* this batch was popped —
|
||||||
|
/// how many still-pending messages the caller could drain next. The
|
||||||
|
/// harness surfaces it to claude ("N more pending") so an in-turn
|
||||||
|
/// `recv` learns whether the inbox is drained, mirroring the count
|
||||||
|
/// the wake prompt already carries.
|
||||||
|
Messages {
|
||||||
|
messages: Vec<DeliveredMessage>,
|
||||||
|
remaining: u64,
|
||||||
|
},
|
||||||
|
/// `Status` result: how many pending messages are in this agent's inbox.
|
||||||
|
Status { unread: u64 },
|
||||||
|
/// `AckUntil` result: how many rows were newly marked handled.
|
||||||
|
Acked { count: u64 },
|
||||||
|
/// `Recent` result: newest-first inbox rows.
|
||||||
|
Recent { rows: Vec<InboxRow> },
|
||||||
|
/// `Ask` result: the queued question id. The answer lands later
|
||||||
|
/// as `HelperEvent::QuestionAnswered` in this agent's inbox.
|
||||||
|
QuestionQueued { id: i64 },
|
||||||
|
/// `GetLooseEnds` result: list of loose ends pending against
|
||||||
|
/// this agent. Ordered newest-first within each kind.
|
||||||
|
LooseEnds { loose_ends: Vec<LooseEnd> },
|
||||||
|
/// `CountPendingReminders` result.
|
||||||
|
PendingRemindersCount { count: u64 },
|
||||||
|
/// `ReminderRollup` result: reminder activity stats for the agent.
|
||||||
|
ReminderRollup(ReminderStats),
|
||||||
|
/// `GetAgentMeta` result. Per-field semantics + serde defaults
|
||||||
|
/// live in `docs/conventions.md::Agent metadata`.
|
||||||
|
AgentMeta {
|
||||||
|
name: String,
|
||||||
|
#[serde(default = "default_true")]
|
||||||
|
running: bool,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
hyperhive_rev: Option<String>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
status_text: Option<String>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
status_set_at: Option<i64>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
hive_name: Option<String>,
|
||||||
|
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||||
|
swarm_name: Option<String>,
|
||||||
|
/// Matrix identities this agent can act as (one per configured +
|
||||||
|
/// live account). Empty for agents with no matrix provisioning.
|
||||||
|
#[serde(default, skip_serializing_if = "Vec::is_empty")]
|
||||||
|
matrix_accounts: Vec<MatrixIdentity>,
|
||||||
|
},
|
||||||
|
/// `GetLogs` result: journal lines for the requested container.
|
||||||
|
/// Returned on the manager socket only.
|
||||||
|
Logs { content: String },
|
||||||
|
/// `GetHostJournal` result: host journal lines matching the
|
||||||
|
/// requested filters. Returned on the agent socket when the agent
|
||||||
|
/// holds the `read_host_journal` capability.
|
||||||
|
HostJournal { content: String },
|
||||||
|
/// `ListSchedules` result. Snapshot of every schedule.
|
||||||
|
/// Returned on the manager socket only.
|
||||||
|
Schedules { schedules: Vec<WireSchedule> },
|
||||||
|
/// `CreateRepo` result: the new repo's full name (`agents/<repo>`)
|
||||||
|
/// and clone URL, so the agent can immediately `git clone` it.
|
||||||
|
RepoCreated {
|
||||||
|
full_name: String,
|
||||||
|
clone_url: String,
|
||||||
|
},
|
||||||
|
/// `ListDescendants` result: all descendant containers, with running
|
||||||
|
/// status. Ordered by topology depth (parents before children), then
|
||||||
|
/// alphabetically within each depth tier.
|
||||||
|
Containers { containers: Vec<ContainerInfo> },
|
||||||
|
/// `Recv` result when a graceful stop is pending for this agent
|
||||||
|
/// (set by hive-c0re's `GracefulStop` orchestration). Returned in
|
||||||
|
/// place of `Messages` — it doubles as the inbound fence: the harness
|
||||||
|
/// stops consuming normal inbox messages and instead runs one
|
||||||
|
/// stop-checkpoint turn (flush durable `/state`), then reports
|
||||||
|
/// `GracefulStopComplete` and exits its serve loop so the container
|
||||||
|
/// can be stopped cleanly. New sends keep queueing in the broker for
|
||||||
|
/// the agent's next start.
|
||||||
|
GracefulStop,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Backwards-compatible response aliases.
|
||||||
|
pub type AgentResponse = Response;
|
||||||
|
pub type ManagerResponse = Response;
|
||||||
|
|
@ -13,7 +13,7 @@ workspace = true
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow.workspace = true
|
anyhow.workspace = true
|
||||||
clap.workspace = true
|
clap.workspace = true
|
||||||
hive-sh4re.workspace = true
|
hive-agent-sock.workspace = true
|
||||||
serde.workspace = true
|
serde.workspace = true
|
||||||
serde_json.workspace = true
|
serde_json.workspace = true
|
||||||
tokio.workspace = true
|
tokio.workspace = true
|
||||||
|
|
|
||||||
|
|
@ -13,7 +13,7 @@ use std::path::PathBuf;
|
||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use hive_sh4re::{AgentRequest, AgentResponse};
|
use hive_agent_sock::{AgentRequest, AgentResponse};
|
||||||
|
|
||||||
/// Per-agent MCP socket, bind-mounted from the host into every container.
|
/// Per-agent MCP socket, bind-mounted from the host into every container.
|
||||||
const DEFAULT_SOCKET: &str = "/run/hive/mcp.sock";
|
const DEFAULT_SOCKET: &str = "/run/hive/mcp.sock";
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ time.workspace = true
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
clap.workspace = true
|
clap.workspace = true
|
||||||
hive-claude.workspace = true
|
hive-claude.workspace = true
|
||||||
|
hive-agent-sock.workspace = true
|
||||||
hive-sh4re.workspace = true
|
hive-sh4re.workspace = true
|
||||||
rmcp.workspace = true
|
rmcp.workspace = true
|
||||||
rusqlite.workspace = true
|
rusqlite.workspace = true
|
||||||
|
|
|
||||||
|
|
@ -1083,11 +1083,11 @@ async fn poll_once(
|
||||||
continue;
|
continue;
|
||||||
};
|
};
|
||||||
|
|
||||||
let req = hive_sh4re::Request::Wake {
|
let req = hive_agent_sock::Request::Wake {
|
||||||
from: "forge".to_owned(),
|
from: "forge".to_owned(),
|
||||||
body,
|
body,
|
||||||
};
|
};
|
||||||
let deliver_result = crate::client::request::<_, hive_sh4re::Response>(socket, &req)
|
let deliver_result = crate::client::request::<_, hive_agent_sock::Response>(socket, &req)
|
||||||
.await
|
.await
|
||||||
.map(|_| ());
|
.map(|_| ());
|
||||||
match deliver_result {
|
match deliver_result {
|
||||||
|
|
|
||||||
|
|
@ -42,7 +42,8 @@ use crate::login::LoginState;
|
||||||
use crate::turn_stats::TurnStats;
|
use crate::turn_stats::TurnStats;
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use hive_sh4re::{AgentRequest, AgentResponse, HelperEvent, SYSTEM_SENDER};
|
use hive_agent_sock::{AgentRequest, AgentResponse};
|
||||||
|
use hive_sh4re::{HelperEvent, SYSTEM_SENDER};
|
||||||
|
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
#[command(name = "hive-agent", about = "hyperhive harness serve loop")]
|
#[command(name = "hive-agent", about = "hyperhive harness serve loop")]
|
||||||
|
|
|
||||||
|
|
@ -23,15 +23,20 @@ pub(super) async fn post_send(
|
||||||
if body.is_empty() {
|
if body.is_empty() {
|
||||||
return error_response(StatusCode::BAD_REQUEST, "send: `body` required");
|
return error_response(StatusCode::BAD_REQUEST, "send: `body` required");
|
||||||
}
|
}
|
||||||
match super::broker_request(&state.socket, &hive_sh4re::Request::OperatorMsg { body }).await {
|
match super::broker_request(
|
||||||
|
&state.socket,
|
||||||
|
&hive_agent_sock::Request::OperatorMsg { body },
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
{
|
||||||
// 200 instead of 303 → the client doesn't refetch /api/state.
|
// 200 instead of 303 → the client doesn't refetch /api/state.
|
||||||
// The operator message becomes a broker `Sent` (already shown
|
// The operator message becomes a broker `Sent` (already shown
|
||||||
// server-side in the dashboard); on the agent side, the
|
// server-side in the dashboard); on the agent side, the
|
||||||
// resulting `TurnStart` SSE event drives the terminal + the
|
// resulting `TurnStart` SSE event drives the terminal + the
|
||||||
// inbox row gets consumed by the time `TurnEnd` fires the
|
// inbox row gets consumed by the time `TurnEnd` fires the
|
||||||
// existing turn-end refresh.
|
// existing turn-end refresh.
|
||||||
Ok(hive_sh4re::Response::Ok) => (axum::http::StatusCode::OK, "ok").into_response(),
|
Ok(hive_agent_sock::Response::Ok) => (axum::http::StatusCode::OK, "ok").into_response(),
|
||||||
Ok(hive_sh4re::Response::Err { message }) => error_response(
|
Ok(hive_agent_sock::Response::Err { message }) => error_response(
|
||||||
StatusCode::INTERNAL_SERVER_ERROR,
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
&format!("send failed: {message}"),
|
&format!("send failed: {message}"),
|
||||||
),
|
),
|
||||||
|
|
|
||||||
|
|
@ -302,11 +302,11 @@ enum BrokerError {
|
||||||
/// web-UI handler that talks to the broker goes through it.
|
/// web-UI handler that talks to the broker goes through it.
|
||||||
async fn broker_request(
|
async fn broker_request(
|
||||||
socket: &Path,
|
socket: &Path,
|
||||||
req: &hive_sh4re::Request,
|
req: &hive_agent_sock::Request,
|
||||||
) -> std::result::Result<hive_sh4re::Response, BrokerError> {
|
) -> std::result::Result<hive_agent_sock::Response, BrokerError> {
|
||||||
match tokio::time::timeout(
|
match tokio::time::timeout(
|
||||||
SOCKET_FETCH_TIMEOUT,
|
SOCKET_FETCH_TIMEOUT,
|
||||||
crate::client::request::<_, hive_sh4re::Response>(socket, req),
|
crate::client::request::<_, hive_agent_sock::Response>(socket, req),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
|
|
|
||||||
|
|
@ -366,8 +366,8 @@ async fn recent_inbox(socket: &std::path::Path) -> Vec<hive_sh4re::InboxRow> {
|
||||||
const LIMIT: u64 = 30;
|
const LIMIT: u64 = 30;
|
||||||
// Deadline-bounded (via `broker_request`): `/api/state` must render even
|
// Deadline-bounded (via `broker_request`): `/api/state` must render even
|
||||||
// when hive-c0re is busy — an empty inbox section beats a hung snapshot.
|
// when hive-c0re is busy — an empty inbox section beats a hung snapshot.
|
||||||
match super::broker_request(socket, &hive_sh4re::Request::Recent { limit: LIMIT }).await {
|
match super::broker_request(socket, &hive_agent_sock::Request::Recent { limit: LIMIT }).await {
|
||||||
Ok(hive_sh4re::Response::Recent { rows }) => rows,
|
Ok(hive_agent_sock::Response::Recent { rows }) => rows,
|
||||||
_ => Vec::new(),
|
_ => Vec::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -35,14 +35,14 @@ async fn fetch_reminder_stats(
|
||||||
) -> Option<hive_sh4re::ReminderStats> {
|
) -> Option<hive_sh4re::ReminderStats> {
|
||||||
match super::broker_request(
|
match super::broker_request(
|
||||||
socket,
|
socket,
|
||||||
&hive_sh4re::Request::ReminderRollup {
|
&hive_agent_sock::Request::ReminderRollup {
|
||||||
since_secs: window_secs,
|
since_secs: window_secs,
|
||||||
agent: None,
|
agent: None,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(hive_sh4re::Response::ReminderRollup(stats)) => Some(stats),
|
Ok(hive_agent_sock::Response::ReminderRollup(stats)) => Some(stats),
|
||||||
_ => None,
|
_ => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -57,14 +57,14 @@ async fn fetch_reminder_stats(
|
||||||
pub(super) async fn api_loose_ends(State(state): State<AppState>) -> Response {
|
pub(super) async fn api_loose_ends(State(state): State<AppState>) -> Response {
|
||||||
match super::broker_request(
|
match super::broker_request(
|
||||||
&state.socket,
|
&state.socket,
|
||||||
&hive_sh4re::Request::GetLooseEnds { agent: None },
|
&hive_agent_sock::Request::GetLooseEnds { agent: None },
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(hive_sh4re::Response::LooseEnds { loose_ends }) => {
|
Ok(hive_agent_sock::Response::LooseEnds { loose_ends }) => {
|
||||||
axum::Json(serde_json::json!({ "loose_ends": loose_ends })).into_response()
|
axum::Json(serde_json::json!({ "loose_ends": loose_ends })).into_response()
|
||||||
}
|
}
|
||||||
Ok(hive_sh4re::Response::Err { message }) => error_response(
|
Ok(hive_agent_sock::Response::Err { message }) => error_response(
|
||||||
StatusCode::INTERNAL_SERVER_ERROR,
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
&format!("get_loose_ends: {message}"),
|
&format!("get_loose_ends: {message}"),
|
||||||
),
|
),
|
||||||
|
|
|
||||||
|
|
@ -8,6 +8,7 @@ workspace = true
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow.workspace = true
|
anyhow.workspace = true
|
||||||
|
hive-agent-sock.workspace = true
|
||||||
hive-sh4re.workspace = true
|
hive-sh4re.workspace = true
|
||||||
libc.workspace = true
|
libc.workspace = true
|
||||||
rmcp.workspace = true
|
rmcp.workspace = true
|
||||||
|
|
|
||||||
|
|
@ -760,7 +760,7 @@ pub(crate) async fn send_wake(
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let req = hive_sh4re::AgentRequest::Wake {
|
let req = hive_agent_sock::AgentRequest::Wake {
|
||||||
from: format!("bash-task-{id}"),
|
from: format!("bash-task-{id}"),
|
||||||
body,
|
body,
|
||||||
};
|
};
|
||||||
|
|
|
||||||
|
|
@ -30,6 +30,7 @@ opentelemetry-otlp = { version = "0.32", default-features = false, features = [
|
||||||
"reqwest-rustls",
|
"reqwest-rustls",
|
||||||
] }
|
] }
|
||||||
indicatif.workspace = true
|
indicatif.workspace = true
|
||||||
|
hive-agent-sock.workspace = true
|
||||||
hive-sh4re.workspace = true
|
hive-sh4re.workspace = true
|
||||||
hive-host-sock.workspace = true
|
hive-host-sock.workspace = true
|
||||||
hive-priv-sock.workspace = true
|
hive-priv-sock.workspace = true
|
||||||
|
|
|
||||||
|
|
@ -9,7 +9,7 @@
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use hive_sh4re::AgentResponse;
|
use hive_agent_sock::AgentResponse;
|
||||||
|
|
||||||
use super::require_new_child;
|
use super::require_new_child;
|
||||||
use crate::coordinator::Coordinator;
|
use crate::coordinator::Coordinator;
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,7 @@
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use hive_sh4re::AgentResponse;
|
use hive_agent_sock::AgentResponse;
|
||||||
|
|
||||||
use super::require_descendant;
|
use super::require_descendant;
|
||||||
use crate::coordinator::Coordinator;
|
use crate::coordinator::Coordinator;
|
||||||
|
|
|
||||||
|
|
@ -13,7 +13,8 @@ use std::path::{Path, PathBuf};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use hive_sh4re::{AgentRequest, AgentResponse, MANAGER_AGENT, Message};
|
use hive_agent_sock::{AgentRequest, AgentResponse};
|
||||||
|
use hive_sh4re::{MANAGER_AGENT, Message};
|
||||||
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader};
|
||||||
use tokio::net::{UnixListener, UnixStream};
|
use tokio::net::{UnixListener, UnixStream};
|
||||||
use tokio::task::JoinHandle;
|
use tokio::task::JoinHandle;
|
||||||
|
|
@ -189,24 +190,24 @@ pub(crate) fn recv_timeout(wait_seconds: Option<u64>) -> std::time::Duration {
|
||||||
/// The unified `dispatch` calls this first; the remaining arms (which gate
|
/// The unified `dispatch` calls this first; the remaining arms (which gate
|
||||||
/// on topology / capabilities / tool-groups) are handled there.
|
/// on topology / capabilities / tool-groups) are handled there.
|
||||||
pub(crate) async fn dispatch_shared(
|
pub(crate) async fn dispatch_shared(
|
||||||
req: &hive_sh4re::Request,
|
req: &hive_agent_sock::Request,
|
||||||
agent: &str,
|
agent: &str,
|
||||||
coord: &Arc<Coordinator>,
|
coord: &Arc<Coordinator>,
|
||||||
) -> Option<hive_sh4re::Response> {
|
) -> Option<hive_agent_sock::Response> {
|
||||||
Some(match req {
|
Some(match req {
|
||||||
hive_sh4re::Request::Send {
|
hive_agent_sock::Request::Send {
|
||||||
to,
|
to,
|
||||||
body,
|
body,
|
||||||
in_reply_to,
|
in_reply_to,
|
||||||
} => handle_send(coord, agent, to, body, *in_reply_to),
|
} => handle_send(coord, agent, to, body, *in_reply_to),
|
||||||
hive_sh4re::Request::Recv { wait_seconds, max } => {
|
hive_agent_sock::Request::Recv { wait_seconds, max } => {
|
||||||
handle_recv(coord, agent, *wait_seconds, *max).await
|
handle_recv(coord, agent, *wait_seconds, *max).await
|
||||||
}
|
}
|
||||||
hive_sh4re::Request::Status => handle_status(coord, agent),
|
hive_agent_sock::Request::Status => handle_status(coord, agent),
|
||||||
hive_sh4re::Request::OperatorMsg { body } => handle_operator_msg(coord, agent, body),
|
hive_agent_sock::Request::OperatorMsg { body } => handle_operator_msg(coord, agent, body),
|
||||||
hive_sh4re::Request::Wake { from, body } => handle_wake(coord, agent, from, body),
|
hive_agent_sock::Request::Wake { from, body } => handle_wake(coord, agent, from, body),
|
||||||
hive_sh4re::Request::Recent { limit } => handle_recent(coord, agent, *limit),
|
hive_agent_sock::Request::Recent { limit } => handle_recent(coord, agent, *limit),
|
||||||
hive_sh4re::Request::Ask {
|
hive_agent_sock::Request::Ask {
|
||||||
question,
|
question,
|
||||||
options,
|
options,
|
||||||
multi,
|
multi,
|
||||||
|
|
@ -222,42 +223,42 @@ pub(crate) async fn dispatch_shared(
|
||||||
to.as_deref(),
|
to.as_deref(),
|
||||||
)
|
)
|
||||||
.map_or_else(
|
.map_or_else(
|
||||||
|message| hive_sh4re::Response::Err { message },
|
|message| hive_agent_sock::Response::Err { message },
|
||||||
|id| hive_sh4re::Response::QuestionQueued { id },
|
|id| hive_agent_sock::Response::QuestionQueued { id },
|
||||||
),
|
),
|
||||||
hive_sh4re::Request::Answer { id, answer } => {
|
hive_agent_sock::Request::Answer { id, answer } => {
|
||||||
crate::questions::handle_answer(coord, agent, *id, answer).map_or_else(
|
crate::questions::handle_answer(coord, agent, *id, answer).map_or_else(
|
||||||
|message| hive_sh4re::Response::Err { message },
|
|message| hive_agent_sock::Response::Err { message },
|
||||||
|()| hive_sh4re::Response::Ok,
|
|()| hive_agent_sock::Response::Ok,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
hive_sh4re::Request::Remind {
|
hive_agent_sock::Request::Remind {
|
||||||
message,
|
message,
|
||||||
timing,
|
timing,
|
||||||
file_path,
|
file_path,
|
||||||
} => handle_remind(coord, agent, message, timing, file_path.as_deref()),
|
} => handle_remind(coord, agent, message, timing, file_path.as_deref()),
|
||||||
hive_sh4re::Request::SetStatus { text } => handle_set_status(coord, text),
|
hive_agent_sock::Request::SetStatus { text } => handle_set_status(coord, text),
|
||||||
hive_sh4re::Request::GetAgentMeta { name } => {
|
hive_agent_sock::Request::GetAgentMeta { name } => {
|
||||||
handle_get_agent_meta(coord, agent, name.as_deref()).await
|
handle_get_agent_meta(coord, agent, name.as_deref()).await
|
||||||
}
|
}
|
||||||
hive_sh4re::Request::CancelLooseEnd { kind, id } => {
|
hive_agent_sock::Request::CancelLooseEnd { kind, id } => {
|
||||||
crate::questions::handle_cancel_loose_end(coord, agent, *kind, *id).map_or_else(
|
crate::questions::handle_cancel_loose_end(coord, agent, *kind, *id).map_or_else(
|
||||||
|message| hive_sh4re::Response::Err { message },
|
|message| hive_agent_sock::Response::Err { message },
|
||||||
|()| hive_sh4re::Response::Ok,
|
|()| hive_agent_sock::Response::Ok,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
hive_sh4re::Request::CreateRepo { repo } => handle_create_repo(agent, repo).await,
|
hive_agent_sock::Request::CreateRepo { repo } => handle_create_repo(agent, repo).await,
|
||||||
hive_sh4re::Request::AckTurn => handle_ack_turn(coord, agent),
|
hive_agent_sock::Request::AckTurn => handle_ack_turn(coord, agent),
|
||||||
hive_sh4re::Request::AckUntil { up_to } => handle_ack_until(coord, agent, *up_to),
|
hive_agent_sock::Request::AckUntil { up_to } => handle_ack_until(coord, agent, *up_to),
|
||||||
hive_sh4re::Request::RequeueInflight => handle_requeue_inflight(coord, agent),
|
hive_agent_sock::Request::RequeueInflight => handle_requeue_inflight(coord, agent),
|
||||||
hive_sh4re::Request::GracefulStopComplete => {
|
hive_agent_sock::Request::GracefulStopComplete => {
|
||||||
// Harness drained + is exiting: clear the fence so the
|
// Harness drained + is exiting: clear the fence so the
|
||||||
// `GracefulStop` orchestration (which polls this flag) proceeds
|
// `GracefulStop` orchestration (which polls this flag) proceeds
|
||||||
// to stop the container without waiting out its timeout.
|
// to stop the container without waiting out its timeout.
|
||||||
coord.clear_graceful_stop(agent);
|
coord.clear_graceful_stop(agent);
|
||||||
hive_sh4re::Response::Ok
|
hive_agent_sock::Response::Ok
|
||||||
}
|
}
|
||||||
hive_sh4re::Request::GetHostJournal {
|
hive_agent_sock::Request::GetHostJournal {
|
||||||
unit,
|
unit,
|
||||||
container,
|
container,
|
||||||
lines,
|
lines,
|
||||||
|
|
@ -292,7 +293,7 @@ async fn handle_recv(
|
||||||
agent: &str,
|
agent: &str,
|
||||||
wait_seconds: Option<u64>,
|
wait_seconds: Option<u64>,
|
||||||
max: Option<u32>,
|
max: Option<u32>,
|
||||||
) -> hive_sh4re::Response {
|
) -> hive_agent_sock::Response {
|
||||||
// Graceful-stop fence: while a graceful stop is pending for this agent,
|
// Graceful-stop fence: while a graceful stop is pending for this agent,
|
||||||
// return `GracefulStop` instead of polling the broker. The harness runs
|
// return `GracefulStop` instead of polling the broker. The harness runs
|
||||||
// one stop-checkpoint turn then exits; new sends keep queueing in the
|
// one stop-checkpoint turn then exits; new sends keep queueing in the
|
||||||
|
|
@ -300,7 +301,7 @@ async fn handle_recv(
|
||||||
// so a flag set between polls is seen on the next Recv — the orchestration
|
// so a flag set between polls is seen on the next Recv — the orchestration
|
||||||
// also fires a transient wake to break an in-flight long-poll.
|
// also fires a transient wake to break an in-flight long-poll.
|
||||||
if coord.is_graceful_stop_pending(agent) {
|
if coord.is_graceful_stop_pending(agent) {
|
||||||
return hive_sh4re::Response::GracefulStop;
|
return hive_agent_sock::Response::GracefulStop;
|
||||||
}
|
}
|
||||||
let cap = max.unwrap_or(1).min(RECV_BATCH_MAX) as usize;
|
let cap = max.unwrap_or(1).min(RECV_BATCH_MAX) as usize;
|
||||||
match coord
|
match coord
|
||||||
|
|
@ -314,7 +315,7 @@ async fn handle_recv(
|
||||||
// exactly how many still-pending messages remain to drain. A count
|
// exactly how many still-pending messages remain to drain. A count
|
||||||
// error is non-fatal — fall back to 0 rather than fail the recv.
|
// error is non-fatal — fall back to 0 rather than fail the recv.
|
||||||
let remaining = coord.broker.count_pending(agent).unwrap_or(0);
|
let remaining = coord.broker.count_pending(agent).unwrap_or(0);
|
||||||
hive_sh4re::Response::Messages {
|
hive_agent_sock::Response::Messages {
|
||||||
messages: deliveries
|
messages: deliveries
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|d| hive_sh4re::DeliveredMessage {
|
.map(|d| hive_sh4re::DeliveredMessage {
|
||||||
|
|
@ -328,7 +329,7 @@ async fn handle_recv(
|
||||||
remaining,
|
remaining,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -342,15 +343,15 @@ fn handle_wake(
|
||||||
agent: &str,
|
agent: &str,
|
||||||
from: &str,
|
from: &str,
|
||||||
body: &str,
|
body: &str,
|
||||||
) -> hive_sh4re::Response {
|
) -> hive_agent_sock::Response {
|
||||||
match coord.broker.send(&Message {
|
match coord.broker.send(&Message {
|
||||||
from: from.to_owned(),
|
from: from.to_owned(),
|
||||||
to: agent.to_owned(),
|
to: agent.to_owned(),
|
||||||
body: body.to_owned(),
|
body: body.to_owned(),
|
||||||
in_reply_to: None,
|
in_reply_to: None,
|
||||||
}) {
|
}) {
|
||||||
Ok(()) => hive_sh4re::Response::Ok,
|
Ok(()) => hive_agent_sock::Response::Ok,
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -360,13 +361,13 @@ fn handle_wake(
|
||||||
/// rescan. The harness has already written the status file to its own
|
/// rescan. The harness has already written the status file to its own
|
||||||
/// `state/` dir (it runs as the agent user), so this only refreshes the
|
/// `state/` dir (it runs as the agent user), so this only refreshes the
|
||||||
/// dashboard's view.
|
/// dashboard's view.
|
||||||
fn handle_set_status(coord: &Arc<Coordinator>, text: &str) -> hive_sh4re::Response {
|
fn handle_set_status(coord: &Arc<Coordinator>, text: &str) -> hive_agent_sock::Response {
|
||||||
if let Err(message) = crate::limits::check_status_text(text) {
|
if let Err(message) = crate::limits::check_status_text(text) {
|
||||||
return hive_sh4re::Response::Err { message };
|
return hive_agent_sock::Response::Err { message };
|
||||||
}
|
}
|
||||||
let coord2 = Arc::clone(coord);
|
let coord2 = Arc::clone(coord);
|
||||||
tokio::spawn(async move { coord2.rescan_containers_and_emit().await });
|
tokio::spawn(async move { coord2.rescan_containers_and_emit().await });
|
||||||
hive_sh4re::Response::Ok
|
hive_agent_sock::Response::Ok
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Validate an agent-supplied repo name: a single safe slug segment, no
|
/// Validate an agent-supplied repo name: a single safe slug segment, no
|
||||||
|
|
@ -384,9 +385,9 @@ fn valid_repo_name(name: &str) -> bool {
|
||||||
/// `CreateRepo` — create a repo for `agent` *through hive-c0re* in the
|
/// `CreateRepo` — create a repo for `agent` *through hive-c0re* in the
|
||||||
/// c0re-owned `agents` org with operator-team branch protection.
|
/// c0re-owned `agents` org with operator-team branch protection.
|
||||||
/// The sanctioned create path now that agents can't create repos directly.
|
/// The sanctioned create path now that agents can't create repos directly.
|
||||||
async fn handle_create_repo(agent: &str, repo: &str) -> hive_sh4re::Response {
|
async fn handle_create_repo(agent: &str, repo: &str) -> hive_agent_sock::Response {
|
||||||
if !valid_repo_name(repo) {
|
if !valid_repo_name(repo) {
|
||||||
return hive_sh4re::Response::Err {
|
return hive_agent_sock::Response::Err {
|
||||||
message: format!(
|
message: format!(
|
||||||
"invalid repo name {repo:?} — single segment of letters, digits, '-', '_', '.' \
|
"invalid repo name {repo:?} — single segment of letters, digits, '-', '_', '.' \
|
||||||
(no leading '-'/'.', max 100 chars)"
|
(no leading '-'/'.', max 100 chars)"
|
||||||
|
|
@ -394,16 +395,16 @@ async fn handle_create_repo(agent: &str, repo: &str) -> hive_sh4re::Response {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
let Some(core_token) = crate::forge::core_token() else {
|
let Some(core_token) = crate::forge::core_token() else {
|
||||||
return hive_sh4re::Response::Err {
|
return hive_agent_sock::Response::Err {
|
||||||
message: "forge unavailable (no core token) — cannot create repo".to_owned(),
|
message: "forge unavailable (no core token) — cannot create repo".to_owned(),
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
match crate::forge::create_agent_repo(agent, repo, &core_token).await {
|
match crate::forge::create_agent_repo(agent, repo, &core_token).await {
|
||||||
Ok(full_name) => hive_sh4re::Response::RepoCreated {
|
Ok(full_name) => hive_agent_sock::Response::RepoCreated {
|
||||||
clone_url: format!("{}/{full_name}.git", crate::forge::forge_http_base()),
|
clone_url: format!("{}/{full_name}.git", crate::forge::forge_http_base()),
|
||||||
full_name,
|
full_name,
|
||||||
},
|
},
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("create repo {repo:?} failed: {e:#}"),
|
message: format!("create repo {repo:?} failed: {e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -416,7 +417,7 @@ async fn handle_get_agent_meta(
|
||||||
coord: &Arc<Coordinator>,
|
coord: &Arc<Coordinator>,
|
||||||
agent: &str,
|
agent: &str,
|
||||||
name: Option<&str>,
|
name: Option<&str>,
|
||||||
) -> hive_sh4re::Response {
|
) -> hive_agent_sock::Response {
|
||||||
let target = name.unwrap_or(agent);
|
let target = name.unwrap_or(agent);
|
||||||
// `name` is agent-supplied and flows into filesystem reads below
|
// `name` is agent-supplied and flows into filesystem reads below
|
||||||
// (`read_agent_status_live`, `read_agent_matrix_identities` →
|
// (`read_agent_status_live`, `read_agent_matrix_identities` →
|
||||||
|
|
@ -425,14 +426,14 @@ async fn handle_get_agent_meta(
|
||||||
// (`target == agent`) is the caller's own authenticated name, already
|
// (`target == agent`) is the caller's own authenticated name, already
|
||||||
// valid — but validating unconditionally is simplest and harmless.
|
// valid — but validating unconditionally is simplest and harmless.
|
||||||
if let Some(reason) = crate::dashboard::validate_agent_name(target) {
|
if let Some(reason) = crate::dashboard::validate_agent_name(target) {
|
||||||
return hive_sh4re::Response::Err {
|
return hive_agent_sock::Response::Err {
|
||||||
message: format!("get_agent_meta: invalid agent name {target:?}: {reason}"),
|
message: format!("get_agent_meta: invalid agent name {target:?}: {reason}"),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
let (status_text, status_set_at, running) =
|
let (status_text, status_set_at, running) =
|
||||||
crate::container_view::read_agent_status_live(target).await;
|
crate::container_view::read_agent_status_live(target).await;
|
||||||
let (hive_name, swarm_name) = crate::container_view::hive_swarm_names();
|
let (hive_name, swarm_name) = crate::container_view::hive_swarm_names();
|
||||||
hive_sh4re::Response::AgentMeta {
|
hive_agent_sock::Response::AgentMeta {
|
||||||
name: target.to_owned(),
|
name: target.to_owned(),
|
||||||
running,
|
running,
|
||||||
hyperhive_rev: crate::auto_update::current_flake_rev(&coord.hyperhive_flake),
|
hyperhive_rev: crate::auto_update::current_flake_rev(&coord.hyperhive_flake),
|
||||||
|
|
@ -466,10 +467,10 @@ fn read_agent_matrix_identities(agent: &str) -> Vec<hive_sh4re::MatrixIdentity>
|
||||||
}
|
}
|
||||||
|
|
||||||
/// `Status` — count of pending (unread) inbox messages for `agent`.
|
/// `Status` — count of pending (unread) inbox messages for `agent`.
|
||||||
fn handle_status(coord: &Arc<Coordinator>, agent: &str) -> hive_sh4re::Response {
|
fn handle_status(coord: &Arc<Coordinator>, agent: &str) -> hive_agent_sock::Response {
|
||||||
match coord.broker.count_pending(agent) {
|
match coord.broker.count_pending(agent) {
|
||||||
Ok(unread) => hive_sh4re::Response::Status { unread },
|
Ok(unread) => hive_agent_sock::Response::Status { unread },
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -477,15 +478,19 @@ fn handle_status(coord: &Arc<Coordinator>, agent: &str) -> hive_sh4re::Response
|
||||||
|
|
||||||
/// `OperatorMsg` — deliver an operator-authored message into `agent`'s
|
/// `OperatorMsg` — deliver an operator-authored message into `agent`'s
|
||||||
/// inbox (from the `operator` recipient).
|
/// inbox (from the `operator` recipient).
|
||||||
fn handle_operator_msg(coord: &Arc<Coordinator>, agent: &str, body: &str) -> hive_sh4re::Response {
|
fn handle_operator_msg(
|
||||||
|
coord: &Arc<Coordinator>,
|
||||||
|
agent: &str,
|
||||||
|
body: &str,
|
||||||
|
) -> hive_agent_sock::Response {
|
||||||
match coord.broker.send(&Message {
|
match coord.broker.send(&Message {
|
||||||
from: hive_sh4re::OPERATOR_RECIPIENT.to_owned(),
|
from: hive_sh4re::OPERATOR_RECIPIENT.to_owned(),
|
||||||
to: agent.to_owned(),
|
to: agent.to_owned(),
|
||||||
body: body.to_owned(),
|
body: body.to_owned(),
|
||||||
in_reply_to: None,
|
in_reply_to: None,
|
||||||
}) {
|
}) {
|
||||||
Ok(()) => hive_sh4re::Response::Ok,
|
Ok(()) => hive_agent_sock::Response::Ok,
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -493,10 +498,10 @@ fn handle_operator_msg(coord: &Arc<Coordinator>, agent: &str, body: &str) -> hiv
|
||||||
|
|
||||||
/// `Recent` — the last `limit` inbox rows for `agent` (read-only,
|
/// `Recent` — the last `limit` inbox rows for `agent` (read-only,
|
||||||
/// doesn't consume).
|
/// doesn't consume).
|
||||||
fn handle_recent(coord: &Arc<Coordinator>, agent: &str, limit: u64) -> hive_sh4re::Response {
|
fn handle_recent(coord: &Arc<Coordinator>, agent: &str, limit: u64) -> hive_agent_sock::Response {
|
||||||
match coord.broker.recent_for(agent, limit) {
|
match coord.broker.recent_for(agent, limit) {
|
||||||
Ok(rows) => hive_sh4re::Response::Recent { rows },
|
Ok(rows) => hive_agent_sock::Response::Recent { rows },
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -504,10 +509,10 @@ fn handle_recent(coord: &Arc<Coordinator>, agent: &str, limit: u64) -> hive_sh4r
|
||||||
|
|
||||||
/// `AckTurn` — mark `agent`'s in-flight delivered messages acked so
|
/// `AckTurn` — mark `agent`'s in-flight delivered messages acked so
|
||||||
/// they don't redeliver on the next turn.
|
/// they don't redeliver on the next turn.
|
||||||
fn handle_ack_turn(coord: &Arc<Coordinator>, agent: &str) -> hive_sh4re::Response {
|
fn handle_ack_turn(coord: &Arc<Coordinator>, agent: &str) -> hive_agent_sock::Response {
|
||||||
match coord.broker.ack_turn(agent) {
|
match coord.broker.ack_turn(agent) {
|
||||||
Ok(_n) => hive_sh4re::Response::Ok,
|
Ok(_n) => hive_agent_sock::Response::Ok,
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -515,10 +520,14 @@ fn handle_ack_turn(coord: &Arc<Coordinator>, agent: &str) -> hive_sh4re::Respons
|
||||||
|
|
||||||
/// `AckUntil` — bulk-ack every message addressed to `agent` with row
|
/// `AckUntil` — bulk-ack every message addressed to `agent` with row
|
||||||
/// id `<= up_to` (the agent-side backlog-triage escape hatch).
|
/// id `<= up_to` (the agent-side backlog-triage escape hatch).
|
||||||
fn handle_ack_until(coord: &Arc<Coordinator>, agent: &str, up_to: i64) -> hive_sh4re::Response {
|
fn handle_ack_until(
|
||||||
|
coord: &Arc<Coordinator>,
|
||||||
|
agent: &str,
|
||||||
|
up_to: i64,
|
||||||
|
) -> hive_agent_sock::Response {
|
||||||
match coord.broker.ack_until(agent, up_to) {
|
match coord.broker.ack_until(agent, up_to) {
|
||||||
Ok(count) => hive_sh4re::Response::Acked { count },
|
Ok(count) => hive_agent_sock::Response::Acked { count },
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
@ -526,15 +535,15 @@ fn handle_ack_until(coord: &Arc<Coordinator>, agent: &str, up_to: i64) -> hive_s
|
||||||
|
|
||||||
/// `RequeueInflight` — resurface `agent`'s unacked in-flight messages
|
/// `RequeueInflight` — resurface `agent`'s unacked in-flight messages
|
||||||
/// (crash recovery on harness boot).
|
/// (crash recovery on harness boot).
|
||||||
fn handle_requeue_inflight(coord: &Arc<Coordinator>, agent: &str) -> hive_sh4re::Response {
|
fn handle_requeue_inflight(coord: &Arc<Coordinator>, agent: &str) -> hive_agent_sock::Response {
|
||||||
match coord.broker.requeue_inflight(agent) {
|
match coord.broker.requeue_inflight(agent) {
|
||||||
Ok(n) => {
|
Ok(n) => {
|
||||||
if n > 0 {
|
if n > 0 {
|
||||||
tracing::info!(%agent, requeued = %n, "requeued in-flight messages");
|
tracing::info!(%agent, requeued = %n, "requeued in-flight messages");
|
||||||
}
|
}
|
||||||
hive_sh4re::Response::Ok
|
hive_agent_sock::Response::Ok
|
||||||
}
|
}
|
||||||
Err(e) => hive_sh4re::Response::Err {
|
Err(e) => hive_agent_sock::Response::Err {
|
||||||
message: format!("{e:#}"),
|
message: format!("{e:#}"),
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,7 @@
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use hive_sh4re::AgentResponse;
|
use hive_agent_sock::AgentResponse;
|
||||||
|
|
||||||
use crate::coordinator::Coordinator;
|
use crate::coordinator::Coordinator;
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,7 @@
|
||||||
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use hive_sh4re::AgentResponse;
|
use hive_agent_sock::AgentResponse;
|
||||||
|
|
||||||
use crate::coordinator::Coordinator;
|
use crate::coordinator::Coordinator;
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -24,7 +24,7 @@ use tokio::net::UnixStream;
|
||||||
/// failure; callers log + ignore so a wake delivery hiccup doesn't tear
|
/// failure; callers log + ignore so a wake delivery hiccup doesn't tear
|
||||||
/// down the matrix sync loop.
|
/// down the matrix sync loop.
|
||||||
///
|
///
|
||||||
/// Wire format matches `hive_sh4re::Request` tagged with `"cmd"` per
|
/// Wire format matches `hive_agent_sock::Request` tagged with `"cmd"` per
|
||||||
/// `#[serde(tag = "cmd", rename_all = "snake_case")]`. Must be `"cmd"`,
|
/// `#[serde(tag = "cmd", rename_all = "snake_case")]`. Must be `"cmd"`,
|
||||||
/// not `"kind"` — the harness deserialises against the hive-sh4re type
|
/// not `"kind"` — the harness deserialises against the hive-sh4re type
|
||||||
/// and silently discards requests that don't match.
|
/// and silently discards requests that don't match.
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,7 @@ pub mod jobs;
|
||||||
pub mod paths;
|
pub mod paths;
|
||||||
pub mod wire_time;
|
pub mod wire_time;
|
||||||
|
|
||||||
/// Server-side hard cap on `Recv.max` (see `AgentRequest::Recv`). Bounds
|
/// Server-side hard cap on `Recv.max` (see the `Recv` request). Bounds
|
||||||
/// the size of a single round-trip so a confused caller can't drain the
|
/// the size of a single round-trip so a confused caller can't drain the
|
||||||
/// entire inbox in one go and blow past wire-buffer sizes; everything
|
/// entire inbox in one go and blow past wire-buffer sizes; everything
|
||||||
/// above the cap silently clamps. 5 keeps individual turns small — a big
|
/// above the cap silently clamps. 5 keeps individual turns small — a big
|
||||||
|
|
@ -352,401 +352,6 @@ pub struct TaskFile {
|
||||||
pub stderr_tail: Option<String>,
|
pub stderr_tail: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Unified request enum for both agent and manager sockets. The agent's
|
|
||||||
/// identity is the socket it arrived on. Privileged variants are marked
|
|
||||||
/// `*(privileged)*` — an agent socket returns `Err` for them server-side.
|
|
||||||
///
|
|
||||||
/// `AgentRequest` and `ManagerRequest` are type aliases for this enum;
|
|
||||||
/// existing callers continue to compile unchanged.
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
#[serde(tag = "cmd", rename_all = "snake_case")]
|
|
||||||
pub enum Request {
|
|
||||||
/// Send a message to another agent.
|
|
||||||
Send {
|
|
||||||
to: String,
|
|
||||||
body: String,
|
|
||||||
/// Optional id of the message being replied to. Stored in the
|
|
||||||
/// broker DB and returned on `Recv` so the dashboard can render
|
|
||||||
/// threads. Ignored if the id is unknown or out of retention.
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
in_reply_to: Option<i64>,
|
|
||||||
},
|
|
||||||
/// Pop pending messages from this agent's inbox.
|
|
||||||
/// Delivery + ack cycle: see
|
|
||||||
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
|
||||||
Recv {
|
|
||||||
#[serde(default)]
|
|
||||||
wait_seconds: Option<u64>,
|
|
||||||
#[serde(default)]
|
|
||||||
max: Option<u32>,
|
|
||||||
},
|
|
||||||
/// Non-mutating: how many pending messages are addressed to me?
|
|
||||||
/// Used by the harness to render a status line after each tool call.
|
|
||||||
Status,
|
|
||||||
/// Operator-injected message TO this agent (from this agent's own web
|
|
||||||
/// UI). Recipient is implicit — `from` is `"operator"`. Effectively the
|
|
||||||
/// per-agent equivalent of the old dashboard T4LK form, but scoped to
|
|
||||||
/// the agent whose page the operator is on.
|
|
||||||
OperatorMsg { body: String },
|
|
||||||
/// Wake-up event injected from inside the container. Recipient is
|
|
||||||
/// implicit (this agent); `from` is caller-chosen. See
|
|
||||||
/// `docs/conventions.md::Wake injection` for the trust model and
|
|
||||||
/// typical callers. The wake is persisted in the sqlite broker
|
|
||||||
/// like any other message — the agent can ack it via `AckUntil`.
|
|
||||||
Wake { from: String, body: String },
|
|
||||||
/// Last `limit` messages addressed to this agent, newest-first.
|
|
||||||
/// Non-mutating — pulls from the broker without delivering. The
|
|
||||||
/// per-agent web UI uses this to render its own inbox section.
|
|
||||||
Recent { limit: u64 },
|
|
||||||
/// Surface a question to either the operator or another agent.
|
|
||||||
/// Routing + shape: see
|
|
||||||
/// `docs/conventions.md::Question routing (Ask / Answer)`.
|
|
||||||
Ask {
|
|
||||||
question: String,
|
|
||||||
#[serde(default)]
|
|
||||||
options: Vec<String>,
|
|
||||||
#[serde(default)]
|
|
||||||
multi: bool,
|
|
||||||
#[serde(default)]
|
|
||||||
ttl_seconds: Option<u64>,
|
|
||||||
#[serde(default)]
|
|
||||||
to: Option<String>,
|
|
||||||
},
|
|
||||||
/// Answer a question previously routed to this agent via
|
|
||||||
/// `HelperEvent::QuestionAsked`. Authorised callers + threading
|
|
||||||
/// back via `HelperEvent::QuestionAnswered`: see
|
|
||||||
/// `docs/conventions.md::Question routing (Ask / Answer)`.
|
|
||||||
Answer { id: i64, answer: String },
|
|
||||||
/// Schedule a reminder message to be delivered to this agent at a
|
|
||||||
/// future time. The reminder lands in the agent's inbox as an auto-sent
|
|
||||||
/// message from `"reminder"`. Use for agent follow-ups (e.g. check task
|
|
||||||
/// status, retry failed operation). Message length is limited; pass
|
|
||||||
/// `file_path` to store in a file and get a path-reference message
|
|
||||||
/// instead.
|
|
||||||
Remind {
|
|
||||||
message: String,
|
|
||||||
timing: ReminderTiming,
|
|
||||||
#[serde(default)]
|
|
||||||
file_path: Option<String>,
|
|
||||||
},
|
|
||||||
/// Loose-ends view. On the agent socket: `None` = self; direct
|
|
||||||
/// children are always accessible; non-children require the
|
|
||||||
/// `query_agent_state` capability — rejected with an error otherwise;
|
|
||||||
/// `"*"` is always rejected (use the manager socket). On the manager
|
|
||||||
/// socket: `None` = manager self, `"*"` = hive-wide, any name =
|
|
||||||
/// that agent. See `docs/conventions.md::Loose-ends wire shape`.
|
|
||||||
GetLooseEnds {
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
agent: Option<String>,
|
|
||||||
},
|
|
||||||
/// Upsert a *todo* (loose-ends v2) from an in-container
|
|
||||||
/// subsystem (matrix / forge / bash). `subsystem` is the producer
|
|
||||||
/// marker; `key` is the optional subsystem-specific dedup key (a
|
|
||||||
/// matrix room id, a bash task id). Re-pushing an identical keyed
|
|
||||||
/// todo is a no-op; a new-or-changed one coalesces a wake to the
|
|
||||||
/// agent. Keyless todos always insert as one-offs.
|
|
||||||
UpsertTodo {
|
|
||||||
subsystem: String,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
key: Option<String>,
|
|
||||||
summary: String,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
source: Option<String>,
|
|
||||||
},
|
|
||||||
/// Clear producer-resolved todo(s) by `(subsystem, key)`. `key =
|
|
||||||
/// Some(k)` clears the one keyed row; `key = None` clears **all** of
|
|
||||||
/// the subsystem's keyless todos (rows with no key can't be told
|
|
||||||
/// apart — clear a specific one via `MarkTodoDone` by id). `all =
|
|
||||||
/// true` wipes the producer's whole set (cancel-and-recreate on
|
|
||||||
/// daemon restart).
|
|
||||||
ClearTodo {
|
|
||||||
subsystem: String,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
key: Option<String>,
|
|
||||||
#[serde(default)]
|
|
||||||
all: bool,
|
|
||||||
},
|
|
||||||
/// List todos, optionally filtered to one `subsystem` (a producer
|
|
||||||
/// enumerating its own set). `None` = all.
|
|
||||||
ListTodos {
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
subsystem: Option<String>,
|
|
||||||
},
|
|
||||||
/// The agent marks one of its own todos done, by id.
|
|
||||||
MarkTodoDone { id: i64 },
|
|
||||||
/// Count of pending (un-delivered) reminders. On the agent socket:
|
|
||||||
/// same target rules as `GetLooseEnds` (self/children free;
|
|
||||||
/// non-children require `query_agent_state`; `"*"` rejected).
|
|
||||||
/// On the manager socket: `None` = self, any name = that agent.
|
|
||||||
/// Used by the harness's per-turn stats sink.
|
|
||||||
CountPendingReminders {
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
agent: Option<String>,
|
|
||||||
},
|
|
||||||
/// Reminder statistics: counts of scheduled, delivered, and pending
|
|
||||||
/// reminders over a time window. `since_secs` filters to reminders
|
|
||||||
/// created in the last N seconds (0 = all). On the agent socket:
|
|
||||||
/// same target rules as `GetLooseEnds` (self/children free;
|
|
||||||
/// non-children require `query_agent_state`; `"*"` rejected).
|
|
||||||
/// On the manager socket: `None` = self, any name = that agent.
|
|
||||||
ReminderRollup {
|
|
||||||
/// Only count reminders created in the last N seconds from now.
|
|
||||||
/// Pass 0 to include all reminders.
|
|
||||||
#[serde(default)]
|
|
||||||
since_secs: u64,
|
|
||||||
/// Whose reminders to roll up. `None` = the caller's own.
|
|
||||||
/// `Some("<name>")` = that agent's (requires `query_agent_state`
|
|
||||||
/// capability on the agent socket; always available on the manager
|
|
||||||
/// socket).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
agent: Option<String>,
|
|
||||||
},
|
|
||||||
/// Set a free-text status string visible on the dashboard. The harness
|
|
||||||
/// writes `{state_dir}/hyperhive-status` locally before sending this
|
|
||||||
/// request; hive-c0re just triggers a dashboard rescan on receipt.
|
|
||||||
/// Pass an empty string to clear the status.
|
|
||||||
SetStatus { text: String },
|
|
||||||
/// Fetch identity + status for an agent. `name = None` =
|
|
||||||
/// self-introspection; `Some(<agent>)` = target query. See
|
|
||||||
/// `docs/conventions.md::Agent metadata`.
|
|
||||||
GetAgentMeta {
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
name: Option<String>,
|
|
||||||
},
|
|
||||||
/// Cancel an open thread the agent owns. Authorisation +
|
|
||||||
/// per-kind semantics in
|
|
||||||
/// `docs/conventions.md::Loose-ends wire shape`.
|
|
||||||
CancelLooseEnd { kind: CancelLooseEndKind, id: i64 },
|
|
||||||
/// Create a git repo *through hive-c0re*. Agents can't create
|
|
||||||
/// repos with their own forge token (`max_repo_creation = 0`); this is
|
|
||||||
/// the sanctioned path. hive-c0re creates `repo` in the c0re-owned
|
|
||||||
/// `agents` org, adds the calling agent as a write collaborator (not
|
|
||||||
/// owner), and applies operator-team branch protection so the author
|
|
||||||
/// can't merge its own PRs. Returns the new repo's full name.
|
|
||||||
CreateRepo { repo: String },
|
|
||||||
/// Mark every message popped since the last `AckTurn` as handled.
|
|
||||||
/// Harness↔broker pairing fired after `TurnOutcome::Ok`. See
|
|
||||||
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
|
||||||
AckTurn,
|
|
||||||
/// Mark every inbox message with broker row id `<= up_to` as
|
|
||||||
/// handled (`acked_at` set), whether still pending or already
|
|
||||||
/// delivered. The agent-facing bulk-triage escape hatch for a
|
|
||||||
/// redelivered / accumulated backlog: instead of popping and
|
|
||||||
/// re-reading dozens of already-handled messages one turn at a
|
|
||||||
/// time, the agent acks everything up to the id it has seen.
|
|
||||||
/// Recipient-scoped — an agent can only ack its own rows. See
|
|
||||||
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
|
||||||
AckUntil { up_to: i64 },
|
|
||||||
/// Requeue every popped-but-unacked message back into the inbox.
|
|
||||||
/// Harness fires this once at boot to recover from
|
|
||||||
/// crashed-mid-turn sessions. See
|
|
||||||
/// `docs/conventions.md::Broker delivery + ack cycle`.
|
|
||||||
RequeueInflight,
|
|
||||||
/// Harness → c0re: "I saw the `GracefulStop` signal, ran my
|
|
||||||
/// stop-checkpoint turn (durable `/state` flushed) and am exiting my
|
|
||||||
/// serve loop now." Lets the `GracefulStop` orchestration stop the
|
|
||||||
/// container immediately instead of waiting out its timeout fallback.
|
|
||||||
GracefulStopComplete,
|
|
||||||
/// *(capability-gated: `read_host_journal`)* Fetch recent lines
|
|
||||||
/// from the host journal. Filters are all optional; omitting all
|
|
||||||
/// returns the last `lines` entries from the global journal.
|
|
||||||
GetHostJournal {
|
|
||||||
/// Filter to a specific systemd unit (e.g. `hive-c0re.service`).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
unit: Option<String>,
|
|
||||||
/// Machine name to pass to journalctl `-M` verbatim (e.g. `h-iris`).
|
|
||||||
/// The caller is responsible for the correct nspawn machine name.
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
container: Option<String>,
|
|
||||||
/// Number of journal lines to return (default 30, max 100).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
lines: Option<u32>,
|
|
||||||
/// Minimum syslog priority level.
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
priority: Option<JournalPriority>,
|
|
||||||
/// Regex to match against log message fields (journalctl `--grep`).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
grep: Option<String>,
|
|
||||||
/// Show entries on or newer than this timestamp (journalctl `--since`).
|
|
||||||
/// ISO 8601 or journalctl-accepted relative strings (e.g. `"-1h"`).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
since: Option<String>,
|
|
||||||
/// Show entries on or older than this timestamp (journalctl `--until`).
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
until: Option<String>,
|
|
||||||
},
|
|
||||||
|
|
||||||
// ---- privileged (manager socket only for now) ---------------------------
|
|
||||||
/// *(privileged)* Initialise a brand-new agent's proposed config repo
|
|
||||||
/// and queue an approval for the operator to review.
|
|
||||||
RequestInitConfig {
|
|
||||||
name: String,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
description: Option<String>,
|
|
||||||
},
|
|
||||||
/// *(privileged)* Stop a sub-agent (graceful).
|
|
||||||
Kill { name: String },
|
|
||||||
/// *(privileged)* Start a previously-stopped sub-agent container.
|
|
||||||
Start { name: String },
|
|
||||||
/// *(privileged)* Restart a sub-agent container (stop + start).
|
|
||||||
Restart { name: String },
|
|
||||||
/// *(privileged)* Rebuild a sub-agent against the current hyperhive
|
|
||||||
/// flake + agent.nix. No approval required.
|
|
||||||
Update { name: String },
|
|
||||||
/// *(privileged)* Fetch recent journal lines for a sub-agent container.
|
|
||||||
GetLogs {
|
|
||||||
agent: String,
|
|
||||||
#[serde(default)]
|
|
||||||
lines: Option<u32>,
|
|
||||||
},
|
|
||||||
/// *(privileged)* Queue an approval to run `nix flake update [inputs...]`.
|
|
||||||
RequestUpdateMetaInputs {
|
|
||||||
#[serde(default)]
|
|
||||||
inputs: Vec<String>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
description: Option<String>,
|
|
||||||
},
|
|
||||||
/// *(privileged)* Queue an approval to add a scheduled prompt.
|
|
||||||
RequestSchedulePrompt(SchedulePromptPayload),
|
|
||||||
/// *(privileged)* Cancel a scheduled prompt.
|
|
||||||
CancelSchedule {
|
|
||||||
id: i64,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
targets: Option<Vec<String>>,
|
|
||||||
},
|
|
||||||
/// *(privileged)* List every schedule in the queue.
|
|
||||||
ListSchedules,
|
|
||||||
/// List all containers that are topological descendants of the calling
|
|
||||||
/// agent (direct children + their subtrees). Scoped to the caller's
|
|
||||||
/// subtree; gated by the `lifecycle` tool group. The result includes all
|
|
||||||
/// known descendants regardless of whether the container is currently
|
|
||||||
/// running — use `running` to distinguish.
|
|
||||||
ListDescendants,
|
|
||||||
/// *(privileged)* Fire a scheduled prompt out of band immediately.
|
|
||||||
FireScheduleNow { id: i64 },
|
|
||||||
/// *(privileged)* Edit an existing schedule's mutable fields.
|
|
||||||
EditSchedule {
|
|
||||||
id: i64,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
body: Option<String>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
description: Option<Option<String>>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
interval_seconds: Option<Option<u64>>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
next_fire_at_unix: Option<i64>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
targets_add: Option<Vec<String>>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
targets_remove: Option<Vec<String>>,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Backwards-compatible aliases. Both sockets now speak the unified `Request`
|
|
||||||
/// / `Response` wire; the server-side privilege gate rejects privileged
|
|
||||||
/// variants on agent sockets with `Err { message: "privileged variant..." }`.
|
|
||||||
pub type AgentRequest = Request;
|
|
||||||
pub type ManagerRequest = Request;
|
|
||||||
|
|
||||||
/// Unified response enum for both agent and manager sockets. Privileged
|
|
||||||
/// variants (`Logs`, `Schedules`) are never returned on agent sockets.
|
|
||||||
///
|
|
||||||
/// `AgentResponse` and `ManagerResponse` are type aliases for this enum.
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
||||||
#[serde(tag = "kind", rename_all = "snake_case")]
|
|
||||||
pub enum Response {
|
|
||||||
/// `Send` succeeded.
|
|
||||||
Ok,
|
|
||||||
/// Either `Send` failed or `Recv` errored.
|
|
||||||
Err { message: String },
|
|
||||||
/// `Recv` result: zero or more messages, FIFO-ordered, never
|
|
||||||
/// longer than the `max` the caller passed. Empty vec = nothing
|
|
||||||
/// pending (the "(empty)" path for the formatter). Per-row `id` +
|
|
||||||
/// `redelivered` carry the broker's row id (tracked by the harness
|
|
||||||
/// for `AckTurn`, and surfaced to claude as a `[msg #<id>]` marker
|
|
||||||
/// so `AckUntil` has something to reference) and the "previously
|
|
||||||
/// popped, not acked" flag — see `DeliveredMessage` for details.
|
|
||||||
/// `remaining` is the inbox depth *after* this batch was popped —
|
|
||||||
/// how many still-pending messages the caller could drain next. The
|
|
||||||
/// harness surfaces it to claude ("N more pending") so an in-turn
|
|
||||||
/// `recv` learns whether the inbox is drained, mirroring the count
|
|
||||||
/// the wake prompt already carries.
|
|
||||||
Messages {
|
|
||||||
messages: Vec<DeliveredMessage>,
|
|
||||||
remaining: u64,
|
|
||||||
},
|
|
||||||
/// `Status` result: how many pending messages are in this agent's inbox.
|
|
||||||
Status { unread: u64 },
|
|
||||||
/// `AckUntil` result: how many rows were newly marked handled.
|
|
||||||
Acked { count: u64 },
|
|
||||||
/// `Recent` result: newest-first inbox rows.
|
|
||||||
Recent { rows: Vec<InboxRow> },
|
|
||||||
/// `Ask` result: the queued question id. The answer lands later
|
|
||||||
/// as `HelperEvent::QuestionAnswered` in this agent's inbox.
|
|
||||||
QuestionQueued { id: i64 },
|
|
||||||
/// `GetLooseEnds` result: list of loose ends pending against
|
|
||||||
/// this agent. Ordered newest-first within each kind.
|
|
||||||
LooseEnds { loose_ends: Vec<LooseEnd> },
|
|
||||||
/// `CountPendingReminders` result.
|
|
||||||
PendingRemindersCount { count: u64 },
|
|
||||||
/// `ReminderRollup` result: reminder activity stats for the agent.
|
|
||||||
ReminderRollup(ReminderStats),
|
|
||||||
/// `GetAgentMeta` result. Per-field semantics + serde defaults
|
|
||||||
/// live in `docs/conventions.md::Agent metadata`.
|
|
||||||
AgentMeta {
|
|
||||||
name: String,
|
|
||||||
#[serde(default = "default_true")]
|
|
||||||
running: bool,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
hyperhive_rev: Option<String>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
status_text: Option<String>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
status_set_at: Option<i64>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
hive_name: Option<String>,
|
|
||||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
|
||||||
swarm_name: Option<String>,
|
|
||||||
/// Matrix identities this agent can act as (one per configured +
|
|
||||||
/// live account). Empty for agents with no matrix provisioning.
|
|
||||||
#[serde(default, skip_serializing_if = "Vec::is_empty")]
|
|
||||||
matrix_accounts: Vec<MatrixIdentity>,
|
|
||||||
},
|
|
||||||
/// `GetLogs` result: journal lines for the requested container.
|
|
||||||
/// Returned on the manager socket only.
|
|
||||||
Logs { content: String },
|
|
||||||
/// `GetHostJournal` result: host journal lines matching the
|
|
||||||
/// requested filters. Returned on the agent socket when the agent
|
|
||||||
/// holds the `read_host_journal` capability.
|
|
||||||
HostJournal { content: String },
|
|
||||||
/// `ListSchedules` result. Snapshot of every schedule.
|
|
||||||
/// Returned on the manager socket only.
|
|
||||||
Schedules { schedules: Vec<WireSchedule> },
|
|
||||||
/// `CreateRepo` result: the new repo's full name (`agents/<repo>`)
|
|
||||||
/// and clone URL, so the agent can immediately `git clone` it.
|
|
||||||
RepoCreated {
|
|
||||||
full_name: String,
|
|
||||||
clone_url: String,
|
|
||||||
},
|
|
||||||
/// `ListDescendants` result: all descendant containers, with running
|
|
||||||
/// status. Ordered by topology depth (parents before children), then
|
|
||||||
/// alphabetically within each depth tier.
|
|
||||||
Containers { containers: Vec<ContainerInfo> },
|
|
||||||
/// `Recv` result when a graceful stop is pending for this agent
|
|
||||||
/// (set by hive-c0re's `GracefulStop` orchestration). Returned in
|
|
||||||
/// place of `Messages` — it doubles as the inbound fence: the harness
|
|
||||||
/// stops consuming normal inbox messages and instead runs one
|
|
||||||
/// stop-checkpoint turn (flush durable `/state`), then reports
|
|
||||||
/// `GracefulStopComplete` and exits its serve loop so the container
|
|
||||||
/// can be stopped cleanly. New sends keep queueing in the broker for
|
|
||||||
/// the agent's next start.
|
|
||||||
GracefulStop,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Backwards-compatible response aliases.
|
|
||||||
pub type AgentResponse = Response;
|
|
||||||
pub type ManagerResponse = Response;
|
|
||||||
|
|
||||||
/// One entry in a `ListDescendants` result.
|
/// One entry in a `ListDescendants` result.
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct ContainerInfo {
|
pub struct ContainerInfo {
|
||||||
|
|
@ -802,13 +407,6 @@ pub struct MatrixIdentity {
|
||||||
pub homeserver: String,
|
pub homeserver: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Serde default for the `running` field; keeps wire backwards-compat
|
|
||||||
/// with pre-running-field payloads. See
|
|
||||||
/// `docs/conventions.md::Agent metadata`.
|
|
||||||
fn default_true() -> bool {
|
|
||||||
true
|
|
||||||
}
|
|
||||||
|
|
||||||
// -----------------------------------------------------------------------------
|
// -----------------------------------------------------------------------------
|
||||||
// Manager socket — /run/hyperhive/manager/mcp.sock on the host, bind-mounted
|
// Manager socket — /run/hyperhive/manager/mcp.sock on the host, bind-mounted
|
||||||
// into the manager container at /run/hive/mcp.sock.
|
// into the manager container at /run/hive/mcp.sock.
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue