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

   1 defmodule Pleroma.Web.ActivityPub.Transmogrifier do
   2   @moduledoc """
   3   A module to handle coding from internal to wire ActivityPub and back.
   4   """
   5   alias Pleroma.User
   6   alias Pleroma.Object
   7   alias Pleroma.Activity
   8   alias Pleroma.Repo
   9   alias Pleroma.Web.ActivityPub.ActivityPub
  10   alias Pleroma.Web.ActivityPub.Utils
  11
  12   import Ecto.Query
  13
  14   require Logger
  15
  16   def get_actor(%{"actor" => actor}) when is_binary(actor) do
  17     actor
  18   end
  19
  20   def get_actor(%{"actor" => actor}) when is_list(actor) do
  21     if is_binary(Enum.at(actor, 0)) do
  22       Enum.at(actor, 0)
  23     else
  24       Enum.find(actor, fn %{"type" => type} -> type in ["Person", "Service", "Application"] end)
  25       |> Map.get("id")
  26     end
  27   end
  28
  29   def get_actor(%{"actor" => %{"id" => id}}) when is_bitstring(id) do
  30     id
  31   end
  32
  33   @doc """
  34   Checks that an imported AP object's actor matches the domain it came from.
  35   """
  36   def contain_origin(id, %{"actor" => actor} = params) do
  37     id_uri = URI.parse(id)
  38     actor_uri = URI.parse(get_actor(params))
  39
  40     if id_uri.host == actor_uri.host do
  41       :ok
  42     else
  43       :error
  44     end
  45   end
  46
  47   @doc """
  48   Modifies an incoming AP object (mastodon format) to our internal format.
  49   """
  50   def fix_object(object) do
  51     object
  52     |> fix_actor
  53     |> fix_attachments
  54     |> fix_context
  55     |> fix_in_reply_to
  56     |> fix_emoji
  57     |> fix_tag
  58     |> fix_content_map
  59     |> fix_likes
  60     |> fix_addressing
  61   end
  62
  63   def fix_addressing_list(map, field) do
  64     if is_binary(map[field]) do
  65       map
  66       |> Map.put(field, [map[field]])
  67     else
  68       map
  69     end
  70   end
  71
  72   def fix_addressing(map) do
  73     map
  74     |> fix_addressing_list("to")
  75     |> fix_addressing_list("cc")
  76     |> fix_addressing_list("bto")
  77     |> fix_addressing_list("bcc")
  78   end
  79
  80   def fix_actor(%{"attributedTo" => actor} = object) do
  81     object
  82     |> Map.put("actor", get_actor(%{"actor" => actor}))
  83   end
  84
  85   def fix_likes(%{"likes" => likes} = object)
  86       when is_bitstring(likes) do
  87     # Check for standardisation
  88     # This is what Peertube does
  89     # curl -H 'Accept: application/activity+json' $likes | jq .totalItems
  90     object
  91     |> Map.put("likes", [])
  92     |> Map.put("like_count", 0)
  93   end
  94
  95   def fix_likes(object) do
  96     object
  97   end
  98
  99   def fix_in_reply_to(%{"inReplyTo" => in_reply_to} = object)
 100       when not is_nil(in_reply_to) do
 101     in_reply_to_id =
 102       cond do
 103         is_bitstring(in_reply_to) -> in_reply_to
 104         is_map(in_reply_to) && is_bitstring(in_reply_to["id"]) -> in_reply_to["id"]
 105         is_list(in_reply_to) && is_bitstring(Enum.at(in_reply_to, 0)) -> Enum.at(in_reply_to, 0)
 106         # Maybe I should output an error too?
 107         true -> ""
 108       end
 109
 110     case ActivityPub.fetch_object_from_id(in_reply_to_id) do
 111       {:ok, replied_object} ->
 112         with %Activity{} = activity <-
 113                Activity.get_create_activity_by_object_ap_id(replied_object.data["id"]) do
 114           object
 115           |> Map.put("inReplyTo", replied_object.data["id"])
 116           |> Map.put("inReplyToAtomUri", object["inReplyToAtomUri"] || in_reply_to_id)
 117           |> Map.put("inReplyToStatusId", activity.id)
 118           |> Map.put("conversation", replied_object.data["context"] || object["conversation"])
 119           |> Map.put("context", replied_object.data["context"] || object["conversation"])
 120         else
 121           e ->
 122             Logger.error("Couldn't fetch #{object["inReplyTo"]} #{inspect(e)}")
 123             object
 124         end
 125
 126       e ->
 127         Logger.error("Couldn't fetch #{object["inReplyTo"]} #{inspect(e)}")
 128         object
 129     end
 130   end
 131
 132   def fix_in_reply_to(object), do: object
 133
 134   def fix_context(object) do
 135     context = object["context"] || object["conversation"] || Utils.generate_context_id()
 136
 137     object
 138     |> Map.put("context", context)
 139     |> Map.put("conversation", context)
 140   end
 141
 142   def fix_attachments(%{"attachment" => attachment} = object) when is_list(attachment) do
 143     attachments =
 144       attachment
 145       |> Enum.map(fn data ->
 146         url = [%{"type" => "Link", "mediaType" => data["mediaType"], "href" => data["url"]}]
 147         Map.put(data, "url", url)
 148       end)
 149
 150     object
 151     |> Map.put("attachment", attachments)
 152   end
 153
 154   def fix_attachments(%{"attachment" => attachment} = object) when is_map(attachment) do
 155     attachment =
 156       Map.put(attachment, "url", [
 157         %{"type" => "Link", "mediaType" => attachment["mediaType"], "href" => attachment["url"]}
 158       ])
 159
 160     Map.put(object, "attachment", attachment)
 161   end
 162
 163   def fix_attachments(object) do
 164     object
 165   end
 166
 167   def fix_emoji(%{"tag" => tags} = object) when is_list(tags) do
 168     emoji = tags |> Enum.filter(fn data -> data["type"] == "Emoji" and data["icon"] end)
 169
 170     emoji =
 171       emoji
 172       |> Enum.reduce(%{}, fn data, mapping ->
 173         name = String.trim(data["name"], ":")
 174
 175         mapping |> Map.put(name, data["icon"]["url"])
 176       end)
 177
 178     # we merge mastodon and pleroma emoji into a single mapping, to allow for both wire formats
 179     emoji = Map.merge(object["emoji"] || %{}, emoji)
 180
 181     object
 182     |> Map.put("emoji", emoji)
 183   end
 184
 185   def fix_emoji(%{"tag" => %{"type" => "Emoji"} = tag} = object) do
 186     name = String.trim(tag["name"], ":")
 187     emoji = %{name => tag["icon"]["url"]}
 188
 189     object
 190     |> Map.put("emoji", emoji)
 191   end
 192
 193   def fix_emoji(object) do
 194     object
 195   end
 196
 197   def fix_tag(%{"tag" => tag} = object) when is_list(tag) do
 198     tags =
 199       tag
 200       |> Enum.filter(fn data -> data["type"] == "Hashtag" and data["name"] end)
 201       |> Enum.map(fn data -> String.slice(data["name"], 1..-1) end)
 202
 203     combined = tag ++ tags
 204
 205     object
 206     |> Map.put("tag", combined)
 207   end
 208
 209   def fix_tag(%{"tag" => %{"type" => "Hashtag", "name" => hashtag} = tag} = object) do
 210     combined = [tag, String.slice(hashtag, 1..-1)]
 211
 212     object
 213     |> Map.put("tag", combined)
 214   end
 215
 216   def fix_tag(object) do
 217     object
 218   end
 219
 220   # content map usually only has one language so this will do for now.
 221   def fix_content_map(%{"contentMap" => content_map} = object) do
 222     content_groups = Map.to_list(content_map)
 223     {_, content} = Enum.at(content_groups, 0)
 224
 225     object
 226     |> Map.put("content", content)
 227   end
 228
 229   def fix_content_map(object), do: object
 230
 231   # disallow objects with bogus IDs
 232   def handle_incoming(%{"id" => nil}), do: :error
 233   def handle_incoming(%{"id" => ""}), do: :error
 234   # length of https:// = 8, should validate better, but good enough for now.
 235   def handle_incoming(%{"id" => id}) when not (is_binary(id) and length(id) > 8), do: :error
 236
 237   # TODO: validate those with a Ecto scheme
 238   # - tags
 239   # - emoji
 240   def handle_incoming(%{"type" => "Create", "object" => %{"type" => objtype} = object} = data)
 241       when objtype in ["Article", "Note", "Video"] do
 242     actor = get_actor(data)
 243
 244     data =
 245       Map.put(data, "actor", actor)
 246       |> fix_addressing
 247
 248     with nil <- Activity.get_create_activity_by_object_ap_id(object["id"]),
 249          %User{} = user <- User.get_or_fetch_by_ap_id(data["actor"]) do
 250       object = fix_object(data["object"])
 251
 252       params = %{
 253         to: data["to"],
 254         object: object,
 255         actor: user,
 256         context: object["conversation"],
 257         local: false,
 258         published: data["published"],
 259         additional:
 260           Map.take(data, [
 261             "cc",
 262             "id"
 263           ])
 264       }
 265
 266       ActivityPub.create(params)
 267     else
 268       %Activity{} = activity -> {:ok, activity}
 269       _e -> :error
 270     end
 271   end
 272
 273   def handle_incoming(
 274         %{"type" => "Follow", "object" => followed, "actor" => follower, "id" => id} = data
 275       ) do
 276     with %User{local: true} = followed <- User.get_cached_by_ap_id(followed),
 277          %User{} = follower <- User.get_or_fetch_by_ap_id(follower),
 278          {:ok, activity} <- ActivityPub.follow(follower, followed, id, false) do
 279       if not User.locked?(followed) do
 280         ActivityPub.accept(%{
 281           to: [follower.ap_id],
 282           actor: followed.ap_id,
 283           object: data,
 284           local: true
 285         })
 286
 287         User.follow(follower, followed)
 288       end
 289
 290       {:ok, activity}
 291     else
 292       _e -> :error
 293     end
 294   end
 295
 296   defp mastodon_follow_hack(%{"id" => id, "actor" => follower_id}, followed) do
 297     with true <- id =~ "follows",
 298          %User{local: true} = follower <- User.get_cached_by_ap_id(follower_id),
 299          %Activity{} = activity <- Utils.fetch_latest_follow(follower, followed) do
 300       {:ok, activity}
 301     else
 302       _ -> {:error, nil}
 303     end
 304   end
 305
 306   defp mastodon_follow_hack(_), do: {:error, nil}
 307
 308   defp get_follow_activity(follow_object, followed) do
 309     with object_id when not is_nil(object_id) <- Utils.get_ap_id(follow_object),
 310          {_, %Activity{} = activity} <- {:activity, Activity.get_by_ap_id(object_id)} do
 311       {:ok, activity}
 312     else
 313       # Can't find the activity. This might a Mastodon 2.3 "Accept"
 314       {:activity, nil} ->
 315         mastodon_follow_hack(follow_object, followed)
 316
 317       _ ->
 318         {:error, nil}
 319     end
 320   end
 321
 322   def handle_incoming(
 323         %{"type" => "Accept", "object" => follow_object, "actor" => actor, "id" => id} = data
 324       ) do
 325     with %User{} = followed <- User.get_or_fetch_by_ap_id(actor),
 326          {:ok, follow_activity} <- get_follow_activity(follow_object, followed),
 327          %User{local: true} = follower <- User.get_cached_by_ap_id(follow_activity.data["actor"]),
 328          {:ok, activity} <-
 329            ActivityPub.accept(%{
 330              to: follow_activity.data["to"],
 331              type: "Accept",
 332              actor: followed.ap_id,
 333              object: follow_activity.data["id"],
 334              local: false
 335            }) do
 336       if not User.following?(follower, followed) do
 337         {:ok, follower} = User.follow(follower, followed)
 338       end
 339
 340       {:ok, activity}
 341     else
 342       _e -> :error
 343     end
 344   end
 345
 346   def handle_incoming(
 347         %{"type" => "Reject", "object" => follow_object, "actor" => actor, "id" => id} = data
 348       ) do
 349     with %User{} = followed <- User.get_or_fetch_by_ap_id(actor),
 350          {:ok, follow_activity} <- get_follow_activity(follow_object, followed),
 351          %User{local: true} = follower <- User.get_cached_by_ap_id(follow_activity.data["actor"]),
 352          {:ok, activity} <-
 353            ActivityPub.accept(%{
 354              to: follow_activity.data["to"],
 355              type: "Accept",
 356              actor: followed.ap_id,
 357              object: follow_activity.data["id"],
 358              local: false
 359            }) do
 360       User.unfollow(follower, followed)
 361
 362       {:ok, activity}
 363     else
 364       _e -> :error
 365     end
 366   end
 367
 368   def handle_incoming(
 369         %{"type" => "Like", "object" => object_id, "actor" => actor, "id" => id} = _data
 370       ) do
 371     with %User{} = actor <- User.get_or_fetch_by_ap_id(actor),
 372          {:ok, object} <-
 373            get_obj_helper(object_id) || ActivityPub.fetch_object_from_id(object_id),
 374          {:ok, activity, _object} <- ActivityPub.like(actor, object, id, false) do
 375       {:ok, activity}
 376     else
 377       _e -> :error
 378     end
 379   end
 380
 381   def handle_incoming(
 382         %{"type" => "Announce", "object" => object_id, "actor" => actor, "id" => id} = _data
 383       ) do
 384     with %User{} = actor <- User.get_or_fetch_by_ap_id(actor),
 385          {:ok, object} <-
 386            get_obj_helper(object_id) || ActivityPub.fetch_object_from_id(object_id),
 387          {:ok, activity, _object} <- ActivityPub.announce(actor, object, id, false) do
 388       {:ok, activity}
 389     else
 390       _e -> :error
 391     end
 392   end
 393
 394   def handle_incoming(
 395         %{"type" => "Update", "object" => %{"type" => object_type} = object, "actor" => actor_id} =
 396           data
 397       )
 398       when object_type in ["Person", "Application", "Service", "Organization"] do
 399     with %User{ap_id: ^actor_id} = actor <- User.get_by_ap_id(object["id"]) do
 400       {:ok, new_user_data} = ActivityPub.user_data_from_user_object(object)
 401
 402       banner = new_user_data[:info]["banner"]
 403       locked = new_user_data[:info]["locked"] || false
 404
 405       update_data =
 406         new_user_data
 407         |> Map.take([:name, :bio, :avatar])
 408         |> Map.put(:info, Map.merge(actor.info, %{"banner" => banner, "locked" => locked}))
 409
 410       actor
 411       |> User.upgrade_changeset(update_data)
 412       |> User.update_and_set_cache()
 413
 414       ActivityPub.update(%{
 415         local: false,
 416         to: data["to"] || [],
 417         cc: data["cc"] || [],
 418         object: object,
 419         actor: actor_id
 420       })
 421     else
 422       e ->
 423         Logger.error(e)
 424         :error
 425     end
 426   end
 427
 428   # TODO: Make secure.
 429   def handle_incoming(
 430         %{"type" => "Delete", "object" => object_id, "actor" => actor, "id" => _id} = _data
 431       ) do
 432     object_id = Utils.get_ap_id(object_id)
 433
 434     with %User{} = _actor <- User.get_or_fetch_by_ap_id(actor),
 435          {:ok, object} <-
 436            get_obj_helper(object_id) || ActivityPub.fetch_object_from_id(object_id),
 437          {:ok, activity} <- ActivityPub.delete(object, false) do
 438       {:ok, activity}
 439     else
 440       _e -> :error
 441     end
 442   end
 443
 444   def handle_incoming(
 445         %{
 446           "type" => "Undo",
 447           "object" => %{"type" => "Announce", "object" => object_id},
 448           "actor" => actor,
 449           "id" => id
 450         } = _data
 451       ) do
 452     with %User{} = actor <- User.get_or_fetch_by_ap_id(actor),
 453          {:ok, object} <-
 454            get_obj_helper(object_id) || ActivityPub.fetch_object_from_id(object_id),
 455          {:ok, activity, _} <- ActivityPub.unannounce(actor, object, id, false) do
 456       {:ok, activity}
 457     else
 458       _e -> :error
 459     end
 460   end
 461
 462   def handle_incoming(
 463         %{
 464           "type" => "Undo",
 465           "object" => %{"type" => "Follow", "object" => followed},
 466           "actor" => follower,
 467           "id" => id
 468         } = _data
 469       ) do
 470     with %User{local: true} = followed <- User.get_cached_by_ap_id(followed),
 471          %User{} = follower <- User.get_or_fetch_by_ap_id(follower),
 472          {:ok, activity} <- ActivityPub.unfollow(follower, followed, id, false) do
 473       User.unfollow(follower, followed)
 474       {:ok, activity}
 475     else
 476       e -> :error
 477     end
 478   end
 479
 480   @ap_config Application.get_env(:pleroma, :activitypub)
 481   @accept_blocks Keyword.get(@ap_config, :accept_blocks)
 482
 483   def handle_incoming(
 484         %{
 485           "type" => "Undo",
 486           "object" => %{"type" => "Block", "object" => blocked},
 487           "actor" => blocker,
 488           "id" => id
 489         } = _data
 490       ) do
 491     with true <- @accept_blocks,
 492          %User{local: true} = blocked <- User.get_cached_by_ap_id(blocked),
 493          %User{} = blocker <- User.get_or_fetch_by_ap_id(blocker),
 494          {:ok, activity} <- ActivityPub.unblock(blocker, blocked, id, false) do
 495       User.unblock(blocker, blocked)
 496       {:ok, activity}
 497     else
 498       e -> :error
 499     end
 500   end
 501
 502   def handle_incoming(
 503         %{"type" => "Block", "object" => blocked, "actor" => blocker, "id" => id} = data
 504       ) do
 505     with true <- @accept_blocks,
 506          %User{local: true} = blocked = User.get_cached_by_ap_id(blocked),
 507          %User{} = blocker = User.get_or_fetch_by_ap_id(blocker),
 508          {:ok, activity} <- ActivityPub.block(blocker, blocked, id, false) do
 509       User.unfollow(blocker, blocked)
 510       User.block(blocker, blocked)
 511       {:ok, activity}
 512     else
 513       e -> :error
 514     end
 515   end
 516
 517   def handle_incoming(
 518         %{
 519           "type" => "Undo",
 520           "object" => %{"type" => "Like", "object" => object_id},
 521           "actor" => actor,
 522           "id" => id
 523         } = _data
 524       ) do
 525     with %User{} = actor <- User.get_or_fetch_by_ap_id(actor),
 526          {:ok, object} <-
 527            get_obj_helper(object_id) || ActivityPub.fetch_object_from_id(object_id),
 528          {:ok, activity, _, _} <- ActivityPub.unlike(actor, object, id, false) do
 529       {:ok, activity}
 530     else
 531       _e -> :error
 532     end
 533   end
 534
 535   def handle_incoming(_), do: :error
 536
 537   def get_obj_helper(id) do
 538     if object = Object.normalize(id), do: {:ok, object}, else: nil
 539   end
 540
 541   def set_reply_to_uri(%{"inReplyTo" => inReplyTo} = object) do
 542     with false <- String.starts_with?(inReplyTo, "http"),
 543          {:ok, %{data: replied_to_object}} <- get_obj_helper(inReplyTo) do
 544       Map.put(object, "inReplyTo", replied_to_object["external_url"] || inReplyTo)
 545     else
 546       _e -> object
 547     end
 548   end
 549
 550   def set_reply_to_uri(obj), do: obj
 551
 552   # Prepares the object of an outgoing create activity.
 553   def prepare_object(object) do
 554     object
 555     |> set_sensitive
 556     |> add_hashtags
 557     |> add_mention_tags
 558     |> add_emoji_tags
 559     |> add_attributed_to
 560     |> prepare_attachments
 561     |> set_conversation
 562     |> set_reply_to_uri
 563   end
 564
 565   #  @doc
 566   #  """
 567   #  internal -> Mastodon
 568   #  """
 569
 570   def prepare_outgoing(%{"type" => "Create", "object" => %{"type" => "Note"} = object} = data) do
 571     object =
 572       object
 573       |> prepare_object
 574
 575     data =
 576       data
 577       |> Map.put("object", object)
 578       |> Map.put("@context", "https://www.w3.org/ns/activitystreams")
 579
 580     {:ok, data}
 581   end
 582
 583   # Mastodon Accept/Reject requires a non-normalized object containing the actor URIs,
 584   # because of course it does.
 585   def prepare_outgoing(%{"type" => "Accept"} = data) do
 586     with follow_activity <- Activity.normalize(data["object"]) do
 587       object = %{
 588         "actor" => follow_activity.actor,
 589         "object" => follow_activity.data["object"],
 590         "id" => follow_activity.data["id"],
 591         "type" => "Follow"
 592       }
 593
 594       data =
 595         data
 596         |> Map.put("object", object)
 597         |> Map.put("@context", "https://www.w3.org/ns/activitystreams")
 598
 599       {:ok, data}
 600     end
 601   end
 602
 603   def prepare_outgoing(%{"type" => "Reject"} = data) do
 604     with follow_activity <- Activity.normalize(data["object"]) do
 605       object = %{
 606         "actor" => follow_activity.actor,
 607         "object" => follow_activity.data["object"],
 608         "id" => follow_activity.data["id"],
 609         "type" => "Follow"
 610       }
 611
 612       data =
 613         data
 614         |> Map.put("object", object)
 615         |> Map.put("@context", "https://www.w3.org/ns/activitystreams")
 616
 617       {:ok, data}
 618     end
 619   end
 620
 621   def prepare_outgoing(%{"type" => _type} = data) do
 622     data =
 623       data
 624       |> maybe_fix_object_url
 625       |> Map.put("@context", "https://www.w3.org/ns/activitystreams")
 626
 627     {:ok, data}
 628   end
 629
 630   def maybe_fix_object_url(data) do
 631     if is_binary(data["object"]) and not String.starts_with?(data["object"], "http") do
 632       case ActivityPub.fetch_object_from_id(data["object"]) do
 633         {:ok, relative_object} ->
 634           if relative_object.data["external_url"] do
 635             _data =
 636               data
 637               |> Map.put("object", relative_object.data["external_url"])
 638           else
 639             data
 640           end
 641
 642         e ->
 643           Logger.error("Couldn't fetch #{data["object"]} #{inspect(e)}")
 644           data
 645       end
 646     else
 647       data
 648     end
 649   end
 650
 651   def add_hashtags(object) do
 652     tags =
 653       (object["tag"] || [])
 654       |> Enum.map(fn tag ->
 655         %{
 656           "href" => Pleroma.Web.Endpoint.url() <> "/tags/#{tag}",
 657           "name" => "##{tag}",
 658           "type" => "Hashtag"
 659         }
 660       end)
 661
 662     object
 663     |> Map.put("tag", tags)
 664   end
 665
 666   def add_mention_tags(object) do
 667     recipients = object["to"] ++ (object["cc"] || [])
 668
 669     mentions =
 670       recipients
 671       |> Enum.map(fn ap_id -> User.get_cached_by_ap_id(ap_id) end)
 672       |> Enum.filter(& &1)
 673       |> Enum.map(fn user ->
 674         %{"type" => "Mention", "href" => user.ap_id, "name" => "@#{user.nickname}"}
 675       end)
 676
 677     tags = object["tag"] || []
 678
 679     object
 680     |> Map.put("tag", tags ++ mentions)
 681   end
 682
 683   # TODO: we should probably send mtime instead of unix epoch time for updated
 684   def add_emoji_tags(object) do
 685     tags = object["tag"] || []
 686     emoji = object["emoji"] || []
 687
 688     out =
 689       emoji
 690       |> Enum.map(fn {name, url} ->
 691         %{
 692           "icon" => %{"url" => url, "type" => "Image"},
 693           "name" => ":" <> name <> ":",
 694           "type" => "Emoji",
 695           "updated" => "1970-01-01T00:00:00Z",
 696           "id" => url
 697         }
 698       end)
 699
 700     object
 701     |> Map.put("tag", tags ++ out)
 702   end
 703
 704   def set_conversation(object) do
 705     Map.put(object, "conversation", object["context"])
 706   end
 707
 708   def set_sensitive(object) do
 709     tags = object["tag"] || []
 710     Map.put(object, "sensitive", "nsfw" in tags)
 711   end
 712
 713   def add_attributed_to(object) do
 714     attributedTo = object["attributedTo"] || object["actor"]
 715
 716     object
 717     |> Map.put("attributedTo", attributedTo)
 718   end
 719
 720   def prepare_attachments(object) do
 721     attachments =
 722       (object["attachment"] || [])
 723       |> Enum.map(fn data ->
 724         [%{"mediaType" => media_type, "href" => href} | _] = data["url"]
 725         %{"url" => href, "mediaType" => media_type, "name" => data["name"], "type" => "Document"}
 726       end)
 727
 728     object
 729     |> Map.put("attachment", attachments)
 730   end
 731
 732   defp user_upgrade_task(user) do
 733     old_follower_address = User.ap_followers(user)
 734
 735     q =
 736       from(
 737         u in User,
 738         where: ^old_follower_address in u.following,
 739         update: [
 740           set: [
 741             following:
 742               fragment(
 743                 "array_replace(?,?,?)",
 744                 u.following,
 745                 ^old_follower_address,
 746                 ^user.follower_address
 747               )
 748           ]
 749         ]
 750       )
 751
 752     Repo.update_all(q, [])
 753
 754     maybe_retire_websub(user.ap_id)
 755
 756     # Only do this for recent activties, don't go through the whole db.
 757     # Only look at the last 1000 activities.
 758     since = (Repo.aggregate(Activity, :max, :id) || 0) - 1_000
 759
 760     q =
 761       from(
 762         a in Activity,
 763         where: ^old_follower_address in a.recipients,
 764         where: a.id > ^since,
 765         update: [
 766           set: [
 767             recipients:
 768               fragment(
 769                 "array_replace(?,?,?)",
 770                 a.recipients,
 771                 ^old_follower_address,
 772                 ^user.follower_address
 773               )
 774           ]
 775         ]
 776       )
 777
 778     Repo.update_all(q, [])
 779   end
 780
 781   def upgrade_user_from_ap_id(ap_id, async \\ true) do
 782     with %User{local: false} = user <- User.get_by_ap_id(ap_id),
 783          {:ok, data} <- ActivityPub.fetch_and_prepare_user_from_ap_id(ap_id) do
 784       data =
 785         data
 786         |> Map.put(:info, Map.merge(user.info, data[:info]))
 787
 788       already_ap = User.ap_enabled?(user)
 789
 790       {:ok, user} =
 791         User.upgrade_changeset(user, data)
 792         |> Repo.update()
 793
 794       if !already_ap do
 795         # This could potentially take a long time, do it in the background
 796         if async do
 797           Task.start(fn ->
 798             user_upgrade_task(user)
 799           end)
 800         else
 801           user_upgrade_task(user)
 802         end
 803       end
 804
 805       {:ok, user}
 806     else
 807       e -> e
 808     end
 809   end
 810
 811   def maybe_retire_websub(ap_id) do
 812     # some sanity checks
 813     if is_binary(ap_id) && String.length(ap_id) > 8 do
 814       q =
 815         from(
 816           ws in Pleroma.Web.Websub.WebsubClientSubscription,
 817           where: fragment("? like ?", ws.topic, ^"#{ap_id}%")
 818         )
 819
 820       Repo.delete_all(q)
 821     end
 822   end
 823
 824   def maybe_fix_user_url(data) do
 825     if is_map(data["url"]) do
 826       Map.put(data, "url", data["url"]["href"])
 827     else
 828       data
 829     end
 830   end
 831
 832   def maybe_fix_user_object(data) do
 833     data
 834     |> maybe_fix_user_url
 835   end
 836 end