Skip to main content

taskfleet_core/
lib.rs

1//! Core library for taskfleet.
2//!
3//! See `issues/taskfleet-mvp/design.md` for the canonical schema and
4//! protocol references. This crate provides:
5//!
6//! - The on-disk schema types ([`Manifest`], [`Node`], [`Event`]).
7//! - Atomic write helpers ([`atomic`]) and per-run advisory `flock`
8//!   ([`RunLock`]).
9//! - The canonical mutation entry point
10//!   ([`append_and_apply_event`]): append one
11//!   event and fold it into the projections under the run's `flock`.
12//!
13//! Higher-level supervisor and CLI logic live in their own crates / issues.
14//!
15//! `taskfleet-core` is the canonical library surface, so public items are required
16//! to carry doc comments (`#![warn(missing_docs)]`). Lint-level policy
17//! otherwise lives in the workspace `[workspace.lints]` table (pedantic clippy).
18#![warn(missing_docs)]
19
20pub mod atomic;
21pub mod cancel;
22pub mod envelope;
23pub mod error;
24pub mod events;
25pub mod ids;
26pub mod lock;
27pub mod paths;
28pub mod projections;
29pub mod reducer;
30pub mod report;
31pub mod schema;
32pub mod telemetry;
33
34#[cfg(test)]
35mod stress_tests;
36
37pub use cancel::{
38    cancel_node, cancel_node_unlocked, cancel_run, cancel_run_unlocked, read_node_status_facts,
39    read_node_statuses, CancelOutcome, NodeCancelOutcome, NodeStatusFact,
40};
41pub use envelope::SCHEMA_VERSION;
42pub use error::{Error, Result};
43pub use events::{
44    append_and_apply_event, append_and_apply_idempotent, append_and_apply_unlocked,
45    find_prior_with_key, quarantine_corrupt_lines, quarantine_corrupt_lines_unlocked,
46    read_all_events, recover_last_seq, replay_unapplied_unlocked, AppendOutcome, AppendResult,
47    PriorEvent, Quarantine,
48};
49pub use ids::{format_node_id, new_op_id, new_run_id};
50pub use lock::{Exclusive, LockedRun, RunLock, Shared};
51pub use paths::{nofollow, run_dir, validate_run_id, RunPaths};
52pub use projections::{read_manifest, read_manifest_opt, read_node, read_node_opt, write_node};
53pub use reducer::{plan_projections, validate_event, KIND_MERGE_ABORTED, KIND_MERGE_STARTED};
54pub use report::{
55    sanitize_report_advisory, validate_report_payload, AdvisoryWarning, ReportOrigin,
56    ReportValidationError, SanitizedReport, REPORT_ORIGIN_KEY, VIA_EXPLICIT_MERGE,
57};
58pub use schema::aggregate_terminal_status;
59pub use schema::{
60    is_run_id_prefix, AgentOwner, AgentSelection, AwaitingInput, ChildRef, Event, EvidenceStatus,
61    IdValidationError, Kind, Lifecycle, Manifest, MergeTxn, Node, NodeId, RetainedDisplay, RunId,
62    SelectedAgentCandidate, SkippedAgentCandidate, Status, TmuxIdentity, TmuxRetentionPolicy,
63    WorkerEvidence, WorkerExit, STATE_SCHEMA_VERSION, SUPPORTED_STATE_SCHEMAS,
64};
65pub use telemetry::{
66    parse_telemetry_update, read_all_telemetry, read_all_telemetry_with_clock, read_telemetry,
67    read_telemetry_with_clock, update_telemetry, update_telemetry_with_clock, SystemTelemetryClock,
68    TelemetryAccepted, TelemetryClock, TelemetryError, TelemetrySampleStatus, TelemetryState,
69    TelemetryUpdate, TelemetryView, TELEMETRY_FRESHNESS_SECS, TELEMETRY_MAX_BYTES,
70    TELEMETRY_PROTOCOL_VERSION, TELEMETRY_SCHEMA_VERSION,
71};
72
73/// Ensure the taskfleet root directory exists (`<root>/runs`,
74/// `<root>/logs`). Idempotent.
75pub fn ensure_root(root: &std::path::Path) -> Result<()> {
76    for sub in ["runs", "logs"] {
77        let p = root.join(sub);
78        std::fs::create_dir_all(&p).map_err(|e| Error::io(&p, e))?;
79    }
80    Ok(())
81}