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

   1 defmodule Pleroma.Web.ActivityPub.MRF.SimplePolicy do
   2   alias Pleroma.User
   3   @behaviour Pleroma.Web.ActivityPub.MRF
   4
   5   defp check_accept(%{host: actor_host} = _actor_info, object) do
   6     accepts = Pleroma.Config.get([:mrf_simple, :accept])
   7
   8     cond do
   9       accepts == [] -> {:ok, object}
  10       Enum.member?(accepts, actor_host) -> {:ok, object}
  11       true -> {:reject, nil}
  12     end
  13   end
  14
  15   defp check_reject(%{host: actor_host} = _actor_info, object) do
  16     if Enum.member?(Pleroma.Config.get([:mrf_simple, :reject]), actor_host) do
  17       {:reject, nil}
  18     else
  19       {:ok, object}
  20     end
  21   end
  22
  23   defp check_media_removal(
  24          %{host: actor_host} = _actor_info,
  25          %{"type" => "Create", "object" => %{"attachement" => child_attachement}} = object
  26        )
  27        when length(child_attachement) > 0 do
  28     object =
  29       if Enum.member?(Pleroma.Config.get([:mrf_simple, :media_removal]), actor_host) do
  30         child_object = Map.delete(object["object"], "attachment")
  31         Map.put(object, "object", child_object)
  32       else
  33         object
  34       end
  35
  36     {:ok, object}
  37   end
  38
  39   defp check_media_removal(_actor_info, object), do: {:ok, object}
  40
  41   defp check_media_nsfw(
  42          %{host: actor_host} = _actor_info,
  43          %{
  44            "type" => "Create",
  45            "object" => %{"attachment" => child_attachment} = child_object
  46          } = object
  47        )
  48        when length(child_attachment) > 0 do
  49     object =
  50       if Enum.member?(Pleroma.Config.get([:mrf_simple, :media_nsfw]), actor_host) do
  51         tags = (child_object["tag"] || []) ++ ["nsfw"]
  52         child_object = Map.put(child_object, "tags", tags)
  53         child_object = Map.put(child_object, "sensitive", true)
  54         Map.put(object, "object", child_object)
  55       else
  56         object
  57       end
  58
  59     {:ok, object}
  60   end
  61
  62   defp check_media_nsfw(_actor_info, object), do: {:ok, object}
  63
  64   defp check_ftl_removal(%{host: actor_host} = _actor_info, object) do
  65     object =
  66       with true <-
  67              Enum.member?(
  68                Pleroma.Config.get([:mrf_simple, :federated_timeline_removal]),
  69                actor_host
  70              ),
  71            user <- User.get_by_ap_id(object["actor"]),
  72            true <- "https://www.w3.org/ns/activitystreams#Public" in object["to"],
  73            true <- user.follower_address in object["cc"] do
  74         to =
  75           List.delete(object["to"], "https://www.w3.org/ns/activitystreams#Public") ++
  76             [user.follower_address]
  77
  78         cc =
  79           List.delete(object["cc"], user.follower_address) ++
  80             ["https://www.w3.org/ns/activitystreams#Public"]
  81
  82         object
  83         |> Map.put("to", to)
  84         |> Map.put("cc", cc)
  85       else
  86         _ -> object
  87       end
  88
  89     {:ok, object}
  90   end
  91
  92   @impl true
  93   def filter(object) do
  94     actor_info = URI.parse(object["actor"])
  95
  96     with {:ok, object} <- check_accept(actor_info, object),
  97          {:ok, object} <- check_reject(actor_info, object),
  98          {:ok, object} <- check_media_removal(actor_info, object),
  99          {:ok, object} <- check_media_nsfw(actor_info, object),
 100          {:ok, object} <- check_ftl_removal(actor_info, object) do
 101       {:ok, object}
 102     else
 103       _e -> {:reject, nil}
 104     end
 105   end
 106 end