107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package bot
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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 || cfg == nil {
|
|
b.log.Warn("cannot get %s settings: %v", evt.RoomID, err)
|
|
}
|
|
|
|
if cfg.Mailbox == "" {
|
|
b.Error(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.rooms[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.log.Warn("cannot get settings: %v", err)
|
|
}
|
|
if cfg == nil {
|
|
cfg = &settings{}
|
|
}
|
|
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)
|
|
}
|
|
}
|