From c971287624913c8555fe288af0df1175701e6209 Mon Sep 17 00:00:00 2001
From: Thomas Citharel <tcit@tcit.fr>
Date: Thu, 1 Jun 2023 14:49:17 +0200
Subject: [PATCH] feat(anti-spam): allow to only scan for spam in profiles or
 events

Signed-off-by: Thomas Citharel <tcit@tcit.fr>
---
 .../mobilizon/maintenance/detect_spam.ex      | 46 +++++++++++--------
 1 file changed, 28 insertions(+), 18 deletions(-)

diff --git a/lib/mix/tasks/mobilizon/maintenance/detect_spam.ex b/lib/mix/tasks/mobilizon/maintenance/detect_spam.ex
index 1e3ee5c1d..1af1902aa 100644
--- a/lib/mix/tasks/mobilizon/maintenance/detect_spam.ex
+++ b/lib/mix/tasks/mobilizon/maintenance/detect_spam.ex
@@ -23,13 +23,17 @@ defmodule Mix.Tasks.Mobilizon.Maintenance.DetectSpam do
           dry_run: :boolean,
           verbose: :boolean,
           forward_reports: :boolean,
-          local_only: :boolean
+          local_only: :boolean,
+          only_profiles: :boolean,
+          only_events: :boolean
         ],
         aliases: [
           d: :dry_run,
           v: :verbose,
           f: :forward_reports,
-          l: :local_only
+          l: :local_only,
+          p: :only_profiles,
+          e: :only_events
         ]
       )
 
@@ -41,23 +45,27 @@ defmodule Mix.Tasks.Mobilizon.Maintenance.DetectSpam do
 
     anonymous_actor_id = Config.anonymous_actor_id()
 
-    options
-    |> Keyword.get(:local_only, false)
-    |> profiles()
-    |> Stream.flat_map(& &1)
-    |> Stream.each(fn profile ->
-      process_profile(profile, Keyword.put(options, :anonymous_actor_id, anonymous_actor_id))
-    end)
-    |> Stream.run()
+    unless only_events?(options) do
+      options
+      |> Keyword.get(:local_only, false)
+      |> profiles()
+      |> Stream.flat_map(& &1)
+      |> Stream.each(fn profile ->
+        process_profile(profile, Keyword.put(options, :anonymous_actor_id, anonymous_actor_id))
+      end)
+      |> Stream.run()
+    end
 
-    options
-    |> Keyword.get(:local_only, false)
-    |> events()
-    |> Stream.flat_map(& &1)
-    |> Stream.each(fn event ->
-      process_event(event, Keyword.put(options, :anonymous_actor_id, anonymous_actor_id))
-    end)
-    |> Stream.run()
+    unless only_profiles?(options) do
+      options
+      |> Keyword.get(:local_only, false)
+      |> events()
+      |> Stream.flat_map(& &1)
+      |> Stream.each(fn event ->
+        process_event(event, Keyword.put(options, :anonymous_actor_id, anonymous_actor_id))
+      end)
+      |> Stream.run()
+    end
   end
 
   defp profiles(local_only) do
@@ -180,6 +188,8 @@ defmodule Mix.Tasks.Mobilizon.Maintenance.DetectSpam do
 
   defp verbose?(options), do: Keyword.get(options, :verbose, false)
   defp dry_run?(options), do: Keyword.get(options, :dry_run, false)
+  defp only_profiles?(options), do: Keyword.get(options, :only_profiles, false)
+  defp only_events?(options), do: Keyword.get(options, :only_events, false)
 
   defp anti_spam, do: AntiSpam.service()
 end