use crate::messenger::{Messenger, MessengerError, OPERATOR};
use crate::provision::{read_ledger, LedgerEntry, NodeStatus};
use crate::tasks::{Refused, Session, TaskStore};
use serde_json::{json, Value};
use std::path::{Path, PathBuf};
use std::sync::{Condvar, Mutex};
use std::time::{Duration, Instant};
#[derive(Debug)]
pub struct HubError {
pub status: u16,
pub message: String,
}
impl HubError {
fn new(status: u16, message: impl Into<String>) -> Self {
Self { status, message: message.into() }
}
}
impl From<MessengerError> for HubError {
fn from(e: MessengerError) -> Self {
match e {
MessengerError::Forbidden => Self::new(403, "refused"),
MessengerError::Bad(m) => Self::new(400, m),
MessengerError::Internal(m) => Self::new(500, m),
}
}
}
pub struct Hub {
data_dir: PathBuf,
messenger: Mutex<Messenger>,
tasks: Mutex<TaskStore>,
mail_signal: (Mutex<u64>, Condvar),
advice: Mutex<crate::routing::AdviceGate>,
m4a: Option<Box<dyn crate::m4a_gateway::M4a>>,
}
fn refused(r: Refused) -> HubError {
match r {
Refused::NotFound => HubError::new(404, "not found"),
Refused::Conflict(m) => HubError::new(409, m),
Refused::Invalid(m) => HubError::new(400, m),
}
}
fn int(e: crate::error::ServerError) -> HubError {
HubError::new(500, e.to_string())
}
fn opt_str<'a>(v: &'a Value, name: &str) -> Option<&'a str> {
v.get(name).and_then(Value::as_str)
}
fn query_param<'a>(query: &'a str, name: &str) -> Option<&'a str> {
query.split('&').find_map(|kv| kv.strip_prefix(name).and_then(|r| r.strip_prefix('='))).filter(|v| !v.is_empty())
}
fn str_field<'a>(v: &'a Value, name: &str) -> Result<&'a str, HubError> {
v.get(name).and_then(Value::as_str).ok_or_else(|| HubError::new(400, format!("missing field: {name}")))
}
fn now_ms() -> u64 {
std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap_or_default().as_millis() as u64
}
impl Hub {
fn routing(&self) -> Result<(crate::routing::RoutingConfig, std::collections::BTreeMap<(String, String), crate::routing::SeatState>), HubError> {
let cfg = crate::routing::RoutingConfig::load(&self.data_dir).map_err(|e| HubError::new(500, e))?;
let tasks = self.tasks.lock().expect("tasks");
let mut all = Vec::new();
let mut after = None;
loop {
let page = tasks.signals(None, after, 500).map_err(int)?;
let done = page.len() < 500;
after = page.last().map(|s| s.id).or(after);
all.extend(page);
if done {
break;
}
}
let state = crate::routing::fold_signals(&all, &cfg);
Ok((cfg, state))
}
pub fn with_m4a(mut self, m4a: Box<dyn crate::m4a_gateway::M4a>) -> Self {
self.m4a = Some(m4a);
self
}
pub fn m4a_enabled(&self) -> bool {
self.m4a.is_some()
}
pub fn poll_m4a(&self) -> usize {
let Some(m4a) = &self.m4a else { return 0 };
let runs = match self.tasks.lock().expect("tasks").active_runs_with_nick() {
Ok(r) => r,
Err(e) => {
tracing::warn!(error = %e, "m4a reader: runs unavailable");
return 0;
}
};
let mut closed = 0;
for (run, nick) in runs {
let letters = match m4a.read_dm(&nick, 20) {
Ok(l) => l,
Err(e) => {
tracing::debug!(error = %e, "m4a reader: read failed");
continue;
}
};
for letter in letters {
let from = crate::m4a_gateway::bare_nick(&letter.from);
if from != crate::m4a_gateway::bare_nick(&nick) {
continue; }
let Some(cmd) = crate::m4a_gateway::parse_close(&letter.body) else { continue };
let op = format!("m4a:{}", letter.event_id);
let tasks = self.tasks.lock().expect("tasks");
if tasks.replay(Some(&op)).ok().flatten().is_some() {
continue;
}
let outcome = if cmd.run_id != run.id {
json!({ "ignored": "letter names another run" })
} else {
match tasks.close_run(run.id, &cmd.status, "session", Some(&run.session_id), cmd.branch.as_deref(), cmd.commit.as_deref(), Some(&letter.event_id)) {
Ok(Ok(r)) => {
closed += 1;
tracing::info!(run = r.id, status = %r.status, "run closed by its session via m4a");
json!({ "run": r })
}
Ok(Err(why)) => json!({ "refused": format!("{why:?}") }),
Err(e) => json!({ "error": e.to_string() }),
}
};
let _ = tasks.record(Some(&op), &outcome);
}
}
closed
}
pub fn recommend(&self, role: &str) -> Result<Value, HubError> {
let (cfg, state) = self.routing()?;
let rec = crate::routing::recommend(&cfg, &state, role, now_ms()).map_err(|e| HubError::new(400, e))?;
Ok(json!({ "recommendation": rec }))
}
pub fn open(data_dir: &Path) -> Result<Self, crate::error::ServerError> {
Ok(Self {
data_dir: data_dir.to_owned(),
messenger: Mutex::new(Messenger::open(data_dir)?),
tasks: Mutex::new(TaskStore::open(data_dir)?),
mail_signal: (Mutex::new(0), Condvar::new()),
advice: Mutex::new(Default::default()),
m4a: None,
})
}
fn bump_mail(&self) {
*self.mail_signal.0.lock().expect("signal") += 1;
self.mail_signal.1.notify_all();
}
fn entry(&self, node_id: &str) -> Option<LedgerEntry> {
read_ledger(&self.data_dir).ok()?.into_iter().find(|e| e.node_id == node_id)
}
fn admitted(&self, node_id: &str, capability: &str) -> bool {
node_id != OPERATOR
&& self.entry(node_id).is_some_and(|e| {
matches!(e.status, NodeStatus::Member | NodeStatus::Trusted) && e.grants.capabilities.iter().any(|c| c == capability)
})
}
pub fn node_call(&self, node_id: &str, op: &str, payload: &Value) -> Result<Value, HubError> {
let capability = match op.split_once('.').map(|p| p.0) {
Some("mail") => "mail",
Some("task") | Some("run") => "tasks",
_ => return Err(HubError::new(404, "unknown operation")),
};
if !self.admitted(node_id, capability) {
return Err(HubError::new(403, "refused"));
}
match op {
"mail.send" => {
let key = payload.get("idempotency_key").and_then(Value::as_str).map(str::to_owned);
let id = self.messenger.lock().expect("messenger").send(node_id, str_field(payload, "to")?, str_field(payload, "subject")?, str_field(payload, "body")?, key)?;
self.bump_mail();
Ok(json!({ "message_id": id }))
}
"mail.inbox" => {
let since = payload.get("since_ms").and_then(Value::as_u64).unwrap_or(0);
let limit = payload.get("limit").and_then(Value::as_u64).unwrap_or(50).min(256) as u16;
let mail = self.messenger.lock().expect("messenger").inbox(node_id, since, limit)?;
Ok(json!({ "messages": mail }))
}
"mail.ack" => {
self.messenger.lock().expect("messenger").ack(node_id, str_field(payload, "message_id")?)?;
Ok(json!({ "ok": true }))
}
"task.list" => {
let tasks = self.tasks.lock().expect("tasks").list(Some(node_id), true).map_err(|e| HubError::new(500, e.to_string()))?;
Ok(json!({ "tasks": tasks }))
}
"task.update" => {
let id = payload.get("id").and_then(Value::as_i64).ok_or_else(|| HubError::new(400, "missing field: id"))?;
let status = str_field(payload, "status")?;
if !matches!(status, "in_progress" | "done") {
return Err(HubError::new(400, "status must be in_progress or done"));
}
let changed = self.tasks.lock().expect("tasks").set_status(id, status, Some(node_id)).map_err(|e| HubError::new(500, e.to_string()))?;
if changed { Ok(json!({ "ok": true })) } else { Err(HubError::new(403, "refused")) }
}
"task.route" => self.recommend(str_field(payload, "role")?),
"task.advice_gate" => {
let (cfg, _) = self.routing()?;
let advisor = format!("{node_id}/{}", opt_str(payload, "advisor").unwrap_or("advisor"));
match self.advice.lock().expect("advice").check(&cfg.advisor, &advisor, str_field(payload, "topic")?, now_ms()) {
Ok(()) => Ok(json!({ "allow": true })),
Err(why) => Ok(json!({ "allow": false, "reason": why })),
}
}
"task.session_signal" => {
let instance = payload.get("instance_id").and_then(Value::as_i64).ok_or_else(|| HubError::new(400, "missing field: instance_id"))?;
let sig = self
.tasks
.lock()
.expect("tasks")
.add_signal(node_id, str_field(payload, "workspace_id")?, instance, str_field(payload, "provider")?, str_field(payload, "kind")?, opt_str(payload, "hint"))
.map_err(int)?
.map_err(refused)?;
tracing::info!(node = node_id, provider = %sig.provider, kind = %sig.kind, "session signal");
Ok(json!({ "signal_id": sig.id }))
}
"task.close_run" | "run.close" => {
let run_id = payload.get("run_id").and_then(Value::as_i64).ok_or_else(|| HubError::new(400, "missing field: run_id"))?;
let tasks = self.tasks.lock().expect("tasks");
let run = tasks.run(run_id).map_err(int)?.ok_or_else(|| HubError::new(404, "not found"))?;
if run.node_id != node_id {
return Err(HubError::new(403, "refused"));
}
let run = tasks
.close_run(run_id, str_field(payload, "status")?, "session", opt_str(payload, "session_id"), opt_str(payload, "branch"), opt_str(payload, "commit"), opt_str(payload, "result_message_id"))
.map_err(int)?
.map_err(refused)?;
Ok(json!({ "run": run }))
}
_ => Err(HubError::new(404, "unknown operation")),
}
}
pub fn operator_call(&self, method: &str, path: &str, body: &Value) -> Result<Value, HubError> {
let (path, query) = path.split_once('?').unwrap_or((path, ""));
let parts: Vec<&str> = path.trim_matches('/').split('/').collect();
match (method, parts.as_slice()) {
("POST", ["v1", "tasks"]) => {
let assignee = opt_str(body, "assignee");
if let Some(node) = assignee {
if !self.admitted(node, "tasks") {
return Err(HubError::new(403, "assignee does not hold the tasks capability"));
}
}
let tasks = self.tasks.lock().expect("tasks");
let op = opt_str(body, "op_id");
if let Some(done) = tasks.replay(op).map_err(int)? {
return Ok(done);
}
let status = opt_str(body, "status");
if let Some(st) = status {
if !matches!(st, "backlog" | "open" | "assigned") {
return Err(HubError::new(400, "initial status must be backlog, open or assigned"));
}
}
let task = tasks
.create_full(str_field(body, "title")?, opt_str(body, "body").unwrap_or(""), assignee, opt_str(body, "role"), body.get("parent_id").and_then(Value::as_i64), status)
.map_err(int)?;
let out = json!({ "task": task });
tasks.record(op, &out).map_err(int)?;
Ok(out)
}
("GET", ["v1", "tasks"]) => {
let roles: Option<Vec<String>> = query_param(query, "role").map(|r| r.split(',').map(str::to_owned).collect());
let tasks = self.tasks.lock().expect("tasks").list_filtered(query_param(query, "assignee"), roles.as_deref(), false).map_err(int)?;
Ok(json!({ "tasks": tasks }))
}
("GET", ["v1", "tasks", id]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
let tasks = self.tasks.lock().expect("tasks");
let task = tasks.get(id).map_err(int)?.ok_or_else(|| HubError::new(404, "not found"))?;
Ok(json!({ "task": task, "runs": tasks.runs(id).map_err(int)? }))
}
("POST", ["v1", "tasks", id, "move"]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
let tasks = self.tasks.lock().expect("tasks");
let op = opt_str(body, "op_id");
if let Some(done) = tasks.replay(op).map_err(int)? {
return Ok(done);
}
let task = tasks
.move_to(id, str_field(body, "status")?, body.get("expected_revision").and_then(Value::as_i64))
.map_err(int)?
.map_err(refused)?;
let out = json!({ "task": task });
tasks.record(op, &out).map_err(int)?;
Ok(out)
}
("POST", ["v1", "tasks", id, "runs"]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
let run = self.tasks.lock().expect("tasks").start_run(id, str_field(body, "session_id")?).map_err(int)?.map_err(refused)?;
Ok(json!({ "run": run }))
}
("POST", ["v1", "runs", id, "close"]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
let run = self
.tasks
.lock()
.expect("tasks")
.close_run(id, str_field(body, "status")?, "operator", None, opt_str(body, "branch"), opt_str(body, "commit"), opt_str(body, "result_message_id"))
.map_err(int)?
.map_err(refused)?;
Ok(json!({ "run": run }))
}
("GET", ["v1", "routing"]) => {
let (cfg, state) = self.routing()?;
let seats: Vec<Value> = cfg
.seats
.iter()
.map(|seat| {
let st = state.get(&(seat.node.clone(), seat.provider.clone())).cloned().unwrap_or_default();
json!({ "seat": seat, "state": st })
})
.collect();
Ok(json!({ "roles": cfg.roles, "seats": seats, "advisor": cfg.advisor }))
}
("POST", ["v1", "routing", "recommend"]) => self.recommend(str_field(body, "role")?),
("GET", ["v1", "signals"]) => {
let after = query_param(query, "after").and_then(|v| v.parse().ok());
let limit = query_param(query, "limit").and_then(|v| v.parse().ok()).unwrap_or(100);
let signals = self.tasks.lock().expect("tasks").signals(query_param(query, "node"), after, limit).map_err(int)?;
Ok(json!({ "signals": signals }))
}
("GET", ["v1", "sessions"]) => {
let sessions = self.tasks.lock().expect("tasks").sessions(query_param(query, "node")).map_err(int)?;
Ok(json!({ "sessions": sessions }))
}
("POST", ["v1", "sessions", id, "wake"]) => {
let Some(m4a) = &self.m4a else { return Err(HubError::new(503, "core m4a is not configured")) };
let nick = self.tasks.lock().expect("tasks").session_nick(id).map_err(int)?.ok_or_else(|| HubError::new(404, "session has no m4a nick"))?;
let text = opt_str(body, "text").map(str::trim).filter(|t| !t.is_empty()).unwrap_or("wake: check your tasks and m4a inbox");
if text.len() > 16 * 1024 {
return Err(HubError::new(400, "text too long"));
}
m4a.send(&nick, text).map_err(|e| HubError::new(502, e))?;
Ok(json!({ "woke": id, "via": "m4a" }))
}
("POST", ["v1", "sessions"]) => {
let session = Session {
session_id: str_field(body, "session_id")?.to_owned(),
node_id: str_field(body, "node_id")?.to_owned(),
provider: str_field(body, "provider")?.to_owned(),
workspace_id: opt_str(body, "workspace_id").map(str::to_owned),
role: opt_str(body, "role").map(str::to_owned),
state: opt_str(body, "state").unwrap_or("live").to_owned(),
created_at_ms: 0,
updated_at_ms: 0,
};
let tasks = self.tasks.lock().expect("tasks");
let session = tasks.upsert_session(&session).map_err(int)?;
if let Some(nick) = opt_str(body, "nick").filter(|n| !n.is_empty() && n.len() <= 128) {
tasks.set_session_nick(&session.session_id, nick).map_err(int)?;
}
Ok(json!({ "session": session }))
}
("POST", ["v1", "tasks", id, "assign"]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
let node = str_field(body, "assignee")?;
if !self.admitted(node, "tasks") {
return Err(HubError::new(403, "assignee does not hold the tasks capability"));
}
if self.tasks.lock().expect("tasks").assign(id, node).map_err(|e| HubError::new(500, e.to_string()))? { Ok(json!({ "ok": true })) } else { Err(HubError::new(409, "task cannot be assigned")) }
}
("POST", ["v1", "tasks", id, "cancel"]) => {
let id: i64 = id.parse().map_err(|_| HubError::new(400, "bad id"))?;
if self.tasks.lock().expect("tasks").set_status(id, "cancelled", None).map_err(|e| HubError::new(500, e.to_string()))? { Ok(json!({ "ok": true })) } else { Err(HubError::new(409, "task is final")) }
}
("POST", ["v1", "mail", "send"]) => {
let id = self.messenger.lock().expect("messenger").send(OPERATOR, str_field(body, "to")?, str_field(body, "subject")?, str_field(body, "body")?, None)?;
self.bump_mail();
Ok(json!({ "message_id": id }))
}
("POST", ["v1", "mail", "wait"]) => {
let since = body.get("since_ms").and_then(Value::as_u64).unwrap_or(0);
let deadline = Instant::now() + Duration::from_secs(body.get("timeout_secs").and_then(Value::as_u64).unwrap_or(25).min(30));
loop {
let seen = *self.mail_signal.0.lock().expect("signal");
let mail: Vec<_> = self.messenger.lock().expect("messenger").inbox(OPERATOR, since, 100)?.into_iter().filter(|m| m.created_at_ms > since).collect();
if !mail.is_empty() {
return Ok(json!({ "messages": mail }));
}
let now = Instant::now();
if now >= deadline {
return Ok(json!({ "messages": [] }));
}
let guard = self.mail_signal.0.lock().expect("signal");
if *guard == seen {
let _ = self.mail_signal.1.wait_timeout(guard, deadline - now).expect("signal wait");
}
}
}
("POST", ["v1", "mail", "thread"]) => {
let mail = self.messenger.lock().expect("messenger").thread(str_field(body, "node")?, 200)?;
Ok(json!({ "messages": mail }))
}
("POST", ["v1", "mail", "inbox"]) => {
let mail = self.messenger.lock().expect("messenger").inbox(OPERATOR, 0, 100)?;
Ok(json!({ "messages": mail }))
}
_ => Err(HubError::new(404, "not found")),
}
}
}
#[cfg(test)]
mod task_api_tests {
use super::*;
use crate::m4a_gateway::{Letter, M4a};
use std::sync::{Arc, Mutex as StdMutex};
#[derive(Clone, Default)]
struct FakeM4a {
sent: Arc<StdMutex<Vec<(String, String)>>>,
inbox: Arc<StdMutex<Vec<Letter>>>,
}
impl M4a for FakeM4a {
fn send(&self, to: &str, text: &str) -> Result<(), String> {
self.sent.lock().unwrap().push((to.into(), text.into()));
Ok(())
}
fn read_dm(&self, _nick: &str, _n: u32) -> Result<Vec<Letter>, String> {
Ok(self.inbox.lock().unwrap().clone())
}
fn me(&self) -> &str {
"hq-core"
}
}
#[test]
fn wake_sends_m4a_and_reader_closes_run_once() {
let dir = tempfile::tempdir().unwrap();
let fake = FakeM4a::default();
let hub = Hub::open(dir.path()).unwrap().with_m4a(Box::new(fake.clone()));
let op = |m: &str, p: &str, b: Value| hub.operator_call(m, p, &b);
op("POST", "/v1/sessions", json!({"session_id": "s1", "node_id": "alpha", "provider": "codex", "nick": "s1nick"})).unwrap();
let t = op("POST", "/v1/tasks", json!({"title": "t"})).unwrap();
let tid = t["task"]["id"].as_i64().unwrap();
let rid = op("POST", &format!("/v1/tasks/{tid}/runs"), json!({"session_id": "s1"})).unwrap()["run"]["id"].as_i64().unwrap();
op("POST", "/v1/sessions/s1/wake", json!({"text": "task #1: fix tests"})).unwrap();
assert_eq!(fake.sent.lock().unwrap()[0], ("s1nick".to_owned(), "task #1: fix tests".to_owned()));
assert_eq!(op("POST", "/v1/sessions/nope/wake", json!({})).unwrap_err().status, 404);
fake.inbox.lock().unwrap().extend([
Letter { event_id: "$1".into(), from: "@mallory:h".into(), body: format!("run.close {rid} done") },
Letter { event_id: "$2".into(), from: "@s1nick:h".into(), body: "run.close 999 done".into() },
Letter { event_id: "$3".into(), from: "@s1nick:h".into(), body: format!("run.close {rid} review feat/x abc") },
]);
assert_eq!(hub.poll_m4a(), 1);
assert_eq!(hub.poll_m4a(), 0); let got = op("GET", &format!("/v1/tasks/{tid}"), json!({})).unwrap();
assert_eq!(got["runs"][0]["status"], "review");
assert_eq!(got["runs"][0]["branch"], "feat/x");
assert_eq!(got["runs"][0]["result_message_id"], "$3");
}
#[test]
fn operator_task_create_move_cancel_and_runs() {
let dir = tempfile::tempdir().unwrap();
let hub = Hub::open(dir.path()).unwrap();
let op = |m: &str, p: &str, b: Value| hub.operator_call(m, p, &b);
let t = op("POST", "/v1/tasks", json!({"title": "port", "role": "dev", "status": "backlog", "op_id": "c1"})).unwrap();
let id = t["task"]["id"].as_i64().unwrap();
assert_eq!(op("POST", "/v1/tasks", json!({"title": "port", "op_id": "c1"})).unwrap(), t);
assert_eq!(op("GET", "/v1/tasks?role=advisor", json!({})).unwrap()["tasks"].as_array().unwrap().len(), 0);
assert_eq!(op("GET", "/v1/tasks?role=dev", json!({})).unwrap()["tasks"].as_array().unwrap().len(), 1);
let moved = op("POST", &format!("/v1/tasks/{id}/move"), json!({"status": "open", "expected_revision": 1})).unwrap();
assert_eq!(moved["task"]["revision"], 2);
assert_eq!(op("POST", &format!("/v1/tasks/{id}/move"), json!({"status": "open", "expected_revision": 1})).unwrap_err().status, 409);
assert_eq!(op("POST", &format!("/v1/tasks/{id}/runs"), json!({"session_id": "s1"})).unwrap_err().status, 400);
op("POST", "/v1/sessions", json!({"session_id": "s1", "node_id": "alpha", "provider": "codex"})).unwrap();
let run = op("POST", &format!("/v1/tasks/{id}/runs"), json!({"session_id": "s1"})).unwrap();
let rid = run["run"]["id"].as_i64().unwrap();
assert_eq!(hub.node_call("alpha", "run.close", &json!({"run_id": rid, "status": "done", "session_id": "s1"})).unwrap_err().status, 403);
let closed = op("POST", &format!("/v1/runs/{rid}/close"), json!({"status": "cancelled"})).unwrap();
assert_eq!(closed["run"]["closed_by"], "operator");
let got = op("GET", &format!("/v1/tasks/{id}"), json!({})).unwrap();
assert_eq!(got["task"]["status"], "cancelled");
assert_eq!(got["runs"].as_array().unwrap().len(), 1);
assert_eq!(op("POST", &format!("/v1/tasks/{id}/move"), json!({"status": "open"})).unwrap_err().status, 409);
assert_eq!(op("GET", "/v1/sessions?node=alpha", json!({})).unwrap()["sessions"].as_array().unwrap().len(), 1);
}
}