distant/src/cli/manager.rs

use crate::{
    config::{AccessControl, NetworkConfig},
    paths::{global as global_paths, user as user_paths},
};
use anyhow::Context;
use distant_core::{net::PlainCodec, DistantManager, DistantManagerConfig, DistantManagerRef};
use log::*;

pub struct Manager {
    pub access: AccessControl,
    pub config: DistantManagerConfig,
    pub network: NetworkConfig,
}

impl Manager {
    /// Begin listening on the network interface specified within [`NetworkConfig`]
    pub async fn listen(self) -> anyhow::Result<DistantManagerRef> {
        let user = self.config.user;

        #[cfg(unix)]
        {
            let socket_path = self.network.unix_socket.as_deref().unwrap_or({
                if user {
                    user_paths::UNIX_SOCKET_PATH.as_path()
                } else {
                    global_paths::UNIX_SOCKET_PATH.as_path()
                }
            });

            // Ensure that the path to the socket exists
            if let Some(parent) = socket_path.parent() {
                tokio::fs::create_dir_all(parent)
                    .await
                    .with_context(|| format!("Failed to create socket directory {parent:?}"))?;
            }

            let boxed_ref = DistantManager::start_unix_socket_with_permissions(
                self.config,
                socket_path,
                PlainCodec,
                self.access.into_mode(),
            )
            .await
            .with_context(|| format!("Failed to start manager at socket {socket_path:?}"))?
            .into_inner()
            .into_boxed_server_ref()
            .map_err(|_| anyhow::anyhow!("Got wrong server ref"))?;

            info!("Manager listening using unix socket @ {:?}", socket_path);
            Ok(*boxed_ref)
        }

        #[cfg(windows)]
        {
            let pipe_name = self.network.windows_pipe.as_deref().unwrap_or(if user {
                user_paths::WINDOWS_PIPE_NAME.as_str()
            } else {
                global_paths::WINDOWS_PIPE_NAME.as_str()
            });
            let boxed_ref =
                DistantManager::start_local_named_pipe(self.config, pipe_name, PlainCodec)
                    .await
                    .with_context(|| {
                        format!("Failed to start manager with pipe named '{pipe_name}'")
                    })?
                    .into_inner()
                    .into_boxed_server_ref()
                    .map_err(|_| anyhow::anyhow!("Got wrong server ref"))?;

            info!("Manager listening using local named pipe @ {:?}", pipe_name);
            Ok(*boxed_ref)
        }
    }
}