Changeset 0.29.0 (#195)
This commit is contained in:
@@ -5,16 +5,14 @@ import (
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"git.gobha.me/xcaliber/chat-switchboard/database"
|
||||
"git.gobha.me/xcaliber/chat-switchboard/store"
|
||||
)
|
||||
|
||||
// RequireTeamAdmin returns middleware that restricts access to users who are
|
||||
// admins of the team identified by :teamId in the URL path.
|
||||
// System admins (role=admin) are always allowed through.
|
||||
// Must be used after Auth() middleware.
|
||||
func RequireTeamAdmin() gin.HandlerFunc {
|
||||
// RequireTeamAdmin returns middleware that restricts access to team admins.
|
||||
// System admins are always allowed through.
|
||||
// v0.29.0: accepts TeamStore instead of using database.DB directly.
|
||||
func RequireTeamAdmin(teams store.TeamStore) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
// System admins bypass team check
|
||||
role, _ := c.Get("role")
|
||||
if role == "admin" {
|
||||
c.Next()
|
||||
@@ -30,13 +28,8 @@ func RequireTeamAdmin() gin.HandlerFunc {
|
||||
return
|
||||
}
|
||||
|
||||
var teamRole string
|
||||
err := database.DB.QueryRow(`
|
||||
SELECT role FROM team_members
|
||||
WHERE team_id = $1 AND user_id = $2
|
||||
`, teamID, userID).Scan(&teamRole)
|
||||
|
||||
if err != nil || teamRole != "admin" {
|
||||
isAdmin, err := teams.IsTeamAdmin(c.Request.Context(), teamID, userID.(string))
|
||||
if err != nil || !isAdmin {
|
||||
c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
|
||||
"error": "team admin access required",
|
||||
})
|
||||
@@ -50,7 +43,8 @@ func RequireTeamAdmin() gin.HandlerFunc {
|
||||
// RequireTeamMember returns middleware that restricts access to users who
|
||||
// belong to the team identified by :teamId (any role).
|
||||
// System admins are always allowed through.
|
||||
func RequireTeamMember() gin.HandlerFunc {
|
||||
// v0.29.0: accepts TeamStore instead of using database.DB directly.
|
||||
func RequireTeamMember(teams store.TeamStore) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
role, _ := c.Get("role")
|
||||
if role == "admin" {
|
||||
@@ -67,12 +61,8 @@ func RequireTeamMember() gin.HandlerFunc {
|
||||
return
|
||||
}
|
||||
|
||||
var exists bool
|
||||
database.DB.QueryRow(`
|
||||
SELECT EXISTS(SELECT 1 FROM team_members WHERE team_id = $1 AND user_id = $2)
|
||||
`, teamID, userID).Scan(&exists)
|
||||
|
||||
if !exists {
|
||||
isMember, _ := teams.IsMember(c.Request.Context(), teamID, userID.(string))
|
||||
if !isMember {
|
||||
c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
|
||||
"error": "team membership required",
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user