Apply suggestion to lib/pleroma/web/activity_pub/transmogrifier.ex
[akkoma] / lib / pleroma / web / activity_pub / side_effects.ex
1 defmodule Pleroma.Web.ActivityPub.SideEffects do
2 @moduledoc """
3 This module looks at an inserted object and executes the side effects that it
4 implies. For example, a `Like` activity will increase the like count on the
5 liked object, a `Follow` activity will add the user to the follower
6 collection, and so on.
7 """
8 alias Pleroma.Chat
9 alias Pleroma.Notification
10 alias Pleroma.Object
11 alias Pleroma.Repo
12 alias Pleroma.User
13 alias Pleroma.Web.ActivityPub.Pipeline
14 alias Pleroma.Web.ActivityPub.Utils
15
16 def handle(object, meta \\ [])
17
18 # Tasks this handles:
19 # - Add like to object
20 # - Set up notification
21 def handle(%{data: %{"type" => "Like"}} = object, meta) do
22 liked_object = Object.get_by_ap_id(object.data["object"])
23 Utils.add_like_to_object(object, liked_object)
24
25 Notification.create_notifications(object)
26
27 {:ok, object, meta}
28 end
29
30 # Tasks this handles
31 # - Actually create object
32 # - Rollback if we couldn't create it
33 # - Set up notifications
34 def handle(%{data: %{"type" => "Create"}} = activity, meta) do
35 with {:ok, _object, _meta} <- handle_object_creation(meta[:object_data], meta) do
36 Notification.create_notifications(activity)
37 {:ok, activity, meta}
38 else
39 e -> Repo.rollback(e)
40 end
41 end
42
43 # Nothing to do
44 def handle(object, meta) do
45 {:ok, object, meta}
46 end
47
48 def handle_object_creation(%{"type" => "ChatMessage"} = object, meta) do
49 with {:ok, object, meta} <- Pipeline.common_pipeline(object, meta) do
50 actor = User.get_cached_by_ap_id(object.data["actor"])
51 recipient = User.get_cached_by_ap_id(hd(object.data["to"]))
52
53 [[actor, recipient], [recipient, actor]]
54 |> Enum.each(fn [user, other_user] ->
55 if user.local do
56 Chat.bump_or_create(user.id, other_user.ap_id)
57 end
58 end)
59
60 {:ok, object, meta}
61 end
62 end
63
64 # Nothing to do
65 def handle_object_creation(object) do
66 {:ok, object}
67 end
68 end