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