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
   7   def verify_credentials(%{assigns: %{user: user}} = conn, _params) do
   8     response = user |> UserRepresenter.to_json(%{for: user})
   9
  10     conn
  11     |> json_reply(200, response)
  12   end
  13
  14   def status_update(%{assigns: %{user: user}} = conn, status_data) do
  15     media_ids = extract_media_ids(status_data)
  16     {:ok, activity} = TwitterAPI.create_status(user, Map.put(status_data, "media_ids",  media_ids ))
  17     conn
  18     |> json_reply(200, ActivityRepresenter.to_json(activity, %{user: user}))
  19   end
  20
  21   defp extract_media_ids(status_data) do
  22     with media_ids when not is_nil(media_ids) <- status_data["media_ids"],
  23          split_ids <- String.split(media_ids, ","),
  24          clean_ids <- Enum.reject(split_ids, fn (id) -> String.length(id) == 0 end)
  25       do
  26         clean_ids
  27       else _e -> []
  28     end
  29   end
  30
  31   def public_timeline(%{assigns: %{user: user}} = conn, params) do
  32     statuses = TwitterAPI.fetch_public_statuses(user, params)
  33     {:ok, json} = Poison.encode(statuses)
  34
  35     conn
  36     |> json_reply(200, json)
  37   end
  38
  39   def friends_timeline(%{assigns: %{user: user}} = conn, params) do
  40     statuses = TwitterAPI.fetch_friend_statuses(user, params)
  41     {:ok, json} = Poison.encode(statuses)
  42
  43     conn
  44     |> json_reply(200, json)
  45   end
  46
  47   def user_timeline(%{assigns: %{user: user}} = conn, params) do
  48     case TwitterAPI.get_user(user, params) do
  49       {:ok, target_user} ->
  50         params = Map.merge(params, %{"actor_id" => target_user.ap_id})
  51         statuses  = TwitterAPI.fetch_user_statuses(user, params)
  52         conn
  53         |> json_reply(200, statuses |> Poison.encode!)
  54       {:error, msg} ->
  55         bad_request_reply(conn, msg)
  56     end
  57   end
  58
  59   def follow(%{assigns: %{user: user}} = conn, %{ "user_id" => followed_id }) do
  60     { :ok, user, follower, _activity } = TwitterAPI.follow(user, followed_id)
  61
  62     response = follower |> UserRepresenter.to_json(%{for: user})
  63
  64     conn
  65     |> json_reply(200, response)
  66   end
  67
  68   def unfollow(%{assigns: %{user: user}} = conn, %{ "user_id" => followed_id }) do
  69     { :ok, user, follower } = TwitterAPI.unfollow(user, followed_id)
  70
  71     response = follower |> UserRepresenter.to_json(%{for: user})
  72
  73     conn
  74     |> json_reply(200, response)
  75   end
  76
  77   def fetch_status(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
  78     response = TwitterAPI.fetch_status(user, id) |> Poison.encode!
  79
  80     conn
  81     |> json_reply(200, response)
  82   end
  83
  84   def fetch_conversation(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
  85     id = String.to_integer(id)
  86     response = TwitterAPI.fetch_conversation(user, id) |> Poison.encode!
  87
  88     conn
  89     |> json_reply(200, response)
  90   end
  91
  92   def upload(conn, %{"media" => media}) do
  93     response = TwitterAPI.upload(media)
  94     conn
  95     |> put_resp_content_type("application/atom+xml")
  96     |> send_resp(200, response)
  97   end
  98
  99   def upload_json(conn, %{"media" => media}) do
 100     response = TwitterAPI.upload(media, "json")
 101     conn
 102     |> json_reply(200, response)
 103   end
 104
 105   def config(conn, _params) do
 106     response = %{
 107       site: %{
 108         name: Pleroma.Web.base_url,
 109         server: Pleroma.Web.base_url,
 110         textlimit: -1
 111       }
 112     }
 113     |> Poison.encode!
 114
 115     conn
 116     |> json_reply(200, response)
 117   end
 118
 119   def favorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 120     activity = Repo.get(Activity, id)
 121     {:ok, status} = TwitterAPI.favorite(user, activity)
 122     response = Poison.encode!(status)
 123
 124     conn
 125     |> json_reply(200, response)
 126   end
 127
 128   def unfavorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 129     activity = Repo.get(Activity, id)
 130     {:ok, status} = TwitterAPI.unfavorite(user, activity)
 131     response = Poison.encode!(status)
 132
 133     conn
 134     |> json_reply(200, response)
 135   end
 136
 137   def retweet(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 138     activity = Repo.get(Activity, id)
 139     {:ok, status} = TwitterAPI.retweet(user, activity)
 140     response = Poison.encode!(status)
 141
 142     conn
 143     |> json_reply(200, response)
 144   end
 145
 146   def register(conn, params) do
 147     with {:ok, user} <- TwitterAPI.register_user(params) do
 148       conn
 149       |> json_reply(200, Poison.encode!(user))
 150     else
 151       {:error, errors} ->
 152       conn
 153       |> json_reply(400, Poison.encode!(errors))
 154     end
 155   end
 156
 157   defp bad_request_reply(conn, error_message) do
 158     json = Poison.encode!(%{"error" => error_message})
 159     json_reply(conn, 400, json)
 160   end
 161
 162   defp json_reply(conn, status, json) do
 163     conn
 164     |> put_resp_content_type("application/json")
 165     |> send_resp(status, json)
 166   end
 167 end