Add disable auth flag to route

warrant-dev · Mar 15, 2024 · b7ce015 · b7ce015
1 parent 9542553
commit b7ce015
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/pkg/service/route.go b/pkg/service/route.go
@@ -21,13 +21,15 @@ type Route interface {
 	GetMethod() string
 	GetHandler() http.Handler
 	GetOverrideAuthMiddlewareFunc() AuthMiddlewareFunc
+	GetDisableAuth() bool
 }
 
 type WarrantRoute struct {
 	Pattern                    string
 	Method                     string
 	Handler                    http.Handler
 	OverrideAuthMiddlewareFunc AuthMiddlewareFunc
+	DisableAuth                bool
 }
 
 func (route WarrantRoute) GetPattern() string {
@@ -45,3 +47,7 @@ func (route WarrantRoute) GetHandler() http.Handler {
 func (route WarrantRoute) GetOverrideAuthMiddlewareFunc() AuthMiddlewareFunc {
 	return route.OverrideAuthMiddlewareFunc
 }
+
+func (route WarrantRoute) GetDisableAuth() bool {
+	return route.DisableAuth
+}
diff --git a/pkg/service/router.go b/pkg/service/router.go
@@ -96,13 +96,16 @@ func NewRouter(config config.Config, pathPrefix string, routes []Route, authMidd
 		middlewareWrappedHandler := ChainMiddleware(route.GetHandler(), requestMiddlewares...)
 
 		var err error
-		if route.GetOverrideAuthMiddlewareFunc() != nil {
-			middlewareWrappedHandler, err = route.GetOverrideAuthMiddlewareFunc()(config, middlewareWrappedHandler)
-		} else {
-			middlewareWrappedHandler, err = authMiddleware(config, middlewareWrappedHandler)
-		}
-		if err != nil {
-			return nil, err
+		if !route.GetDisableAuth() {
+			if route.GetOverrideAuthMiddlewareFunc() != nil {
+				middlewareWrappedHandler, err = route.GetOverrideAuthMiddlewareFunc()(config, middlewareWrappedHandler)
+			} else {
+				middlewareWrappedHandler, err = authMiddleware(config, middlewareWrappedHandler)
+			}
+
+			if err != nil {
+				return nil, err
+			}
 		}
 
 		router.Handle(routePattern, middlewareWrappedHandler).Methods(route.GetMethod())