1 files changed, 2 insertions, 19 deletions
diff --git a/src/main.rs b/src/main.rs
index 1a3ff00c..13d313d7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,31 +1,14 @@
-pub mod authorization;
+use authzd::create_server;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    use envoy_types::ext_authz::v3::pb::AuthorizationServer;
-    use std::sync::Arc;
-    use tonic::transport::Server;
-
     tracing_subscriber::fmt().json().init();
 
     let addr = std::env::var("BIND_ADDR")
         .unwrap_or_else(|_| "[::1]:50051".to_string())
         .parse()?;
 
-    let (_health_reporter, health_service) = tonic_health::server::health_reporter();
-
-    let authorizer = Arc::new(authorization::CedarAuthorizer::default());
-    let check_service = authorization::CheckService::new(authorizer);
-
-    let server = Server::builder()
-        .add_service(AuthorizationServer::new(check_service))
-        .add_service(health_service)
-        .add_service(
-            tonic_reflection::server::Builder::configure()
-                .register_encoded_file_descriptor_set(tonic_health::pb::FILE_DESCRIPTOR_SET)
-                .build_v1()
-                .unwrap(),
-        );
+    let server = create_server()?;
 
     log::info!("Listening on... {addr}");
     server.serve(addr).await?;