policyeval/execute: Add capability to auto-suspend on prefix

nexy7574 · nexy7574 · commit 2dcf1d0f0547 · 2025-08-19T18:36:57.000+01:00
diff --git a/config/event.go b/config/event.go
@@ -13,13 +13,13 @@ var (
 )
 
 type WatchedPolicyList struct {
-	RoomID       id.RoomID `json:"room_id"`
-	Name         string    `json:"name"`
-	Shortcode    string    `json:"shortcode"`
-	DontApply    bool      `json:"dont_apply"`
-	DontApplyACL bool      `json:"dont_apply_acl"`
-	AutoUnban    bool      `json:"auto_unban"`
-	AutoSuspend  bool      `json:"auto_suspend"`
+	RoomID            id.RoomID `json:"room_id"`
+	Name              string    `json:"name"`
+	Shortcode         string    `json:"shortcode"`
+	DontApply         bool      `json:"dont_apply"`
+	DontApplyACL      bool      `json:"dont_apply_acl"`
+	AutoUnban         bool      `json:"auto_unban"`
+	AutoSuspendPrefix *string   `json:"auto_suspend_prefix,omitempty"`
 
 	DontNotifyOnChange bool `json:"dont_notify_on_change"`
 }
diff --git a/policyeval/execute.go b/policyeval/execute.go
@@ -136,7 +136,7 @@ func (pe *PolicyEvaluator) maybeApplySuspend(ctx context.Context, userID id.User
 		return
 	}
 	plist := pe.GetWatchedListMeta(policy.RoomID)
-	if !plist.AutoSuspend {
+	if plist.AutoSuspendPrefix == nil || !strings.HasPrefix(*plist.AutoSuspendPrefix, policy.Reason) {
 		return
 	}
 	err := pe.Bot.SynapseAdmin.SuspendAccount(ctx, userID, synapseadmin.ReqSuspendUser{Suspend: true})

Original file line number	Diff line number	Diff line change
`@@ -136,7 +136,7 @@ func (pe *PolicyEvaluator) maybeApplySuspend(ctx context.Context, userID id.User`
`136`	`136`	`return`
`137`	`137`	`}`
`138`	`138`	`plist := pe.GetWatchedListMeta(policy.RoomID)`
`139`		`- if !plist.AutoSuspend {`
	`139`	`+ if plist.AutoSuspendPrefix == nil \|\| !strings.HasPrefix(*plist.AutoSuspendPrefix, policy.Reason) {`
`140`	`140`	`return`
`141`	`141`	`}`
`142`	`142`	`err := pe.Bot.SynapseAdmin.SuspendAccount(ctx, userID, synapseadmin.ReqSuspendUser{Suspend: true})`