Initial commit

crates/daemon/src/main.rs

@@ -0,0 +1,35 @@
+use shared::ai::ai_daemon_server::{AiDaemonServer, AiDaemon};
+use shared::ai::{PromptRequest, PromptResponse};
+use tonic::{transport::Server, Response, Request, Status};
+
+#[derive(Default)]
+pub struct DaemonServer {}
+
+#[tonic::async_trait]
+impl AiDaemon for DaemonServer {
+    async fn prompt(
+        &self,
+        request: Request<PromptRequest>,
+    ) -> Result<Response<PromptResponse>, Status> {
+        println!("Request from {:?}: {:?}", request.remote_addr(), request.into_inner().prompt);
+        let reply = PromptResponse {
+            response: "Very nice!".to_string()
+        };
+        Ok(Response::new(reply))
+    }
+}
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let addr = "[::1]:50051".parse().unwrap();
+    let daemon = DaemonServer::default();
+    let reflection_service = tonic_reflection::server::Builder::configure()
+        .register_encoded_file_descriptor_set(shared::ai::FILE_DESCRIPTOR_SET)
+        .build_v1()?;
+    Server::builder()
+        .add_service(AiDaemonServer::new(daemon))
+        .add_service(reflection_service)
+        .serve(addr)
+        .await?;
+    Ok(())
+}