package bot import ( "context" "fmt" "regexp" "sync" "github.com/getsentry/sentry-go" "gitlab.com/etke.cc/go/logger" "gitlab.com/etke.cc/linkpearl" "maunium.net/go/mautrix/event" "maunium.net/go/mautrix/format" "maunium.net/go/mautrix/id" ) // Bot represents matrix bot type Bot struct { prefix string domains []string allowedUsers []*regexp.Regexp allowedAdmins []*regexp.Regexp reservedMailboxes []string commands commandList banlist bglist rooms sync.Map sendmail func(string, string, string) error log *logger.Logger lp *linkpearl.Linkpearl mu map[string]*sync.Mutex handledMembershipEvents sync.Map } // New creates a new matrix bot func New( lp *linkpearl.Linkpearl, log *logger.Logger, prefix string, domains []string, reserved []string, admins []string, ) (*Bot, error) { b := &Bot{ reservedMailboxes: reserved, domains: domains, prefix: prefix, rooms: sync.Map{}, log: log, lp: lp, mu: map[string]*sync.Mutex{}, } users, err := b.initBotUsers() if err != nil { return nil, err } allowedUsers, uerr := parseMXIDpatterns(users, "") if uerr != nil { return nil, uerr } b.allowedUsers = allowedUsers allowedAdmins, aerr := parseMXIDpatterns(admins, "") if aerr != nil { return nil, aerr } b.allowedAdmins = allowedAdmins b.commands = b.initCommands() return b, nil } // Error message to the log and matrix room func (b *Bot) Error(ctx context.Context, roomID id.RoomID, message string, args ...interface{}) { b.log.Error(message, args...) err := fmt.Errorf(message, args...) if hub := sentry.GetHubFromContext(ctx); hub != nil { sentry.GetHubFromContext(ctx).CaptureException(err) } if roomID != "" { b.SendError(ctx, roomID, err.Error()) } } // SendError sends an error message to the matrix room func (b *Bot) SendError(ctx context.Context, roomID id.RoomID, message string) { b.SendNotice(ctx, roomID, "ERROR: "+message) } // SendNotice sends a notice message to the matrix room func (b *Bot) SendNotice(ctx context.Context, roomID id.RoomID, message string) { parsed := format.RenderMarkdown(message, true, true) parsed.MsgType = event.MsgNotice _, err := b.lp.Send(roomID, &event.Content{Parsed: &parsed}) if err != nil { sentry.GetHubFromContext(ctx).CaptureException(err) } } // Start performs matrix /sync func (b *Bot) Start(statusMsg string) error { if err := b.migrate(); err != nil { return err } if err := b.syncRooms(); err != nil { return err } b.syncBanlist() b.initSync() b.log.Info("Postmoogle has been started") return b.lp.Start(statusMsg) } // Stop the bot func (b *Bot) Stop() { err := b.lp.GetClient().SetPresence(event.PresenceOffline) if err != nil { b.log.Error("cannot set presence = offline: %v", err) } b.lp.GetClient().StopSync() }