feat: with ssh agent

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

crates/cuddle-ci/src/rust_service.rs

@@ -410,6 +410,7 @@ mod clap_sanity_test;
 mod docker_cache;
 mod mold;
 mod sqlx;
+mod ssh_agent;
 
 pub mod extensions {
     pub use super::apt::*;
@@ -421,6 +422,7 @@ pub mod extensions {
     pub use super::docker_cache::*;
     pub use super::mold::*;
     pub use super::sqlx::*;
+    pub use super::ssh_agent::*;
 }
 
 #[cfg(test)]

crates/cuddle-ci/src/rust_service/ssh_agent.rs

@@ -0,0 +1,55 @@
+use std::sync::Arc;
+
+use async_trait::async_trait;
+use dagger_sdk::{Container, Socket};
+
+use crate::{dagger_middleware::DaggerMiddleware, leptos_service::LeptosService};
+
+use super::RustService;
+
+pub struct SshAgent {
+    socket: Socket,
+}
+
+impl SshAgent {
+    pub fn new(socket: Socket) -> Self {
+        Self { socket }
+    }
+}
+
+#[async_trait]
+impl DaggerMiddleware for SshAgent {
+    async fn handle(&self, container: Container) -> eyre::Result<Container> {
+        let c = container.with_unix_socket("/tmp/ssh_socket", self.socket);
+
+        Ok(c)
+    }
+}
+
+pub trait SshAgentExt {
+    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+        self
+    }
+}
+
+impl SshAgentExt for RustService {
+    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+        self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
+            SshAgent::new(socket.clone()).extend(deps),
+        )))
+        .with_stage(super::RustServiceStage::AfterBase(Arc::new(
+            SshAgent::new(socket.clone()).extend(deps),
+        )))
+    }
+}
+
+impl SshAgentExt for LeptosService {
+    fn with_socket(&mut self, socket: dagger_sdk::Socket) -> &mut Self {
+        self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
+            SshAgent::new(socket.clone()).extend(deps),
+        )))
+        .with_stage(super::RustServiceStage::AfterBase(Arc::new(
+            SshAgent::new(socket.clone()).extend(deps),
+        )))
+    }
+}