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

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.ActivityPub.Transmogrifier do
   6   @moduledoc """
   7   A module to handle coding from internal to wire ActivityPub and back.
   8   """
   9   alias Pleroma.Activity
  10   alias Pleroma.EarmarkRenderer
  11   alias Pleroma.FollowingRelationship
  12   alias Pleroma.Object
  13   alias Pleroma.Object.Containment
  14   alias Pleroma.Repo
  15   alias Pleroma.User
  16   alias Pleroma.Web.ActivityPub.ActivityPub
  17   alias Pleroma.Web.ActivityPub.ObjectValidator
  18   alias Pleroma.Web.ActivityPub.Pipeline
  19   alias Pleroma.Web.ActivityPub.Utils
  20   alias Pleroma.Web.ActivityPub.Visibility
  21   alias Pleroma.Web.Federator
  22   alias Pleroma.Workers.TransmogrifierWorker
  23
  24   import Ecto.Query
  25
  26   require Logger
  27   require Pleroma.Constants
  28
  29   @doc """
  30   Modifies an incoming AP object (mastodon format) to our internal format.
  31   """
  32   def fix_object(object, options \\ []) do
  33     object
  34     |> strip_internal_fields
  35     |> fix_actor
  36     |> fix_url
  37     |> fix_attachments
  38     |> fix_context
  39     |> fix_in_reply_to(options)
  40     |> fix_emoji
  41     |> fix_tag
  42     |> fix_content_map
  43     |> fix_addressing
  44     |> fix_summary
  45     |> fix_type(options)
  46     |> fix_content
  47   end
  48
  49   def fix_summary(%{"summary" => nil} = object) do
  50     Map.put(object, "summary", "")
  51   end
  52
  53   def fix_summary(%{"summary" => _} = object) do
  54     # summary is present, nothing to do
  55     object
  56   end
  57
  58   def fix_summary(object), do: Map.put(object, "summary", "")
  59
  60   def fix_addressing_list(map, field) do
  61     cond do
  62       is_binary(map[field]) ->
  63         Map.put(map, field, [map[field]])
  64
  65       is_nil(map[field]) ->
  66         Map.put(map, field, [])
  67
  68       true ->
  69         map
  70     end
  71   end
  72
  73   def fix_explicit_addressing(
  74         %{"to" => to, "cc" => cc} = object,
  75         explicit_mentions,
  76         follower_collection
  77       ) do
  78     explicit_to = Enum.filter(to, fn x -> x in explicit_mentions end)
  79
  80     explicit_cc = Enum.filter(to, fn x -> x not in explicit_mentions end)
  81
  82     final_cc =
  83       (cc ++ explicit_cc)
  84       |> Enum.reject(fn x -> String.ends_with?(x, "/followers") and x != follower_collection end)
  85       |> Enum.uniq()
  86
  87     object
  88     |> Map.put("to", explicit_to)
  89     |> Map.put("cc", final_cc)
  90   end
  91
  92   def fix_explicit_addressing(object, _explicit_mentions, _followers_collection), do: object
  93
  94   # if directMessage flag is set to true, leave the addressing alone
  95   def fix_explicit_addressing(%{"directMessage" => true} = object), do: object
  96
  97   def fix_explicit_addressing(object) do
  98     explicit_mentions = Utils.determine_explicit_mentions(object)
  99
 100     %User{follower_address: follower_collection} =
 101       object
 102       |> Containment.get_actor()
 103       |> User.get_cached_by_ap_id()
 104
 105     explicit_mentions =
 106       explicit_mentions ++
 107         [
 108           Pleroma.Constants.as_public(),
 109           follower_collection
 110         ]
 111
 112     fix_explicit_addressing(object, explicit_mentions, follower_collection)
 113   end
 114
 115   # if as:Public is addressed, then make sure the followers collection is also addressed
 116   # so that the activities will be delivered to local users.
 117   def fix_implicit_addressing(%{"to" => to, "cc" => cc} = object, followers_collection) do
 118     recipients = to ++ cc
 119
 120     if followers_collection not in recipients do
 121       cond do
 122         Pleroma.Constants.as_public() in cc ->
 123           to = to ++ [followers_collection]
 124           Map.put(object, "to", to)
 125
 126         Pleroma.Constants.as_public() in to ->
 127           cc = cc ++ [followers_collection]
 128           Map.put(object, "cc", cc)
 129
 130         true ->
 131           object
 132       end
 133     else
 134       object
 135     end
 136   end
 137
 138   def fix_implicit_addressing(object, _), do: object
 139
 140   def fix_addressing(object) do
 141     {:ok, %User{} = user} = User.get_or_fetch_by_ap_id(object["actor"])
 142     followers_collection = User.ap_followers(user)
 143
 144     object
 145     |> fix_addressing_list("to")
 146     |> fix_addressing_list("cc")
 147     |> fix_addressing_list("bto")
 148     |> fix_addressing_list("bcc")
 149     |> fix_explicit_addressing()
 150     |> fix_implicit_addressing(followers_collection)
 151   end
 152
 153   def fix_actor(%{"attributedTo" => actor} = object) do
 154     Map.put(object, "actor", Containment.get_actor(%{"actor" => actor}))
 155   end
 156
 157   def fix_in_reply_to(object, options \\ [])
 158
 159   def fix_in_reply_to(%{"inReplyTo" => in_reply_to} = object, options)
 160       when not is_nil(in_reply_to) do
 161     in_reply_to_id = prepare_in_reply_to(in_reply_to)
 162     object = Map.put(object, "inReplyToAtomUri", in_reply_to_id)
 163     depth = (options[:depth] || 0) + 1
 164
 165     if Federator.allowed_thread_distance?(depth) do
 166       with {:ok, replied_object} <- get_obj_helper(in_reply_to_id, options),
 167            %Activity{} <- Activity.get_create_by_object_ap_id(replied_object.data["id"]) do
 168         object
 169         |> Map.put("inReplyTo", replied_object.data["id"])
 170         |> Map.put("inReplyToAtomUri", object["inReplyToAtomUri"] || in_reply_to_id)
 171         |> Map.put("conversation", replied_object.data["context"] || object["conversation"])
 172         |> Map.put("context", replied_object.data["context"] || object["conversation"])
 173       else
 174         e ->
 175           Logger.error("Couldn't fetch #{inspect(in_reply_to_id)}, error: #{inspect(e)}")
 176           object
 177       end
 178     else
 179       object
 180     end
 181   end
 182
 183   def fix_in_reply_to(object, _options), do: object
 184
 185   defp prepare_in_reply_to(in_reply_to) do
 186     cond do
 187       is_bitstring(in_reply_to) ->
 188         in_reply_to
 189
 190       is_map(in_reply_to) && is_bitstring(in_reply_to["id"]) ->
 191         in_reply_to["id"]
 192
 193       is_list(in_reply_to) && is_bitstring(Enum.at(in_reply_to, 0)) ->
 194         Enum.at(in_reply_to, 0)
 195
 196       true ->
 197         ""
 198     end
 199   end
 200
 201   def fix_context(object) do
 202     context = object["context"] || object["conversation"] || Utils.generate_context_id()
 203
 204     object
 205     |> Map.put("context", context)
 206     |> Map.put("conversation", context)
 207   end
 208
 209   defp add_if_present(map, _key, nil), do: map
 210
 211   defp add_if_present(map, key, value) do
 212     Map.put(map, key, value)
 213   end
 214
 215   def fix_attachments(%{"attachment" => attachment} = object) when is_list(attachment) do
 216     attachments =
 217       Enum.map(attachment, fn data ->
 218         url =
 219           cond do
 220             is_list(data["url"]) -> List.first(data["url"])
 221             is_map(data["url"]) -> data["url"]
 222             true -> nil
 223           end
 224
 225         media_type =
 226           cond do
 227             is_map(url) && is_binary(url["mediaType"]) -> url["mediaType"]
 228             is_binary(data["mediaType"]) -> data["mediaType"]
 229             is_binary(data["mimeType"]) -> data["mimeType"]
 230             true -> nil
 231           end
 232
 233         href =
 234           cond do
 235             is_map(url) && is_binary(url["href"]) -> url["href"]
 236             is_binary(data["url"]) -> data["url"]
 237             is_binary(data["href"]) -> data["href"]
 238           end
 239
 240         attachment_url =
 241           %{"href" => href}
 242           |> add_if_present("mediaType", media_type)
 243           |> add_if_present("type", Map.get(url || %{}, "type"))
 244
 245         %{"url" => [attachment_url]}
 246         |> add_if_present("mediaType", media_type)
 247         |> add_if_present("type", data["type"])
 248         |> add_if_present("name", data["name"])
 249       end)
 250
 251     Map.put(object, "attachment", attachments)
 252   end
 253
 254   def fix_attachments(%{"attachment" => attachment} = object) when is_map(attachment) do
 255     object
 256     |> Map.put("attachment", [attachment])
 257     |> fix_attachments()
 258   end
 259
 260   def fix_attachments(object), do: object
 261
 262   def fix_url(%{"url" => url} = object) when is_map(url) do
 263     Map.put(object, "url", url["href"])
 264   end
 265
 266   def fix_url(%{"type" => object_type, "url" => url} = object)
 267       when object_type in ["Video", "Audio"] and is_list(url) do
 268     first_element = Enum.at(url, 0)
 269
 270     link_element = Enum.find(url, fn x -> is_map(x) and x["mimeType"] == "text/html" end)
 271
 272     object
 273     |> Map.put("attachment", [first_element])
 274     |> Map.put("url", link_element["href"])
 275   end
 276
 277   def fix_url(%{"type" => object_type, "url" => url} = object)
 278       when object_type != "Video" and is_list(url) do
 279     first_element = Enum.at(url, 0)
 280
 281     url_string =
 282       cond do
 283         is_bitstring(first_element) -> first_element
 284         is_map(first_element) -> first_element["href"] || ""
 285         true -> ""
 286       end
 287
 288     Map.put(object, "url", url_string)
 289   end
 290
 291   def fix_url(object), do: object
 292
 293   def fix_emoji(%{"tag" => tags} = object) when is_list(tags) do
 294     emoji =
 295       tags
 296       |> Enum.filter(fn data -> data["type"] == "Emoji" and data["icon"] end)
 297       |> Enum.reduce(%{}, fn data, mapping ->
 298         name = String.trim(data["name"], ":")
 299
 300         Map.put(mapping, name, data["icon"]["url"])
 301       end)
 302
 303     # we merge mastodon and pleroma emoji into a single mapping, to allow for both wire formats
 304     emoji = Map.merge(object["emoji"] || %{}, emoji)
 305
 306     Map.put(object, "emoji", emoji)
 307   end
 308
 309   def fix_emoji(%{"tag" => %{"type" => "Emoji"} = tag} = object) do
 310     name = String.trim(tag["name"], ":")
 311     emoji = %{name => tag["icon"]["url"]}
 312
 313     Map.put(object, "emoji", emoji)
 314   end
 315
 316   def fix_emoji(object), do: object
 317
 318   def fix_tag(%{"tag" => tag} = object) when is_list(tag) do
 319     tags =
 320       tag
 321       |> Enum.filter(fn data -> data["type"] == "Hashtag" and data["name"] end)
 322       |> Enum.map(fn data -> String.slice(data["name"], 1..-1) end)
 323
 324     Map.put(object, "tag", tag ++ tags)
 325   end
 326
 327   def fix_tag(%{"tag" => %{"type" => "Hashtag", "name" => hashtag} = tag} = object) do
 328     combined = [tag, String.slice(hashtag, 1..-1)]
 329
 330     Map.put(object, "tag", combined)
 331   end
 332
 333   def fix_tag(%{"tag" => %{} = tag} = object), do: Map.put(object, "tag", [tag])
 334
 335   def fix_tag(object), do: object
 336
 337   # content map usually only has one language so this will do for now.
 338   def fix_content_map(%{"contentMap" => content_map} = object) do
 339     content_groups = Map.to_list(content_map)
 340     {_, content} = Enum.at(content_groups, 0)
 341
 342     Map.put(object, "content", content)
 343   end
 344
 345   def fix_content_map(object), do: object
 346
 347   def fix_type(object, options \\ [])
 348
 349   def fix_type(%{"inReplyTo" => reply_id, "name" => _} = object, options)
 350       when is_binary(reply_id) do
 351     with true <- Federator.allowed_thread_distance?(options[:depth]),
 352          {:ok, %{data: %{"type" => "Question"} = _} = _} <- get_obj_helper(reply_id, options) do
 353       Map.put(object, "type", "Answer")
 354     else
 355       _ -> object
 356     end
 357   end
 358
 359   def fix_type(object, _), do: object
 360
 361   defp fix_content(%{"mediaType" => "text/markdown", "content" => content} = object)
 362        when is_binary(content) do
 363     html_content =
 364       content
 365       |> Earmark.as_html!(%Earmark.Options{renderer: EarmarkRenderer})
 366       |> Pleroma.HTML.filter_tags()
 367
 368     Map.merge(object, %{"content" => html_content, "mediaType" => "text/html"})
 369   end
 370
 371   defp fix_content(object), do: object
 372
 373   defp mastodon_follow_hack(%{"id" => id, "actor" => follower_id}, followed) do
 374     with true <- id =~ "follows",
 375          %User{local: true} = follower <- User.get_cached_by_ap_id(follower_id),
 376          %Activity{} = activity <- Utils.fetch_latest_follow(follower, followed) do
 377       {:ok, activity}
 378     else
 379       _ -> {:error, nil}
 380     end
 381   end
 382
 383   defp mastodon_follow_hack(_, _), do: {:error, nil}
 384
 385   defp get_follow_activity(follow_object, followed) do
 386     with object_id when not is_nil(object_id) <- Utils.get_ap_id(follow_object),
 387          {_, %Activity{} = activity} <- {:activity, Activity.get_by_ap_id(object_id)} do
 388       {:ok, activity}
 389     else
 390       # Can't find the activity. This might a Mastodon 2.3 "Accept"
 391       {:activity, nil} ->
 392         mastodon_follow_hack(follow_object, followed)
 393
 394       _ ->
 395         {:error, nil}
 396     end
 397   end
 398
 399   # Reduce the object list to find the reported user.
 400   defp get_reported(objects) do
 401     Enum.reduce_while(objects, nil, fn ap_id, _ ->
 402       with %User{} = user <- User.get_cached_by_ap_id(ap_id) do
 403         {:halt, user}
 404       else
 405         _ -> {:cont, nil}
 406       end
 407     end)
 408   end
 409
 410   def handle_incoming(data, options \\ [])
 411
 412   # Flag objects are placed ahead of the ID check because Mastodon 2.8 and earlier send them
 413   # with nil ID.
 414   def handle_incoming(%{"type" => "Flag", "object" => objects, "actor" => actor} = data, _options) do
 415     with context <- data["context"] || Utils.generate_context_id(),
 416          content <- data["content"] || "",
 417          %User{} = actor <- User.get_cached_by_ap_id(actor),
 418          # Reduce the object list to find the reported user.
 419          %User{} = account <- get_reported(objects),
 420          # Remove the reported user from the object list.
 421          statuses <- Enum.filter(objects, fn ap_id -> ap_id != account.ap_id end) do
 422       %{
 423         actor: actor,
 424         context: context,
 425         account: account,
 426         statuses: statuses,
 427         content: content,
 428         additional: %{"cc" => [account.ap_id]}
 429       }
 430       |> ActivityPub.flag()
 431     end
 432   end
 433
 434   # disallow objects with bogus IDs
 435   def handle_incoming(%{"id" => nil}, _options), do: :error
 436   def handle_incoming(%{"id" => ""}, _options), do: :error
 437   # length of https:// = 8, should validate better, but good enough for now.
 438   def handle_incoming(%{"id" => id}, _options) when is_binary(id) and byte_size(id) < 8,
 439     do: :error
 440
 441   # TODO: validate those with a Ecto scheme
 442   # - tags
 443   # - emoji
 444   def handle_incoming(
 445         %{"type" => "Create", "object" => %{"type" => objtype} = object} = data,
 446         options
 447       )
 448       when objtype in ["Article", "Event", "Note", "Video", "Page", "Question", "Answer", "Audio"] do
 449     actor = Containment.get_actor(data)
 450
 451     data =
 452       Map.put(data, "actor", actor)
 453       |> fix_addressing
 454
 455     with nil <- Activity.get_create_by_object_ap_id(object["id"]),
 456          {:ok, %User{} = user} <- User.get_or_fetch_by_ap_id(data["actor"]) do
 457       object = fix_object(object, options)
 458
 459       params = %{
 460         to: data["to"],
 461         object: object,
 462         actor: user,
 463         context: object["conversation"],
 464         local: false,
 465         published: data["published"],
 466         additional:
 467           Map.take(data, [
 468             "cc",
 469             "directMessage",
 470             "id"
 471           ])
 472       }
 473
 474       with {:ok, created_activity} <- ActivityPub.create(params) do
 475         reply_depth = (options[:depth] || 0) + 1
 476
 477         if Federator.allowed_thread_distance?(reply_depth) do
 478           for reply_id <- replies(object) do
 479             Pleroma.Workers.RemoteFetcherWorker.enqueue("fetch_remote", %{
 480               "id" => reply_id,
 481               "depth" => reply_depth
 482             })
 483           end
 484         end
 485
 486         {:ok, created_activity}
 487       end
 488     else
 489       %Activity{} = activity -> {:ok, activity}
 490       _e -> :error
 491     end
 492   end
 493
 494   def handle_incoming(
 495         %{"type" => "Listen", "object" => %{"type" => "Audio"} = object} = data,
 496         options
 497       ) do
 498     actor = Containment.get_actor(data)
 499
 500     data =
 501       Map.put(data, "actor", actor)
 502       |> fix_addressing
 503
 504     with {:ok, %User{} = user} <- User.get_or_fetch_by_ap_id(data["actor"]) do
 505       reply_depth = (options[:depth] || 0) + 1
 506       options = Keyword.put(options, :depth, reply_depth)
 507       object = fix_object(object, options)
 508
 509       params = %{
 510         to: data["to"],
 511         object: object,
 512         actor: user,
 513         context: nil,
 514         local: false,
 515         published: data["published"],
 516         additional: Map.take(data, ["cc", "id"])
 517       }
 518
 519       ActivityPub.listen(params)
 520     else
 521       _e -> :error
 522     end
 523   end
 524
 525   def handle_incoming(
 526         %{"type" => "Follow", "object" => followed, "actor" => follower, "id" => id} = data,
 527         _options
 528       ) do
 529     with %User{local: true} = followed <-
 530            User.get_cached_by_ap_id(Containment.get_actor(%{"actor" => followed})),
 531          {:ok, %User{} = follower} <-
 532            User.get_or_fetch_by_ap_id(Containment.get_actor(%{"actor" => follower})),
 533          {:ok, activity} <- ActivityPub.follow(follower, followed, id, false) do
 534       with deny_follow_blocked <- Pleroma.Config.get([:user, :deny_follow_blocked]),
 535            {_, false} <- {:user_blocked, User.blocks?(followed, follower) && deny_follow_blocked},
 536            {_, false} <- {:user_locked, User.locked?(followed)},
 537            {_, {:ok, follower}} <- {:follow, User.follow(follower, followed)},
 538            {_, {:ok, _}} <-
 539              {:follow_state_update, Utils.update_follow_state_for_all(activity, "accept")},
 540            {:ok, _relationship} <-
 541              FollowingRelationship.update(follower, followed, :follow_accept) do
 542         ActivityPub.accept(%{
 543           to: [follower.ap_id],
 544           actor: followed,
 545           object: data,
 546           local: true
 547         })
 548       else
 549         {:user_blocked, true} ->
 550           {:ok, _} = Utils.update_follow_state_for_all(activity, "reject")
 551           {:ok, _relationship} = FollowingRelationship.update(follower, followed, :follow_reject)
 552
 553           ActivityPub.reject(%{
 554             to: [follower.ap_id],
 555             actor: followed,
 556             object: data,
 557             local: true
 558           })
 559
 560         {:follow, {:error, _}} ->
 561           {:ok, _} = Utils.update_follow_state_for_all(activity, "reject")
 562           {:ok, _relationship} = FollowingRelationship.update(follower, followed, :follow_reject)
 563
 564           ActivityPub.reject(%{
 565             to: [follower.ap_id],
 566             actor: followed,
 567             object: data,
 568             local: true
 569           })
 570
 571         {:user_locked, true} ->
 572           {:ok, _relationship} = FollowingRelationship.update(follower, followed, :follow_pending)
 573           :noop
 574       end
 575
 576       {:ok, activity}
 577     else
 578       _e ->
 579         :error
 580     end
 581   end
 582
 583   def handle_incoming(
 584         %{"type" => "Accept", "object" => follow_object, "actor" => _actor, "id" => id} = data,
 585         _options
 586       ) do
 587     with actor <- Containment.get_actor(data),
 588          {:ok, %User{} = followed} <- User.get_or_fetch_by_ap_id(actor),
 589          {:ok, follow_activity} <- get_follow_activity(follow_object, followed),
 590          {:ok, follow_activity} <- Utils.update_follow_state_for_all(follow_activity, "accept"),
 591          %User{local: true} = follower <- User.get_cached_by_ap_id(follow_activity.data["actor"]),
 592          {:ok, _relationship} <- FollowingRelationship.update(follower, followed, :follow_accept) do
 593       ActivityPub.accept(%{
 594         to: follow_activity.data["to"],
 595         type: "Accept",
 596         actor: followed,
 597         object: follow_activity.data["id"],
 598         local: false,
 599         activity_id: id
 600       })
 601     else
 602       _e -> :error
 603     end
 604   end
 605
 606   def handle_incoming(
 607         %{"type" => "Reject", "object" => follow_object, "actor" => _actor, "id" => id} = data,
 608         _options
 609       ) do
 610     with actor <- Containment.get_actor(data),
 611          {:ok, %User{} = followed} <- User.get_or_fetch_by_ap_id(actor),
 612          {:ok, follow_activity} <- get_follow_activity(follow_object, followed),
 613          {:ok, follow_activity} <- Utils.update_follow_state_for_all(follow_activity, "reject"),
 614          %User{local: true} = follower <- User.get_cached_by_ap_id(follow_activity.data["actor"]),
 615          {:ok, _relationship} <- FollowingRelationship.update(follower, followed, :follow_reject),
 616          {:ok, activity} <-
 617            ActivityPub.reject(%{
 618              to: follow_activity.data["to"],
 619              type: "Reject",
 620              actor: followed,
 621              object: follow_activity.data["id"],
 622              local: false,
 623              activity_id: id
 624            }) do
 625       {:ok, activity}
 626     else
 627       _e -> :error
 628     end
 629   end
 630
 631   @misskey_reactions %{
 632     "like" => "👍",
 633     "love" => "❤️",
 634     "laugh" => "😆",
 635     "hmm" => "🤔",
 636     "surprise" => "😮",
 637     "congrats" => "🎉",
 638     "angry" => "💢",
 639     "confused" => "😥",
 640     "rip" => "😇",
 641     "pudding" => "🍮",
 642     "star" => "⭐"
 643   }
 644
 645   @doc "Rewrite misskey likes into EmojiReacts"
 646   def handle_incoming(
 647         %{
 648           "type" => "Like",
 649           "_misskey_reaction" => reaction
 650         } = data,
 651         options
 652       ) do
 653     data
 654     |> Map.put("type", "EmojiReact")
 655     |> Map.put("content", @misskey_reactions[reaction] || reaction)
 656     |> handle_incoming(options)
 657   end
 658
 659   def handle_incoming(
 660         %{"type" => "Create", "object" => %{"type" => "ChatMessage"}} = data,
 661         _options
 662       ) do
 663     with {:ok, %User{}} <- ObjectValidator.fetch_actor(data),
 664          {:ok, activity, _} <- Pipeline.common_pipeline(data, local: false) do
 665       {:ok, activity}
 666     end
 667   end
 668
 669   def handle_incoming(%{"type" => type} = data, _options) when type in ["Like", "EmojiReact"] do
 670     with :ok <- ObjectValidator.fetch_actor_and_object(data),
 671          {:ok, activity, _meta} <-
 672            Pipeline.common_pipeline(data, local: false) do
 673       {:ok, activity}
 674     else
 675       e -> {:error, e}
 676     end
 677   end
 678
 679   def handle_incoming(
 680         %{"type" => "Announce", "object" => object_id, "actor" => _actor, "id" => id} = data,
 681         _options
 682       ) do
 683     with actor <- Containment.get_actor(data),
 684          {:ok, %User{} = actor} <- User.get_or_fetch_by_ap_id(actor),
 685          {:ok, object} <- get_embedded_obj_helper(object_id, actor),
 686          public <- Visibility.is_public?(data),
 687          {:ok, activity, _object} <- ActivityPub.announce(actor, object, id, false, public) do
 688       {:ok, activity}
 689     else
 690       _e -> :error
 691     end
 692   end
 693
 694   def handle_incoming(
 695         %{"type" => "Update", "object" => %{"type" => object_type} = object, "actor" => actor_id} =
 696           data,
 697         _options
 698       )
 699       when object_type in [
 700              "Person",
 701              "Application",
 702              "Service",
 703              "Organization"
 704            ] do
 705     with %User{ap_id: ^actor_id} = actor <- User.get_cached_by_ap_id(object["id"]) do
 706       {:ok, new_user_data} = ActivityPub.user_data_from_user_object(object)
 707
 708       actor
 709       |> User.remote_user_changeset(new_user_data)
 710       |> User.update_and_set_cache()
 711
 712       ActivityPub.update(%{
 713         local: false,
 714         to: data["to"] || [],
 715         cc: data["cc"] || [],
 716         object: object,
 717         actor: actor_id,
 718         activity_id: data["id"]
 719       })
 720     else
 721       e ->
 722         Logger.error(e)
 723         :error
 724     end
 725   end
 726
 727   def handle_incoming(
 728         %{"type" => "Delete"} = data,
 729         _options
 730       ) do
 731     with {:ok, activity, _} <- Pipeline.common_pipeline(data, local: false) do
 732       {:ok, activity}
 733     end
 734   end
 735
 736   def handle_incoming(
 737         %{
 738           "type" => "Undo",
 739           "object" => %{"type" => "Follow", "object" => followed},
 740           "actor" => follower,
 741           "id" => id
 742         } = _data,
 743         _options
 744       ) do
 745     with %User{local: true} = followed <- User.get_cached_by_ap_id(followed),
 746          {:ok, %User{} = follower} <- User.get_or_fetch_by_ap_id(follower),
 747          {:ok, activity} <- ActivityPub.unfollow(follower, followed, id, false) do
 748       User.unfollow(follower, followed)
 749       {:ok, activity}
 750     else
 751       _e -> :error
 752     end
 753   end
 754
 755   def handle_incoming(
 756         %{
 757           "type" => "Undo",
 758           "object" => %{"type" => type}
 759         } = data,
 760         _options
 761       )
 762       when type in ["Like", "EmojiReact", "Announce", "Block"] do
 763     with {:ok, activity, _} <- Pipeline.common_pipeline(data, local: false) do
 764       {:ok, activity}
 765     end
 766   end
 767
 768   # For Undos that don't have the complete object attached, try to find it in our database.
 769   def handle_incoming(
 770         %{
 771           "type" => "Undo",
 772           "object" => object
 773         } = activity,
 774         options
 775       )
 776       when is_binary(object) do
 777     with %Activity{data: data} <- Activity.get_by_ap_id(object) do
 778       activity
 779       |> Map.put("object", data)
 780       |> handle_incoming(options)
 781     else
 782       _e -> :error
 783     end
 784   end
 785
 786   def handle_incoming(
 787         %{"type" => "Block", "object" => blocked, "actor" => blocker, "id" => id} = _data,
 788         _options
 789       ) do
 790     with %User{local: true} = blocked = User.get_cached_by_ap_id(blocked),
 791          {:ok, %User{} = blocker} = User.get_or_fetch_by_ap_id(blocker),
 792          {:ok, activity} <- ActivityPub.block(blocker, blocked, id, false) do
 793       User.unfollow(blocker, blocked)
 794       User.block(blocker, blocked)
 795       {:ok, activity}
 796     else
 797       _e -> :error
 798     end
 799   end
 800
 801   def handle_incoming(
 802         %{
 803           "type" => "Move",
 804           "actor" => origin_actor,
 805           "object" => origin_actor,
 806           "target" => target_actor
 807         },
 808         _options
 809       ) do
 810     with %User{} = origin_user <- User.get_cached_by_ap_id(origin_actor),
 811          {:ok, %User{} = target_user} <- User.get_or_fetch_by_ap_id(target_actor),
 812          true <- origin_actor in target_user.also_known_as do
 813       ActivityPub.move(origin_user, target_user, false)
 814     else
 815       _e -> :error
 816     end
 817   end
 818
 819   def handle_incoming(_, _), do: :error
 820
 821   @spec get_obj_helper(String.t(), Keyword.t()) :: {:ok, Object.t()} | nil
 822   def get_obj_helper(id, options \\ []) do
 823     case Object.normalize(id, true, options) do
 824       %Object{} = object -> {:ok, object}
 825       _ -> nil
 826     end
 827   end
 828
 829   @spec get_embedded_obj_helper(String.t() | Object.t(), User.t()) :: {:ok, Object.t()} | nil
 830   def get_embedded_obj_helper(%{"attributedTo" => attributed_to, "id" => object_id} = data, %User{
 831         ap_id: ap_id
 832       })
 833       when attributed_to == ap_id do
 834     with {:ok, activity} <-
 835            handle_incoming(%{
 836              "type" => "Create",
 837              "to" => data["to"],
 838              "cc" => data["cc"],
 839              "actor" => attributed_to,
 840              "object" => data
 841            }) do
 842       {:ok, Object.normalize(activity)}
 843     else
 844       _ -> get_obj_helper(object_id)
 845     end
 846   end
 847
 848   def get_embedded_obj_helper(object_id, _) do
 849     get_obj_helper(object_id)
 850   end
 851
 852   def set_reply_to_uri(%{"inReplyTo" => in_reply_to} = object) when is_binary(in_reply_to) do
 853     with false <- String.starts_with?(in_reply_to, "http"),
 854          {:ok, %{data: replied_to_object}} <- get_obj_helper(in_reply_to) do
 855       Map.put(object, "inReplyTo", replied_to_object["external_url"] || in_reply_to)
 856     else
 857       _e -> object
 858     end
 859   end
 860
 861   def set_reply_to_uri(obj), do: obj
 862
 863   @doc """
 864   Serialized Mastodon-compatible `replies` collection containing _self-replies_.
 865   Based on Mastodon's ActivityPub::NoteSerializer#replies.
 866   """
 867   def set_replies(obj_data) do
 868     replies_uris =
 869       with limit when limit > 0 <-
 870              Pleroma.Config.get([:activitypub, :note_replies_output_limit], 0),
 871            %Object{} = object <- Object.get_cached_by_ap_id(obj_data["id"]) do
 872         object
 873         |> Object.self_replies()
 874         |> select([o], fragment("?->>'id'", o.data))
 875         |> limit(^limit)
 876         |> Repo.all()
 877       else
 878         _ -> []
 879       end
 880
 881     set_replies(obj_data, replies_uris)
 882   end
 883
 884   defp set_replies(obj, []) do
 885     obj
 886   end
 887
 888   defp set_replies(obj, replies_uris) do
 889     replies_collection = %{
 890       "type" => "Collection",
 891       "items" => replies_uris
 892     }
 893
 894     Map.merge(obj, %{"replies" => replies_collection})
 895   end
 896
 897   def replies(%{"replies" => %{"first" => %{"items" => items}}}) when not is_nil(items) do
 898     items
 899   end
 900
 901   def replies(%{"replies" => %{"items" => items}}) when not is_nil(items) do
 902     items
 903   end
 904
 905   def replies(_), do: []
 906
 907   # Prepares the object of an outgoing create activity.
 908   def prepare_object(object) do
 909     object
 910     |> set_sensitive
 911     |> add_hashtags
 912     |> add_mention_tags
 913     |> add_emoji_tags
 914     |> add_attributed_to
 915     |> prepare_attachments
 916     |> set_conversation
 917     |> set_reply_to_uri
 918     |> set_replies
 919     |> strip_internal_fields
 920     |> strip_internal_tags
 921     |> set_type
 922   end
 923
 924   #  @doc
 925   #  """
 926   #  internal -> Mastodon
 927   #  """
 928
 929   def prepare_outgoing(%{"type" => activity_type, "object" => object_id} = data)
 930       when activity_type in ["Create", "Listen"] do
 931     object =
 932       object_id
 933       |> Object.normalize()
 934       |> Map.get(:data)
 935       |> prepare_object
 936
 937     data =
 938       data
 939       |> Map.put("object", object)
 940       |> Map.merge(Utils.make_json_ld_header())
 941       |> Map.delete("bcc")
 942
 943     {:ok, data}
 944   end
 945
 946   def prepare_outgoing(%{"type" => "Announce", "actor" => ap_id, "object" => object_id} = data) do
 947     object =
 948       object_id
 949       |> Object.normalize()
 950
 951     data =
 952       if Visibility.is_private?(object) && object.data["actor"] == ap_id do
 953         data |> Map.put("object", object |> Map.get(:data) |> prepare_object)
 954       else
 955         data |> maybe_fix_object_url
 956       end
 957
 958     data =
 959       data
 960       |> strip_internal_fields
 961       |> Map.merge(Utils.make_json_ld_header())
 962       |> Map.delete("bcc")
 963
 964     {:ok, data}
 965   end
 966
 967   # Mastodon Accept/Reject requires a non-normalized object containing the actor URIs,
 968   # because of course it does.
 969   def prepare_outgoing(%{"type" => "Accept"} = data) do
 970     with follow_activity <- Activity.normalize(data["object"]) do
 971       object = %{
 972         "actor" => follow_activity.actor,
 973         "object" => follow_activity.data["object"],
 974         "id" => follow_activity.data["id"],
 975         "type" => "Follow"
 976       }
 977
 978       data =
 979         data
 980         |> Map.put("object", object)
 981         |> Map.merge(Utils.make_json_ld_header())
 982
 983       {:ok, data}
 984     end
 985   end
 986
 987   def prepare_outgoing(%{"type" => "Reject"} = data) do
 988     with follow_activity <- Activity.normalize(data["object"]) do
 989       object = %{
 990         "actor" => follow_activity.actor,
 991         "object" => follow_activity.data["object"],
 992         "id" => follow_activity.data["id"],
 993         "type" => "Follow"
 994       }
 995
 996       data =
 997         data
 998         |> Map.put("object", object)
 999         |> Map.merge(Utils.make_json_ld_header())
