97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package bot
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/getsentry/sentry-go"
|
|
"gitlab.com/etke.cc/postmoogle/utils"
|
|
"maunium.net/go/mautrix/event"
|
|
)
|
|
|
|
type sanitizerFunc func(string) string
|
|
|
|
// sanitizers is map of option name => sanitizer function
|
|
var sanitizers = map[string]sanitizerFunc{
|
|
"mailbox": utils.Mailbox,
|
|
"nosender": utils.SanitizeBoolString,
|
|
}
|
|
|
|
func (b *Bot) handleOption(ctx context.Context, evt *event.Event, command []string) {
|
|
if len(command) == 1 {
|
|
b.getOption(ctx, evt, command[0])
|
|
return
|
|
}
|
|
b.setOption(ctx, evt, command[0], command[1])
|
|
}
|
|
|
|
func (b *Bot) getOption(ctx context.Context, evt *event.Event, name string) {
|
|
msg := "`%s` of this room is %s"
|
|
span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("getOption"))
|
|
defer span.Finish()
|
|
|
|
cfg, err := b.getSettings(span.Context(), evt.RoomID)
|
|
if err != nil {
|
|
b.Error(span.Context(), evt.RoomID, "failed to retrieve settings: %v", err)
|
|
return
|
|
}
|
|
|
|
value := cfg.Get(name)
|
|
if value == "" {
|
|
b.Notice(span.Context(), evt.RoomID, "`%s` is not set", name)
|
|
return
|
|
}
|
|
|
|
if name == "mailbox" {
|
|
msg = msg + "@" + b.domain
|
|
}
|
|
|
|
b.Notice(span.Context(), evt.RoomID, msg, name, value)
|
|
}
|
|
|
|
func (b *Bot) setOption(ctx context.Context, evt *event.Event, name, value string) {
|
|
span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("setOption"))
|
|
defer span.Finish()
|
|
msg := "`%s` of this room set to %s"
|
|
|
|
sanitizer, ok := sanitizers[name]
|
|
if ok {
|
|
value = sanitizer(value)
|
|
}
|
|
|
|
if name == "mailbox" {
|
|
existingID, ok := b.GetMapping(ctx, value)
|
|
if ok && existingID != "" && existingID != evt.RoomID {
|
|
b.Notice(span.Context(), evt.RoomID, "Mailbox %s@%s already taken", value, b.domain)
|
|
return
|
|
}
|
|
}
|
|
|
|
cfg, err := b.getSettings(span.Context(), evt.RoomID)
|
|
if err != nil {
|
|
b.Error(span.Context(), evt.RoomID, "failed to retrieve settings: %v", err)
|
|
return
|
|
}
|
|
|
|
if !cfg.Allowed(b.noowner, evt.Sender) {
|
|
b.Notice(span.Context(), evt.RoomID, "you don't have permission to do that")
|
|
return
|
|
}
|
|
|
|
cfg.Set(name, value)
|
|
if name == "mailbox" {
|
|
msg = msg + "@" + b.domain
|
|
cfg.Set("owner", evt.Sender.String())
|
|
b.roomsmu.Lock()
|
|
b.rooms[value] = evt.RoomID
|
|
b.roomsmu.Unlock()
|
|
}
|
|
|
|
err = b.setSettings(span.Context(), evt.RoomID, cfg)
|
|
if err != nil {
|
|
b.Error(span.Context(), evt.RoomID, "cannot update settings: %v", err)
|
|
return
|
|
}
|
|
|
|
b.Notice(span.Context(), evt.RoomID, msg, name, value)
|
|
}
|