add: features for winebridge and server

Author: edfloreshz

Cargo.toml

@@ -3,6 +3,11 @@ name = "bottles-core"
 version = "0.1.0"
 edition = "2021"
 
+[features]
+default = []
+winebridge = ["dep:windows"]
+server = ["default"]
+
 [dependencies]
 serde = { version = "1.0.217", features = ["derive"] }
 serde_json = "1.0.134"
@@ -12,6 +17,17 @@ tonic = "0.12.3"
 tokio = { version = "1.28.2", features = ["full"] }
 prost = "0.13.4"
 
+[dependencies.windows]
+version = "0.58.0"
+optional = true
+features = [
+    "Data_Xml_Dom",
+    "Win32_Foundation",
+    "Win32_Security",
+    "Win32_System_Threading",
+    "Win32_UI_WindowsAndMessaging",
+]
+
 [build-dependencies]
 tonic-build = "0.12.3"
 

src/lib.rs

@@ -5,3 +5,9 @@ pub mod proto {
     tonic::include_proto!("winebridge");
     tonic::include_proto!("bottles");
 }
+
+#[cfg(feature = "winebridge")]
+pub mod winebridge;
+
+#[cfg(feature = "server")]
+pub mod server;

src/server.rs

@@ -0,0 +1,36 @@
+use crate::proto::{self, bottles_server::Bottles, wine_bridge_client::WineBridgeClient};
+
+#[derive(Debug, Default)]
+pub struct BottlesService;
+
+#[tonic::async_trait]
+impl Bottles for BottlesService {
+    async fn health(
+        &self,
+        request: tonic::Request<proto::HealthRequest>,
+    ) -> Result<tonic::Response<proto::HealthResponse>, tonic::Status> {
+        let request = request.get_ref();
+        println!("Received request: {:?}", request);
+        Ok(tonic::Response::new(proto::HealthResponse { ok: true }))
+    }
+
+    async fn notify(
+        &self,
+        request: tonic::Request<proto::NotifyRequest>,
+    ) -> Result<tonic::Response<proto::NotifyResponse>, tonic::Status> {
+        let request = request.get_ref();
+        println!("Received request: {:?}", request);
+        let mut client = WineBridgeClient::connect("http://[::1]:50051")
+            .await
+            .map_err(|e| tonic::Status::from_error(Box::new(e)))?;
+
+        let request = proto::MessageRequest {
+            message: request.message.clone(),
+        };
+        let response = client.message(request).await?;
+        let response = response.get_ref();
+        Ok(tonic::Response::new(proto::NotifyResponse {
+            success: response.success,
+        }))
+    }
+}

src/winebridge.rs

@@ -0,0 +1,33 @@
+use super::proto::{self, wine_bridge_server::WineBridge};
+#[cfg(feature = "winebridge")]
+use windows::{
+    core::{s, PCSTR},
+    Win32::UI::WindowsAndMessaging::{MessageBoxA, MB_OK},
+};
+
+#[derive(Debug, Default)]
+pub struct WineBridgeService;
+
+#[tonic::async_trait]
+impl WineBridge for WineBridgeService {
+    async fn message(
+        &self,
+        request: tonic::Request<proto::MessageRequest>,
+    ) -> Result<tonic::Response<proto::MessageResponse>, tonic::Status> {
+        let request = request.get_ref();
+        println!("Got a request: {:?}", request);
+        let message = request.message.as_str();
+        let c_message = std::ffi::CString::new(message).unwrap();
+        unsafe {
+            MessageBoxA(
+                None,
+                PCSTR(c_message.as_ptr() as *const u8),
+                s!("Hello"),
+                MB_OK,
+            );
+        }
+        Ok(tonic::Response::new(proto::MessageResponse {
+            success: true,
+        }))
+    }
+}