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

   1 defmodule Pleroma.Web.TwitterAPI.TwitterAPI do
   2   alias Pleroma.{User, Activity, Repo, Object}
   3   alias Pleroma.Web.ActivityPub.ActivityPub
   4   alias Pleroma.Web.TwitterAPI.Representers.ActivityRepresenter
   5
   6   import Ecto.Query
   7
   8   def create_status(user = %User{}, data = %{}) do
   9     date = DateTime.utc_now() |> DateTime.to_iso8601
  10
  11     attachments = Enum.map(data["media_ids"] || [], fn (media_id) ->
  12       Repo.get(Object, media_id).data
  13     end)
  14
  15     context = ActivityPub.generate_context_id
  16     activity = %{
  17       "type" => "Create",
  18       "to" => [
  19         User.ap_followers(user),
  20         "https://www.w3.org/ns/activitystreams#Public"
  21       ],
  22       "actor" => User.ap_id(user),
  23       "object" => %{
  24         "type" => "Note",
  25         "content" => data["status"],
  26         "published" => date,
  27         "context" => context,
  28         "attachment" => attachments
  29       },
  30       "published" => date,
  31       "context" => context
  32     }
  33
  34     # Wire up reply info.
  35     activity = with inReplyToId when not is_nil(inReplyToId) <- data["in_reply_to_status_id"],
  36                     inReplyTo <- Repo.get(Activity, inReplyToId),
  37                     context <- inReplyTo.data["context"]
  38                do
  39                activity
  40                |> put_in(["context"], context)
  41                |> put_in(["object", "context"], context)
  42                |> put_in(["object", "inReplyTo"], inReplyTo.data["object"]["id"])
  43                |> put_in(["object", "inReplyToStatusId"], inReplyToId)
  44                |> put_in(["statusnetConversationId"], inReplyTo.data["statusnetConversationId"])
  45                |> put_in(["object", "statusnetConversationId"], inReplyTo.data["statusnetConversationId"])
  46                else _e ->
  47                  activity
  48                end
  49
  50     with {:ok, activity} <- ActivityPub.insert(activity) do
  51       add_conversation_id(activity)
  52     end
  53   end
  54
  55   def fetch_friend_statuses(user, opts \\ %{}) do
  56     ActivityPub.fetch_activities(user.following, opts)
  57     |> activities_to_statuses(%{for: user})
  58   end
  59
  60   def fetch_public_statuses(user, opts \\ %{}) do
  61     ActivityPub.fetch_public_activities(opts)
  62     |> activities_to_statuses(%{for: user})
  63   end
  64
  65   def fetch_conversation(user, id) do
  66     query = from activity in Activity,
  67       where: fragment("? @> ?", activity.data, ^%{ statusnetConversationId: id}),
  68       limit: 1
  69
  70     with %Activity{} = activity <- Repo.one(query),
  71          context <- activity.data["context"],
  72          activities <- ActivityPub.fetch_activities_for_context(context),
  73          statuses <- activities |> activities_to_statuses(%{for: user})
  74     do
  75       statuses
  76     else e ->
  77         IO.inspect(e)
  78       []
  79     end
  80   end
  81
  82   def fetch_status(user, id) do
  83     with %Activity{} = activity <- Repo.get(Activity, id) do
  84       activity_to_status(activity, %{for: user})
  85     end
  86   end
  87
  88   def follow(%User{} = follower, followed_id) do
  89     with %User{} = followed <- Repo.get(User, followed_id),
  90          { :ok, follower } <- User.follow(follower, followed)
  91     do
  92       { :ok, follower, followed }
  93     end
  94   end
  95
  96   def unfollow(%User{} = follower, followed_id) do
  97     with %User{} = followed <- Repo.get(User, followed_id),
  98          { :ok, follower } <- User.unfollow(follower, followed)
  99     do
 100       { :ok, follower, followed }
 101     end
 102   end
 103
 104   def upload(%Plug.Upload{} = file) do
 105     {:ok, object} = ActivityPub.upload(file)
 106
 107     url = List.first(object.data["url"])
 108     href = url["href"]
 109     type = url["mediaType"]
 110
 111     # Fake this as good as possible...
 112     """
 113     <?xml version="1.0" encoding="UTF-8"?>
 114     <rsp stat="ok" xmlns:atom="http://www.w3.org/2005/Atom">
 115       <mediaid>#{object.id}</mediaid>
 116       <media_id>#{object.id}</media_id>
 117       <media_id_string>#{object.id}</media_id_string>
 118       <media_url>#{href}</media_url>
 119       <mediaurl>#{href}</mediaurl>
 120       <atom:link rel="enclosure" href="#{href}" type="#{type}"></atom:link>
 121     </rsp>
 122     """
 123   end
 124
 125   defp add_conversation_id(activity) do
 126     if is_integer(activity.data["statusnetConversationId"]) do
 127       {:ok, activity}
 128     else
 129       data = activity.data
 130       |> put_in(["object", "statusnetConversationId"], activity.id)
 131       |> put_in(["statusnetConversationId"], activity.id)
 132
 133       changeset = Ecto.Changeset.change(activity, data: data)
 134       Repo.update(changeset)
 135     end
 136   end
 137
 138   defp activities_to_statuses(activities, opts) do
 139     Enum.map(activities, fn(activity) ->
 140       activity_to_status(activity, opts)
 141     end)
 142   end
 143
 144   defp activity_to_status(activity, opts) do
 145     actor = get_in(activity.data, ["actor"])
 146     user = Repo.get_by!(User, ap_id: actor)
 147     ActivityRepresenter.to_map(activity, Map.merge(opts, %{user: user}))
 148   end
 149 end