use std::sync::Arc;

use sea_orm::DatabaseConnection;
use tonic::transport::Server;

pub mod ssh;

#[async_trait::async_trait]
pub trait AgentConnectorTrait: Send + Sync {
    async fn start_server(
        &mut self,
        settings: &crate::config::settings::Settings,
        cert_service: Arc<dyn crate::service::certificate::CertificateService>,
        connection: DatabaseConnection,
    ) -> Result<(), Box<dyn std::error::Error + Send + Sync>>;
}

pub struct AgentConnector {
    connector: Box<dyn AgentConnectorTrait>,
}

impl AgentConnector {
    pub fn new(connector: Box<dyn AgentConnectorTrait>) -> Self {
        Self { connector }
    }
}

#[async_trait::async_trait]
impl AgentConnectorTrait for AgentConnector {
    async fn start_server(
        &mut self,
        settings: &crate::config::settings::Settings,
        cert_service: Arc<dyn crate::service::certificate::CertificateService>,
        connection: DatabaseConnection,
    ) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
        self.connector
            .start_server(settings, cert_service, connection)
            .await
    }
}