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     statuses = TwitterAPI.fetch_user_statuses(user, params)
  49     conn
  50     |> json_reply(200, statuses |> Poison.encode!)
  51   end
  52
  53   def follow(%{assigns: %{user: user}} = conn, %{ "user_id" => followed_id }) do
  54     { :ok, user, follower, _activity } = TwitterAPI.follow(user, followed_id)
  55
  56     response = follower |> UserRepresenter.to_json(%{for: user})
  57
  58     conn
  59     |> json_reply(200, response)
  60   end
  61
  62   def unfollow(%{assigns: %{user: user}} = conn, %{ "user_id" => followed_id }) do
  63     { :ok, user, follower } = TwitterAPI.unfollow(user, followed_id)
  64
  65     response = follower |> UserRepresenter.to_json(%{for: user})
  66
  67     conn
  68     |> json_reply(200, response)
  69   end
  70
  71   def fetch_status(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
  72     response = TwitterAPI.fetch_status(user, id) |> Poison.encode!
  73
  74     conn
  75     |> json_reply(200, response)
  76   end
  77
  78   def fetch_conversation(%{assigns: %{user: user}} = conn, %{ "id" => id }) do
  79     id = String.to_integer(id)
  80     response = TwitterAPI.fetch_conversation(user, id) |> Poison.encode!
  81
  82     conn
  83     |> json_reply(200, response)
  84   end
  85
  86   def upload(conn, %{"media" => media}) do
  87     response = TwitterAPI.upload(media)
  88     conn
  89     |> put_resp_content_type("application/atom+xml")
  90     |> send_resp(200, response)
  91   end
  92
  93   def upload_json(conn, %{"media" => media}) do
  94     response = TwitterAPI.upload(media, "json")
  95     conn
  96     |> json_reply(200, response)
  97   end
  98
  99   def config(conn, _params) do
 100     response = %{
 101       site: %{
 102         name: Pleroma.Web.base_url,
 103         server: Pleroma.Web.base_url,
 104         textlimit: -1
 105       }
 106     }
 107     |> Poison.encode!
 108
 109     conn
 110     |> json_reply(200, response)
 111   end
 112
 113   def favorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 114     activity = Repo.get(Activity, id)
 115     {:ok, status} = TwitterAPI.favorite(user, activity)
 116     response = Poison.encode!(status)
 117
 118     conn
 119     |> json_reply(200, response)
 120   end
 121
 122   def unfavorite(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 123     activity = Repo.get(Activity, id)
 124     {:ok, status} = TwitterAPI.unfavorite(user, activity)
 125     response = Poison.encode!(status)
 126
 127     conn
 128     |> json_reply(200, response)
 129   end
 130
 131   def retweet(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 132     activity = Repo.get(Activity, id)
 133     {:ok, status} = TwitterAPI.retweet(user, activity)
 134     response = Poison.encode!(status)
 135
 136     conn
 137     |> json_reply(200, response)
 138   end
 139
 140   def register(conn, params) do
 141     with {:ok, user} <- TwitterAPI.register_user(params) do
 142       conn
 143       |> json_reply(200, Poison.encode!(user))
 144     else
 145       {:error, errors} ->
 146       conn
 147       |> json_reply(400, Poison.encode!(errors))
 148     end
 149   end
 150
 151   defp json_reply(conn, status, json) do
 152     conn
 153     |> put_resp_content_type("application/json")
 154     |> send_resp(status, json)
 155   end
 156 end