fix: make sure files embedded during build

pactus-project · Jun 29, 2024 · fd6f6aa · fd6f6aa
1 parent a1fe02a
commit fd6f6aa
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/www/grpc/gateway.go b/www/grpc/gateway.go
@@ -26,12 +26,16 @@ type GatewayConfig struct {
 
 // getOpenAPIHandler serves an OpenAPI UI.
 func (*Server) getOpenAPIHandler() (http.Handler, error) {
-	swagger, err := fs.Sub(swaggerFS, "swaggerFS-ui")
-	if err != nil {
-		return nil, err
+	if _, err := swaggerFS.ReadFile("swagger-ui/index.html"); err == nil {
+		swagger, err := fs.Sub(swaggerFS, "swagger-ui")
+		if err != nil {
+			return nil, err
+		}
+
+		return http.FileServer(http.FS(swagger)), nil
 	}
 
-	return http.FileServer(http.FS(swagger)), nil
+	return http.FileServer(http.Dir("swagger-ui")), nil
 }
 
 func (s *Server) startGateway(grpcAddr string) error {