hyperlog/crates/hyperlog-server/src/external_grpc.rs

use std::{collections::HashMap, net::SocketAddr};

use hyperlog_protos::hyperlog::{
    graph_server::{Graph, GraphServer},
    *,
};
use tonic::{transport, Response};

use crate::{
    querier::{Querier, QuerierExt},
    state::SharedState,
};

pub struct Server {
    querier: Querier,
}

impl Server {
    pub fn new(querier: Querier) -> Self {
        Self { querier }
    }
}

#[tonic::async_trait]
impl Graph for Server {
    async fn get(
        &self,
        request: tonic::Request<GetRequest>,
    ) -> std::result::Result<tonic::Response<GetReply>, tonic::Status> {
        let msg = request.get_ref();

        tracing::trace!("get: req({:?})", msg);

        Ok(Response::new(GetReply {
            item: Some(GraphItem {
                path: "some.path".into(),
                contents: Some(graph_item::Contents::Item(ItemGraphItem {
                    title: "some-title".into(),
                    description: "some-description".into(),
                    item_state: Some(item_graph_item::ItemState::Done(ItemStateDone {})),
                })),
            }),
        }))
    }
}

pub trait ServerExt {
    fn grpc_server(&self) -> Server;
}

impl ServerExt for SharedState {
    fn grpc_server(&self) -> Server {
        Server::new(self.querier())
    }
}

pub async fn serve(state: &SharedState, host: SocketAddr) -> anyhow::Result<()> {
    tracing::info!("listening on {}", host);

    let graph_server = state.grpc_server();

    transport::Server::builder()
        .add_service(GraphServer::new(graph_server))
        .serve(host)
        .await?;

    Ok(())
}