1 files changed, 57 insertions, 0 deletions
diff --git a/crates/configuration/src/server.rs b/crates/configuration/src/server.rs
new file mode 100644
index 0000000..1131144
--- /dev/null
+++ b/crates/configuration/src/server.rs
@@ -0,0 +1,57 @@
+mod error;
+mod http_svc;
+mod interceptor;
+mod version;
+
+use axum::http::header::CONTENT_TYPE;
+use http_svc::build_router;
+use interceptor::MyInterceptor;
+use tonic::service::Routes;
+use tower::{make::Shared, steer::Steer};
+use warden_core::{
+    FILE_DESCRIPTOR_SET,
+    configuration::routing::{
+        mutate_routing_server::MutateRoutingServer, query_routing_server::QueryRoutingServer,
+    },
+};
+
+use crate::state::AppHandle;
+
+pub async fn serve<S>(state: AppHandle) -> anyhow::Result<Shared<S>> {
+    let app = build_router(state.clone());
+
+    let service = QueryRoutingServer::with_interceptor(state.clone(), MyInterceptor);
+
+    let routing_reflector = tonic_reflection::server::Builder::configure()
+        .register_encoded_file_descriptor_set(FILE_DESCRIPTOR_SET)
+        .build_v1()?;
+
+    let grpc_server = Routes::new(service)
+        .add_service(MutateRoutingServer::with_interceptor(
+            state.clone(),
+            MyInterceptor,
+        ))
+        .add_service(routing_reflector)
+        .into_axum_router();
+
+    let service = Steer::new(
+        vec![app, grpc_server],
+        |req: &axum::extract::Request, _services: &[_]| {
+            if req
+                .headers()
+                .get(CONTENT_TYPE)
+                .map(|content_type| content_type.as_bytes())
+                .filter(|content_type| content_type.starts_with(b"application/grpc"))
+                .is_some()
+            {
+                // grpc service
+                1
+            } else {
+                // http service
+                0
+            }
+        },
+    );
+
+    Ok(Shared::new(service))
+}