loopd: heartbeat tick on the background slot

Co-Authored-By: ornith-1.5-35b-a3b <agent@wntrmute.dev>

crates/loopd/src/scheduler.rs

@@ -0,0 +1,176 @@
+//! Isolated heartbeat (and, from task 07, cron) turns (M5b spec, section 4).
+
+use crate::baseline::Baseline;
+use crate::channel::{Busy, Context};
+use crate::compact;
+use crate::jobs::{self, Run, RunStatus};
+use crate::session::{Session, SessionError};
+use crate::turn::{Runtime, run_turn};
+use bxcore::{SessionId, Timestamp, TurnEvent};
+use std::fs;
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub enum TickWork {
+    Heartbeat,
+    SkippedBusy { session: SessionId },
+    HeartbeatUnreadable,
+    Job { name: String, session: SessionId },
+    SkippedJob { name: String },
+}
+
+pub struct Scheduler {
+    started: Timestamp,
+    last_heartbeat: Option<Timestamp>,
+    leftovers_closed: bool,
+    heartbeat_unreadable_logged: bool,
+}
+
+impl Scheduler {
+    pub fn new(started: Timestamp) -> Scheduler {
+        Scheduler {
+            started,
+            last_heartbeat: None,
+            leftovers_closed: false,
+            heartbeat_unreadable_logged: false,
+        }
+    }
+
+    pub fn tick(&mut self, ctx: &Context, now: Timestamp) -> Vec<TickWork> {
+        let mut work = Vec::new();
+        if !self.leftovers_closed {
+            close_idle_leftovers(ctx, now);
+            self.leftovers_closed = true;
+        }
+        self.heartbeat(ctx, now, &mut work);
+        work
+    }
+
+    fn heartbeat(&mut self, ctx: &Context, now: Timestamp, work: &mut Vec<TickWork>) {
+        if ctx.cfg.heartbeat.every_s == 0 {
+            return;
+        }
+        let last = self.last_heartbeat.unwrap_or(self.started);
+        let due = now.unix_millis().saturating_sub(last.unix_millis())
+            >= ctx.cfg.heartbeat.every_s.saturating_mul(1000);
+        if !due {
+            return;
+        }
+
+        let id = match SessionId::new("heartbeat") {
+            Ok(id) => id,
+            Err(_) => return,
+        };
+
+        let prompt = ctx.cfg.heartbeat_prompt();
+        let file = match fs::read_to_string(&prompt) {
+            Ok(file) => file,
+            Err(e) => {
+                if !self.heartbeat_unreadable_logged {
+                    eprintln!(
+                        "loopd: {}: {e}; see docs/runbook.md#heartbeat-prompt-unreadable",
+                        prompt.display()
+                    );
+                    self.heartbeat_unreadable_logged = true;
+                }
+                work.push(TickWork::HeartbeatUnreadable);
+                return;
+            }
+        };
+
+        let held = match ctx.hold(id.clone()) {
+            Ok(held) => held,
+            Err(Busy) => {
+                work.push(TickWork::SkippedBusy { session: id });
+                return;
+            }
+        };
+
+        let mut session = match Session::open(&ctx.cfg.paths.home, id.clone()) {
+            Ok(session) => session,
+            Err(SessionError::NotFound(_)) => {
+                let baseline = match Baseline::assemble(&ctx.cfg, &ctx.registry) {
+                    Ok(baseline) => baseline,
+                    Err(e) => {
+                        eprintln!("loopd: heartbeat baseline: {e}");
+                        return;
+                    }
+                };
+                match Session::create(
+                    &ctx.cfg.paths.home,
+                    id.clone(),
+                    baseline,
+                    ctx.cfg.slots.background,
+                ) {
+                    Ok(session) => session,
+                    Err(e) => {
+                        eprintln!("loopd: heartbeat session {}: {e}", id.as_str());
+                        return;
+                    }
+                }
+            }
+            Err(e) => {
+                eprintln!("loopd: heartbeat session {}: {e}", id.as_str());
+                return;
+            }
+        };
+
+        let runtime = Runtime {
+            cfg: &ctx.cfg,
+            client: &ctx.client,
+            port: &*ctx.port,
+            registry: &ctx.registry,
+        };
+
+        let text = format!("It is {}.\n\n{file}", now.to_rfc3339());
+        let mut emit = |_event: &TurnEvent| {};
+        let outcome = run_turn(&mut session, &runtime, &text, None, &mut emit);
+        let outcome = compact::finish_turn(
+            &mut session,
+            &runtime,
+            &*ctx.memory,
+            &text,
+            None,
+            &mut emit,
+            outcome,
+        );
+        if outcome.is_ok() {
+            self.last_heartbeat = Some(now);
+            work.push(TickWork::Heartbeat);
+        }
+        drop(held);
+    }
+}
+
+fn close_idle_leftovers(ctx: &Context, now: Timestamp) {
+    let jobs_dir = ctx.cfg.paths.home.join("jobs");
+    let runs_path = jobs_dir.join("runs.jsonl");
+    let jobs = match jobs::load(&jobs_dir) {
+        Ok(jobs) => jobs,
+        Err(e) => {
+            eprintln!("loopd: cannot read the jobs: {e}");
+            return;
+        }
+    };
+    for job in &jobs {
+        let session = match jobs::live_session(&runs_path, &job.name) {
+            Ok(Some(session)) => session,
+            Ok(None) => continue,
+            Err(e) => {
+                eprintln!("loopd: cannot read the live session for {}: {e}", job.name);
+                continue;
+            }
+        };
+        if let Ok(held) = ctx.hold(session.clone()) {
+            drop(held);
+            let _ = jobs::append(
+                &runs_path,
+                &Run {
+                    time: now,
+                    name: job.name.clone(),
+                    session,
+                    status: RunStatus::Failed,
+                },
+            );
+        }
+    }
+}