feat: Adding support to multiple ids

Signed-off-by: Vincent Boutour <[email protected]>

cmd/sweeper/config.go

@@ -13,8 +13,8 @@ type configuration struct {
 	logger  *logger.Config
 	discord *discord.Config
 
-	currentUser *string
-	usernames   *[]string
+	userIDs   *[]string
+	usernames *[]string
 }
 
 func newConfiguration() configuration {
@@ -25,8 +25,8 @@ func newConfiguration() configuration {
 		logger:  logger.Flags(fs, "logger"),
 		discord: discord.Flags(fs, ""),
 
-		currentUser: flags.New("currentUser", "Current User ID to clean").DocPrefix("sweeper").String(fs, "", nil),
-		usernames:   flags.New("username", "Username of messages to clean").DocPrefix("sweeper").StringSlice(fs, nil, nil),
+		userIDs:   flags.New("user", "User ID to clean").DocPrefix("sweeper").StringSlice(fs, nil, nil),
+		usernames: flags.New("username", "Username to clean").DocPrefix("sweeper").StringSlice(fs, nil, nil),
 	}
 
 	_ = fs.Parse(os.Args[1:])

cmd/sweeper/main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"context"
-	"fmt"
 	"log/slog"
 	"runtime"
 	"strings"
@@ -62,16 +61,17 @@ func main() {
 			continue
 		}
 
-		if shouldDelete(*config.currentUser, message, *config.usernames) {
-			fmt.Println(message.String())
+		if shouldDelete(message, *config.userIDs, *config.usernames) {
 			logger.FatalfOnErr(ctx, services.discord.DeleteMessage(ctx, req, message), "delete")
 		}
 	}
 }
 
-func shouldDelete(currentUser string, message discord.Message, usernames []string) bool {
-	if message.Author.Bot && strings.Contains(message.Content, currentUser) {
-		return true
+func shouldDelete(message discord.Message, userIDs, usernames []string) bool {
+	for _, userID := range userIDs {
+		if message.Author.Bot && strings.Contains(message.Content, userID) {
+			return true
+		}
 	}
 
 	for _, username := range usernames {