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   @instance Application.get_env(:pleroma, :instance)
  43
  44   def masto_instance(conn, _params) do
  45     response = %{
  46       uri: Web.base_url,
  47       title: Keyword.get(@instance, :name),
  48       description: "A Pleroma instance, an alternative fediverse server",
  49       version: Keyword.get(@instance, :version),
  50       email: Keyword.get(@instance, :email),
  51       urls: %{
  52         streaming_api: String.replace(Web.base_url, ["http","https"], "wss")
  53       },
  54       stats: %{
  55         user_count: 1,
  56         status_count: 2,
  57         domain_count: 3
  58       }
  59     }
  60
  61     json(conn, response)
  62   end
  63
  64   defp add_link_headers(conn, method, activities) do
  65     last = List.last(activities)
  66     first = List.first(activities)
  67     if last do
  68       min = last.id
  69       max = first.id
  70       next_url = mastodon_api_url(Pleroma.Web.Endpoint, method, max_id: min)
  71       prev_url = mastodon_api_url(Pleroma.Web.Endpoint, method, since_id: max)
  72       conn
  73       |> put_resp_header("link", "<#{next_url}>; rel=\"next\", <#{prev_url}>; rel=\"prev\"")
  74     else
  75       conn
  76     end
  77   end
  78
  79   def home_timeline(%{assigns: %{user: user}} = conn, params) do
  80     activities = ActivityPub.fetch_activities([user.ap_id | user.following], Map.put(params, "type", "Create"))
  81     |> Enum.reverse
  82
  83     conn
  84     |> add_link_headers(:home_timeline, activities)
  85     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
  86   end
  87
  88   def public_timeline(%{assigns: %{user: user}} = conn, params) do
  89     params = params
  90     |> Map.put("type", "Create")
  91     |> Map.put("local_only", !!params["local"])
  92
  93     activities = ActivityPub.fetch_public_activities(params)
  94     |> Enum.reverse
  95
  96     conn
  97     |> add_link_headers(:public_timeline, activities)
  98     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
  99   end
 100
 101   # TODO: Link headers
 102   def user_statuses(%{assigns: %{user: user}} = conn, params) do
 103     with %User{ap_id: ap_id} <- Repo.get(User, params["id"]) do
 104       params = params
 105       |> Map.put("type", "Create")
 106       |> Map.put("actor_id", ap_id)
 107
 108       activities = ActivityPub.fetch_activities([], params)
 109       |> Enum.reverse
 110
 111       render conn, StatusView, "index.json", %{activities: activities, for: user, as: :activity}
 112     end
 113   end
 114
 115   def get_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 116     with %Activity{} = activity <- Repo.get(Activity, id) do
 117       render conn, StatusView, "status.json", %{activity: activity, for: user}
 118     end
 119   end
 120
 121   def get_context(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 122     with %Activity{} = activity <- Repo.get(Activity, id),
 123          activities <- ActivityPub.fetch_activities_for_context(activity.data["object"]["context"]),
 124          activities <- activities |> Enum.filter(fn (%{id: aid}) -> to_string(aid) != to_string(id) end),
 125          grouped_activities <- Enum.group_by(activities, fn (%{id: id}) -> id < activity.id end) do
 126       result = %{
 127         ancestors: StatusView.render("index.json", for: user, activities: grouped_activities[true] || [], as: :activity) |> Enum.reverse,
 128         descendants: StatusView.render("index.json", for: user, activities: grouped_activities[false] || [], as: :activity) |> Enum.reverse,
 129       }
 130
 131       json(conn, result)
 132     end
 133   end
 134
 135   def post_status(%{assigns: %{user: user}} = conn, %{"status" => status} = params) do
 136     l = status |> String.trim |> String.length
 137
 138     params = params
 139     |> Map.put("in_reply_to_status_id", params["in_reply_to_id"])
 140
 141     if l > 0 && l < 5000 do
 142       {:ok, activity} = TwitterAPI.create_status(user, params)
 143       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 144     end
 145   end
 146
 147   def delete_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 148     with {:ok, %Activity{}} <- CommonAPI.delete(id, user) do
 149       json(conn, %{})
 150     else
 151       _e ->
 152         conn
 153         |> put_status(403)
 154         |> json(%{error: "Can't delete this post"})
 155     end
 156   end
 157
 158   def reblog_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 159     with {:ok, _announce, %{data: %{"id" => id}}} = CommonAPI.repeat(ap_id_or_id, user),
 160          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 161       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 162     end
 163   end
 164
 165   def fav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 166     with {:ok, _fav, %{data: %{"id" => id}}} = CommonAPI.favorite(ap_id_or_id, user),
 167          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 168       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 169     end
 170   end
 171
 172   def unfav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 173     with {:ok, %{data: %{"id" => id}}} = CommonAPI.unfavorite(ap_id_or_id, user),
 174          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 175       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 176     end
 177   end
 178
 179   def notifications(%{assigns: %{user: user}} = conn, params) do
 180     notifications = Notification.for_user(user, params)
 181     result = Enum.map(notifications, fn (%{id: id, activity: activity, inserted_at: created_at}) ->
 182       actor = User.get_cached_by_ap_id(activity.data["actor"])
 183       created_at = NaiveDateTime.to_iso8601(created_at)
 184       |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
 185       case activity.data["type"] do
 186         "Create" ->
 187           %{id: id, type: "mention", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: activity})}
 188         "Like" ->
 189           liked_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 190           %{id: id, type: "favourite", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: liked_activity})}
 191         "Announce" ->
 192           announced_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 193           %{id: id, type: "reblog", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: announced_activity})}
 194         "Follow" ->
 195           %{id: id, type: "follow", created_at: created_at, account: AccountView.render("account.json", %{user: actor})}
 196         _ -> nil
 197       end
 198     end)
 199     |> Enum.filter(&(&1))
 200
 201     conn
 202     |> add_link_headers(:notifications, notifications)
 203     |> json(result)
 204   end
 205
 206   def relationships(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 207     id = List.wrap(id)
 208     q = from u in User,
 209       where: u.id in ^id
 210     targets = Repo.all(q)
 211     render conn, AccountView, "relationships.json", %{user: user, targets: targets}
 212   end
 213
 214   def upload(%{assigns: %{user: user}} = conn, %{"file" => file}) do
 215     with {:ok, object} <- ActivityPub.upload(file) do
 216       data = object.data
 217       |> Map.put("id", object.id)
 218
 219       render conn, StatusView, "attachment.json", %{attachment: data}
 220     end
 221   end
 222
 223   def favourited_by(conn, %{"id" => id}) do
 224     with %Activity{data: %{"object" => %{"likes" => likes} = data}} <- Repo.get(Activity, id) do
 225       q = from u in User,
 226         where: u.ap_id in ^likes
 227       users = Repo.all(q)
 228       render conn, AccountView, "accounts.json", %{users: users, as: :user}
 229     else
 230       _ -> json(conn, [])
 231     end
 232   end
 233
 234   def reblogged_by(conn, %{"id" => id}) do
 235     with %Activity{data: %{"object" => %{"announcements" => announces}}} <- Repo.get(Activity, id) do
 236       q = from u in User,
 237         where: u.ap_id in ^announces
 238       users = Repo.all(q)
 239       render conn, AccountView, "accounts.json", %{users: users, as: :user}
 240     else
 241       _ -> json(conn, [])
 242     end
 243   end
 244
 245   # TODO: Link headers
 246   def hashtag_timeline(%{assigns: %{user: user}} = conn, params) do
 247     params = params
 248     |> Map.put("type", "Create")
 249     |> Map.put("local_only", !!params["local"])
 250
 251     activities = ActivityPub.fetch_public_activities(params)
 252     |> Enum.reverse
 253
 254     conn
 255     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 256   end
 257
 258   # TODO: Pagination
 259   def followers(conn, %{"id" => id}) do
 260     with %User{} = user <- Repo.get(User, id),
 261          {:ok, followers} <- User.get_followers(user) do
 262       render conn, AccountView, "accounts.json", %{users: followers, as: :user}
 263     end
 264   end
 265
 266   def following(conn, %{"id" => id}) do
 267     with %User{} = user <- Repo.get(User, id),
 268          {:ok, followers} <- User.get_friends(user) do
 269       render conn, AccountView, "accounts.json", %{users: followers, as: :user}
 270     end
 271   end
 272
 273   def follow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 274     with %User{} = followed <- Repo.get(User, id),
 275        {:ok, follower} <- User.follow(follower, followed),
 276        {:ok, activity} <- ActivityPub.follow(follower, followed) do
 277       render conn, AccountView, "relationship.json", %{user: follower, target: followed}
 278     end
 279   end
 280
 281   # TODO: Clean up and unify
 282   def unfollow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 283     with %User{} = followed <- Repo.get(User, id),
 284          { :ok, follower, follow_activity } <- User.unfollow(follower, followed),
 285          { :ok, _activity } <- ActivityPub.insert(%{
 286            "type" => "Undo",
 287            "actor" => follower.ap_id,
 288            "object" => follow_activity.data["id"] # get latest Follow for these users
 289          }) do
 290       render conn, AccountView, "relationship.json", %{user: follower, target: followed}
 291     end
 292   end
 293
 294   def relationship_noop(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 295     Logger.debug("Unimplemented, returning unmodified relationship")
 296     with %User{} = target <- Repo.get(User, id) do
 297       render conn, AccountView, "relationship.json", %{user: user, target: target}
 298     end
 299   end
 300
 301   def empty_array(conn, _) do
 302     Logger.debug("Unimplemented, returning an empty array")
 303     json(conn, [])
 304   end
 305 end