feat: refactor sgclaw around zeroclaw compat runtime

This commit is contained in:
zyl
2026-03-26 16:23:31 +08:00
parent bca5b75801
commit ff0771a83f
1059 changed files with 409460 additions and 23 deletions

1016
third_party/zeroclaw/src/cron/mod.rs vendored Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,306 @@
use crate::cron::Schedule;
use anyhow::{Context, Result};
use chrono::{DateTime, Duration as ChronoDuration, Utc};
use cron::Schedule as CronExprSchedule;
use std::str::FromStr;
pub fn next_run_for_schedule(schedule: &Schedule, from: DateTime<Utc>) -> Result<DateTime<Utc>> {
match schedule {
Schedule::Cron { expr, tz } => {
let normalized = normalize_expression(expr)?;
let cron = CronExprSchedule::from_str(&normalized)
.with_context(|| format!("Invalid cron expression: {expr}"))?;
if let Some(tz_name) = tz {
let timezone = chrono_tz::Tz::from_str(tz_name)
.with_context(|| format!("Invalid IANA timezone: {tz_name}"))?;
let localized_from = from.with_timezone(&timezone);
let next_local = cron.after(&localized_from).next().ok_or_else(|| {
anyhow::anyhow!("No future occurrence for expression: {expr}")
})?;
Ok(next_local.with_timezone(&Utc))
} else {
cron.after(&from)
.next()
.ok_or_else(|| anyhow::anyhow!("No future occurrence for expression: {expr}"))
}
}
Schedule::At { at } => Ok(*at),
Schedule::Every { every_ms } => {
if *every_ms == 0 {
anyhow::bail!("Invalid schedule: every_ms must be > 0");
}
let ms = i64::try_from(*every_ms).context("every_ms is too large")?;
let delta = ChronoDuration::milliseconds(ms);
from.checked_add_signed(delta)
.ok_or_else(|| anyhow::anyhow!("every_ms overflowed DateTime"))
}
}
}
pub fn validate_schedule(schedule: &Schedule, now: DateTime<Utc>) -> Result<()> {
match schedule {
Schedule::Cron { expr, .. } => {
let _ = normalize_expression(expr)?;
let _ = next_run_for_schedule(schedule, now)?;
Ok(())
}
Schedule::At { at } => {
if *at <= now {
anyhow::bail!("Invalid schedule: 'at' must be in the future");
}
Ok(())
}
Schedule::Every { every_ms } => {
if *every_ms == 0 {
anyhow::bail!("Invalid schedule: every_ms must be > 0");
}
Ok(())
}
}
}
pub fn schedule_cron_expression(schedule: &Schedule) -> Option<String> {
match schedule {
Schedule::Cron { expr, .. } => Some(expr.clone()),
_ => None,
}
}
pub fn normalize_expression(expression: &str) -> Result<String> {
let expression = expression.trim();
let field_count = expression.split_whitespace().count();
match field_count {
// standard crontab syntax: minute hour day month weekday
// Normalize weekday field from standard crontab semantics (0/7=Sun, 1=Mon, …, 6=Sat)
// to cron-crate semantics (1=Sun, 2=Mon, …, 7=Sat).
5 => {
let mut fields: Vec<&str> = expression.split_whitespace().collect();
let weekday = fields[4];
let normalized_weekday = normalize_weekday_field(weekday)?;
fields[4] = &normalized_weekday;
Ok(format!(
"0 {} {} {} {} {}",
fields[0], fields[1], fields[2], fields[3], fields[4]
))
}
// crate-native syntax includes seconds (+ optional year)
6 | 7 => Ok(expression.to_string()),
_ => anyhow::bail!(
"Invalid cron expression: {expression} (expected 5, 6, or 7 fields, got {field_count})"
),
}
}
/// Translate a single numeric weekday value from standard crontab semantics
/// (0 or 7 = Sunday, 1 = Monday, …, 6 = Saturday) to cron-crate semantics
/// (1 = Sunday, 2 = Monday, …, 7 = Saturday).
fn translate_weekday_value(val: u8) -> Result<u8> {
match val {
0 | 7 => Ok(1), // Sunday
1..=6 => Ok(val + 1),
_ => anyhow::bail!("Invalid weekday value: {val} (expected 0-7)"),
}
}
/// Normalize the weekday field of a 5-field cron expression from standard
/// crontab numbering to cron-crate numbering. Passes through `*`, named days
/// (e.g. `MON`, `MON-FRI`), and already-valid tokens unchanged.
fn normalize_weekday_field(field: &str) -> Result<String> {
// Asterisk and wildcard variants pass through unchanged.
if field == "*" || field == "?" {
return Ok(field.to_string());
}
// If the field contains any alphabetic character it uses named days
// (e.g. MON-FRI) which the cron crate handles natively.
if field.chars().any(|c| c.is_ascii_alphabetic()) {
return Ok(field.to_string());
}
// The field may be a comma-separated list of items, where each item is
// either a single value, a range (start-end), or a range/value with a
// step (/N).
let parts: Vec<&str> = field.split(',').collect();
let mut result_parts = Vec::with_capacity(parts.len());
for part in parts {
// Split off optional step suffix first (e.g. "1-5/2" → "1-5" + "2").
let (range_part, step) = if let Some((r, s)) = part.split_once('/') {
(r, Some(s))
} else {
(part, None)
};
let translated = if let Some((start_s, end_s)) = range_part.split_once('-') {
let start: u8 = start_s
.parse()
.with_context(|| format!("Invalid weekday in range: {start_s}"))?;
let end: u8 = end_s
.parse()
.with_context(|| format!("Invalid weekday in range: {end_s}"))?;
let new_start = translate_weekday_value(start)?;
let new_end = translate_weekday_value(end)?;
format!("{new_start}-{new_end}")
} else if range_part == "*" {
"*".to_string()
} else {
let val: u8 = range_part
.parse()
.with_context(|| format!("Invalid weekday value: {range_part}"))?;
translate_weekday_value(val)?.to_string()
};
if let Some(s) = step {
result_parts.push(format!("{translated}/{s}"));
} else {
result_parts.push(translated);
}
}
Ok(result_parts.join(","))
}
#[cfg(test)]
mod tests {
use super::*;
use chrono::{Datelike, TimeZone};
#[test]
fn next_run_for_schedule_supports_every_and_at() {
let now = Utc::now();
let every = Schedule::Every { every_ms: 60_000 };
let next = next_run_for_schedule(&every, now).unwrap();
assert!(next > now);
let at = now + ChronoDuration::minutes(10);
let at_schedule = Schedule::At { at };
let next_at = next_run_for_schedule(&at_schedule, now).unwrap();
assert_eq!(next_at, at);
}
#[test]
fn next_run_for_schedule_supports_timezone() {
let from = Utc.with_ymd_and_hms(2026, 2, 16, 0, 0, 0).unwrap();
let schedule = Schedule::Cron {
expr: "0 9 * * *".into(),
tz: Some("America/Los_Angeles".into()),
};
let next = next_run_for_schedule(&schedule, from).unwrap();
assert_eq!(next, Utc.with_ymd_and_hms(2026, 2, 16, 17, 0, 0).unwrap());
}
#[test]
fn normalize_weekday_field_translates_standard_crontab_values() {
// Single values: standard crontab → cron crate
assert_eq!(normalize_weekday_field("0").unwrap(), "1"); // Sun
assert_eq!(normalize_weekday_field("1").unwrap(), "2"); // Mon
assert_eq!(normalize_weekday_field("5").unwrap(), "6"); // Fri
assert_eq!(normalize_weekday_field("6").unwrap(), "7"); // Sat
assert_eq!(normalize_weekday_field("7").unwrap(), "1"); // Sun (alias)
}
#[test]
fn normalize_weekday_field_translates_ranges() {
// 1-5 (Mon-Fri) → 2-6
assert_eq!(normalize_weekday_field("1-5").unwrap(), "2-6");
// 0-6 (Sun-Sat) → 1-7
assert_eq!(normalize_weekday_field("0-6").unwrap(), "1-7");
}
#[test]
fn normalize_weekday_field_translates_lists() {
// 0,6 (Sun,Sat) → 1,7
assert_eq!(normalize_weekday_field("0,6").unwrap(), "1,7");
// 1,3,5 (Mon,Wed,Fri) → 2,4,6
assert_eq!(normalize_weekday_field("1,3,5").unwrap(), "2,4,6");
}
#[test]
fn normalize_weekday_field_translates_steps() {
// 1-5/2 (Mon-Fri every other) → 2-6/2
assert_eq!(normalize_weekday_field("1-5/2").unwrap(), "2-6/2");
// */2 (every other day) → */2
assert_eq!(normalize_weekday_field("*/2").unwrap(), "*/2");
}
#[test]
fn normalize_weekday_field_passes_through_wildcards_and_names() {
assert_eq!(normalize_weekday_field("*").unwrap(), "*");
assert_eq!(normalize_weekday_field("?").unwrap(), "?");
assert_eq!(normalize_weekday_field("MON-FRI").unwrap(), "MON-FRI");
assert_eq!(
normalize_weekday_field("MON,WED,FRI").unwrap(),
"MON,WED,FRI"
);
}
#[test]
fn normalize_expression_applies_weekday_fix_to_5_field() {
// "0 9 * * 1-5" should become "0 0 9 * * 2-6"
let result = normalize_expression("0 9 * * 1-5").unwrap();
assert_eq!(result, "0 0 9 * * 2-6");
}
#[test]
fn normalize_expression_does_not_modify_6_field() {
// 6-field expressions already use cron-crate semantics
let result = normalize_expression("0 0 9 * * 1-5").unwrap();
assert_eq!(result, "0 0 9 * * 1-5");
}
#[test]
fn weekday_1_5_schedules_monday_through_friday() {
// 2026-02-16 is a Monday. With "0 9 * * 1-5" (Mon-Fri at 09:00 UTC),
// the next run from Sunday 2026-02-15 should be Monday 2026-02-16.
let sunday = Utc.with_ymd_and_hms(2026, 2, 15, 0, 0, 0).unwrap();
let schedule = Schedule::Cron {
expr: "0 9 * * 1-5".into(),
tz: None,
};
let next = next_run_for_schedule(&schedule, sunday).unwrap();
// Should be Monday 2026-02-16 at 09:00 UTC (weekday = Mon)
assert_eq!(next, Utc.with_ymd_and_hms(2026, 2, 16, 9, 0, 0).unwrap());
assert_eq!(next.weekday(), chrono::Weekday::Mon);
}
#[test]
fn weekday_1_5_does_not_fire_on_saturday_or_sunday() {
// From Friday evening, next run should skip Sat/Sun → Monday
let friday_evening = Utc.with_ymd_and_hms(2026, 2, 20, 18, 0, 0).unwrap();
let schedule = Schedule::Cron {
expr: "0 9 * * 1-5".into(),
tz: None,
};
let next = next_run_for_schedule(&schedule, friday_evening).unwrap();
// Should be Monday 2026-02-23 at 09:00 UTC
assert_eq!(next, Utc.with_ymd_and_hms(2026, 2, 23, 9, 0, 0).unwrap());
assert_eq!(next.weekday(), chrono::Weekday::Mon);
}
#[test]
fn weekday_0_means_sunday() {
// "0 10 * * 0" should fire on Sunday only
let monday = Utc.with_ymd_and_hms(2026, 2, 16, 0, 0, 0).unwrap();
let schedule = Schedule::Cron {
expr: "0 10 * * 0".into(),
tz: None,
};
let next = next_run_for_schedule(&schedule, monday).unwrap();
assert_eq!(next.weekday(), chrono::Weekday::Sun);
}
#[test]
fn weekday_7_means_sunday() {
// "0 10 * * 7" should also fire on Sunday (alias)
let monday = Utc.with_ymd_and_hms(2026, 2, 16, 0, 0, 0).unwrap();
let schedule = Schedule::Cron {
expr: "0 10 * * 7".into(),
tz: None,
};
let next = next_run_for_schedule(&schedule, monday).unwrap();
assert_eq!(next.weekday(), chrono::Weekday::Sun);
}
}

