Modify auth middleware

Signed-off-by: Asai Neko <sugar@sne.moe>
2026-01-02 13:00:02 +08:00
parent bd8eecbc7d
commit 4dfd4cd529
10 changed files with 41 additions and 89 deletions
--- a/middleware/jwt.go
+++ b/middleware/jwt.go
@@ -6,7 +6,7 @@ import (
 	"github.com/gin-gonic/gin"
 )

-func JWTAuth() gin.HandlerFunc {
+func JWTAuth(required bool) gin.HandlerFunc {

 	return func(c *gin.Context) {
 		auth := c.GetHeader("Authorization")
@@ -18,6 +18,12 @@ func JWTAuth() gin.HandlerFunc {
 			return
 		}

+		if required == true {
+			c.JSON(401, gin.H{"status": "unauthorized"})
+			c.Next()
+			return
+		}
+
 		if uid == "" {
 			c.Set("user_id", "")
 			c.Next()