hello world

endpoints/authorization.go

@@ -0,0 +1,52 @@
+package endpoints
+
+import (
+	"strings"
+
+	"git.zomo.dev/zomo/discord-retokenizer/storage"
+	"github.com/gin-gonic/gin"
+)
+
+type AuthorizationScope int
+const (
+	AuthorizationScopeNone AuthorizationScope = iota
+	AuthorizationScopeUser
+	AuthorizationScopeBot
+)
+
+func getAuthorization(c *gin.Context) (AuthorizationScope, string) {
+	header := c.GetHeader("Authorization")
+	if header == "" {
+		return AuthorizationScopeNone, ""
+	}
+	headerSpl := strings.Split(header, " ")
+	if len(headerSpl) != 2 {
+		return AuthorizationScopeNone, ""
+	}
+	if headerSpl[0] == "Bearer" {
+		if storage.CheckLoginToken(headerSpl[1]) {
+			return AuthorizationScopeUser, headerSpl[1]
+		}
+	}
+	if headerSpl[0] == "Bot" {
+		// TODO check bot token
+		if true {
+			return AuthorizationScopeBot, headerSpl[1]
+		}
+	}
+	return AuthorizationScopeNone, ""
+
+}
+
+func isUserAuthorized(c *gin.Context) bool {
+	scope, _ := getAuthorization(c)
+	return scope == AuthorizationScopeUser
+}
+
+func userIsAuthorized(c *gin.Context) {
+	if isUserAuthorized(c) {
+		c.Next()
+	} else {
+		c.AbortWithStatus(401)
+	}
+}