feat: add common queue

Signed-off-by: kjuulh <contact@kjuulh.io>

crates/churn/src/agent/refresh.rs

@@ -1,10 +1,15 @@
+use std::collections::BTreeMap;
+
 use anyhow::Context;
 
-use super::agent_state::AgentState;
+use crate::agent::models::Commands;
+
+use super::{agent_state::AgentState, queue::AgentQueue};
 
 #[derive(Clone)]
 pub struct AgentRefresh {
     process_host: String,
+    queue: AgentQueue,
 }
 
 impl AgentRefresh {
@@ -12,6 +17,7 @@ impl AgentRefresh {
         let state: AgentState = state.into();
         Self {
             process_host: state.discovery.process_host.clone(),
+            queue: state.queue.clone(),
         }
     }
 }
@@ -44,80 +50,12 @@ impl nodrift::Drifter for AgentRefresh {
     async fn execute(&self, _token: tokio_util::sync::CancellationToken) -> anyhow::Result<()> {
         tracing::info!(process_host = self.process_host, "refreshing agent");
 
-        // Get plan
-        let plan = Plan::new();
-        let tasks = plan.tasks().await?;
-
-        // For task
-        for task in tasks {
-            // Check idempotency rules
-            if !task.should_run().await? {
-                tracing::debug!(task = task.id(), "skipping run");
-                continue;
-            }
-
-            // Run task if not valid
-            tracing::info!(task = task.id(), "executing task");
-            task.execute().await?;
-        }
-
-        Ok(())
-    }
-}
-
-pub struct Plan {}
-impl Plan {
-    pub fn new() -> Self {
-        Self {}
-    }
-
-    pub async fn tasks(&self) -> anyhow::Result<Vec<Task>> {
-        Ok(vec![Task::new()])
-    }
-}
-
-pub struct Task {}
-
-impl Task {
-    pub fn new() -> Self {
-        Self {}
-    }
-
-    pub fn id(&self) -> String {
-        "apt".into()
-    }
-
-    pub async fn should_run(&self) -> anyhow::Result<bool> {
-        Ok(true)
-    }
-
-    pub async fn execute(&self) -> anyhow::Result<()> {
-        let mut cmd = tokio::process::Command::new("apt-get");
-        cmd.args(["update", "-q"]);
-        let output = cmd.output().await.context("failed to run apt update")?;
-        match output.status.success() {
-            true => tracing::info!("successfully ran apt update"),
-            false => {
-                anyhow::bail!(
-                    "failed to run apt update: {}",
-                    std::str::from_utf8(&output.stderr)?
-                );
-            }
-        }
-
-        let mut cmd = tokio::process::Command::new("apt-get");
-        cmd.env("DEBIAN_FRONTEND", "noninteractive")
-            .args(["upgrade", "-y"]);
-        let output = cmd.output().await.context("failed to run apt upgrade")?;
-        match output.status.success() {
-            true => tracing::info!("successfully ran apt upgrade"),
-            false => {
-                anyhow::bail!(
-                    "failed to run apt upgrade: {}",
-                    std::str::from_utf8(&output.stderr)?
-                );
-            }
-        }
+        self.queue
+            .publish(Commands::ScheduleTask {
+                task: "update".into(),
+                properties: BTreeMap::default(),
+            })
+            .await?;
 
         Ok(())
     }