1000
1001       {:ok, data}
1002     end
1003   end
1004
1005   def prepare_outgoing(%{"type" => _type} = data) do
1006     data =
1007       data
1008       |> strip_internal_fields
1009       |> maybe_fix_object_url
1010       |> Map.merge(Utils.make_json_ld_header())
1011
1012     {:ok, data}
1013   end
1014
1015   def maybe_fix_object_url(%{"object" => object} = data) when is_binary(object) do
1016     with false <- String.starts_with?(object, "http"),
1017          {:fetch, {:ok, relative_object}} <- {:fetch, get_obj_helper(object)},
1018          %{data: %{"external_url" => external_url}} when not is_nil(external_url) <-
1019            relative_object do
1020       Map.put(data, "object", external_url)
1021     else
1022       {:fetch, e} ->
1023         Logger.error("Couldn't fetch #{object} #{inspect(e)}")
1024         data
1025
1026       _ ->
1027         data
1028     end
1029   end
1030
1031   def maybe_fix_object_url(data), do: data
1032
1033   def add_hashtags(object) do
1034     tags =
1035       (object["tag"] || [])
1036       |> Enum.map(fn
1037         # Expand internal representation tags into AS2 tags.
1038         tag when is_binary(tag) ->
1039           %{
1040             "href" => Pleroma.Web.Endpoint.url() <> "/tags/#{tag}",
1041             "name" => "##{tag}",
1042             "type" => "Hashtag"
1043           }
1044
1045         # Do not process tags which are already AS2 tag objects.
1046         tag when is_map(tag) ->
1047           tag
1048       end)
1049
1050     Map.put(object, "tag", tags)
1051   end
1052
1053   def add_mention_tags(object) do
1054     {enabled_receivers, disabled_receivers} = Utils.get_notified_from_object(object)
1055     potential_receivers = enabled_receivers ++ disabled_receivers
1056     mentions = Enum.map(potential_receivers, &build_mention_tag/1)
1057
1058     tags = object["tag"] || []
1059     Map.put(object, "tag", tags ++ mentions)
1060   end
1061
1062   defp build_mention_tag(%{ap_id: ap_id, nickname: nickname} = _) do
1063     %{"type" => "Mention", "href" => ap_id, "name" => "@#{nickname}"}
1064   end
1065
1066   def take_emoji_tags(%User{emoji: emoji}) do
1067     emoji
1068     |> Map.to_list()
1069     |> Enum.map(&build_emoji_tag/1)
1070   end
1071
1072   # TODO: we should probably send mtime instead of unix epoch time for updated
1073   def add_emoji_tags(%{"emoji" => emoji} = object) do
1074     tags = object["tag"] || []
1075
1076     out = Enum.map(emoji, &build_emoji_tag/1)
1077
1078     Map.put(object, "tag", tags ++ out)
1079   end
1080
1081   def add_emoji_tags(object), do: object
1082
1083   defp build_emoji_tag({name, url}) do
1084     %{
1085       "icon" => %{"url" => url, "type" => "Image"},
1086       "name" => ":" <> name <> ":",
1087       "type" => "Emoji",
1088       "updated" => "1970-01-01T00:00:00Z",
1089       "id" => url
1090     }
1091   end
1092
1093   def set_conversation(object) do
1094     Map.put(object, "conversation", object["context"])
1095   end
1096
1097   def set_sensitive(%{"sensitive" => true} = object) do
1098     object
1099   end
1100
1101   def set_sensitive(object) do
1102     tags = object["tag"] || []
1103     Map.put(object, "sensitive", "nsfw" in tags)
1104   end
1105
1106   def set_type(%{"type" => "Answer"} = object) do
1107     Map.put(object, "type", "Note")
1108   end
1109
1110   def set_type(object), do: object
1111
1112   def add_attributed_to(object) do
1113     attributed_to = object["attributedTo"] || object["actor"]
1114     Map.put(object, "attributedTo", attributed_to)
1115   end
1116
1117   # TODO: Revisit this
1118   def prepare_attachments(%{"type" => "ChatMessage"} = object), do: object
1119
1120   def prepare_attachments(object) do
1121     attachments =
1122       object
1123       |> Map.get("attachment", [])
1124       |> Enum.map(fn data ->
1125         [%{"mediaType" => media_type, "href" => href} | _] = data["url"]
1126
1127         %{
1128           "url" => href,
1129           "mediaType" => media_type,
1130           "name" => data["name"],
1131           "type" => "Document"
1132         }
1133       end)
1134
1135     Map.put(object, "attachment", attachments)
1136   end
1137
1138   def strip_internal_fields(object) do
1139     Map.drop(object, Pleroma.Constants.object_internal_fields())
1140   end
1141
1142   defp strip_internal_tags(%{"tag" => tags} = object) do
1143     tags = Enum.filter(tags, fn x -> is_map(x) end)
1144
1145     Map.put(object, "tag", tags)
1146   end
1147
1148   defp strip_internal_tags(object), do: object
1149
1150   def perform(:user_upgrade, user) do
1151     # we pass a fake user so that the followers collection is stripped away
1152     old_follower_address = User.ap_followers(%User{nickname: user.nickname})
1153
1154     from(
1155       a in Activity,
1156       where: ^old_follower_address in a.recipients,
1157       update: [
1158         set: [
1159           recipients:
1160             fragment(
1161               "array_replace(?,?,?)",
1162               a.recipients,
1163               ^old_follower_address,
1164               ^user.follower_address
1165             )
1166         ]
1167       ]
1168     )
1169     |> Repo.update_all([])
1170   end
1171
1172   def upgrade_user_from_ap_id(ap_id) do
1173     with %User{local: false} = user <- User.get_cached_by_ap_id(ap_id),
1174          {:ok, data} <- ActivityPub.fetch_and_prepare_user_from_ap_id(ap_id),
1175          {:ok, user} <- update_user(user, data) do
1176       TransmogrifierWorker.enqueue("user_upgrade", %{"user_id" => user.id})
1177       {:ok, user}
1178     else
1179       %User{} = user -> {:ok, user}
1180       e -> e
1181     end
1182   end
1183
1184   defp update_user(user, data) do
1185     user
1186     |> User.remote_user_changeset(data)
1187     |> User.update_and_set_cache()
1188   end
1189
1190   def maybe_fix_user_url(%{"url" => url} = data) when is_map(url) do
1191     Map.put(data, "url", url["href"])
1192   end
1193
1194   def maybe_fix_user_url(data), do: data
1195
1196   def maybe_fix_user_object(data), do: maybe_fix_user_url(data)
1197 end