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

   1 defmodule Pleroma.Web.TwitterAPI.Controller do
   2   use Pleroma.Web, :controller
   3   alias Pleroma.Web.TwitterAPI.TwitterAPI
   4   alias Pleroma.Web.TwitterAPI.Representers.{UserRepresenter, ActivityRepresenter}
   5   alias Pleroma.{Repo, Activity}
   6   alias Pleroma.Web.ActivityPub.ActivityPub
   7
   8   def verify_credentials(%{assigns: %{user: user}} = conn, _params) do
   9     response = user |> UserRepresenter.to_json(%{for: user})
  10
  11     conn
  12     |> json_reply(200, response)
  13   end
  14
  15   def status_update(conn, %{"status" => ""} = _status_data) do
  16     empty_status_reply(conn)
  17   end
  18
  19   def status_update(%{assigns: %{user: user}} = conn, %{"status" => status_text} = status_data) do
  20     if status_text |> String.trim |> String.length != 0 do
  21       media_ids = extract_media_ids(status_data)
  22       {:ok, activity} = TwitterAPI.create_status(user, Map.put(status_data, "media_ids",  media_ids ))
  23       conn
  24       |> json_reply(200, ActivityRepresenter.to_json(activity, %{user: user}))
  25     else
  26       empty_status_reply(conn)
  27     end
  28   end
  29
  30   def status_update(conn, _status_data) do
  31     empty_status_reply(conn)
  32   end
  33
  34   defp empty_status_reply(conn) do
  35     bad_request_reply(conn, "Client must provide a 'status' parameter with a value.")
  36   end
  37
  38   defp extract_media_ids(status_data) do
  39     with media_ids when not is_nil(media_ids) <- status_data["media_ids"],
  40          split_ids <- String.split(media_ids, ","),
  41          clean_ids <- Enum.reject(split_ids, fn (id) -> String.length(id) == 0 end)
  42       do
  43         clean_ids
  44       else _e -> []
  45     end
  46   end
  47
  48   def public_timeline(%{assigns: %{user: user}} = conn, params) do
  49     statuses = TwitterAPI.fetch_public_statuses(user, params)
  50     {:ok, json} = Poison.encode(statuses)
  51
  52     conn
  53     |> json_reply(200, json)
  54   end
  55
  56   def friends_timeline(%{assigns: %{user: user}} = conn, params) do
  57     statuses = TwitterAPI.fetch_friend_statuses(user, params)
  58     {:ok, json} = Poison.encode(statuses)
  59
  60     conn
  61     |> json_reply(200, json)
  62   end
  63
  64   def user_timeline(%{assigns: %{user: user}} = conn, params) do
  65     case TwitterAPI.get_user(user, params) do
  66       {:ok, target_user} ->
  67         params = Map.merge(params, %{"actor_id" => target_user.ap_id})
  68         statuses  = TwitterAPI.fetch_user_statuses(user, params)
  69         conn
  70         |> json_reply(200, statuses |> Poison.encode!)
  71       {:error, msg} ->
  72         bad_request_reply(conn, msg)
  73     end
  74   end
  75
  76   def mentions_timeline(%{assigns: %{user: user}} = conn, params) do
  77     statuses = TwitterAPI.fetch_mentions(user, params)
  78     {:ok, json} = Poison.encode(statuses)
  79
  80     conn
  81     |> json_reply(200, json)
  82   end
  83
  84   def follow(%{assigns: %{user: user}} = conn, params) do
  85     case TwitterAPI.follow(user, params) do
  86       { :ok, user, followed, _activity } ->
  87         response = followed |> UserRepresenter.to_json(%{for: user})
  88         conn
  89         |> json_reply(200, response)
  90       { :error, msg } -> forbidden_json_reply(conn, msg)
  91     end
  92   end
  93
  94   def unfollow(%{assigns: %{user: user}} = conn, params) do
  95     case TwitterAPI.unfollow(user, params) do
  96       { :ok, user, unfollowed, } ->
  97         response = unfollowed |> UserRepresenter.to_json(%{for: user})
  98         conn
  99         |> json_reply(200, response)
 100       { :error, msg } -> forbidden_json_reply(conn, msg)
 101     end
 102   end
 103
 104   def fetch_status(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
 105     response = TwitterAPI.fetch_status(user, id) |> Poison.encode!
 106
 107     conn
 108     |> json_reply(200, response)
 109   end
 110
 111   def fetch_conversation(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
 112     id = String.to_integer(id)
 113     response = TwitterAPI.fetch_conversation(user, id) |> Poison.encode!
 114
 115     conn
 116     |> json_reply(200, response)
 117   end
 118
 119   def upload(conn, %{"media" => media}) do
 120     response = TwitterAPI.upload(media)
 121     conn
 122     |> put_resp_content_type("application/atom+xml")
 123     |> send_resp(200, response)
 124   end
 125
 126   def help_test(conn, _params) do
 127     conn |> json_reply(200, Poison.encode!("ok"))
 128   end
 129
 130   def upload_json(conn, %{"media" => media}) do
 131     response = TwitterAPI.upload(media, "json")
 132     conn
 133     |> json_reply(200, response)
 134   end
 135
 136   def config(conn, _params) do
 137     response = %{
 138       site: %{
 139         name: Pleroma.Web.base_url,
 140         server: Pleroma.Web.base_url,
 141         textlimit: -1
 142       }
 143     }
 144     |> Poison.encode!
 145
 146     conn
 147     |> json_reply(200, response)
 148   end
 149
 150   def favorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 151     activity = Repo.get(Activity, id)
 152     {:ok, status} = TwitterAPI.favorite(user, activity)
 153     response = Poison.encode!(status)
 154
 155     conn
 156     |> json_reply(200, response)
 157   end
 158
 159   def unfavorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 160     activity = Repo.get(Activity, id)
 161     {:ok, status} = TwitterAPI.unfavorite(user, activity)
 162     response = Poison.encode!(status)
 163
 164     conn
 165     |> json_reply(200, response)
 166   end
 167
 168   def retweet(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 169     activity = Repo.get(Activity, id)
 170     {:ok, status} = TwitterAPI.retweet(user, activity)
 171     response = Poison.encode!(status)
 172
 173     conn
 174     |> json_reply(200, response)
 175   end
 176
 177   def register(conn, params) do
 178     with {:ok, user} <- TwitterAPI.register_user(params) do
 179       conn
 180       |> json_reply(200, Poison.encode!(user))
 181     else
 182       {:error, errors} ->
 183       conn
 184       |> json_reply(400, Poison.encode!(errors))
 185     end
 186   end
 187
 188   def update_avatar(%{assigns: %{user: user}} = conn, params) do
 189     {:ok, object} = ActivityPub.upload(params)
 190     change = Ecto.Changeset.change(user, %{avatar: object.data})
 191     {:ok, user} = Repo.update(change)
 192
 193     response = UserRepresenter.to_map(user, %{for: user})
 194     |> Poison.encode!
 195
 196     conn
 197     |> json_reply(200, response)
 198   end
 199
 200   defp bad_request_reply(conn, error_message) do
 201     json = error_json(conn, error_message)
 202     json_reply(conn, 400, json)
 203   end
 204
 205   defp json_reply(conn, status, json) do
 206     conn
 207     |> put_resp_content_type("application/json")
 208     |> send_resp(status, json)
 209   end
 210
 211   defp forbidden_json_reply(conn, error_message) do
 212     json = error_json(conn, error_message)
 213     json_reply(conn, 403, json)
 214   end
 215
 216   defp error_json(conn, error_message) do
 217     %{"error" => error_message, "request" => conn.request_path} |> Poison.encode!
 218   end
 219 end