Add Follow Activity representer

[akkoma] / lib / pleroma / web / twitter_api / representers / activity_representer.ex

diff --git a/lib/pleroma/web/twitter_api/representers/activity_representer.ex b/lib/pleroma/web/twitter_api/representers/activity_representer.ex
index bcb819200f93ba5308a292c7c1b44b586c790cf5..5fe0df359f0a772d07bc61cf3f7f84a97bbd3f15 100644 (file)
--- a/lib/pleroma/web/twitter_api/representers/activity_representer.ex
+++ b/lib/pleroma/web/twitter_api/representers/activity_representer.ex
@@ -3,9 +3,31 @@ defmodule Pleroma.Web.TwitterAPI.Representers.ActivityRepresenter do
   alias Pleroma.Web.TwitterAPI.Representers.{UserRepresenter, ObjectRepresenter}
   alias Pleroma.Activity
 
+  def to_map(%Activity{data: %{"type" => "Follow"}} = activity, %{user: user} = opts) do
+    %{
+      "id" => activity.id,
+      "user" => UserRepresenter.to_map(user, opts),
+      "attentions" => [],
+      "statusnet_html" => "",  # TODO: add summary
+      "text" => "",
+      "is_local" => true,
+      "is_post_verb" => false,
+      "created_at" => get_in(activity.data, ["published"]),
+      "in_reply_to_status_id" => nil,
+    }
+  end
+
   def to_map(%Activity{} = activity, %{user: user} = opts) do
     content = get_in(activity.data, ["object", "content"])
     published = get_in(activity.data, ["object", "published"])
+
+    mentions = opts[:mentioned] || []
+
+    attentions = activity.data["to"]
+    |> Enum.map(fn (ap_id) -> Enum.find(mentions, fn(user) -> ap_id == user.ap_id end) end)
+    |> Enum.filter(&(&1))
+    |> Enum.map(fn (user) -> UserRepresenter.to_map(user, opts) end)
+
     %{
       "id" => activity.id,
       "user" => UserRepresenter.to_map(user, opts),
@@ -17,7 +39,8 @@ defmodule Pleroma.Web.TwitterAPI.Representers.ActivityRepresenter do
       "created_at" => published,
       "in_reply_to_status_id" => activity.data["object"]["inReplyToStatusId"],
       "statusnet_conversation_id" => activity.data["object"]["statusnetConversationId"],
-      "attachments" => (activity.data["attachment"] || []) |> ObjectRepresenter.enum_to_list(opts)
+      "attachments" => (activity.data["object"]["attachment"] || []) |> ObjectRepresenter.enum_to_list(opts),
+      "attentions" => attentions
     }
   end
 end