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

   1 defmodule Pleroma.Web.MastodonAPI.MastodonAPIController do
   2   use Pleroma.Web, :controller
   3   alias Pleroma.{Repo, Activity, User, Notification}
   4   alias Pleroma.Web.OAuth.App
   5   alias Pleroma.Web
   6   alias Pleroma.Web.MastodonAPI.{StatusView, AccountView}
   7   alias Pleroma.Web.ActivityPub.ActivityPub
   8   alias Pleroma.Web.TwitterAPI.TwitterAPI
   9   alias Pleroma.Web.CommonAPI
  10   import Ecto.Query
  11   import Logger
  12
  13   def create_app(conn, params) do
  14     with cs <- App.register_changeset(%App{}, params) |> IO.inspect,
  15          {:ok, app} <- Repo.insert(cs) |> IO.inspect do
  16       res = %{
  17         id: app.id,
  18         client_id: app.client_id,
  19         client_secret: app.client_secret
  20       }
  21
  22       json(conn, res)
  23     end
  24   end
  25
  26   def verify_credentials(%{assigns: %{user: user}} = conn, params) do
  27     account = AccountView.render("account.json", %{user: user})
  28     json(conn, account)
  29   end
  30
  31   def user(conn, %{"id" => id}) do
  32     with %User{} = user <- Repo.get(User, id) do
  33       account = AccountView.render("account.json", %{user: user})
  34       json(conn, account)
  35     else
  36       _e -> conn
  37       |> put_status(404)
  38       |> json(%{error: "Can't find user"})
  39     end
  40   end
  41
  42   def masto_instance(conn, _params) do
  43     response = %{
  44       uri: Web.base_url,
  45       title: Web.base_url,
  46       description: "A Pleroma instance, an alternative fediverse server",
  47       version: "Pleroma Dev"
  48     }
  49
  50     json(conn, response)
  51   end
  52
  53   defp add_link_headers(conn, method, activities) do
  54     last = List.last(activities)
  55     first = List.first(activities)
  56     if last do
  57       min = last.id
  58       max = first.id
  59       next_url = mastodon_api_url(Pleroma.Web.Endpoint, method, max_id: min)
  60       prev_url = mastodon_api_url(Pleroma.Web.Endpoint, method, since_id: max)
  61       conn
  62       |> put_resp_header("link", "<#{next_url}>; rel=\"next\", <#{prev_url}>; rel=\"prev\"")
  63     else
  64       conn
  65     end
  66   end
  67
  68   def home_timeline(%{assigns: %{user: user}} = conn, params) do
  69     activities = ActivityPub.fetch_activities([user.ap_id | user.following], Map.put(params, "type", "Create"))
  70     |> Enum.reverse
  71
  72     conn
  73     |> add_link_headers(:home_timeline, activities)
  74     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
  75   end
  76
  77   def public_timeline(%{assigns: %{user: user}} = conn, params) do
  78     params = params
  79     |> Map.put("type", "Create")
  80     |> Map.put("local_only", !!params["local"])
  81
  82     activities = ActivityPub.fetch_public_activities(params)
  83     |> Enum.reverse
  84
  85     conn
  86     |> add_link_headers(:public_timeline, activities)
  87     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
  88   end
  89
  90   def user_statuses(%{assigns: %{user: user}} = conn, params) do
  91     with %User{ap_id: ap_id} <- Repo.get(User, params["id"]) do
  92       params = params
  93       |> Map.put("type", "Create")
  94       |> Map.put("actor_id", ap_id)
  95
  96       activities = ActivityPub.fetch_activities([], params)
  97       |> Enum.reverse
  98
  99       render conn, StatusView, "index.json", %{activities: activities, for: user, as: :activity}
 100     end
 101   end
 102
 103   def get_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 104     with %Activity{} = activity <- Repo.get(Activity, id) do
 105       render conn, StatusView, "status.json", %{activity: activity, for: user}
 106     end
 107   end
 108
 109   def get_context(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 110     with %Activity{} = activity <- Repo.get(Activity, id),
 111          activities <- ActivityPub.fetch_activities_for_context(activity.data["object"]["context"]),
 112          activities <- activities |> Enum.filter(fn (%{id: aid}) -> to_string(aid) != to_string(id) end),
 113          grouped_activities <- Enum.group_by(activities, fn (%{id: id}) -> id < activity.id end) do
 114       result = %{
 115         ancestors: StatusView.render("index.json", for: user, activities: grouped_activities[true] || [], as: :activity) |> Enum.reverse,
 116         descendants: StatusView.render("index.json", for: user, activities: grouped_activities[false] || [], as: :activity) |> Enum.reverse,
 117       }
 118
 119       json(conn, result)
 120     end
 121   end
 122
 123   def post_status(%{assigns: %{user: user}} = conn, %{"status" => status} = params) do
 124     l = status |> String.trim |> String.length
 125
 126     params = params
 127     |> Map.put("in_reply_to_status_id", params["in_reply_to_id"])
 128
 129     if l > 0 && l < 5000 do
 130       {:ok, activity} = TwitterAPI.create_status(user, params)
 131       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 132     end
 133   end
 134
 135   def delete_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 136     with {:ok, %Activity{}} <- CommonAPI.delete(id, user) do
 137       json(conn, %{})
 138     else
 139       _e ->
 140         conn
 141         |> put_status(403)
 142         |> json(%{error: "Can't delete this post"})
 143     end
 144   end
 145
 146   def reblog_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 147     with {:ok, _announce, %{data: %{"id" => id}}} = CommonAPI.repeat(ap_id_or_id, user),
 148          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 149       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 150     end
 151   end
 152
 153   def fav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 154     with {:ok, _fav, %{data: %{"id" => id}}} = CommonAPI.favorite(ap_id_or_id, user),
 155          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 156       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 157     end
 158   end
 159
 160   def unfav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 161     with {:ok, %{data: %{"id" => id}}} = CommonAPI.unfavorite(ap_id_or_id, user),
 162          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 163       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 164     end
 165   end
 166
 167   def notifications(%{assigns: %{user: user}} = conn, params) do
 168     notifications = Notification.for_user(user, params)
 169     result = Enum.map(notifications, fn (%{id: id, activity: activity, inserted_at: created_at}) ->
 170       actor = User.get_cached_by_ap_id(activity.data["actor"])
 171       created_at = NaiveDateTime.to_iso8601(created_at)
 172       |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
 173       case activity.data["type"] do
 174         "Create" ->
 175           %{id: id, type: "mention", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: activity})}
 176         "Like" ->
 177           liked_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 178           %{id: id, type: "favourite", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: liked_activity})}
 179         "Announce" ->
 180           announced_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 181           %{id: id, type: "reblog", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: announced_activity})}
 182         "Follow" ->
 183           %{id: id, type: "follow", created_at: created_at, account: AccountView.render("account.json", %{user: actor})}
 184         _ -> nil
 185       end
 186     end)
 187     |> Enum.filter(&(&1))
 188
 189     conn
 190     |> add_link_headers(:notifications, notifications)
 191     |> json(result)
 192   end
 193
 194   def relationships(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 195     id = List.wrap(id)
 196     q = from u in User,
 197       where: u.id in ^id
 198     targets = Repo.all(q)
 199     render conn, AccountView, "relationships.json", %{user: user, targets: targets}
 200   end
 201
 202   def empty_array(conn, _) do
 203     Logger.debug("Unimplemented, returning an empty array")
 204     json(conn, [])
 205   end
 206 end