aranya-client 6.0.0

//! Client-daemon connection.

mod device;
mod label;
mod object;
mod role;
mod team;

use std::{fmt::Debug, io, path::Path};

use anyhow::Context as _;
use aranya_crypto::{Csprng, Rng};
#[doc(inline)]
pub use aranya_daemon_api::ChanOp;
#[doc(inline)]
pub use aranya_daemon_api::Perm as Permission;
use aranya_daemon_api::{
    crypto::{
        txp::{self, LengthDelimitedCodec},
        PublicApiKey,
    },
    DaemonApiClient, Version, CS,
};
use aranya_util::{error::ReportExt, Addr};
use tokio::{fs, net::UnixStream};
use tracing::{debug, error, info};
#[cfg(feature = "afc")]
use {
    crate::afc::{ChannelKeys as AfcChannelKeys, Channels as AfcChannels},
    std::sync::Arc,
};

#[doc(inline)]
#[expect(deprecated)]
pub use self::device::KeyBundle;
#[doc(inline)]
pub use self::{
    device::{Device, DeviceId, Devices, PublicKeyBundle},
    label::{Label, LabelId, Labels},
    object::{ObjectId, Rank},
    role::{Role, RoleId, Roles},
    team::{Team, TeamId},
};
use crate::{
    config::{AddTeamConfig, CreateTeamConfig},
    error::{self, aranya_error, InvalidArg, IpcError, Result},
    util::{rpc_context, ApiConv as _},
};

/// Builds a [`Client`].
#[derive(Debug, Default)]
pub struct ClientBuilder<'a> {
    /// The UDS that the daemon is listening on.
    #[cfg(unix)]
    daemon_uds_path: Option<&'a Path>,
}

impl ClientBuilder<'_> {
    /// Creates a new client builder.
    pub fn new() -> Self {
        Self::default()
    }

    /// Creates a client connection to the daemon.
    ///
    /// # Example
    /// ```rust,no_run
    /// use std::net::Ipv4Addr;
    /// # use aranya_client::Client;
    /// # #[tokio::main]
    /// # async fn main() -> anyhow::Result<()> {
    /// let client = Client::builder()
    ///     .with_daemon_uds_path("/var/run/aranya/uds.sock".as_ref())
    ///     .connect()
    ///     .await?;
    /// #    Ok(())
    /// # }
    pub async fn connect(self) -> Result<Client> {
        let Some(uds_path) = self.daemon_uds_path else {
            return Err(IpcError::new(InvalidArg::new(
                "daemon_uds_path",
                "must specify the daemon's UDS path",
            ))
            .into());
        };

        async {
            info!(path = ?uds_path, "connecting to daemon");

            let daemon = {
                let pk = {
                    // The public key is located next to the socket.
                    let api_pk_path = uds_path.parent().unwrap_or(uds_path).join("api.pk");
                    let bytes = fs::read(&api_pk_path)
                        .await
                        .with_context(|| "unable to read daemon API public key")
                        .map_err(IpcError::new)?;
                    PublicApiKey::<CS>::decode(&bytes)
                        .context("unable to decode public API key")
                        .map_err(IpcError::new)?
                };

                let uds_path = uds_path
                    .canonicalize()
                    .context("could not canonicalize uds_path")
                    .map_err(error::other)?;
                let sock = UnixStream::connect(&uds_path)
                    .await
                    .context("unable to connect to UDS path")
                    .map_err(IpcError::new)?;
                let info = uds_path.as_os_str().as_encoded_bytes();
                let codec = LengthDelimitedCodec::builder()
                    .max_frame_length(usize::MAX)
                    .new_codec();
                let transport = txp::client(sock, codec, Rng, pk, info);

                DaemonApiClient::new(tarpc::client::Config::default(), transport).spawn()
            };
            debug!("connected to daemon");

            let got = daemon
                .version(rpc_context())
                .await
                .map_err(IpcError::new)?
                .context("unable to retrieve daemon version")
                .map_err(error::other)?;
            let want = Version::parse(env!("CARGO_PKG_VERSION"))
                .context("unable to parse `CARGO_PKG_VERSION`")
                .map_err(error::other)?;
            if got.major != want.major || got.minor != want.minor {
                return Err(IpcError::new(io::Error::new(
                    io::ErrorKind::Unsupported,
                    format!("version mismatch: `{got}` != `{want}`"),
                ))
                .into());
            }
            debug!(client = ?want, daemon = ?got, "versions");

            #[cfg(feature = "afc")]
            let afc_keys = {
                let afc_shm_info = daemon
                    .afc_shm_info(rpc_context())
                    .await
                    .map_err(IpcError::new)?
                    .context("unable to retrieve afc shm info")
                    .map_err(error::other)?;
                Arc::new(AfcChannelKeys::new(&afc_shm_info)?)
            };

            let client = Client {
                daemon,
                #[cfg(feature = "afc")]
                afc_keys,
            };

            Ok(client)
        }
        .await
        .inspect_err(
            |err: &crate::Error| error!(error = %err.report(), "unable to connect to daemon"),
        )
    }
}

