Fix openapi base url for PaaS envs

UKGovernmentBEIS · Nov 22, 2023 · dad8ab3 · dad8ab3
1 parent 9cb9319
commit dad8ab3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/java/uk/gov/beis/els/api/openapi/OpenApiConfig.java b/src/main/java/uk/gov/beis/els/api/openapi/OpenApiConfig.java
@@ -6,6 +6,7 @@
 import java.util.Map;
 import java.util.TreeMap;
 import org.springdoc.core.customizers.OpenApiCustomizer;
+import org.springdoc.core.customizers.ServerBaseUrlCustomizer;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
@@ -31,4 +32,14 @@ public OpenApiCustomizer sortSchemasAlphabetically() {
     };
   }
 
+  @Bean
+  ServerBaseUrlCustomizer serverBaseUrlCustomizer() {
+    return baseUrl -> {
+      if (baseUrl.startsWith("http://localhost")) {
+        return baseUrl;
+      }
+      return baseUrl.replace("http://", "https://");
+    };
+  }
+
 }