DeveloperDurp/DurpAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
DeveloperDurp
2023-05-20 09:52:00 -05:00
parent ca99c8850d
commit 1b34c79386
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.go
View File

@@ -48,7 +48,7 @@ func main() {
} }
unraid := v1.Group("/unraid") unraid := v1.Group("/unraid")
{ {
groups := []string{"grafana"} groups := []string{"grafana"}
unraid.Use(authMiddleware(groups)) unraid.Use(authMiddleware(groups))
unraid.GET("powerusage", c.UnraidPowerUsage) unraid.GET("powerusage", c.UnraidPowerUsage)
} }
@@ -60,7 +60,7 @@ func main() {
func authMiddleware(allowedGroups []string) gin.HandlerFunc { func authMiddleware(allowedGroups []string) gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
// Get the user groups from the request headers // Get the user groups from the request headers
groupsHeader := c.GetHeader("X-Forwarded-Groups") groupsHeader := c.GetHeader("X-authentik-groups")
// Split the groups header value into individual groups // Split the groups header value into individual groups
groups := strings.Split(groupsHeader, ",") groups := strings.Split(groupsHeader, ",")
@@ -82,9 +82,9 @@ func authMiddleware(allowedGroups []string) gin.HandlerFunc {
// If the user is not in any of the allowed groups, respond with unauthorized access // If the user is not in any of the allowed groups, respond with unauthorized access
if !isAllowed { if !isAllowed {
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{ c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{
"message": "Unauthorized access", "message": "Unauthorized access",
"groups": groupsHeader, "groups": groupsHeader,
}) })
return return
} }