fix geoip

.github/workflows/test.yml

@@ -11,17 +11,17 @@ env:
 
 jobs:
   test:
-    runs-on: bazel-runner-small
+    runs-on: dind-small
     container:
-      image: ghcr.io/catthehacker/ubuntu:full-22.04
+      image: ghcr.io/catthehacker/ubuntu:rust-22.04
 
     steps:
     - uses: actions/checkout@v4
 
     - uses: Swatinem/rust-cache@v2
 
     - name: Install protoc
-      run: sudo apt-get update && sudo apt-get install protobuf-compiler
+      run: sudo apt-get update && sudo apt-get -y install protobuf-compiler
 
     - name: Run tests
       run: cargo test

src/policy/mod.rs

@@ -1,6 +1,5 @@
 pub mod denylist;
 pub mod domain_canister;
-mod test;
 
 use std::{fs, path::PathBuf};
 

src/routing/middleware/geoip.rs

@@ -48,8 +48,16 @@ pub async fn middleware(
 ) -> Response {
     // Lookup code
     let country_code = geoip.lookup(conn_info.remote_addr.ip());
-    request.extensions_mut().insert(country_code.clone());
+
+    if let Some(v) = &country_code {
+        request.extensions_mut().insert(v.clone());
+    }
+
     let mut response = next.run(request).await;
-    response.extensions_mut().insert(country_code);
+
+    if let Some(v) = country_code {
+        response.extensions_mut().insert(v);
+    }
+
     response
 }