use std::io::BufRead;
use std::ops::{Deref, DerefMut};
use std::path::Path;
use std::time::Duration;
use anyhow::{anyhow, Result};
use sha2::{Digest, Sha256};
use tokio::sync::mpsc;
use crate::chat::types::{ConversationEvent, ConversationItem, Lifecycle};
use crate::child::ChildRef;
use crate::controller::wave::playhead::{
BodyProvenance, Playhead, PlayheadEvent, QueuedInvocation, StepKind, StepOutcome,
};
use crate::durable::{FlowPosition, Steer, WorkRef};
use crate::harness::{drain_turn_failure_reason, ApprovalPolicy, Harness};
use crate::planning::ProjectPlan;
use crate::store::SharedStore;
use crate::work::project::Project;
use crate::work::task::{PrPhase, Task, TaskEventKind, TaskId};
use crate::work::wave::Wave;
mod state;
pub(crate) use state::State;
pub use state::{TaskGateProposal, TaskLifecyclePhase, TaskLifecyclePlan, TaskPhasePlan};
const STEER_POLL_INTERVAL: Duration = Duration::from_secs(5);
#[derive(Debug)]
struct PreparedTaskStep {
turn: crate::lf::commands::run::PreparedHarnessTurn,
position: StepPosition,
seeded_steer_id: i64,
interrupt_id: i64,
}
#[derive(Debug, Clone, Copy)]
struct ControlCursors {
steer: i64,
interrupt: i64,
}
impl ControlCursors {
fn from_prepared(prepared: &PreparedTaskStep) -> Self {
Self {
steer: prepared.seeded_steer_id,
interrupt: prepared.interrupt_id,
}
}
}
#[derive(Debug)]
struct StepPosition {
human: bool,
}
#[derive(Debug, Clone)]
struct ControlledTask {
work: Task,
state: State,
}
impl Deref for ControlledTask {
type Target = Task;
fn deref(&self) -> &Self::Target {
&self.work
}
}
impl DerefMut for ControlledTask {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.work
}
}
fn task_controller_state(task: &ControlledTask) -> &State {
&task.state
}
fn task_controller_state_mut(task: &mut ControlledTask) -> &mut State {
&mut task.state
}
async fn controlled_task(store: &SharedStore, task_id: &TaskId) -> Result<ControlledTask> {
let work = store
.get_task(task_id)
.await?
.ok_or_else(|| anyhow!("Task {task_id} not found"))?;
let state = store
.task_controller_state(task_id)
.await?
.ok_or_else(|| anyhow!("Task {task_id} has no end-to-end controller state"))?;
Ok(ControlledTask { work, state })
}
pub(crate) async fn run(store: SharedStore, task_id: TaskId) -> Result<()> {
let result = run_task_with(
store.clone(),
task_id.clone(),
Box::new(crate::harness::default_create_harness),
)
.await;
if let Err(error) = &result {
record_unhandled_failure(&store, &task_id, error).await;
}
result
}
fn take_task_launch_env(key: &'static str, label: &str) -> Result<Option<String>> {
let Some(value) = std::env::var_os(key) else {
return Ok(None);
};
std::env::remove_var(key);
value
.into_string()
.map(Some)
.map_err(|_| anyhow!("{label} is not valid UTF-8"))
}
async fn owning_wave(store: &SharedStore, task: &ControlledTask) -> Result<Wave> {
store
.get_wave(&task.wave_id)
.await?
.ok_or_else(|| anyhow!("owning Wave {} is not registered", task.wave_id))
}
async fn owning_project(store: &SharedStore, task: &ControlledTask) -> Result<Project> {
store
.get_project(&task.project_id)
.await?
.ok_or_else(|| anyhow!("owning Project {} is not registered", task.project_id))
}
async fn run_task_with(
store: SharedStore,
task_id: TaskId,
create_harness: crate::harness::CreateHarness,
) -> Result<()> {
let mut task = controlled_task(&store, &task_id).await?;
let wave = owning_wave(&store, &task).await?;
let project = owning_project(&store, &task).await?;
store
.append_task_event(&task.id, &TaskEventKind::Started)
.await?;
let mut flow = resume_task_phase(&task)?;
let Some(mut prepared) =
prepare_task_flow_step(&store, &mut task, wave.name(), &mut flow).await?
else {
return Ok(());
};
let (harness_name, _) = crate::engine::config::parse_agent(&task_controller_state(&task).agent);
let capture = crate::run_record::CaptureHandle::begin_with_context(
crate::run_record::RunSpec {
harness: prepared.turn.harness.clone(),
model: prepared.turn.model.clone(),
surface: "headless".to_string(),
cwd: Path::new(&task.worktree).to_path_buf(),
repo: Some(Path::new(wave.repo()).to_path_buf()),
worktree: Some(Path::new(&task.worktree).to_path_buf()),
skill: flow.current().map(|step| step.step.clone()),
subjects: vec![
crate::run_record::SubjectAttribution::declared(format!("wave:{}", wave.name())),
crate::run_record::SubjectAttribution::declared(format!(
"project:{}",
project.plan.slug
)),
crate::run_record::SubjectAttribution::declared(format!(
"task:{}",
task.plan.identifier
)),
],
},
&prepared.turn.context,
)?;
capture.record_input("initial", &prepared.turn.input);
prepared.turn.config.env.extend(capture.environment());
capture.mark_spawn_requested();
let capture = Some(capture);
let (event_tx, mut event_rx) = mpsc::unbounded_channel();
let mut harness = create_harness(&harness_name, ApprovalPolicy::AutoApprove, event_tx)
.inspect_err(|_| {
finish_capture(capture.as_ref(), "failed");
})?;
let requested_account =
take_task_launch_env(crate::ops::TASK_ACCOUNT_ID_ENV, "Task provider account id")?
.map(|value| crate::store::ProviderAccountId::parse(&value))
.transpose()
.map_err(|reason| anyhow!("invalid Task provider account route: {reason}"))?;
harness.set_provider_account_id(requested_account);
harness.set_provider_session_id(
take_task_launch_env(crate::ops::TASK_RESUME_TOKEN_ENV, "Task resume token")?
.or_else(|| task_controller_state(&task).provider_session_id.clone()),
);
if let Err(error) = harness.start(&prepared.turn.config).await {
finish_capture(capture.as_ref(), "failed");
return Err(error);
}
{
let resident = task_controller_state_mut(&mut task);
resident.provider = harness_name;
resident.provider_session_id = harness.provider_session_id();
resident.updated_at = time::OffsetDateTime::now_utc();
}
if let Err(error) = store.put_task_controller_state(&task.state).await {
let _ = harness.stop().await;
return Err(error.into());
}
let mut state_fingerprint = task_state_fingerprint(&task)?;
let mut gate_fingerprint =
if task_controller_state(&task).lifecycle_phase == TaskLifecyclePhase::Finally {
Some(task_gate_fingerprint(&task)?)
} else {
None
};
if let Some(capture) = &capture {
capture.set_provider_session_id(task_controller_state(&task).provider_session_id.clone());
}
let mut control_cursors = ControlCursors::from_prepared(&prepared);
start_prepared_task_step(&mut task, harness.as_mut(), &mut flow, None, prepared).await?;
let (attachment_tx, mut attachment_rx) = mpsc::unbounded_channel();
std::thread::spawn(move || {
for line in std::io::stdin().lock().lines() {
let Ok(line) = line else { break };
if attachment_tx.send(line).is_err() {
break;
}
}
});
println!(
"task {}> attached; /status, /interrupt, /detach, or type a message/instruction",
task.plan.identifier
);
let mut last_text = String::new();
let mut command_failures = Vec::new();
let work = WorkRef::Task(task.id.clone());
let mut steer_tick = tokio::time::interval(STEER_POLL_INTERVAL);
steer_tick.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);
'runner: loop {
tokio::select! {
_ = steer_tick.tick() => {
crate::ops::child::inject_live_steers(
&store, &work, harness.as_mut(), &mut control_cursors.steer,
).await;
crate::ops::child::observe_interrupt(
&store, &work, harness.as_mut(), &mut control_cursors.interrupt,
).await;
}
line = attachment_rx.recv() => {
if let Some(line) = line {
handle_attachment(
&store,
&task,
harness.as_mut(),
line,
).await?;
}
}
event = event_rx.recv() => {
let Some(event) = event else {
return finish_failed(
&store,
&mut task,
harness.as_mut(),
"provider event stream closed",
capture.as_ref(),
).await;
};
if let Some(capture) = &capture {
capture.record_conversation(event.clone());
}
let provider_session_id = harness.provider_session_id();
if provider_session_id != task_controller_state(&task).provider_session_id {
let resident = task_controller_state_mut(&mut task);
resident.provider_session_id = provider_session_id;
resident.updated_at = time::OffsetDateTime::now_utc();
store.put_task_controller_state(&task.state).await?;
}
match event {
ConversationEvent::TextDelta { content, .. } => last_text.push_str(&content),
ConversationEvent::TurnStarted { .. } => {
command_failures.clear();
}
ConversationEvent::ItemCompleted { item, .. } => {
if let ConversationItem::Command {
command,
status,
output,
exit_code,
..
} = &item
{
if let Some(failure) = completed_boundary_failure(
command,
*status,
output.as_deref(),
*exit_code,
) {
if !command_failures.contains(&failure) {
command_failures.push(failure);
}
}
}
}
ConversationEvent::TurnCompleted { status, .. } => {
if status == Lifecycle::Failed {
let reason = drain_turn_failure_reason(
&mut event_rx,
"provider turn failed",
);
return finish_body_failure(
&store,
&mut task,
harness.as_mut(),
&reason,
capture.as_ref(),
)
.await;
}
if execution_blocker_at_handoff(status, &command_failures).is_some() {
return finish_execution_blocked(
&store,
&mut task,
harness.as_mut(),
&command_failures,
capture.as_ref(),
)
.await;
}
let mut flow_iteration_completed =
finish_task_flow_turn(&mut flow, status)?;
let mut finally_ops_ran = false;
if !flow_iteration_completed
&& flow.current().is_some_and(|step| step.kind == StepKind::Op)
{
let current_fingerprint = task_gate_fingerprint(&task)?;
if gate_fingerprint.as_ref() == Some(¤t_fingerprint) {
flow_iteration_completed =
run_task_flow_ops(&task, &mut flow).await?;
finally_ops_ran = true;
} else {
flow_iteration_completed = true;
}
}
let latest = controlled_task(&store, &task.id).await?;
sync_task_state(&mut task, &latest);
record_task_flow_position(&mut task, &flow)?;
store.put_task_controller_state(&task.state).await?;
if status == Lifecycle::Interrupted {
let Some(prepared) = prepare_task_flow_step(
&store,
&mut task,
wave.name(),
&mut flow,
)
.await?
else {
return park_task_at_human(
Some(harness.as_mut()),
capture.as_ref(),
)
.await;
};
control_cursors = ControlCursors::from_prepared(&prepared);
start_prepared_task_step(
&mut task,
harness.as_mut(),
&mut flow,
capture.as_ref(),
prepared,
)
.await?;
last_text.clear();
continue 'runner;
}
if flow_iteration_completed
&& task_controller_state(&task).lifecycle_phase == TaskLifecyclePhase::First
{
task_controller_state_mut(&mut task).enter_loop()?;
store.put_task_controller_state(&task.state).await?;
flow = resume_task_phase(&task)?;
flow_iteration_completed = false;
state_fingerprint = task_state_fingerprint(&task)?;
gate_fingerprint = None;
last_text.clear();
}
let approved_gate = if flow_iteration_completed
&& task_controller_state(&task).lifecycle_phase == TaskLifecyclePhase::Finally
{
let next_gate_fingerprint = task_gate_fingerprint(&task)?;
if !finally_ops_ran
&& gate_fingerprint.as_ref() != Some(&next_gate_fingerprint)
{
state_fingerprint = task_state_fingerprint(&task)?;
gate_fingerprint = None;
task_controller_state_mut(&mut task).enter_loop()?;
store.put_task_controller_state(&task.state).await?;
let Some(()) = start_resumed_task_phase(
&store,
&mut task,
harness.as_mut(),
&mut flow,
wave.name(),
capture.as_ref(),
&mut control_cursors,
)
.await?
else {
return park_task_at_human(
Some(harness.as_mut()),
capture.as_ref(),
)
.await;
};
last_text.clear();
continue 'runner;
}
Some(task_controller_state(&task).approved_gate_proposal()?)
} else {
None
};
if !flow_iteration_completed {
let Some(prepared) = prepare_task_flow_step(
&store,
&mut task,
wave.name(),
&mut flow,
)
.await?
else {
return park_task_at_human(
Some(harness.as_mut()),
capture.as_ref(),
)
.await;
};
control_cursors = ControlCursors::from_prepared(&prepared);
start_prepared_task_step(
&mut task,
harness.as_mut(),
&mut flow,
capture.as_ref(),
prepared,
)
.await?;
continue 'runner;
}
let summary = progress_summary(&last_text);
let latest = controlled_task(&store, &task.id).await?;
sync_task_state(&mut task, &latest);
let observed_pr = crate::ops::task::reconcile_task_pr(
&store,
&mut task,
)
.await
.map_err(|error| anyhow!(error.to_string()))?;
let (stopped_done, stopped_reason) = if let Some(proposal) = approved_gate {
(proposal.done, proposal.reason)
} else if let Some(pr) = observed_pr
.as_ref()
.filter(|pr| pr.phase() == PrPhase::Open)
{
(false, crate::ops::task::open_pr_wait_reason(pr))
} else {
let next_fingerprint = task_state_fingerprint(&task)?;
if next_fingerprint != state_fingerprint {
state_fingerprint = next_fingerprint;
store.put_task_controller_state(&task.state).await?;
let Some(prepared) = prepare_task_flow_step(
&store,
&mut task,
wave.name(),
&mut flow,
)
.await?
else {
return park_task_at_human(
Some(harness.as_mut()),
capture.as_ref(),
)
.await;
};
control_cursors = ControlCursors::from_prepared(&prepared);
start_prepared_task_step(
&mut task,
harness.as_mut(),
&mut flow,
capture.as_ref(),
prepared,
)
.await?;
last_text.clear();
continue 'runner;
}
(
false,
"Task flow completed without a PR or any worktree change; another automatic iteration would spin".to_string(),
)
};
if task_controller_state(&task).lifecycle_phase == TaskLifecyclePhase::Loop {
task_controller_state_mut(&mut task).enter_finally(TaskGateProposal {
done: stopped_done,
reason: stopped_reason,
})?;
gate_fingerprint = Some(task_gate_fingerprint(&task)?);
store.put_task_controller_state(&task.state).await?;
let Some(()) = start_resumed_task_phase(
&store,
&mut task,
harness.as_mut(),
&mut flow,
wave.name(),
capture.as_ref(),
&mut control_cursors,
)
.await?
else {
return park_task_at_human(
Some(harness.as_mut()),
capture.as_ref(),
)
.await;
};
last_text.clear();
continue 'runner;
}
store.put_task_controller_state(&task.state).await?;
let _ = harness.stop().await;
finish_capture(capture.as_ref(), "completed");
if !summary.is_empty() {
store.append_task_event(
&task.id,
&TaskEventKind::Progress {
summary: summary.clone(),
},
).await?;
}
if stopped_done {
store.complete_task(&task, None).await?;
}
return Ok(());
}
ConversationEvent::Error { code, message, .. } => {
let reason = format!("{code}: {message}");
return finish_body_failure(
&store,
&mut task,
harness.as_mut(),
&reason,
capture.as_ref(),
)
.await;
}
ConversationEvent::ItemStarted { .. }
| ConversationEvent::ItemUpdated { .. }
| ConversationEvent::ReasoningDelta { .. }
| ConversationEvent::DiffUpdated { .. }
| ConversationEvent::UsageCheckpoint { .. }
| ConversationEvent::SuggestedActions { .. }
| ConversationEvent::StatusChanged { .. } => {}
}
}
}
}
}
async fn prepare_task_flow_step_once(
store: &SharedStore,
task: &mut ControlledTask,
wave_name: &str,
flow: &Playhead,
) -> Result<PreparedTaskStep> {
let work = store
.work_for_child(&ChildRef::Task(task.id.clone()))
.await?;
let steers = store.work_steers(&work).await?;
let seeded_steer_id = steers.last().map_or(0, |steer| steer.id);
let interrupt_id = store.latest_interrupt_id(&work).await?;
let step = flow
.current()
.ok_or_else(|| anyhow!("Task flow has no current step"))?;
if step.kind != StepKind::Skill {
anyhow::bail!(
"Task flow step {} is {:?}; durable Task flows currently require skills",
step.step,
step.kind
);
}
store.put_task_controller_state(&task.state).await?;
let pr = store
.active_task_pr(&task.id)
.await?
.ok_or_else(|| anyhow!("Task {} has no active PR", task.id))?;
let project = owning_project(store, task).await?;
let seed = format!(
"{}\n\n{}",
task_seed(task, &project.plan, &pr, wave_name, &steers),
crate::ops::task::task_workspace_context(task, &pr)
.map_err(|error| anyhow!(error.to_string()))?
);
let mut prepared = crate::lf::commands::run::prepare_harness_turn_at(
&step.step,
&seed,
wave_name,
None,
&task.worktree,
)?;
prepared.config.agent = Some(task_controller_state(task).agent.clone());
prepared.config.write_scope = crate::engine::agent::AgentWriteScope::Worktree;
prepared.config.execution_boundary = Some(
crate::ops::task::task_execution_boundary(
&task.worktree,
&task_controller_state(task).agent,
)
.map_err(|error| anyhow!(error.to_string()))?,
);
prepared.config.skip_permissions = true;
let position = StepPosition {
human: step.policy.human,
};
Ok(PreparedTaskStep {
turn: prepared,
position,
seeded_steer_id,
interrupt_id,
})
}
async fn prepare_task_flow_step(
store: &SharedStore,
task: &mut ControlledTask,
wave_name: &str,
flow: &mut Playhead,
) -> Result<Option<PreparedTaskStep>> {
let step = flow
.current()
.ok_or_else(|| anyhow!("Task flow has no current step"))?;
if !step.policy.human {
return prepare_task_flow_step_once(store, task, wave_name, flow)
.await
.map(Some);
}
if step.kind != StepKind::Skill {
anyhow::bail!(
"Task flow step {} is {:?}; durable Task flows currently require skills",
step.step,
step.kind
);
}
let node_id = step
.policy
.id
.clone()
.ok_or_else(|| anyhow!("human Task flow step has no stable node id"))?;
let work = store
.work_for_child(&ChildRef::Task(task.id.clone()))
.await?;
store.put_task_controller_state(&task.state).await?;
let mut position = FlowPosition {
work,
flow: task_controller_state(task).phase_plan().flow.clone(),
step: step.step.clone(),
node_id: Some(node_id.clone()),
human: true,
session_run_id: None,
ready_summary: None,
step_index: step.index,
iteration: step.iteration,
updated_at: time::OffsetDateTime::now_utc(),
};
if let Some(previous) = store.flow_position(&position.work).await? {
if previous.flow == position.flow
&& previous.step == position.step
&& previous.node_id == position.node_id
&& previous.iteration == position.iteration
{
position.session_run_id = previous.session_run_id;
position.ready_summary = previous.ready_summary;
}
}
checkpoint_worktree_before_human(task, &node_id).await;
store
.set_flow_position(&position.work, position.clone())
.await?;
if let Err(error) = crate::ops::human_session::prepare(store, task, &position).await {
tracing::warn!(task = %task.id, node = %node_id, %error, "human flow session did not start; the Task remains waiting");
}
tracing::info!(task = %task.id, node = %node_id, "Task is waiting at a human flow node");
Ok(None)
}
async fn complete_human_task_step(
store: &SharedStore,
task: &mut ControlledTask,
flow: &mut Playhead,
) -> Result<()> {
open_task_flow_body(flow, task)?;
let completed = finish_task_flow_turn(flow, Lifecycle::Completed)?;
if completed && task_controller_state(task).lifecycle_phase == TaskLifecyclePhase::First {
task_controller_state_mut(task).enter_loop()?;
*flow = resume_task_phase(task)?;
} else {
record_task_flow_position(task, flow)?;
}
store.put_task_controller_state(&task.state).await?;
Ok(())
}
pub(crate) async fn decide_human_flow_step(
store: &SharedStore,
token: &crate::ops::human_session::FlowSessionToken,
decision: crate::ops::human_session::FlowDecision,
text: &str,
) -> Result<()> {
let text = text.trim();
if text.is_empty() {
anyhow::bail!("human flow settlement text cannot be empty");
}
if !crate::ops::human_session::token_is_current(store, token).await? {
anyhow::bail!("human flow session is stale");
}
let mut task = controlled_task(store, &token.task_id).await?;
let mut flow = resume_task_phase(&task)?;
let step = flow
.current()
.ok_or_else(|| anyhow!("Task {} has no current flow step", task.id))?;
if !step.policy.human
|| step.policy.id.as_deref() != Some(token.node_id.as_str())
|| step.step != token.skill
|| step.iteration != token.iteration
|| step.flow != token.flow
{
anyhow::bail!("human flow session no longer matches the Task playhead");
}
match decision {
crate::ops::human_session::FlowDecision::Approve => {
store
.append_task_event(
&task.id,
&TaskEventKind::Progress {
summary: text.to_string(),
},
)
.await?;
complete_human_task_step(store, &mut task, &mut flow).await?;
}
crate::ops::human_session::FlowDecision::Iterate => {
let work = WorkRef::Task(task.id.clone());
store
.append_steer(
&work,
crate::durable::Author::User,
&format!(
"Human requested another iteration on flow node {}: {text}",
token.node_id
),
)
.await?;
let state = task_controller_state_mut(&mut task);
state.phase_cursor = preceding_autonomous_step(&flow, step.index)?;
state.phase_iteration += 1;
state.updated_at = time::OffsetDateTime::now_utc();
store.put_task_controller_state(&task.state).await?;
flow = resume_task_phase(&task)?;
}
}
let position = current_flow_position(&task, &flow)?;
let work = position.work.clone();
store.set_flow_position(&work, position).await?;
Ok(())
}
fn current_flow_position(task: &ControlledTask, flow: &Playhead) -> Result<FlowPosition> {
let step = flow
.current()
.ok_or_else(|| anyhow!("Task {} has no next flow step", task.id))?;
Ok(FlowPosition {
work: WorkRef::Task(task.id.clone()),
flow: task_controller_state(task).phase_plan().flow.clone(),
step: step.step,
node_id: step.policy.id,
human: step.policy.human,
session_run_id: None,
ready_summary: None,
step_index: step.index,
iteration: step.iteration,
updated_at: time::OffsetDateTime::now_utc(),
})
}
fn preceding_autonomous_step(flow: &Playhead, current: u32) -> Result<u32> {
let root = flow
.stack
.first()
.ok_or_else(|| anyhow!("Task flow has no root invocation"))?;
root.steps
.iter()
.enumerate()
.take(current as usize)
.rev()
.find(|(_, step)| step.kind == StepKind::Skill && !step.policy.human)
.map(|(index, _)| index as u32)
.ok_or_else(|| anyhow!("human Task flow node has no preceding autonomous skill"))
}
fn open_task_flow_body(flow: &mut Playhead, task: &ControlledTask) -> Result<()> {
let step = flow
.current()
.ok_or_else(|| anyhow!("Task flow has no current step"))?;
if step.kind != StepKind::Skill {
anyhow::bail!("Task flow step {} is not a skill", step.step);
}
flow.start_body(BodyProvenance::for_step(&step, &task.worktree))?;
Ok(())
}
async fn start_task_flow_turn(
task: &mut ControlledTask,
harness: &mut dyn Harness,
flow: &mut Playhead,
prepared: crate::lf::commands::run::PreparedHarnessTurn,
) -> Result<()> {
open_task_flow_body(flow, task)?;
harness.send_input(&prepared.input).await?;
Ok(())
}
async fn start_prepared_task_step(
task: &mut ControlledTask,
harness: &mut dyn Harness,
flow: &mut Playhead,
capture: Option<&crate::run_record::CaptureHandle>,
prepared: PreparedTaskStep,
) -> Result<()> {
debug_assert!(!prepared.position.human);
if let Some(capture) = capture {
capture.record_input("queued", &prepared.turn.input);
}
start_task_flow_turn(task, harness, flow, prepared.turn).await
}
async fn start_resumed_task_phase(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
flow: &mut Playhead,
wave_name: &str,
capture: Option<&crate::run_record::CaptureHandle>,
control_cursors: &mut ControlCursors,
) -> Result<Option<()>> {
*flow = resume_task_phase(task)?;
let Some(prepared) = prepare_task_flow_step(store, task, wave_name, flow).await? else {
return Ok(None);
};
*control_cursors = ControlCursors::from_prepared(&prepared);
start_prepared_task_step(task, harness, flow, capture, prepared).await?;
Ok(Some(()))
}
fn finish_task_flow_turn(flow: &mut Playhead, status: Lifecycle) -> Result<bool> {
let body_id = flow
.active
.as_ref()
.map(|body| body.body_id.clone())
.ok_or_else(|| anyhow!("Task flow turn completed without an active body"))?;
let outcome = match status {
Lifecycle::Completed => StepOutcome::Completed,
Lifecycle::Interrupted => StepOutcome::Interrupted,
_ => anyhow::bail!("Task flow turn ended with unexpected status {status:?}"),
};
let events = flow.finish_body(&body_id, outcome, status.name())?;
Ok(events
.iter()
.any(|event| matches!(event, PlayheadEvent::InvocationCompleted { .. })))
}
async fn run_task_flow_ops(task: &ControlledTask, flow: &mut Playhead) -> Result<bool> {
if task_controller_state(task).lifecycle_phase != TaskLifecyclePhase::Finally {
anyhow::bail!(
"Task {} {} flow reached an op; only finally flows may run mechanical ops",
task.id,
task_controller_state(task).lifecycle_phase.as_str()
);
}
while let Some(step) = flow.current() {
if step.kind != StepKind::Op {
return Ok(false);
}
let definition = crate::engine::load_flow(&step.flow, &task.worktree)?;
let items = crate::engine::expand_flow(&definition, &task.worktree)?;
let op = match items.get(step.index as usize) {
Some(crate::engine::ConcreteStep::Op(op)) => op.clone(),
Some(item) => {
anyhow::bail!(
"Task flow step {} was planned as an op but expanded to {item:?}",
step.step
)
}
None => anyhow::bail!("Task flow op {} is outside its expanded flow", step.step),
};
let body = BodyProvenance::for_step(&step, &task.worktree);
let body_id = body.body_id.clone();
flow.start_body(body)?;
let worktree = task.worktree.clone();
let result = tokio::task::spawn_blocking(move || {
crate::ops::execute_flow_ops(&worktree, &op.item, &crate::ops::NullProgress)
})
.await
.map_err(|error| anyhow!("Task flow op worker failed: {error}"))?;
if let Err(error) = result {
flow.finish_body(&body_id, StepOutcome::Failed, &error.to_string())?;
return Err(anyhow!(error.to_string()));
}
let events = flow.finish_body(&body_id, StepOutcome::Completed, "completed")?;
if events
.iter()
.any(|event| matches!(event, PlayheadEvent::InvocationCompleted { .. }))
{
return Ok(true);
}
}
Ok(true)
}
fn finish_capture(capture: Option<&crate::run_record::CaptureHandle>, outcome: &str) {
let Some(capture) = capture else { return };
if let Err(error) = capture.finish(outcome) {
tracing::warn!(%error, "failed to finish Task Run record");
}
}
async fn checkpoint_worktree_before_human(task: &ControlledTask, node_id: &str) {
if let Err(error) = crate::ops::checkpoint_task_worktree(
task.worktree.clone(),
task.plan.identifier.clone(),
format!("checkpoint: park at human node {node_id}"),
)
.await
{
tracing::warn!(task = %task.id, %error, "Task parks at a human node without a pushed checkpoint");
}
}
async fn park_task_at_human(
harness: Option<&mut dyn Harness>,
capture: Option<&crate::run_record::CaptureHandle>,
) -> Result<()> {
finish_capture(capture, "completed");
if let Some(harness) = harness {
let _ = harness.stop().await;
}
Ok(())
}
fn record_task_flow_position(task: &mut ControlledTask, flow: &Playhead) -> Result<()> {
let root = flow
.stack
.first()
.ok_or_else(|| anyhow!("Task flow has no root invocation"))?;
if root.flow != task_controller_state(task).phase_plan().flow {
anyhow::bail!(
"Task {} {} flow is {:?}, but its playhead is {:?}",
task.id,
task_controller_state(task).lifecycle_phase.as_str(),
task_controller_state(task).phase_plan().flow,
root.flow
);
}
let resident = task_controller_state_mut(task);
resident.phase_cursor = root.cursor;
resident.phase_iteration = root.iteration;
resident.updated_at = time::OffsetDateTime::now_utc();
Ok(())
}
fn resume_task_phase(task: &ControlledTask) -> Result<Playhead> {
let (flow, _) = Playhead::resume_root(
QueuedInvocation::load(
&task.worktree,
&task_controller_state(task).phase_plan().flow,
)?,
task_controller_state(task).phase_cursor,
task_controller_state(task).phase_iteration,
)?;
Ok(flow)
}
fn sync_task_state(task: &mut ControlledTask, latest: &ControlledTask) {
task.pm_writeback = latest.pm_writeback.clone();
if task_controller_state(task).lifecycle_phase == TaskLifecyclePhase::Finally
&& task_controller_state(latest).lifecycle_phase == TaskLifecyclePhase::Finally
{
task_controller_state_mut(task).gate_proposal =
task_controller_state(latest).gate_proposal.clone();
} else if task_controller_state(task).lifecycle_phase != TaskLifecyclePhase::Finally {
task_controller_state_mut(task).gate_proposal = None;
}
}
fn task_state_fingerprint(task: &ControlledTask) -> Result<String> {
let state = crate::engine::git::worktree_state(Path::new(&task.worktree))?;
Ok(hex::encode(Sha256::digest(state.as_bytes())))
}
fn task_gate_fingerprint(task: &ControlledTask) -> Result<String> {
let state = crate::engine::git::material_worktree_state(Path::new(&task.worktree))?;
Ok(hex::encode(Sha256::digest(state.as_bytes())))
}
async fn handle_attachment(
store: &SharedStore,
task: &ControlledTask,
harness: &mut dyn Harness,
line: String,
) -> Result<()> {
let line = line.trim();
if line.is_empty() {
return Ok(());
}
if line == "/status" {
let work = store
.work_for_child(&ChildRef::Task(task.id.clone()))
.await?;
println!(
"{} {:?}",
task.plan.identifier,
store.work_status(&work).await?
);
return Ok(());
}
if line == "/detach" {
let _ = std::process::Command::new("tmux")
.args(["detach-client"])
.status();
return Ok(());
}
let target = ChildRef::Task(task.id.clone());
if line == "/interrupt" {
harness.interrupt().await?;
println!("interrupted active provider turn");
} else {
let work = store.work_for_child(&target).await?;
let steer = store
.append_steer(&work, crate::durable::Author::User, line)
.await?;
harness.send_input(line).await?;
println!("queued {}", steer.id);
}
Ok(())
}
async fn record_unhandled_failure(store: &SharedStore, task_id: &TaskId, error: &anyhow::Error) {
let detail = error.to_string();
let (message, resumable) = unhandled_failure_receipt(&detail);
if store
.latest_task_event(task_id)
.await
.ok()
.flatten()
.is_some_and(|event| {
matches!(
event.kind,
TaskEventKind::Failed {
error,
resumable: recorded_resumable,
} if (error == detail || error == message) && recorded_resumable == resumable
)
})
{
return;
}
let event = TaskEventKind::Failed {
error: message,
resumable,
};
if let Err(persist_error) = store.append_task_event(task_id, &event).await {
tracing::error!(
task = %task_id,
error = %persist_error,
"Task failure receipt did not persist"
);
}
}
fn unhandled_failure_receipt(detail: &str) -> (String, bool) {
match provider_credential_blocker(detail) {
Some(message) => (message, false),
None => (format!("task process failed: {detail}"), true),
}
}
fn provider_credential_blocker(detail: &str) -> Option<String> {
crate::engine::agent::credential_invalidated_failure(detail).map(|_| {
format!(
"Task provider credential capability is blocked: {detail}. Reconnect the named managed account before starting a new Run"
)
})
}
async fn finish_failed(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
error: &str,
capture: Option<&crate::run_record::CaptureHandle>,
) -> Result<()> {
finish_capture(capture, "failed");
let _ = harness.stop().await;
store.put_task_controller_state(&task.state).await?;
store
.append_task_event(
&task.id,
&TaskEventKind::Failed {
error: error.to_string(),
resumable: true,
},
)
.await?;
anyhow::bail!(error.to_string())
}
fn completed_boundary_failure(
command: &[String],
status: Lifecycle,
output: Option<&str>,
exit_code: Option<i32>,
) -> Option<String> {
if status != Lifecycle::Failed && exit_code.is_none_or(|code| code == 0) {
return None;
}
let output = output?;
let lower = output.to_ascii_lowercase();
if ![
"operation not permitted",
"permission denied",
"read-only file system",
"network access is disabled",
"network is unreachable",
]
.iter()
.any(|marker| lower.contains(marker))
{
return None;
}
let command = command.join(" ");
let detail = output
.lines()
.rev()
.find(|line| !line.trim().is_empty())
.map(str::trim)
.unwrap_or("no command error output");
let detail = detail.chars().take(1_000).collect::<String>();
let exit = exit_code
.map(|code| format!(" (exit {code})"))
.unwrap_or_default();
Some(format!("`{command}` failed{exit}: {detail}"))
}
async fn finish_execution_blocked(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
failures: &[String],
capture: Option<&crate::run_record::CaptureHandle>,
) -> Result<()> {
let reason = execution_blocked_reason(failures);
finish_nonresumable(store, task, harness, &reason, capture).await
}
async fn finish_nonresumable(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
reason: &str,
capture: Option<&crate::run_record::CaptureHandle>,
) -> Result<()> {
finish_capture(capture, "failed");
let _ = harness.stop().await;
store.put_task_controller_state(&task.state).await?;
store
.append_task_event(
&task.id,
&TaskEventKind::Failed {
error: reason.to_string(),
resumable: false,
},
)
.await?;
Ok(())
}
fn execution_blocked_reason(failures: &[String]) -> String {
format!(
"Task execution boundary is blocked:\n- {}\nCorrect the named filesystem, control-plane, or network capability before starting a new Run.",
failures.join("\n- ")
)
}
fn execution_blocker_at_handoff(status: Lifecycle, failures: &[String]) -> Option<String> {
(status == Lifecycle::Completed && !failures.is_empty())
.then(|| execution_blocked_reason(failures))
}
fn infra_blocked_reason(capability: &str, detail: &str, pr_number: Option<u32>) -> String {
let pr_note = pr_number
.map(|n| format!(" Pull request #{n} stays attached."))
.unwrap_or_default();
format!("Task execution blocked by {capability}: {detail}.{pr_note}")
}
async fn finish_infra_blocked(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
capability: &str,
detail: &str,
) -> Result<()> {
let _ = harness.stop().await;
let pr_number = store
.active_task_pr(&task.id)
.await?
.and_then(|pr| pr.github().map(|g| g.number));
let reason = infra_blocked_reason(capability, detail, pr_number);
store.put_task_controller_state(&task.state).await?;
store
.append_task_event(
&task.id,
&TaskEventKind::Failed {
error: reason,
resumable: true,
},
)
.await?;
Ok(())
}
async fn finish_body_failure(
store: &SharedStore,
task: &mut ControlledTask,
harness: &mut dyn Harness,
reason: &str,
capture: Option<&crate::run_record::CaptureHandle>,
) -> Result<()> {
if let Some(blocker) = provider_credential_blocker(reason) {
return finish_nonresumable(store, task, harness, &blocker, capture).await;
}
if store.active_task_pr(&task.id).await?.is_some() {
finish_capture(capture, "failed");
return finish_infra_blocked(store, task, harness, "provider", reason).await;
}
finish_failed(store, task, harness, reason, capture).await
}
fn task_seed(
task: &ControlledTask,
project: &ProjectPlan,
pr: &crate::work::task::TaskPr,
wave_name: &str,
steers: &[Steer],
) -> String {
let context = crate::ops::render_task_context(
task,
Some(task_controller_state(task)),
project,
pr,
wave_name,
steers,
);
format!(
"{context}\n\nAdvance this Task through its pinned lifecycle. This PR owns one serial branch. The pinned finally flow owns landing and Task completion. `lf pr abandon` discards only this PR. If this PR already merged out of band and follow-up work remains, `lf pr next [slug]` rotates to the next serial PR, carrying committed and uncommitted follow-up forward. Watched landing owns automatic post-merge rotation."
)
}
fn progress_summary(text: &str) -> String {
const MAX_CHARS: usize = 2_000;
let text = text.trim();
if text.chars().count() <= MAX_CHARS {
return text.to_string();
}
let mut summary: String = text.chars().take(MAX_CHARS - 1).collect();
summary.push('…');
summary
}
#[cfg(test)]
struct TestLfBinGuard {
previous: Option<std::ffi::OsString>,
_lock: std::sync::MutexGuard<'static, ()>,
}
#[cfg(test)]
impl TestLfBinGuard {
fn pin() -> Self {
let lock = crate::journal::test_env_lock();
let previous = std::env::var_os("LF_BIN");
std::env::set_var("LF_BIN", std::env::current_exe().unwrap());
Self {
previous,
_lock: lock,
}
}
}
#[cfg(test)]
impl Drop for TestLfBinGuard {
fn drop(&mut self) {
match &self.previous {
Some(value) => std::env::set_var("LF_BIN", value),
None => std::env::remove_var("LF_BIN"),
}
}
}
#[cfg(test)]
mod planning_tests {
use super::{
completed_boundary_failure, execution_blocker_at_handoff, preceding_autonomous_step,
sync_task_state, task_seed, unhandled_failure_receipt, ControlledTask, State,
TaskGateProposal, TaskLifecyclePhase, TaskLifecyclePlan,
};
use crate::chat::types::Lifecycle;
use crate::controller::wave::playhead::{Playhead, QueuedInvocation, StepKind, StepPlan};
use crate::durable::{Author, RunId, WorkRef};
use crate::engine::agent::AgentConfig;
use crate::engine::OccurrencePolicy;
use crate::harness::{Harness, SendCurrentOutcome};
use crate::planning::{LinearIssueId, LinearProjectId, ProjectPlan, TaskPlan};
use crate::store::{SharedStore, StorageConfig};
use crate::work::project::{Project, ProjectId};
use crate::work::task::{
Observation, PmWritebackState, Task, TaskEventKind, TaskId, TaskPr, TaskPrId,
};
use crate::work::wave::Wave;
#[derive(Default)]
struct UnusedHarness {
stopped: bool,
}
#[derive(Default)]
struct RecordingControlHarness {
steers: Vec<String>,
interrupts: usize,
}
fn step(name: &str, id: Option<&str>, human: bool) -> StepPlan {
StepPlan {
name: name.to_string(),
kind: StepKind::Skill,
policy: OccurrencePolicy {
id: id.map(str::to_string),
human,
},
}
}
#[test]
fn iterate_skips_prior_human_nodes_when_returning_to_autonomous_work() {
let (flow, _) = Playhead::new(QueuedInvocation {
id: "human-iterate-proof".to_string(),
flow: "proof".to_string(),
steps: vec![
step("kickoff", None, false),
step("first-review", Some("first_review"), true),
step("second-review", Some("second_review"), true),
],
});
assert_eq!(preceding_autonomous_step(&flow, 2).unwrap(), 0);
}
#[async_trait::async_trait]
impl Harness for UnusedHarness {
async fn start(&mut self, _config: &AgentConfig) -> anyhow::Result<()> {
anyhow::bail!("unused test harness must not start")
}
async fn send_input(&mut self, _content: &str) -> anyhow::Result<()> {
anyhow::bail!("unused test harness must not receive input")
}
async fn send_current(&mut self, _content: &str) -> SendCurrentOutcome {
SendCurrentOutcome::NotSteerable
}
async fn interrupt(&mut self) -> anyhow::Result<()> {
Ok(())
}
async fn stop(&mut self) -> anyhow::Result<()> {
self.stopped = true;
Ok(())
}
fn provider_session_id(&self) -> Option<String> {
None
}
}
#[async_trait::async_trait]
impl Harness for RecordingControlHarness {
async fn start(&mut self, _config: &AgentConfig) -> anyhow::Result<()> {
Ok(())
}
async fn send_input(&mut self, _content: &str) -> anyhow::Result<()> {
Ok(())
}
async fn send_current(&mut self, content: &str) -> SendCurrentOutcome {
self.steers.push(content.to_string());
SendCurrentOutcome::Sent {
provider_turn_id: "turn-test".to_string(),
}
}
async fn interrupt(&mut self) -> anyhow::Result<()> {
self.interrupts += 1;
Ok(())
}
async fn stop(&mut self) -> anyhow::Result<()> {
Ok(())
}
fn provider_session_id(&self) -> Option<String> {
None
}
}
async fn human_task_fixture() -> (SharedStore, ControlledTask, Playhead) {
let repository =
std::fs::canonicalize(std::path::Path::new(env!("CARGO_MANIFEST_DIR")).join("../.."))
.unwrap();
let worktree = tempfile::tempdir().unwrap().keep();
let git = |args: &[&str]| {
assert!(std::process::Command::new("git")
.current_dir(&worktree)
.args(args)
.output()
.unwrap()
.status
.success());
};
git(&["init", "-q"]);
git(&[
"-c",
"user.email=test@loopflow.dev",
"-c",
"user.name=Loopflow Test",
"commit",
"--allow-empty",
"-q",
"-m",
"init",
]);
let base_commit = String::from_utf8(
std::process::Command::new("git")
.current_dir(&worktree)
.args(["rev-parse", "HEAD"])
.output()
.unwrap()
.stdout,
)
.unwrap()
.trim()
.to_string();
let database = tempfile::tempdir().unwrap().keep();
let database = database.join("registry.db");
let store = std::sync::Arc::new(
crate::store::open_ephemeral_store(&StorageConfig::sqlite(database.clone()))
.await
.unwrap(),
);
let now = time::OffsetDateTime::now_utc();
let wave = Wave::new(
crate::id::WaveId::new(),
"human-task-proof".to_string(),
repository.display().to_string(),
);
let project = Project {
id: ProjectId::new(),
plan: ProjectPlan {
id: LinearProjectId::new("human-task-project").unwrap(),
slug: "human-task-proof".to_string(),
name: "Human Task proof".to_string(),
prompt_context: "Prove durable human flow nodes.".to_string(),
pm_snapshot_synced_at: now.unix_timestamp(),
},
wave_id: wave.id().clone(),
abandon_intent: None,
created_at: now,
updated_at: now,
};
let task = Task {
id: TaskId::new(),
plan: TaskPlan {
id: LinearIssueId::new("human-task-issue").unwrap(),
identifier: "TEST-1".to_string(),
title: "Human Task proof".to_string(),
description: "Stop at review_kickoff.".to_string(),
pm_snapshot_synced_at: now.unix_timestamp(),
},
pm_writeback: PmWritebackState::Current,
wave_id: wave.id().clone(),
project_id: project.id.clone(),
worktree: worktree.clone(),
workspace_slug: "human-task-proof".to_string(),
abandon_intent: None,
created_at: now,
updated_at: now,
observation: Observation::NotRequired,
};
let state = State {
task_id: task.id.clone(),
lifecycle: TaskLifecyclePlan::defaults(),
lifecycle_phase: TaskLifecyclePhase::First,
phase_cursor: 1,
phase_iteration: 0,
gate_cycle: 0,
gate_proposal: None,
agent: "codex".to_string(),
provider: "codex".to_string(),
provider_session_id: None,
updated_at: now,
};
let pr = TaskPr {
id: TaskPrId::new(),
task_id: task.id.clone(),
sequence: 1,
slug: task.workspace_slug.clone(),
branch: "test/human-task-proof".to_string(),
base_commit,
parent_pr_id: None,
publication: None,
merge_commit: None,
abandoned_at: None,
ci_observation: None,
github_observation: None,
linear_attachment_id: None,
linear_comment_id: None,
linear_link_error: None,
created_at: now,
updated_at: now,
};
store.create_wave(&wave).await.unwrap();
store.create_project(&project).await.unwrap();
store.create_task(&task, &pr).await.unwrap();
store.put_task_controller_state(&state).await.unwrap();
let task = ControlledTask { work: task, state };
let flow = super::resume_task_phase(&task).unwrap();
(store, task, flow)
}
#[tokio::test]
async fn provider_failure_records_planning() {
let (store, mut task, _) = human_task_fixture().await;
let mut harness = UnusedHarness::default();
super::finish_body_failure(
&store,
&mut task,
&mut harness,
"opencode_disconnected: provider stream ended",
None,
)
.await
.unwrap();
let events = store.task_events_after(&task.id, 0).await.unwrap();
assert!(matches!(
&events.last().unwrap().kind,
TaskEventKind::Failed {
error,
resumable: true,
} if error.contains("provider stream ended")
));
assert!(harness.stopped);
}
#[test]
fn normal_task_completion_preserves_delivery_permission_and_run_network_failures() {
let commit = completed_boundary_failure(
&["lf".into(), "commit".into(), "-m".into(), "ship".into(), "-p".into()],
Lifecycle::Failed,
Some(
"fatal: Unable to create '/repo/.git/worktrees/task/index.lock': Operation not permitted",
),
Some(128),
)
.unwrap();
let run = completed_boundary_failure(
&[
"lf".into(),
"--as".into(),
"project:proj_1".into(),
":".into(),
"Review this".into(),
],
Lifecycle::Failed,
Some("network access is disabled by policy"),
Some(1),
)
.unwrap();
let reason = execution_blocker_at_handoff(Lifecycle::Completed, &[commit, run])
.expect("normal task_complete with unresolved capability failures is blocked");
assert!(reason.contains(".git/worktrees/task/index.lock"));
assert!(reason.contains("Operation not permitted"));
assert!(reason.contains("network access is disabled by policy"));
assert!(reason.contains("before starting a new Run"));
}
#[test]
fn ordinary_failed_probe_is_not_an_execution_boundary_blocker() {
assert!(completed_boundary_failure(
&["rg".into(), "missing-pattern".into()],
Lifecycle::Failed,
Some(""),
Some(1),
)
.is_none());
}
#[test]
fn rejected_provider_auth_is_a_named_nonresumable_capability_blocker() {
let (reason, resumable) = unhandled_failure_receipt(
"Your authentication token has been invalidated (token_invalidated)",
);
assert!(!resumable);
assert!(reason.contains("provider credential capability is blocked"));
assert!(reason.contains("Reconnect the named managed account"));
}
#[tokio::test]
async fn unhandled_task_failure_is_a_planning_receipt() {
let (store, task, _) = human_task_fixture().await;
super::record_unhandled_failure(
&store,
&task.id,
&anyhow::anyhow!("provider stream closed"),
)
.await;
let events = store.recent_task_events(&task.id, 10).await.unwrap();
assert!(matches!(
&events[0].kind,
TaskEventKind::Failed { error, resumable: true }
if error == "task process failed: provider stream closed"
));
}
#[tokio::test]
#[allow(clippy::await_holding_lock)] async fn control_events_after_seed_preparation_remain_live() {
let _lf_bin = super::TestLfBinGuard::pin();
let (store, mut task, flow) = human_task_fixture().await;
let work = WorkRef::Task(task.id.clone());
let seeded = store
.append_steer(&work, Author::User, "seeded direction")
.await
.unwrap();
let prepared =
super::prepare_task_flow_step_once(&store, &mut task, "human-task-proof", &flow)
.await
.unwrap();
assert_eq!(prepared.seeded_steer_id, seeded.id);
assert!(prepared.turn.input.contains("seeded direction"));
let late = store
.append_steer(&work, Author::User, "late direction")
.await
.unwrap();
let late_interrupt = store.append_interrupt(&work).await.unwrap();
let mut harness = RecordingControlHarness::default();
let mut steer_cursor = prepared.seeded_steer_id;
let mut interrupt_cursor = prepared.interrupt_id;
crate::ops::child::inject_live_steers(&store, &work, &mut harness, &mut steer_cursor).await;
crate::ops::child::observe_interrupt(&store, &work, &mut harness, &mut interrupt_cursor)
.await;
assert_eq!(harness.steers, vec!["late direction"]);
assert_eq!(steer_cursor, late.id);
assert_eq!(harness.interrupts, 1);
assert_eq!(interrupt_cursor, late_interrupt);
}
#[tokio::test]
#[allow(clippy::await_holding_lock)] async fn interrupted_step_restarts_in_place_with_fresh_direction() {
let _lf_bin = super::TestLfBinGuard::pin();
let (store, mut task, mut flow) = human_task_fixture().await;
let interrupted_step = flow.current().unwrap().step.clone();
super::open_task_flow_body(&mut flow, &task).unwrap();
assert!(!super::finish_task_flow_turn(&mut flow, Lifecycle::Interrupted).unwrap());
store
.append_steer(
&WorkRef::Task(task.id.clone()),
Author::User,
"direction after interrupt",
)
.await
.unwrap();
let prepared =
super::prepare_task_flow_step_once(&store, &mut task, "human-task-proof", &flow)
.await
.unwrap();
assert_eq!(flow.current().unwrap().step, interrupted_step);
assert!(prepared.turn.input.contains("direction after interrupt"));
}
async fn park_human_task(
store: &SharedStore,
task: &mut ControlledTask,
flow: &mut Playhead,
) -> crate::ops::human_session::FlowSessionToken {
assert!(
super::prepare_task_flow_step(store, task, "human-task-proof", flow)
.await
.unwrap()
.is_none()
);
let position = store
.flow_position(&WorkRef::Task(task.id.clone()))
.await
.unwrap()
.unwrap();
crate::ops::human_session::FlowSessionToken {
task_id: task.id.clone(),
flow: position.flow,
node_id: position.node_id.unwrap(),
skill: position.step,
iteration: position.iteration,
}
}
#[tokio::test]
async fn restarting_a_human_node_reuses_the_same_task_position() {
let (store, mut task, mut flow) = human_task_fixture().await;
let original = park_human_task(&store, &mut task, &mut flow).await;
let work = WorkRef::Task(task.id.clone());
let run_id = RunId::new();
let mut position = store.flow_position(&work).await.unwrap().unwrap();
position.session_run_id = Some(run_id.clone());
position.ready_summary = Some("ready".to_string());
store.set_flow_position(&work, position).await.unwrap();
let mut restarted_flow = super::resume_task_phase(&task).unwrap();
let recovered = park_human_task(&store, &mut task, &mut restarted_flow).await;
assert_eq!(recovered, original);
assert_eq!(store.human_flow_positions().await.unwrap().len(), 1);
let recovered = store.flow_position(&work).await.unwrap().unwrap();
assert_eq!(recovered.session_run_id, Some(run_id));
assert_eq!(recovered.ready_summary.as_deref(), Some("ready"));
}
#[tokio::test]
async fn approving_a_human_node_advances_the_task_without_an_ask() {
let (store, mut task, mut flow) = human_task_fixture().await;
let token = park_human_task(&store, &mut task, &mut flow).await;
super::decide_human_flow_step(
&store,
&token,
crate::ops::human_session::FlowDecision::Approve,
"design approved",
)
.await
.unwrap();
let task = super::controlled_task(&store, &task.id).await.unwrap();
assert_eq!(task.state.lifecycle_phase, TaskLifecyclePhase::Loop);
assert!(!crate::ops::human_session::token_is_current(&store, &token)
.await
.unwrap());
assert!(store
.recent_task_events(&task.id, 10)
.await
.unwrap()
.iter()
.any(|event| matches!(
&event.kind,
TaskEventKind::Progress { summary } if summary == "design approved"
)));
}
#[tokio::test]
async fn iterating_a_human_node_returns_to_autonomous_work_with_a_steer() {
let (store, mut task, mut flow) = human_task_fixture().await;
let token = park_human_task(&store, &mut task, &mut flow).await;
super::decide_human_flow_step(
&store,
&token,
crate::ops::human_session::FlowDecision::Iterate,
"narrow the design",
)
.await
.unwrap();
let task = super::controlled_task(&store, &task.id).await.unwrap();
assert_eq!(task.state.phase_cursor, 0);
assert_eq!(task.state.phase_iteration, 1);
let work = WorkRef::Task(task.id.clone());
let position = store.flow_position(&work).await.unwrap().unwrap();
assert_eq!(position.step, "kickoff");
assert!(!position.human);
assert!(store
.work_steers(&work)
.await
.unwrap()
.iter()
.any(|steer| steer.text.contains("narrow the design")));
}
#[test]
fn task_state_sync_keeps_gate_proposals_scoped_to_finally() {
let now = time::OffsetDateTime::now_utc();
let first_work = Task {
id: TaskId::new(),
plan: TaskPlan {
id: LinearIssueId::new("incident-issue").unwrap(),
identifier: "ENG-125".to_string(),
title: "Incident".to_string(),
description: String::new(),
pm_snapshot_synced_at: 1,
},
pm_writeback: PmWritebackState::Current,
wave_id: crate::id::WaveId::new(),
project_id: ProjectId::new(),
worktree: "/tmp/incident".into(),
workspace_slug: "incident".to_string(),
abandon_intent: None,
created_at: now,
updated_at: now,
observation: Observation::NotRequired,
};
let first = ControlledTask {
state: State {
task_id: first_work.id.clone(),
lifecycle: TaskLifecyclePlan::standard("incident", "ship-5whys", "ship"),
lifecycle_phase: TaskLifecyclePhase::First,
phase_cursor: 0,
phase_iteration: 0,
gate_cycle: 0,
gate_proposal: None,
agent: "codex".to_string(),
provider: "codex".to_string(),
provider_session_id: None,
updated_at: now,
},
work: first_work,
};
let mut finally = first.clone();
finally.state.enter_loop().unwrap();
finally
.state
.enter_finally(TaskGateProposal {
done: true,
reason: "pull request merged".to_string(),
})
.unwrap();
let mut first_body = first.clone();
sync_task_state(&mut first_body, &finally);
assert!(first_body.state.gate_proposal.is_none());
first_body.validate().unwrap();
let mut loop_body = first;
loop_body.state.enter_loop().unwrap();
sync_task_state(&mut loop_body, &finally);
assert!(loop_body.state.gate_proposal.is_none());
loop_body.validate().unwrap();
let mut finally_body = finally.clone();
finally.state.gate_proposal = Some(TaskGateProposal {
done: false,
reason: "another prevention remains".to_string(),
});
sync_task_state(&mut finally_body, &finally);
assert_eq!(finally_body.state, finally.state);
finally_body.validate().unwrap();
}
#[test]
fn task_seed_uses_the_current_parent_project_definition() {
let now = time::OffsetDateTime::UNIX_EPOCH;
let task_work = Task {
id: TaskId::new(),
plan: TaskPlan {
id: LinearIssueId::new("issue-1").unwrap(),
identifier: "INF-123".to_string(),
title: "Ship it".to_string(),
description: "Task direction".to_string(),
pm_snapshot_synced_at: 11,
},
pm_writeback: PmWritebackState::Current,
wave_id: crate::id::WaveId::new(),
project_id: crate::work::project::ProjectId::new(),
worktree: "/tmp/task".into(),
workspace_slug: "ship-it".to_string(),
abandon_intent: None,
created_at: now,
updated_at: now,
observation: Observation::NotRequired,
};
let task = ControlledTask {
state: State {
task_id: task_work.id.clone(),
lifecycle: TaskLifecyclePlan::standard("task-design", "task", "ship"),
lifecycle_phase: TaskLifecyclePhase::Loop,
phase_cursor: 0,
phase_iteration: 0,
gate_cycle: 0,
gate_proposal: None,
agent: "codex".to_string(),
provider: "codex".to_string(),
provider_session_id: None,
updated_at: now,
},
work: task_work,
};
let pr = TaskPr {
id: TaskPrId::new(),
task_id: task.id.clone(),
sequence: 1,
slug: "ship-it".to_string(),
branch: "jack/ship-it".to_string(),
base_commit: "deadbeef".to_string(),
parent_pr_id: None,
publication: None,
merge_commit: None,
abandoned_at: None,
ci_observation: None,
github_observation: None,
linear_attachment_id: None,
linear_comment_id: None,
linear_link_error: None,
created_at: now,
updated_at: now,
};
let project = ProjectPlan {
id: LinearProjectId::new("project-1").unwrap(),
slug: "runtime".to_string(),
name: "Current project name".to_string(),
prompt_context: "Current project definition".to_string(),
pm_snapshot_synced_at: 22,
};
let seed = task_seed(&task, &project, &pr, "wave", &[]);
assert!(seed.contains("Current project name"));
assert!(seed.contains("Current project definition"));
assert!(seed.contains("Task directive snapshot synced at: 11"));
assert!(seed.contains("Project definition snapshot synced at: 22"));
assert!(!seed.contains("metric-portfolio"));
assert!(!seed.contains("project-owned-metrics"));
}
}