mirror of
https://mau.dev/mautrix/go.git
synced 2026-03-15 14:55:51 +01:00
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
// Copyright (c) 2024 Tulir Asokan
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"maunium.net/go/mautrix/event"
|
|
)
|
|
|
|
type MinimalCommandHandler interface {
|
|
Run(*Event)
|
|
}
|
|
|
|
type MinimalCommandHandlerFunc func(*Event)
|
|
|
|
func (mhf MinimalCommandHandlerFunc) Run(ce *Event) {
|
|
mhf(ce)
|
|
}
|
|
|
|
type CommandState struct {
|
|
Next MinimalCommandHandler
|
|
Action string
|
|
Meta any
|
|
Cancel func()
|
|
}
|
|
|
|
type CommandHandler interface {
|
|
MinimalCommandHandler
|
|
GetName() string
|
|
}
|
|
|
|
type AliasedCommandHandler interface {
|
|
CommandHandler
|
|
GetAliases() []string
|
|
}
|
|
|
|
type FullHandler struct {
|
|
Func func(*Event)
|
|
|
|
Name string
|
|
Aliases []string
|
|
Help HelpMeta
|
|
|
|
RequiresAdmin bool
|
|
RequiresPortal bool
|
|
RequiresLogin bool
|
|
RequiresEventLevel event.Type
|
|
RequiresLoginPermission bool
|
|
}
|
|
|
|
func (fh *FullHandler) GetHelp() HelpMeta {
|
|
fh.Help.Command = fh.Name
|
|
return fh.Help
|
|
}
|
|
|
|
func (fh *FullHandler) GetName() string {
|
|
return fh.Name
|
|
}
|
|
|
|
func (fh *FullHandler) GetAliases() []string {
|
|
return fh.Aliases
|
|
}
|
|
|
|
func (fh *FullHandler) ShowInHelp(ce *Event) bool {
|
|
return true
|
|
//return !fh.RequiresAdmin || ce.User.GetPermissionLevel() >= bridgeconfig.PermissionLevelAdmin
|
|
}
|
|
|
|
func (fh *FullHandler) userHasRoomPermission(ce *Event) bool {
|
|
levels, err := ce.Bridge.Matrix.GetPowerLevels(ce.Ctx, ce.RoomID)
|
|
if err != nil {
|
|
ce.Log.Warn().Err(err).Msg("Failed to check room power levels")
|
|
ce.Reply("Failed to get room power levels to see if you're allowed to use that command")
|
|
return false
|
|
}
|
|
return levels.GetUserLevel(ce.User.MXID) >= levels.GetEventLevel(fh.RequiresEventLevel)
|
|
}
|
|
|
|
func (fh *FullHandler) Run(ce *Event) {
|
|
if fh.RequiresAdmin && !ce.User.Permissions.Admin {
|
|
ce.Reply("That command is limited to bridge administrators.")
|
|
} else if fh.RequiresLoginPermission && !ce.User.Permissions.Login {
|
|
ce.Reply("You do not have permissions to log into this bridge.")
|
|
} else if fh.RequiresEventLevel.Type != "" && !ce.User.Permissions.Admin && !fh.userHasRoomPermission(ce) {
|
|
ce.Reply("That command requires room admin rights.")
|
|
} else if fh.RequiresPortal && ce.Portal == nil {
|
|
ce.Reply("That command can only be ran in portal rooms.")
|
|
} else if fh.RequiresLogin && ce.User.GetDefaultLogin() == nil {
|
|
ce.Reply("That command requires you to be logged in.")
|
|
} else {
|
|
fh.Func(ce)
|
|
}
|
|
}
|