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