Add basic cron scheduler to goose-server (#2621)

crates/goose-server/src/state.rs

@@ -1,21 +1,15 @@
 use goose::agents::Agent;
+use goose::scheduler::Scheduler;
 use std::sync::Arc;
+use tokio::sync::Mutex;
 
-/// Shared reference to an Agent that can be cloned cheaply
-/// without cloning the underlying Agent object
 pub type AgentRef = Arc<Agent>;
 
-/// Thread-safe container for an optional Agent reference
-/// Outer Arc: Allows multiple route handlers to access the same Mutex
-/// - Mutex provides exclusive access for updates
-/// - Option allows for the case where no agent exists yet
-///
-/// Shared application state
 #[derive(Clone)]
 pub struct AppState {
-    // agent: SharedAgentStore,
     agent: Option<AgentRef>,
     pub secret_key: String,
+    pub scheduler: Arc<Mutex<Option<Arc<Scheduler>>>>,
 }
 
 impl AppState {
@@ -23,6 +17,7 @@ impl AppState {
         Arc::new(Self {
             agent: Some(agent.clone()),
             secret_key,
+            scheduler: Arc::new(Mutex::new(None)),
         })
     }
 
@@ -31,4 +26,17 @@ impl AppState {
             .clone()
             .ok_or_else(|| anyhow::anyhow!("Agent needs to be created first."))
     }
+
+    pub async fn set_scheduler(&self, sched: Arc<Scheduler>) {
+        let mut guard = self.scheduler.lock().await;
+        *guard = Some(sched);
+    }
+
+    pub async fn scheduler(&self) -> Result<Arc<Scheduler>, anyhow::Error> {
+        self.scheduler
+            .lock()
+            .await
+            .clone()
+            .ok_or_else(|| anyhow::anyhow!("Scheduler not initialized"))
+    }
 }