DEVELOPMENT ENVIRONMENT

~liljamo/felu

ref: 08a7a55902c758aaf1432ce557f0143715c363ee felu/internal/middlewares/admin.go -rw-r--r-- 958 bytes
08a7a559Jonni Liljamo feat: tsig 17 days ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
/*
 * Copyright (C) 2024 Jonni Liljamo <jonni@liljamo.com>
 *
 * This file is licensed under AGPL-3.0-or-later, see NOTICE and LICENSE for
 * more information.
 */

package middlewares

import (
	"net/http"

	"git.src.quest/~liljamo/felu/internal/db"
	"github.com/gin-gonic/gin"
)

// AdminOnly returns a gin middleware for checking if a user is an admin.
func AdminOnly() gin.HandlerFunc {
	return func(c *gin.Context) {
		userID, exists := c.Get("user_id")
		if exists {
			user, err := db.FetchUserWithID(userID.(string))
			if err == nil {
				if user.IsAdmin {
					c.Next()
				} else {
					c.Redirect(http.StatusTemporaryRedirect, "/manage")
					c.Abort()
				}
			} else {
				c.String(http.StatusInternalServerError, "This should not be possible, but don't quote me on that, S01E02")
				c.Abort()
			}
		} else {
			c.String(http.StatusInternalServerError, "This should not be possible, but don't quote me on that, S01E01")
			c.Abort()
		}
	}
}