//! DAG shape builders — every operation as a template over the shared //! node primitives — plus submit-time cycle validation (petgraph is //! confined to this validation; the runtime store stays the plain //! `Vec` + `deps`). //! //! Every node carries its own `agent` (there is no DAG-level agent) — the //! `node` helper stamps each node's agent. This module holds the *pure* //! shape builders (no I/O). The hive-wide **power ops** (`stop` / `start` / //! `restart`) are NOT here: their per-agent shape depends on each agent's //! live running state (an async `lifecycle::is_running` read), so they are //! assembled dynamically in `submit.rs` out of the shared pure primitives //! this module exports (`node`, `after_ok`, `rebuild_nodes`) — one //! independent per-agent subgraph each, concurrent on its own lease, ONE //! DAG for the whole hive-wide op. `stop`/`start` write the durable `wanted` //! intent via a head `SetWanted(w)` node (holding the agent lease, so //! intent+reconcile is atomic per-agent); `restart` writes no intent — it //! bounces the container and lets the tail `Reconcile` converge to the //! agent's existing `wanted`. //! //! ```text //! rebuild(a): MetaSync(a) → Prebuild(a) → StopForUpdate(a) → Swap(a) →(ok) PostSwap(a) →(any) Reconcile(a) //! rebuild(a) graceful: … Prebuild(a) → Signal(a) → Drain(a) → StopForUpdate(a) → … [boot sweep only] //! spawn(a): Provision(a) → Create(a) → WriteDropin(a) → Reconcile(a) [wanted=Up at approve] //! perm-change(a): WritePermFile(a) → «rebuild subgraph» //! meta-update(inp): MetaLock(inp) →«in-DAG rebuild subgraph per affected a» //! reparent(moves): Reparent(moves) [no rebuild — topology.json is read live] //! ``` //! //! For the dynamic power-op shapes (`stop` / `start` / `restart`, built from //! live online/offline state), see `submit.rs`. use anyhow::{Result, bail}; use hive_jobq::{DepWhen, TerminalState}; use super::model::{DagSpec, Dep, NodeKind, NodeSpec, PermPayload, Source}; /// After-ok edge on the previous node — the common chain link. Shared with /// the async power-op builders in `submit.rs` (which assemble per-agent /// chains dynamically from live container state). pub(crate) fn after_ok(on: u64) -> Vec { vec![Dep { on, when: DepWhen::AFTER_OK, }] } /// `AfterOk` edges onto every one of a DAG's **group-roots** — the success /// branch of a per-outcome tail pair, and the aggregator the failure branch /// keys off. /// /// Group-roots are the right granularity, not "every node": a root's state *is* /// its subtree's roll-up, so edging the roots covers every descendant while /// keeping the dep list small and stable as subtrees grow. Because every edge is /// `AFTER_OK`, this node runs only if *all* of them succeeded — and is ruled out /// ([`TerminalState::Skipped`]) the moment one doesn't, which is precisely the /// signal [`on_elimination_of`] waits for. pub(crate) fn after_ok_all(ons: &[u64]) -> Vec { ons.iter() .map(|&on| Dep { on, when: DepWhen::AFTER_OK, }) .collect() } /// `AFTER_ANY` edges onto every group-root — "wait for all of these to finish, /// however they went". Ordering only; it accepts any outcome except the DAG /// being dropped. pub(crate) fn after_any_all(ons: &[u64]) -> Vec { ons.iter() .map(|&on| Dep { on, when: DepWhen::AFTER_ANY, }) .collect() } /// A single edge satisfied only when `on` was **ruled out** by its own edges. /// /// Dependency edges are conjunctive, so "any one of these several nodes failed" /// cannot be written directly. This is the composition that expresses it: point /// the success branch at every root with [`after_ok_all`], then hang the failure /// branch off *that* node's elimination. Exactly one of the pair ever runs. /// /// Note it accepts `Skipped` and **not** `Cancelled`: if the whole DAG was /// dropped before it started, the success branch is marked `Cancelled` directly /// and this branch is ruled out too — a job nobody ran reports nothing. pub(crate) fn on_elimination_of(on: u64) -> Vec { vec![Dep { on, when: DepWhen::of(&[TerminalState::Skipped]), }] } /// A single edge satisfied only by the listed outcomes of `on` — for the /// one-tail-per-outcome shape an approval DAG uses. pub(crate) fn on_outcome(on: u64, outcomes: &[TerminalState]) -> Vec { vec![Dep { on, when: DepWhen::of(outcomes), }] } /// The `Rebuilt`-reporting tail pair for a rebuild-shaped DAG: the success node /// gated on every group-root in `roots`, and the failure node gated on *its* /// elimination. `base` is the spec index the pair starts at. /// /// Exactly one runs on a DAG that executed, and neither runs on one the operator /// dropped — see [`on_elimination_of`]. fn emit_rebuilt_tails(agent: &str, roots: &[u64], base: u64) -> Vec { // The failure branch needs *both*: the ok branch being ruled out (that is the // "something went wrong" signal) **and** every root actually finished. The // second half is easy to forget and gets the ordering wrong without it — a // failed `Prebuild` eliminates the ok branch immediately, while the recovery // `Reconcile` is still bringing the container back up, so reporting straight // off the elimination would announce the failure mid-recovery. let mut on_fail = after_any_all(roots); on_fail.extend(on_elimination_of(base)); vec![ node( NodeKind::EmitRebuilt { agent: agent.to_owned(), ok: true, }, after_ok_all(roots), ), node( NodeKind::EmitRebuilt { agent: agent.to_owned(), ok: false, }, on_fail, ), ] } /// The approval-resolving tails for an approval-carrying DAG: one per outcome of /// the DAG's single group-root `root`, each accepting only its own. /// /// The `Cancelled` node is what keeps a dropped approval DAG from dangling its /// row forever — its edge is the only one [`super::JobQueue::cancel`] spares. fn resolve_approval_tails(approval_id: i64, root: u64) -> Vec { [ TerminalState::Done, TerminalState::Failed, TerminalState::Cancelled, ] .into_iter() .map(|outcome| { node( NodeKind::ResolveApproval { approval_id, outcome, }, on_outcome(root, &[outcome]), ) }) .collect() } /// Build one **top-level (group-root)** node — `parent = None`. `kind` carries /// the agent it targets ([`NodeKind`] is the payload directly). Shared with /// `submit.rs`'s dynamic power-op builders. A root owns whatever resource it /// declares for its whole subtree; its descendants borrow it (agent-lease / /// build-slot continuity). Ordering vs other nodes is `deps`; grouping is /// `parent`. pub(crate) fn node(kind: NodeKind, deps: Vec) -> NodeSpec { NodeSpec { kind, deps, parent: None, } } /// Build a **child** node whose structural parent is spec-index `parent`. The /// child runs once its parent reaches `Finishing` (the parent gate), so it must /// NOT `deps` on `parent` (dep-scope validation rejects a dep on one's own /// parent). `deps` here order the child against its *siblings* only. pub(crate) fn child(parent: u64, kind: NodeKind, deps: Vec) -> NodeSpec { NodeSpec { kind, deps, parent: Some(parent), } } /// Knobs for [`rebuild_nodes`]. A struct rather than two positional `bool`s so /// a call site cannot silently swap them. #[derive(Debug, Clone, Copy)] pub(crate) struct RebuildOpts { /// Re-lock the meta flake inside `MetaSync`. pub relock: bool, /// Give the agent its `Signal` → `Drain` window to finish the turn in /// flight before the container is stopped, instead of stopping it /// outright. Costs up to one `GRACEFUL_STOP_TIMEOUT` per subgraph, and /// those overlap across agents. pub graceful: bool, } /// The rebuild node subtree (nested, three group roots). `base` is the spec /// index of the first node (`MetaSync`). Structure: /// - `MetaSync` (base+0, **root**): the meta-repo preamble (dir prep, agent /// sync, optional relock). Owns the global `MetaWindow` — and *only* for its /// own short duration, which is why it is a sibling root rather than /// `Prebuild`'s parent: a resource is held across the holder's whole subtree, /// so parenting the build under it would extend a hive-global window over /// every rebuild's nix build. /// - `Prebuild` (base+1, **root**): `AfterOk` `MetaSync`. Owns the build slot /// for the whole mechanical subtree below it. Lease-exempt — the nix build /// overlaps other DAGs on the same agent. /// - the **stop root** (base+2, child of `Prebuild`): owns the agent lease and /// runs once `Prebuild` reaches `Finishing` (parent gate). Non-graceful that /// is `StopForUpdate` itself; graceful it is `Signal`, with `Drain` and then /// `StopForUpdate` as its children so the lease stays continuous across the /// whole stop — siblings would each take the lease separately and leave a gap /// another DAG could claim the agent in, mid-bounce. /// - `Swap` (child of `StopForUpdate`): borrows the agent lease from its /// ancestors and the build slot from `Prebuild` — both continuous. /// - `PostSwap` (child of `StopForUpdate`): the swap's Ok-only /// bookkeeping tail (rev marker, forge/matrix sync, kick, rescan), `AfterOk` /// its sibling `Swap`. /// - `Reconcile` (**last, root**): `AfterAny` `Prebuild`, which rolls up /// terminal only once its whole mechanical subtree (SFU→Swap→PostSwap) has /// settled — so `Reconcile` runs after the swap regardless of outcome, and as /// a top-level root it survives the cancel-cascade of a failed `Prebuild` /// (recovery-start invariant, which also covers a failed `MetaSync`: that /// cancel-cascades `Prebuild`, i.e. terminal, so the tail still runs). It /// takes a fresh lease; the tiny gap is harmless — `Reconcile` converges to /// the persisted `wanted` idempotently. pub(crate) fn rebuild_nodes(agent: &str, opts: RebuildOpts, base: u64) -> Vec { let a = || agent.to_owned(); let RebuildOpts { relock, graceful } = opts; let mut nodes = vec![ node( NodeKind::MetaSync { agent: a(), relock }, if base == 0 { Vec::new() } else { after_ok(base - 1) }, ), node(NodeKind::Prebuild { agent: a() }, after_ok(base)), ]; // The stop root hangs off `Prebuild` and owns the agent lease for // everything below it. let stop_root = base + 2; if graceful { nodes.push(child(base + 1, NodeKind::Signal { agent: a() }, Vec::new())); // `Drain` is a *child* of `Signal`, so the parent gate already orders // it — a child must not dep on its own parent (dep-scope). nodes.push(child(stop_root, NodeKind::Drain { agent: a() }, Vec::new())); nodes.push(child( stop_root, NodeKind::StopForUpdate { agent: a() }, after_ok(stop_root + 1), )); } else { nodes.push(child( base + 1, NodeKind::StopForUpdate { agent: a() }, Vec::new(), )); } // Index of `StopForUpdate`, which parents the swap pair either way. let sfu = if graceful { stop_root + 2 } else { stop_root }; nodes.push(child(sfu, NodeKind::Swap { agent: a() }, Vec::new())); nodes.push(child( sfu, NodeKind::PostSwap { agent: a() }, after_ok(sfu + 1), )); nodes.push(node( NodeKind::Reconcile { agent: a() }, vec![Dep { on: base + 1, when: DepWhen::AFTER_ANY, }], )); nodes } /// The rebuild subgraph a [`NodeKind::DeployApply`] grows into its own DAG once /// the merge has landed and `prepare_deploy` has staged the lock, plus the /// [`NodeKind::FinalizeDeploy`] that closes the window behind it. /// /// `relock = false` is the whole reason this composes: `prepare_deploy` already /// relocked and staged `flake.lock`, so the appended `MetaSync` must do the dir /// prep + `sync_agents` *without* re-locking over it. /// /// `FinalizeDeploy` waits on **two** siblings, which together reproduce the gate /// the old fused node had around its inline `rebuild_no_meta` call: /// - `AfterOk` `Prebuild` — a parent's state is its roll-up, so this is `Done` /// only once `StopForUpdate` → `Swap` → `PostSwap` all are (a failed *or* /// cancelled child rolls the parent up `Failed`). That's the old /// `build_result`. /// - `AfterOk` `Reconcile` — the old call passed `deferred_start = false` on /// purpose: the container had to come back up *before* the deploy was /// finalized. `Reconcile` alone would not do, being `AfterAny` — it reaches /// `Done` even after a failed `Swap`. /// /// Appended, not submitted: the roots below become children of the emitting /// `DeployApply` (see [`super::JobQueue::append_subgraph`]), which puts them /// inside the `DeployWindow`'s subtree — so the `MetaWindow` this subgraph's /// `MetaSync` and `FinalizeDeploy` declare is re-entered from the ancestor /// already holding it rather than deadlocking against it. pub(crate) fn deploy_rebuild_nodes(agent: &str, approval_id: i64) -> Vec { let mut nodes = rebuild_nodes( agent, RebuildOpts { relock: false, graceful: false, }, 0, ); let reconcile = reconcile_index(&nodes, 0); nodes.push(node( NodeKind::FinalizeDeploy { agent: agent.to_owned(), approval_id, }, vec![ Dep { on: 1, when: DepWhen::AFTER_OK, }, Dep { on: reconcile, when: DepWhen::AFTER_OK, }, ], )); nodes } /// Spec index of the `Reconcile` root a [`rebuild_nodes`] subgraph ends on, /// for callers that gate a tail on it. Read off the emitted list rather than /// hard-coded, because the subgraph's length depends on [`RebuildOpts`]. fn reconcile_index(rebuild: &[NodeSpec], base: u64) -> u64 { base + u64::try_from(rebuild.len()).unwrap_or(0).saturating_sub(1) } /// One uniform rebuild shape — no `was_running` branch. `StopForUpdate` /// noops when already down; the tail `Reconcile` auto-noops the start /// when `wanted = Offline` (a rebuild of a deliberately-stopped agent /// leaves it stopped). `relock = false` only for meta-update cascade /// children. /// /// Closed by an [`NodeKind::EmitRebuilt`] tail edged onto all three group-roots /// (`MetaSync`, `Prebuild`, `Reconcile`) — `Prebuild`'s roll-up carries the /// whole `StopForUpdate`→`Swap`→`PostSwap` subtree, so those three cover every /// node. Edging `Reconcile` alone would not do: it is `AfterAny` `Prebuild`, so /// it reaches `Done` even after a failed swap and the tail would report success. pub fn rebuild(agent: &str, source: Source, reason: String, relock: bool) -> DagSpec { let mut nodes = rebuild_nodes( agent, RebuildOpts { relock, graceful: false, }, 0, ); let reconcile = reconcile_index(&nodes, 0); let tail_base = u64::try_from(nodes.len()).unwrap_or(0); nodes.extend(emit_rebuilt_tails(agent, &[0, 1, reconcile], tail_base)); DagSpec { source, reason, nodes, } } /// Approval-driven deploy (`MergeConfigPr`) as a phase subtree rather than the /// single opaque node it used to be. Structure: /// - `DeployWindow` (0, **root**): the resource holder — global meta window, /// agent lease, build slot — held across every child below. No work of its /// own; it reaches `Finishing` immediately and the children run inside it. /// - `MergeVerify` (1, child): drift-gate + fetch + eval-verify. Mutates /// nothing, so a failure here cancel-cascades its siblings with the forge and /// the applied repo exactly as they were. /// - `DeployApply` (2, child, `AfterOk` `MergeVerify`): the irreversible half — /// ff-merge + `prepare_deploy`. It doesn't rebuild inline; it grows /// [`deploy_rebuild_nodes`] into this DAG as its own children, so the build /// and the closing `FinalizeDeploy` are real nodes under the same window. /// - `DeployTail` (3, child, `AfterAny` `DeployApply`): the compensation + /// bookkeeping tail — rollback when a merge landed unfinalized, forge tag /// mirror, PR failure comment (see [`NodeKind::DeployTail`]). /// /// - `ResolveApproval` (4, **root**, `AfterAny` `DeployWindow`): resolves the /// approval row. A root rather than another child, so it isn't inside the /// window's resource subtree — it runs once the window has released the meta /// window, lease and build slot. One edge suffices here: `DeployWindow` is the /// DAG's only other group-root, so its roll-up already *is* the whole /// pipeline's outcome. /// /// The window still spans the container build, as it must: `prepare_deploy` /// leaves `flake.lock` staged-uncommitted for the build's whole duration. pub fn approval_deploy(agent: &str, approval_id: i64, reason: String) -> DagSpec { let a = || agent.to_owned(); DagSpec { source: Source::Approval, reason, nodes: vec![ node( NodeKind::DeployWindow { agent: a(), approval_id, }, Vec::new(), ), child( 0, NodeKind::MergeVerify { agent: a(), approval_id, }, Vec::new(), ), child( 0, NodeKind::DeployApply { agent: a(), approval_id, }, after_ok(1), ), child( 0, NodeKind::DeployTail { agent: a(), approval_id, }, vec![Dep { on: 2, when: DepWhen::AFTER_ANY, }], ), ] .into_iter() .chain(resolve_approval_tails(approval_id, 0)) .collect(), } } /// A single `Reconcile` node that converges observed power state to the /// persisted intent — `wanted` is untouched (no `SetWanted`), unlike the /// operator `start`/`stop` templates. Test-only helper now (used to build /// single-node lifecycle DAGs that exercise per-agent lease serialization /// in the queue tests); production paths no longer emit a bare reconcile. #[cfg(test)] pub fn reconcile_only(agent: &str, source: Source, reason: String) -> DagSpec { DagSpec { source, reason, nodes: vec![node( NodeKind::Reconcile { agent: agent.to_owned(), }, Vec::new(), )], } } /// First-deploy spawn (approval-driven): `Provision` (proposed/applied /// repos, state subvolume, meta registration) then `Create` /// (`nixos-container create`), drop-in write, then `Reconcile` starts /// the container (`wanted = Up` written at approve time). All-or-nothing: /// `Provision` (lease-exempt, precedes the container) is the group root; /// `Create` (child) owns the agent lease; `WriteDropin` + `Reconcile` /// (children of `Create`) borrow it. A failure cancel-cascades the rest — /// unlike rebuild there's no recovery-reconcile (nothing to converge if the /// container was never created). Closed by a `ResolveApproval` tail root edged /// `AfterAny` onto `Provision` — the DAG's only other group-root, so its roll-up /// already carries the whole cascade. pub fn spawn(agent: &str, approval_id: i64, reason: String) -> DagSpec { DagSpec { source: Source::Approval, reason, nodes: { let a = || agent.to_owned(); vec![ node(NodeKind::Provision { agent: a() }, Vec::new()), child(0, NodeKind::Create { agent: a() }, Vec::new()), child(1, NodeKind::WriteDropin { agent: a() }, Vec::new()), child(1, NodeKind::Reconcile { agent: a() }, after_ok(2)), ] .into_iter() .chain(resolve_approval_tails(approval_id, 0)) .collect() }, } } /// Perm change: commit the JSON file(s), then the rebuild subgraph so /// the updated `HIVE_TOOL_GROUPS` / `HIVE_CAPABILITIES` env var takes /// effect in the container. Group-roots are `WritePermFile`(0) plus the rebuild /// subgraph's `MetaSync`(1) / `Prebuild`(2) / `Reconcile`(6), so the /// `EmitRebuilt` tail edges all four. pub fn perm_change(agent: &str, source: Source, reason: String, payload: PermPayload) -> DagSpec { let mut nodes = vec![node( NodeKind::WritePermFile { agent: agent.to_owned(), payload, }, Vec::new(), )]; let rebuild = rebuild_nodes( agent, RebuildOpts { relock: true, graceful: false, }, 1, ); let reconcile = reconcile_index(&rebuild, 1); nodes.extend(rebuild); let tail_base = u64::try_from(nodes.len()).unwrap_or(0); nodes.extend(emit_rebuilt_tails(agent, &[0, 1, 2, reconcile], tail_base)); DagSpec { source, reason, nodes, } } /// Meta-input lock bump. The `MetaLock` executor grows one rebuild subgraph /// per affected agent into *this same* DAG on completion (via /// `append_subgraph`) — appended *after* the bump lands so their prebuilds /// run against the post-bump lock, and a failed bump appends nothing /// (replacing the old fan-out-child-DAGs dance). /// `transient = Rebuilding` because those appended subgraphs are rebuilds: /// it's applied per-agent at claim time (the `MetaLock` head needs no lease, /// so the "hyperhive" pseudo-agent gets no pill), giving each cascade agent /// crash-watch suppression during its `Swap` — the property the old child /// `Rebuild` DAGs carried via their own transient. pub fn meta_update( inputs: Vec, source: Source, reason: String, approval_id: Option, ) -> DagSpec { let mut nodes = vec![node( NodeKind::MetaLock { sweep: false, fanout: None, inputs, }, Vec::new(), )]; // The bump itself has no side effect, so an operator-driven one ends at the // `MetaLock`; an approval-driven one still has its row to resolve and gets the // per-outcome tails edged onto that single group-root — whose roll-up covers // the rebuild subgraphs `MetaLock` grows into itself. if let Some(approval_id) = approval_id { nodes.extend(resolve_approval_tails(approval_id, 0)); } DagSpec { source, reason, nodes, } } /// Topology move(s) as a single-node DAG. `moves` is `(child, new_parent)` /// pairs — len 1 for `set-parent`, len N for `set-parent-bulk`, applied /// uniformly by the one [`NodeKind::Reparent`] node (which holds the global /// meta window for its duration, same precedent as [`NodeKind::WritePermFile`]). /// No rebuild subgraph: `topology.json` is read live by every consumer /// (dashboard tree, ``/`` sentinel routing, permission /// checks), so a parent move needs no container rebuild to take effect. /// No transient pill either — the node is agentless (no lease to hang one /// off of) and near-instant. No tail node: the write is the whole effect. pub fn reparent( moves: Vec<(hive_types::Ident, Option)>, source: Source, reason: String, ) -> DagSpec { DagSpec { source, reason, nodes: vec![node(NodeKind::Reparent { moves }, Vec::new())], } } // The boot is assembled inline in `workers/auto_update.rs::submit_boot_tree` // as ONE `Boot` DAG (a sweep `MetaLock` root that grows rebuild subgraphs // in-DAG, plus a `Reconcile` root per drifted agent) — no anchor node and no // per-agent child DAGs. /// Validate a spec before it enters the queue: node ids are dense /// (index = id), deps + parents reference existing *earlier* nodes, and the /// dep graph is acyclic (petgraph `toposort`). Rejecting cycles here fixes the /// old queue's documented "circular dep silently deadlocks forever" caveat. pub fn validate(spec: &DagSpec) -> Result<()> { if spec.nodes.is_empty() { bail!("dag spec {:?} has no nodes", spec.reason); } let n = spec.nodes.len(); let mut graph = petgraph::graph::DiGraph::::new(); let idx: Vec<_> = (0..n) .map(|i| graph.add_node(u32::try_from(i).unwrap_or(u32::MAX))) .collect(); for (i, node) in spec.nodes.iter().enumerate() { // A `parent` must index an earlier node — `insert_group` resolves it to // an already-inserted `NodeId`, so a forward/out-of-bounds parent would // otherwise panic there. if let Some(p) = node.parent && usize::try_from(p).is_ok_and(|p| p >= i) { bail!( "dag spec {:?} node {i} has invalid parent {p} (must be an earlier node)", spec.reason ); } for dep in &node.deps { let Some(&dep_idx) = usize::try_from(dep.on).ok().and_then(|i| idx.get(i)) else { bail!( "dag spec {:?} node {i} depends on unknown node {}", spec.reason, dep.on ); }; graph.add_edge(dep_idx, idx[i], ()); } } if petgraph::algo::toposort(&graph, None).is_err() { bail!("dag spec {:?} contains a dependency cycle", spec.reason); } Ok(()) }