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

use std::sync::Arc;

use async_trait::async_trait;
use dagger_sdk::{Container, ContainerWithNewFileOptsBuilder, Socket};
use eyre::Context;

use crate::{dagger_middleware::DaggerMiddleware, leptos_service::LeptosService};

use super::RustService;

pub struct SshAgent {
    client: dagger_sdk::Query,
}

impl SshAgent {
    pub fn new(client: dagger_sdk::Query) -> Self {
        Self { client }
    }
}

#[async_trait]
impl DaggerMiddleware for SshAgent {
    async fn handle(&self, container: Container) -> eyre::Result<Container> {
        let sock_var =
            std::env::var("SSH_AUTH_SOCK").context("failed to find variable SSH_AUTH_SOCK")?;

        let socket = self.client.host().unix_socket(&sock_var);

        let home = container.env_variable("HOME").await.unwrap_or("".into());

        let c = container
            .with_new_file_opts(
                format!("{home}/.ssh/config"),
                ContainerWithNewFileOptsBuilder::default()
                    .contents(
                        r#"
Host *
   StrictHostKeyChecking no
   UserKnownHostsFile=/dev/null
                "#,
                    )
                    .build()?,
            )
            .with_unix_socket(&sock_var, socket)
            .with_env_variable("SSH_AUTH_SOCK", &sock_var);

        Ok(c)
    }
}

pub trait SshAgentExt {
    fn with_ssh_agent(&mut self) -> &mut Self {
        self
    }
}

impl SshAgentExt for RustService {
    fn with_ssh_agent(&mut self) -> &mut Self {
        let client = self.client.clone();

        self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
            SshAgent::new(client.clone()),
        )))
        .with_stage(super::RustServiceStage::AfterBase(Arc::new(SshAgent::new(
            client,
        ))))
    }
}

impl SshAgentExt for LeptosService {
    fn with_ssh_agent(&mut self) -> &mut Self {
        let client = self.client.clone();

        self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
            SshAgent::new(client.clone()),
        )))
        .with_stage(super::RustServiceStage::AfterBase(Arc::new(SshAgent::new(
            client,
        ))))
    }
}