impl<'a> ClientBuilder<'a> {
    /// Specifies the UDS socket path the daemon is listening on.
    #[cfg(unix)]
    #[cfg_attr(docsrs, doc(cfg(unix)))]
    pub fn with_daemon_uds_path(mut self, sock: &'a Path) -> Self {
        self.daemon_uds_path = Some(sock);
        self
    }
}

/// A client for invoking actions on and processing effects from
/// the Aranya graph.
///
/// `Client` interacts with the [Aranya daemon] over
/// a platform-specific IPC mechanism.
///
/// [Aranya daemon]: https://crates.io/crates/aranya-daemon
#[derive(Debug)]
pub struct Client {
    /// RPC connection to the daemon
    pub(crate) daemon: DaemonApiClient,
    /// AFC channel keys.
    #[cfg(feature = "afc")]
    afc_keys: Arc<AfcChannelKeys>,
}

impl Client {
    /// Returns a builder for `Client`.
    pub fn builder<'a>() -> ClientBuilder<'a> {
        ClientBuilder::new()
    }

    /// Returns the address that the Aranya sync server is bound to.
    pub async fn local_addr(&self) -> Result<Addr> {
        self.daemon
            .aranya_local_addr(rpc_context())
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)
    }

    /// See [`Self::get_public_key_bundle`].
    #[deprecated(note = "Use `get_public_key_bundle`")]
    pub async fn get_key_bundle(&self) -> Result<PublicKeyBundle> {
        self.get_public_key_bundle().await
    }

    /// Gets the public key bundle for this device.
    pub async fn get_public_key_bundle(&self) -> Result<PublicKeyBundle> {
        self.daemon
            .get_public_key_bundle(rpc_context())
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)
            .map(PublicKeyBundle::from_api)
    }

    /// Gets the public device ID for this device.
    pub async fn get_device_id(&self) -> Result<DeviceId> {
        self.daemon
            .get_device_id(rpc_context())
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)
            .map(DeviceId::from_api)
    }

    /// Returns a pair of trace IDs `(client_sent, daemon_received)` for
    /// validating RPC trace propagation.
    #[cfg(feature = "test-utils")]
    pub async fn test_trace_id(&self) -> Result<(String, String)> {
        let ctx = rpc_context();
        let client_trace_id = ctx.trace_context.trace_id.to_string();
        let daemon_trace_id = self
            .daemon
            .test_trace_id(ctx)
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)?;
        Ok((client_trace_id, daemon_trace_id))
    }

    /// Create a new graph/team with the current device as the owner.
    pub async fn create_team(&self, cfg: CreateTeamConfig) -> Result<Team<'_>> {
        let team_id = self
            .daemon
            .create_team(rpc_context(), cfg.into())
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)
            .map(TeamId::from_api)?;
        Ok(Team {
            client: self,
            id: team_id.into_api(),
        })
    }

    /// Generate random bytes from a CSPRNG.
    /// Can be used to generate IKM for a generating a PSK seed.
    pub async fn rand(&self, buf: &mut [u8]) {
        <Rng as Csprng>::fill_bytes(&Rng, buf);
    }

    /// Get an existing team.
    pub fn team(&self, team_id: TeamId) -> Team<'_> {
        Team {
            client: self,
            id: team_id.into_api(),
        }
    }

    /// Add a team to local device storage.
    pub async fn add_team(&self, cfg: AddTeamConfig) -> Result<Team<'_>> {
        let cfg = aranya_daemon_api::AddTeamConfig::from(cfg);
        let team_id = TeamId::from_api(cfg.team_id);

        self.daemon
            .add_team(rpc_context(), cfg)
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)?;
        Ok(Team {
            client: self,
            id: team_id.into_api(),
        })
    }

    /// Remove a team from local device storage.
    pub async fn remove_team(&self, team_id: TeamId) -> Result<()> {
        self.daemon
            .remove_team(rpc_context(), team_id.into_api())
            .await
            .map_err(IpcError::new)?
            .map_err(aranya_error)
    }

    /// Get access to Aranya Fast Channels.
    #[cfg(feature = "afc")]
    #[cfg_attr(docsrs, doc(cfg(feature = "afc")))]
    pub fn afc(&self) -> AfcChannels {
        AfcChannels::new(self.daemon.clone(), self.afc_keys.clone())
    }
}