wings/router/tokens/websocket.go

31 lines
656 B
Go
Raw Normal View History

package tokens
2020-04-06 01:00:33 +00:00
import (
"encoding/json"
"github.com/gbrlsnchs/jwt/v3"
"strings"
2020-04-06 01:00:33 +00:00
)
type WebsocketPayload struct {
2020-04-06 01:00:33 +00:00
jwt.Payload
UserID json.Number `json:"user_id"`
ServerUUID string `json:"server_uuid"`
Permissions []string `json:"permissions"`
}
// Returns the JWT payload.
func (p *WebsocketPayload) GetPayload() *jwt.Payload {
return &p.Payload
}
2020-04-06 01:00:33 +00:00
// Checks if the given token payload has a permission string.
func (p *WebsocketPayload) HasPermission(permission string) bool {
2020-04-06 01:00:33 +00:00
for _, k := range p.Permissions {
if k == permission || (!strings.HasPrefix(permission, "admin") && k == "*") {
2020-04-06 01:00:33 +00:00
return true
}
}
return false
}