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