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
   4   @mrf_policy Application.get_env(:pleroma, :mrf_simple)
   5
   6   @reject Keyword.get(@mrf_policy, :reject)
   7   defp check_reject(actor_info, object) do
   8     if actor_info.host in @reject do
   9       {:reject, nil}
  10     else
  11       {:ok, object}
  12     end
  13   end
  14
  15   @media_removal Keyword.get(@mrf_policy, :media_removal)
  16   defp check_media_removal(actor_info, object) do
  17     if actor_info.host in @media_removal do
  18       child_object = Map.delete(object["object"], "attachment")
  19       object = Map.put(object, "object", child_object)
  20       {:ok, object}
  21     else
  22       {:ok, object}
  23     end
  24   end
  25
  26   @media_nsfw Keyword.get(@mrf_policy, :media_nsfw)
  27   defp check_media_nsfw(actor_info, object) do
  28     child_object = object["object"]
  29
  30     if actor_info.host in @media_nsfw and child_object["attachment"] != nil and
  31          length(child_object["attachment"]) > 0 do
  32       tags = (child_object["tag"] || []) ++ ["nsfw"]
  33       child_object = Map.put(child_object, "tags", tags)
  34       child_object = Map.put(child_object, "sensitive", true)
  35       object = Map.put(object, "object", child_object)
  36       {:ok, object}
  37     else
  38       {:ok, object}
  39     end
  40   end
  41
  42   @ftl_removal Keyword.get(@mrf_policy, :federated_timeline_removal)
  43   defp check_ftl_removal(actor_info, object) do
  44     if actor_info.host in @ftl_removal do
  45       user = User.get_by_ap_id(object["actor"])
  46
  47       # flip to/cc relationship to make the post unlisted
  48       object =
  49         if "https://www.w3.org/ns/activitystreams#Public" in object["to"] and
  50              user.follower_address in object["cc"] do
  51           to =
  52             List.delete(object["to"], "https://www.w3.org/ns/activitystreams#Public") ++
  53               [user.follower_address]
  54
  55           cc =
  56             List.delete(object["cc"], user.follower_address) ++
  57               ["https://www.w3.org/ns/activitystreams#Public"]
  58
  59           object
  60           |> Map.put("to", to)
  61           |> Map.put("cc", cc)
  62         else
  63           object
  64         end
  65
  66       {:ok, object}
  67     else
  68       {:ok, object}
  69     end
  70   end
  71
  72   def filter(object) do
  73     actor_info = URI.parse(object["actor"])
  74
  75     with {:ok, object} <- check_reject(actor_info, object),
  76          {:ok, object} <- check_media_removal(actor_info, object),
  77          {:ok, object} <- check_media_nsfw(actor_info, object),
  78          {:ok, object} <- check_ftl_removal(actor_info, object) do
  79       {:ok, object}
  80     else
  81       _e -> {:reject, nil}
  82     end
  83   end
  84 end