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_statuses,
39    CancelOutcome, NodeCancelOutcome,
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, AppendOutcome, AppendResult, PriorEvent, Quarantine,
47};
48pub use ids::{format_node_id, new_op_id, new_run_id};
49pub use lock::{Exclusive, LockedRun, RunLock, Shared};
50pub use paths::{nofollow, run_dir, validate_run_id, RunPaths};
51pub use projections::{read_manifest, read_manifest_opt, read_node, read_node_opt, write_node};
52pub use reducer::{plan_projections, validate_event, KIND_MERGE_ABORTED, KIND_MERGE_STARTED};
53pub use report::{
54    sanitize_report_advisory, validate_report_payload, AdvisoryWarning, ReportOrigin,
55    ReportValidationError, SanitizedReport, REPORT_ORIGIN_KEY, VIA_EXPLICIT_MERGE,
56};
57pub use schema::aggregate_terminal_status;
58pub use schema::{
59    is_run_id_prefix, AgentSelection, AwaitingInput, ChildRef, Event, IdValidationError, Kind,
60    Lifecycle, Manifest, MergeTxn, Node, NodeId, RunId, SelectedAgentCandidate,
61    SkippedAgentCandidate, Status, WorkerExit, STATE_SCHEMA_VERSION, SUPPORTED_STATE_SCHEMAS,
62};
63pub use telemetry::{
64    parse_telemetry_update, read_all_telemetry, read_all_telemetry_with_clock, read_telemetry,
65    read_telemetry_with_clock, update_telemetry, update_telemetry_with_clock, SystemTelemetryClock,
66    TelemetryAccepted, TelemetryClock, TelemetryError, TelemetrySampleStatus, TelemetryState,
67    TelemetryUpdate, TelemetryView, TELEMETRY_FRESHNESS_SECS, TELEMETRY_MAX_BYTES,
68    TELEMETRY_PROTOCOL_VERSION, TELEMETRY_SCHEMA_VERSION,
69};
70
71/// Ensure the taskfleet root directory exists (`<root>/runs`,
72/// `<root>/logs`). Idempotent.
73pub fn ensure_root(root: &std::path::Path) -> Result<()> {
74    for sub in ["runs", "logs"] {
75        let p = root.join(sub);
76        std::fs::create_dir_all(&p).map_err(|e| Error::io(&p, e))?;
77    }
78    Ok(())
79}