heimoshuiyu/postmoogle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

automatically remove mailboxes in empty rooms, fixes #3

Browse Source
This commit is contained in:
Aine
2022-08-26 16:22:44 +03:00
parent 2dcba843cc
commit 2da1aacc7a
3 changed files with 37 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bot/bot.go
View File

@@ -26,7 +26,7 @@ type Bot struct {
rooms sync.Map rooms sync.Map
log *logger.Logger log *logger.Logger
lp *linkpearl.Linkpearl lp *linkpearl.Linkpearl
handledJoinEvents sync.Map handledMembershipEvents sync.Map
} }
// New creates a new matrix bot // New creates a new matrix bot

6
bot/command.go
View File

@@ -112,7 +112,7 @@ func (b *Bot) handleCommand(ctx context.Context, evt *event.Event, command []str
case "help": case "help":
b.sendHelp(ctx, evt.RoomID) b.sendHelp(ctx, evt.RoomID)
case "stop": case "stop":
b.runStop(ctx) b.runStop(ctx, true)
default: default:
b.handleOption(ctx, command) b.handleOption(ctx, command)
} }
@@ -159,7 +159,7 @@ func (b *Bot) sendHelp(ctx context.Context, roomID id.RoomID) {
b.Notice(ctx, roomID, msg.String()) b.Notice(ctx, roomID, msg.String())
} }
func (b *Bot) runStop(ctx context.Context) { func (b *Bot) runStop(ctx context.Context, checkAllowed bool) {
evt := eventFromContext(ctx) evt := eventFromContext(ctx)
cfg, err := b.getSettings(evt.RoomID) cfg, err := b.getSettings(evt.RoomID)
if err != nil { if err != nil {
@@ -167,7 +167,7 @@ func (b *Bot) runStop(ctx context.Context) {
return return
} }
if !cfg.Allowed(b.noowner, evt.Sender) { if checkAllowed && !cfg.Allowed(b.noowner, evt.Sender) {
b.Notice(ctx, evt.RoomID, "you don't have permission to do that") b.Notice(ctx, evt.RoomID, "you don't have permission to do that")
return return
} }

28
bot/sync.go
View File

@@ -29,11 +29,16 @@ func (b *Bot) initSync() {
func (b *Bot) onMembership(evt *event.Event) { func (b *Bot) onMembership(evt *event.Event) {
ctx := newContext(evt) ctx := newContext(evt)
if evt.Content.AsMember().Membership == event.MembershipJoin && evt.Sender == b.lp.GetClient().UserID { evtType := evt.Content.AsMember().Membership
if evtType == event.MembershipJoin && evt.Sender == b.lp.GetClient().UserID {
b.onBotJoin(ctx) b.onBotJoin(ctx)
return return
} }
if evtType == event.MembershipBan || evtType == event.MembershipLeave && evt.Sender != b.lp.GetClient().UserID {
b.onLeave(ctx)
}
// Potentially handle other membership events in the future // Potentially handle other membership events in the future
} }
@@ -68,7 +73,7 @@ func (b *Bot) onBotJoin(ctx context.Context) {
evt := eventFromContext(ctx) evt := eventFromContext(ctx)
// Workaround for membership=join events which are delivered to us twice, // Workaround for membership=join events which are delivered to us twice,
// as described in this bug report: https://github.com/matrix-org/synapse/issues/9768 // as described in this bug report: https://github.com/matrix-org/synapse/issues/9768
_, ok := b.handledJoinEvents.LoadOrStore(evt.ID, true) _, ok := b.handledMembershipEvents.LoadOrStore(evt.ID, true)
if ok { if ok {
b.log.Info("Suppressing already handled event %s", evt.ID) b.log.Info("Suppressing already handled event %s", evt.ID)
return return
@@ -77,3 +82,22 @@ func (b *Bot) onBotJoin(ctx context.Context) {
b.sendIntroduction(ctx, evt.RoomID) b.sendIntroduction(ctx, evt.RoomID)
b.sendHelp(ctx, evt.RoomID) b.sendHelp(ctx, evt.RoomID)
} }
func (b *Bot) onLeave(ctx context.Context) {
evt := eventFromContext(ctx)
_, ok := b.handledMembershipEvents.LoadOrStore(evt.ID, true)
if ok {
b.log.Info("Suppressing already handled event %s", evt.ID)
return
}
members := b.lp.GetStore().GetRoomMembers(evt.RoomID)
count := len(members)
if count == 1 && members[0] == b.lp.GetClient().UserID {
b.log.Info("no more users left in the %s room", evt.RoomID)
b.runStop(ctx, false)
_, err := b.lp.GetClient().LeaveRoom(evt.RoomID)
if err != nil {
b.Error(ctx, evt.RoomID, "cannot leave empty room: %v", err)
}
}
}