1429
third_party/zeroclaw/src/cron/scheduler.rs vendored Normal file

File diff suppressed because it is too large Load Diff

1696
third_party/zeroclaw/src/cron/store.rs vendored Normal file

File diff suppressed because it is too large Load Diff

244
third_party/zeroclaw/src/cron/types.rs vendored Normal file
View File

@@ -0,0 +1,244 @@
use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
/// Try to deserialize a `serde_json::Value` as `T`. If the value is a JSON
/// string that looks like an object (i.e. the LLM double-serialized it), parse
/// the inner string first and then deserialize the resulting object. This
/// provides backward-compatible handling for both `Value::Object` and
/// `Value::String` representations.
pub fn deserialize_maybe_stringified<T: serde::de::DeserializeOwned>(
v: &serde_json::Value,
) -> Result<T, serde_json::Error> {
// Fast path: value is already the right shape (object, array, etc.)
match serde_json::from_value::<T>(v.clone()) {
Ok(parsed) => Ok(parsed),
Err(first_err) => {
// If it's a string, try parsing the string as JSON first.
if let Some(s) = v.as_str() {
let s = s.trim();
if s.starts_with('{') || s.starts_with('[') {
if let Ok(inner) = serde_json::from_str::<serde_json::Value>(s) {
return serde_json::from_value::<T>(inner);
}
}
}
Err(first_err)
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
#[serde(rename_all = "lowercase")]
pub enum JobType {
#[default]
Shell,
Agent,
}
impl From<JobType> for &'static str {
fn from(value: JobType) -> Self {
match value {
JobType::Shell => "shell",
JobType::Agent => "agent",
}
}
}
impl TryFrom<&str> for JobType {
type Error = String;
fn try_from(value: &str) -> Result<Self, Self::Error> {
match value.to_lowercase().as_str() {
"shell" => Ok(JobType::Shell),
"agent" => Ok(JobType::Agent),
_ => Err(format!(
"Invalid job type '{}'. Expected one of: 'shell', 'agent'",
value
)),
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
#[serde(rename_all = "lowercase")]
pub enum SessionTarget {
#[default]
Isolated,
Main,
}
impl SessionTarget {
pub(crate) fn as_str(&self) -> &'static str {
match self {
Self::Isolated => "isolated",
Self::Main => "main",
}
}
pub(crate) fn parse(raw: &str) -> Self {
if raw.eq_ignore_ascii_case("main") {
Self::Main
} else {
Self::Isolated
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(tag = "kind", rename_all = "lowercase")]
pub enum Schedule {
Cron {
expr: String,
#[serde(default)]
tz: Option<String>,
},
At {
at: DateTime<Utc>,
},
Every {
every_ms: u64,
},
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct DeliveryConfig {
#[serde(default)]
pub mode: String,
#[serde(default)]
pub channel: Option<String>,
#[serde(default)]
pub to: Option<String>,
#[serde(default = "default_true")]
pub best_effort: bool,
}
impl Default for DeliveryConfig {
fn default() -> Self {
Self {
mode: "none".to_string(),
channel: None,
to: None,
best_effort: true,
}
}
}
fn default_true() -> bool {
true
}
fn default_source() -> String {
"imperative".to_string()
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CronJob {
pub id: String,
pub expression: String,
pub schedule: Schedule,
pub command: String,
pub prompt: Option<String>,
pub name: Option<String>,
pub job_type: JobType,
pub session_target: SessionTarget,
pub model: Option<String>,
pub enabled: bool,
pub delivery: DeliveryConfig,
pub delete_after_run: bool,
/// Optional allowlist of tool names this cron job may use.
/// When `Some(list)`, only tools whose name is in the list are available.
/// When `None`, all tools are available (backward compatible default).
#[serde(default, skip_serializing_if = "Option::is_none")]
pub allowed_tools: Option<Vec<String>>,
/// How the job was created: `"imperative"` (CLI/API) or `"declarative"` (config).
#[serde(default = "default_source")]
pub source: String,
pub created_at: DateTime<Utc>,
pub next_run: DateTime<Utc>,
pub last_run: Option<DateTime<Utc>>,
pub last_status: Option<String>,
pub last_output: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CronRun {
pub id: i64,
pub job_id: String,
pub started_at: DateTime<Utc>,
pub finished_at: DateTime<Utc>,
pub status: String,
pub output: Option<String>,
pub duration_ms: Option<i64>,
}
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct CronJobPatch {
pub schedule: Option<Schedule>,
pub command: Option<String>,
pub prompt: Option<String>,
pub name: Option<String>,
pub enabled: Option<bool>,
pub delivery: Option<DeliveryConfig>,
pub model: Option<String>,
pub session_target: Option<SessionTarget>,
pub delete_after_run: Option<bool>,
pub allowed_tools: Option<Vec<String>>,
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn deserialize_schedule_from_object() {
let val = serde_json::json!({"kind": "cron", "expr": "*/5 * * * *"});
let sched = deserialize_maybe_stringified::<Schedule>(&val).unwrap();
assert!(matches!(sched, Schedule::Cron { ref expr, .. } if expr == "*/5 * * * *"));
}
#[test]
fn deserialize_schedule_from_string() {
let val = serde_json::Value::String(r#"{"kind":"cron","expr":"*/5 * * * *"}"#.to_string());
let sched = deserialize_maybe_stringified::<Schedule>(&val).unwrap();
assert!(matches!(sched, Schedule::Cron { ref expr, .. } if expr == "*/5 * * * *"));
}
#[test]
fn deserialize_schedule_string_with_tz() {
let val = serde_json::Value::String(
r#"{"kind":"cron","expr":"*/30 9-15 * * 1-5","tz":"Asia/Shanghai"}"#.to_string(),
);
let sched = deserialize_maybe_stringified::<Schedule>(&val).unwrap();
match sched {
Schedule::Cron { tz, .. } => assert_eq!(tz.as_deref(), Some("Asia/Shanghai")),
_ => panic!("expected Cron variant"),
}
}
#[test]
fn deserialize_every_from_string() {
let val = serde_json::Value::String(r#"{"kind":"every","every_ms":60000}"#.to_string());
let sched = deserialize_maybe_stringified::<Schedule>(&val).unwrap();
assert!(matches!(sched, Schedule::Every { every_ms: 60000 }));
}
#[test]
fn deserialize_invalid_string_returns_error() {
let val = serde_json::Value::String("not json at all".to_string());
assert!(deserialize_maybe_stringified::<Schedule>(&val).is_err());
}
#[test]
fn job_type_try_from_accepts_known_values_case_insensitive() {
assert_eq!(JobType::try_from("shell").unwrap(), JobType::Shell);
assert_eq!(JobType::try_from("SHELL").unwrap(), JobType::Shell);
assert_eq!(JobType::try_from("agent").unwrap(), JobType::Agent);
assert_eq!(JobType::try_from("AgEnT").unwrap(), JobType::Agent);
}
#[test]
fn job_type_try_from_rejects_invalid_values() {
assert!(JobType::try_from("").is_err());
assert!(JobType::try_from("unknown").is_err());
}
}