package bot import ( "context" "fmt" "github.com/getsentry/sentry-go" "gitlab.com/etke.cc/postmoogle/utils" "maunium.net/go/mautrix/event" "maunium.net/go/mautrix/format" "maunium.net/go/mautrix/id" ) func (b *Bot) syncRooms(ctx context.Context) error { b.roomsmu.Lock() defer b.roomsmu.Unlock() span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("syncRooms")) defer span.Finish() resp, err := b.lp.GetClient().JoinedRooms() if err != nil { return err } b.rooms = make(map[string]id.RoomID, len(resp.JoinedRooms)) for _, roomID := range resp.JoinedRooms { cfg, serr := b.getSettings(span.Context(), roomID) if serr != nil { b.log.Warn("cannot get %s settings: %v", roomID, err) continue } if cfg.Mailbox != "" { b.rooms[cfg.Mailbox] = roomID } } return nil } func (b *Bot) handleMailbox(ctx context.Context, evt *event.Event, command []string) { if len(command) == 1 { b.getMailbox(ctx, evt) return } b.setMailbox(ctx, evt, command[1]) } func (b *Bot) getMailbox(ctx context.Context, evt *event.Event) { span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("getMailbox")) defer span.Finish() cfg, err := b.getSettings(span.Context(), evt.RoomID) if err != nil { b.Error(span.Context(), evt.RoomID, "failed to retrieve setting: %v", err) return } if cfg.Mailbox == "" { b.Notice(span.Context(), evt.RoomID, "mailbox name is not set") return } content := format.RenderMarkdown("Mailbox of this room is **"+cfg.Mailbox+"@"+b.domain+"**", true, true) content.MsgType = event.MsgNotice _, err = b.lp.Send(evt.RoomID, content) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err) } } func (b *Bot) setMailbox(ctx context.Context, evt *event.Event, mailbox string) { span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName("setMailbox")) defer span.Finish() mailbox = utils.Mailbox(mailbox) existingID, ok := b.GetMapping(ctx, mailbox) if ok && existingID != "" && existingID != evt.RoomID { content := format.RenderMarkdown("Mailbox "+mailbox+"@"+b.domain+" already taken", true, true) content.MsgType = event.MsgNotice _, err := b.lp.Send(evt.RoomID, content) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err) } } cfg, err := b.getSettings(span.Context(), evt.RoomID) if err != nil { b.Error(span.Context(), evt.RoomID, "failed to retrieve setting: %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.Owner = evt.Sender cfg.Mailbox = mailbox err = b.setSettings(span.Context(), evt.RoomID, cfg) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot update settings: %v", err) return } b.roomsmu.Lock() b.rooms[mailbox] = evt.RoomID b.roomsmu.Unlock() content := format.RenderMarkdown("Mailbox of this room set to **"+cfg.Mailbox+"@"+b.domain+"**", true, true) content.MsgType = event.MsgNotice _, err = b.lp.Send(evt.RoomID, content) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err) } } func (b *Bot) handleHideSenderAddress(ctx context.Context, evt *event.Event, command []string) { getter := func(entity settings) bool { return entity.HideSenderAddress } setter := func(entity *settings, value bool) error { entity.HideSenderAddress = value return nil } b.handleBooleanConfigurationKey(ctx, evt, command, "hide-sender-address", getter, setter) } func (b *Bot) handleBooleanConfigurationKey( ctx context.Context, evt *event.Event, command []string, configKey string, getter func(entity settings) bool, setter func(entity *settings, value bool) error, ) { if len(command) == 1 { b.getBooleanConfigurationKey(ctx, evt, configKey, getter, setter) return } b.setBooleanConfigurationKey(ctx, evt, configKey, command[1], getter, setter) } func (b *Bot) getBooleanConfigurationKey( ctx context.Context, evt *event.Event, configKey string, getter func(entity settings) bool, setter func(entity *settings, value bool) error, ) { span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName(fmt.Sprintf("getBooleanConfigurationKey.%s", configKey))) defer span.Finish() cfg, err := b.getSettings(span.Context(), evt.RoomID) if err != nil { b.log.Warn("cannot get %s settings: %v", evt.RoomID, err) return } value := getter(cfg) content := format.RenderMarkdown(fmt.Sprintf("`%s` configuration setting for this room is currently set to `%v`", configKey, value), true, true) content.MsgType = event.MsgNotice _, err = b.lp.Send(evt.RoomID, content) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err) } } func (b *Bot) setBooleanConfigurationKey( ctx context.Context, evt *event.Event, configKey string, value string, getter func(entity settings) bool, setter func(entity *settings, value bool) error, ) { var actualValue bool if value == "true" { actualValue = true } else if value == "false" { actualValue = false } else { b.Notice(ctx, evt.RoomID, "you are supposed to send a true or false value") return } span := sentry.StartSpan(ctx, "http.server", sentry.TransactionName(fmt.Sprintf("setBooleanConfigurationKey.%s", configKey))) defer span.Finish() cfg, err := b.getSettings(span.Context(), evt.RoomID) if err != nil { b.Error(span.Context(), evt.RoomID, "failed to retrieve setting: %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 } setter(&cfg, actualValue) err = b.setSettings(span.Context(), evt.RoomID, cfg) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot update settings: %v", err) return } content := format.RenderMarkdown(fmt.Sprintf("`%s` configuration setting for this room has been set to `%v`", configKey, actualValue), true, true) content.MsgType = event.MsgNotice _, err = b.lp.Send(evt.RoomID, content) if err != nil { b.Error(span.Context(), evt.RoomID, "cannot send message: %v", err) } }