use std::{ops::Deref, pin::Pin}; use futures::{Stream, StreamExt}; use nix_compat::store_path::StorePath; use tonic::async_trait; use tvix_castore::{proto::node::Node, Error}; use crate::pathinfoservice::PathInfoService; /// Provides an interface for looking up root nodes in tvix-castore by given /// a lookup key (usually the basename), and optionally allow a listing. /// #[async_trait] pub trait RootNodes: Send + Sync { /// Looks up a root CA node based on the basename of the node in the root /// directory of the filesystem. async fn get_by_basename(&self, name: &[u8]) -> Result<Option<Node>, Error>; /// Lists all root CA nodes in the filesystem. An error can be returned /// in case listing is not allowed fn list(&self) -> Pin<Box<dyn Stream<Item = Result<Node, Error>> + Send>>; } /// Implements root node lookup for any [PathInfoService]. This represents a flat /// directory structure like /nix/store where each entry in the root filesystem /// directory corresponds to a CA node. #[async_trait] impl<T> RootNodes for T where T: Deref<Target = dyn PathInfoService> + Send + Sync, { async fn get_by_basename(&self, name: &[u8]) -> Result<Option<Node>, Error> { let Ok(store_path) = StorePath::from_bytes(name) else { return Ok(None); }; Ok(self .deref() .get(*store_path.digest()) .await? .map(|path_info| { path_info .node .expect("missing root node") .node .expect("empty node") })) } fn list(&self) -> Pin<Box<dyn Stream<Item = Result<Node, Error>> + Send>> { Box::pin(self.deref().list().map(|result| { result.map(|path_info| { path_info .node .expect("missing root node") .node .expect("empty node") }) })) } }