Use discordgo's permission calculator

This commit is contained in:
Tulir Asokan
2023-01-29 14:58:58 +02:00
parent d78d4c4da5
commit 840e95b388
3 changed files with 15 additions and 43 deletions

View File

@@ -1,29 +1,23 @@
package main package main
import ( import (
"encoding/json"
"errors" "errors"
"github.com/bwmarrin/discordgo" "github.com/bwmarrin/discordgo"
) )
func ptrBool(val bool) *bool {
return &val
}
func (user *User) channelIsBridgeable(channel *discordgo.Channel) bool { func (user *User) channelIsBridgeable(channel *discordgo.Channel) bool {
switch channel.Type { switch channel.Type {
case discordgo.ChannelTypeGuildText, discordgo.ChannelTypeGuildNews: case discordgo.ChannelTypeGuildText, discordgo.ChannelTypeGuildNews:
// allowed // allowed
case discordgo.ChannelTypeDM, discordgo.ChannelTypeGroupDM:
// DMs are always bridgeable, no need for permission checks
return true
default: default:
// everything else is not allowed // everything else is not allowed
return false return false
} }
hasRole := map[string]bool{
channel.GuildID: true,
}
var roles []string
member, err := user.Session.State.Member(channel.GuildID, user.DiscordID) member, err := user.Session.State.Member(channel.GuildID, user.DiscordID)
if errors.Is(err, discordgo.ErrStateNotFound) { if errors.Is(err, discordgo.ErrStateNotFound) {
user.log.Debugfln("Fetching own membership in %s to check own roles", channel.GuildID) user.log.Debugfln("Fetching own membership in %s to check own roles", channel.GuildID)
@@ -39,37 +33,15 @@ func (user *User) channelIsBridgeable(channel *discordgo.Channel) bool {
} else if err != nil { } else if err != nil {
user.log.Warnfln("Failed to get own membership in %s from cache to determine own roles for bridging %s: %v", channel.GuildID, channel.ID, err) user.log.Warnfln("Failed to get own membership in %s from cache to determine own roles for bridging %s: %v", channel.GuildID, channel.ID, err)
} }
if member != nil { err = user.Session.State.ChannelAdd(channel)
roles = member.Roles if err != nil {
for _, role := range member.Roles { user.log.Warnfln("Failed to add channel %s/%s to cache: %v", channel.GuildID, channel.ID, err)
hasRole[role] = true
}
} }
var userAllowed, roleAllowed *bool perms, err := user.Session.State.UserChannelPermissions(user.DiscordID, channel.ID)
for _, override := range channel.PermissionOverwrites { if err != nil {
if override.Type == discordgo.PermissionOverwriteTypeMember && override.ID == user.DiscordID { user.log.Warnfln("Failed to get permissions in %s/%s to determine if it's bridgeable: %v", channel.GuildID, channel.ID, err)
if override.Allow&discordgo.PermissionViewChannel > 0 { return true
userAllowed = ptrBool(true)
} else if override.Deny&discordgo.PermissionViewChannel > 0 {
userAllowed = ptrBool(false)
}
} else if override.Type == discordgo.PermissionOverwriteTypeRole && hasRole[override.ID] {
if override.Allow&discordgo.PermissionViewChannel > 0 {
roleAllowed = ptrBool(true)
} else if override.Deny&discordgo.PermissionViewChannel > 0 {
roleAllowed = ptrBool(false)
}
}
} }
allowed := true user.log.Debugfln("Computed permissions in %s/%s: %d (view channel: %t)", channel.GuildID, channel.ID, perms, perms&discordgo.PermissionViewChannel > 0)
if userAllowed != nil { return perms&discordgo.PermissionViewChannel > 0
allowed = *userAllowed
} else if roleAllowed != nil {
allowed = *roleAllowed
}
if !allowed {
dat, _ := json.Marshal(channel.PermissionOverwrites)
user.log.Debugfln("Permission overwrites (%s) resulted in %s/%s not being allowed to bridge with roles %+v", dat, channel.GuildID, channel.ID, roles)
}
return allowed
} }

2
go.mod
View File

@@ -34,4 +34,4 @@ require (
maunium.net/go/mauflag v1.0.0 // indirect maunium.net/go/mauflag v1.0.0 // indirect
) )
replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20230129113304-139f45f429a0 replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20230129125832-37978ff8e399

4
go.sum
View File

@@ -1,6 +1,6 @@
github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
github.com/beeper/discordgo v0.0.0-20230129113304-139f45f429a0 h1:RrF9ffkMyEsUtZqWR/m/KXSrYbpyT7bkuL+KY8pexSE= github.com/beeper/discordgo v0.0.0-20230129125832-37978ff8e399 h1:3GZhhiyeXo/r40NmaQddBpCfosSSIrSrqZBLXJWrtYc=
github.com/beeper/discordgo v0.0.0-20230129113304-139f45f429a0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= github.com/beeper/discordgo v0.0.0-20230129125832-37978ff8e399/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
github.com/coreos/go-systemd/v22 v22.3.3-0.20220203105225-a9a7ef127534/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc= github.com/coreos/go-systemd/v22 v22.3.3-0.20220203105225-a9a7ef127534/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=