lakehouse/crates/catalogd/src/grpc.rs

use proto::lakehouse::{
    catalog_service_server::CatalogService,
    CreateDatasetRequest, DatasetResponse, GetDatasetByNameRequest,
    GetDatasetRequest, ListDatasetsRequest, ListDatasetsResponse,
    ObjectRef as ProtoObjectRef,
};
use shared::types::{DatasetId, ObjectRef, SchemaFingerprint};
use tonic::{Request, Response, Status};
use uuid::Uuid;

use crate::registry::Registry;

pub struct CatalogGrpc {
    registry: Registry,
}

impl CatalogGrpc {
    pub fn new(registry: Registry) -> Self {
        Self { registry }
    }
}

fn manifest_to_proto(m: &shared::types::DatasetManifest) -> DatasetResponse {
    DatasetResponse {
        id: m.id.to_string(),
        name: m.name.clone(),
        schema_fingerprint: m.schema_fingerprint.0.clone(),
        objects: m.objects.iter().map(|o| ProtoObjectRef {
            bucket: o.bucket.clone(),
            key: o.key.clone(),
            size_bytes: o.size_bytes,
            created_at: o.created_at.to_rfc3339(),
        }).collect(),
        created_at: m.created_at.to_rfc3339(),
        updated_at: m.updated_at.to_rfc3339(),
    }
}

#[tonic::async_trait]
impl CatalogService for CatalogGrpc {
    async fn create_dataset(
        &self,
        request: Request<CreateDatasetRequest>,
    ) -> Result<Response<DatasetResponse>, Status> {
        let req = request.into_inner();
        let now = chrono::Utc::now();
        let objects: Vec<ObjectRef> = req.objects.into_iter().map(|o| ObjectRef {
            bucket: o.bucket,
            key: o.key,
            size_bytes: o.size_bytes,
            created_at: now,
        }).collect();

        let manifest = self.registry
            .register(req.name, SchemaFingerprint(req.schema_fingerprint), objects)
            .await
            .map_err(|e| Status::internal(e))?;

        Ok(Response::new(manifest_to_proto(&manifest)))
    }

    async fn get_dataset(
        &self,
        request: Request<GetDatasetRequest>,
    ) -> Result<Response<DatasetResponse>, Status> {
        let req = request.into_inner();
        let uuid = Uuid::parse_str(&req.id).map_err(|e| Status::invalid_argument(e.to_string()))?;
        let id = DatasetId(uuid);

        match self.registry.get(&id).await {
            Some(m) => Ok(Response::new(manifest_to_proto(&m))),
            None => Err(Status::not_found(format!("dataset not found: {}", req.id))),
        }
    }

    async fn get_dataset_by_name(
        &self,
        request: Request<GetDatasetByNameRequest>,
    ) -> Result<Response<DatasetResponse>, Status> {
        let req = request.into_inner();
        match self.registry.get_by_name(&req.name).await {
            Some(m) => Ok(Response::new(manifest_to_proto(&m))),
            None => Err(Status::not_found(format!("dataset not found: {}", req.name))),
        }
    }

    async fn list_datasets(
        &self,
        _request: Request<ListDatasetsRequest>,
    ) -> Result<Response<ListDatasetsResponse>, Status> {
        let datasets = self.registry.list().await;
        let responses = datasets.iter().map(manifest_to_proto).collect();
        Ok(Response::new(ListDatasetsResponse { datasets: responses }))
    }
}