git.squeep.com Git - akkoma/blob - lib/pleroma/web/activity_pub/mrf/keyword_policy.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.ActivityPub.MRF.KeywordPolicy do
   6   @behaviour Pleroma.Web.ActivityPub.MRF
   7   defp string_matches?(string, pattern) when is_binary(pattern) do
   8     String.contains?(string, pattern)
   9   end
  10
  11   defp string_matches?(string, pattern) do
  12     String.match?(string, pattern)
  13   end
  14
  15   defp check_reject(%{"object" => %{"content" => content}} = message) do
  16     if Enum.any?(Pleroma.Config.get([:mrf_keyword, :reject]), fn pattern ->
  17          string_matches?(content, pattern)
  18        end) do
  19       {:reject, nil}
  20     else
  21       {:ok, message}
  22     end
  23   end
  24
  25   defp check_ftl_removal(%{"to" => to, "object" => %{"content" => content}} = message) do
  26     if "https://www.w3.org/ns/activitystreams#Public" in to and
  27          Enum.any?(Pleroma.Config.get([:mrf_keyword, :federated_timeline_removal]), fn pattern ->
  28            string_matches?(content, pattern)
  29          end) do
  30       to = List.delete(to, "https://www.w3.org/ns/activitystreams#Public")
  31       cc = ["https://www.w3.org/ns/activitystreams#Public" | message["cc"] || []]
  32
  33       message =
  34         message
  35         |> Map.put("to", to)
  36         |> Map.put("cc", cc)
  37
  38       IO.inspect(message)
  39       {:ok, message}
  40     else
  41       {:ok, message}
  42     end
  43   end
  44
  45   defp check_replace(%{"object" => %{"content" => content}} = message) do
  46     content =
  47       Enum.reduce(Pleroma.Config.get([:mrf_keyword, :replace]), content, fn {pattern, replacement},
  48                                                                             acc ->
  49         String.replace(acc, pattern, replacement)
  50       end)
  51
  52     {:ok, put_in(message["object"]["content"], content)}
  53   end
  54
  55   @impl true
  56   def filter(%{"object" => %{"content" => nil}} = message) do
  57     {:ok, message}
  58   end
  59
  60   @impl true
  61   def filter(%{"type" => "Create", "object" => %{"content" => _content}} = message) do
  62     with {:ok, message} <- check_reject(message),
  63          {:ok, message} <- check_ftl_removal(message),
  64          {:ok, message} <- check_replace(message) do
  65       {:ok, message}
  66     else
  67       _e ->
  68         {:reject, nil}
  69     end
  70   end
  71
  72   @impl true
  73   def filter(message), do: {:ok, message}
  74 end