update

makeless · Sep 28, 2021 · 17593c2 · 17593c2
1 parent c942c79
commit 17593c2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/authenticator/authenticator.go b/authenticator/authenticator.go
@@ -12,6 +12,7 @@ type Authenticator interface {
 	SetMiddleware(middleware *jwt.GinJWTMiddleware)
 	GetMiddleware() *jwt.GinJWTMiddleware
 	CreateMiddleware() error
+	AuthMiddleware() gin.HandlerFunc
 	GetSecurity() makeless_go_security.Security
 	GetRealm() string
 	GetKey() []byte

diff --git a/authenticator/basic/authenticator.go b/authenticator/basic/authenticator.go
@@ -2,6 +2,7 @@ package makeless_go_authenticator_basic
 
 import (
 	"github.com/appleboy/gin-jwt/v2"
+	"github.com/gin-gonic/gin"
 	"github.com/makeless/makeless-go/security"
 	"net/http"
 	"sync"
@@ -36,6 +37,10 @@ func (authenticator *Authenticator) GetMiddleware() *jwt.GinJWTMiddleware {
 	return authenticator.Middlware
 }
 
+func (authenticator *Authenticator) AuthMiddleware() gin.HandlerFunc {
+	return authenticator.GetMiddleware().MiddlewareFunc()
+}
+
 func (authenticator *Authenticator) GetSecurity() makeless_go_security.Security {
 	authenticator.RLock()
 	defer authenticator.RUnlock()

diff --git a/http/basic/router.go b/http/basic/router.go
@@ -12,6 +12,7 @@ type Router struct {
 }
 
 func (router *Router) Init(http makeless_go_http.Http) error {
+	gin.SetMode(http.GetMode())
 	r := gin.Default()
 	r.Use(http.CorsMiddleware(http.GetOrigins(), http.GetOriginsFunc(), http.GetHeaders()))