git.squeep.com Git - akkoma/blob - lib/pleroma/web/activity_pub/object_validators/common_fixes.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.ActivityPub.ObjectValidators.CommonFixes do
   6   alias Pleroma.EctoType.ActivityPub.ObjectValidators
   7   alias Pleroma.Object.Containment
   8   alias Pleroma.User
   9   alias Pleroma.Web.ActivityPub.Transmogrifier
  10   alias Pleroma.Web.ActivityPub.Utils
  11
  12   def cast_recipients(message, field, field_fallback \\ []) do
  13     {:ok, data} = ObjectValidators.Recipients.cast(message[field] || field_fallback)
  14
  15     Map.put(message, field, data)
  16   end
  17
  18   def fix_object_defaults(data) do
  19     %{data: %{"id" => context}, id: context_id} =
  20       Utils.create_context(data["context"] || data["conversation"])
  21
  22     %User{follower_address: follower_collection} = User.get_cached_by_ap_id(data["attributedTo"])
  23
  24     data
  25     |> Map.put("context", context)
  26     |> Map.put("context_id", context_id)
  27     |> cast_recipients("to")
  28     |> cast_recipients("cc")
  29     |> cast_recipients("bto")
  30     |> cast_recipients("bcc")
  31     |> Transmogrifier.fix_explicit_addressing(follower_collection)
  32     |> Transmogrifier.fix_implicit_addressing(follower_collection)
  33   end
  34
  35   def fix_activity_addressing(activity, _meta) do
  36     %User{follower_address: follower_collection} = User.get_cached_by_ap_id(activity["actor"])
  37
  38     activity
  39     |> cast_recipients("to")
  40     |> cast_recipients("cc")
  41     |> cast_recipients("bto")
  42     |> cast_recipients("bcc")
  43     |> Transmogrifier.fix_explicit_addressing(follower_collection)
  44     |> Transmogrifier.fix_implicit_addressing(follower_collection)
  45   end
  46
  47   def fix_actor(data) do
  48     actor =
  49       data
  50       |> Map.put_new("actor", data["attributedTo"])
  51       |> Containment.get_actor()
  52
  53     data
  54     |> Map.put("actor", actor)
  55     |> Map.put("attributedTo", actor)
  56   end
  57 end