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   def custom_emojis(conn, _params) do
  65     mastodon_emoji = Pleroma.Formatter.get_custom_emoji()
  66     |> Enum.map(fn {shortcode, relative_url} ->
  67       url = to_string URI.merge(Web.base_url(), relative_url)
  68       %{
  69         "shortcode" => shortcode,
  70         "static_url" => url,
  71         "url" => url
  72       }
  73     end)
  74     json conn, mastodon_emoji
  75   end
  76
  77   defp add_link_headers(conn, method, activities) do
  78     last = List.last(activities)
  79     first = List.first(activities)
  80     if last do
  81       min = last.id
  82       max = first.id
  83       next_url = mastodon_api_url(Pleroma.Web.Endpoint, method, max_id: min)
  84       prev_url = mastodon_api_url(Pleroma.Web.Endpoint, method, since_id: max)
  85       conn
  86       |> put_resp_header("link", "<#{next_url}>; rel=\"next\", <#{prev_url}>; rel=\"prev\"")
  87     else
  88       conn
  89     end
  90   end
  91
  92   def home_timeline(%{assigns: %{user: user}} = conn, params) do
  93     params = params
  94     |> Map.put("type", ["Create", "Announce"])
  95     |> Map.put("blocking_user", user)
  96
  97     activities = ActivityPub.fetch_activities([user.ap_id | user.following], params)
  98     |> Enum.reverse
  99
 100     conn
 101     |> add_link_headers(:home_timeline, activities)
 102     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 103   end
 104
 105   def public_timeline(%{assigns: %{user: user}} = conn, params) do
 106     params = params
 107     |> Map.put("type", ["Create", "Announce"])
 108     |> Map.put("local_only", !!params["local"])
 109     |> Map.put("blocking_user", user)
 110
 111     activities = ActivityPub.fetch_public_activities(params)
 112     |> Enum.reverse
 113
 114     conn
 115     |> add_link_headers(:public_timeline, activities)
 116     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 117   end
 118
 119   # TODO: Link headers
 120   def user_statuses(%{assigns: %{user: user}} = conn, params) do
 121     with %User{ap_id: ap_id} <- Repo.get(User, params["id"]) do
 122       params = params
 123       |> Map.put("type", ["Create", "Announce"])
 124       |> Map.put("actor_id", ap_id)
 125       |> Map.put("whole_db", true)
 126
 127       activities = ActivityPub.fetch_activities([], params)
 128       |> Enum.reverse
 129
 130       render conn, StatusView, "index.json", %{activities: activities, for: user, as: :activity}
 131     end
 132   end
 133
 134   def get_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 135     with %Activity{} = activity <- Repo.get(Activity, id) do
 136       render conn, StatusView, "status.json", %{activity: activity, for: user}
 137     end
 138   end
 139
 140   def get_context(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 141     with %Activity{} = activity <- Repo.get(Activity, id),
 142          activities <- ActivityPub.fetch_activities_for_context(activity.data["object"]["context"], %{"blocking_user" => user}),
 143          activities <- activities |> Enum.filter(fn (%{id: aid}) -> to_string(aid) != to_string(id) end),
 144          grouped_activities <- Enum.group_by(activities, fn (%{id: id}) -> id < activity.id end) do
 145       result = %{
 146         ancestors: StatusView.render("index.json", for: user, activities: grouped_activities[true] || [], as: :activity) |> Enum.reverse,
 147         descendants: StatusView.render("index.json", for: user, activities: grouped_activities[false] || [], as: :activity) |> Enum.reverse,
 148       }
 149
 150       json(conn, result)
 151     end
 152   end
 153
 154   def post_status(%{assigns: %{user: user}} = conn, %{"status" => status} = params) do
 155     params = params
 156     |> Map.put("in_reply_to_status_id", params["in_reply_to_id"])
 157
 158     {:ok, activity} = CommonAPI.post(user, params)
 159     render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 160   end
 161
 162   def delete_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 163     with {:ok, %Activity{}} <- CommonAPI.delete(id, user) do
 164       json(conn, %{})
 165     else
 166       _e ->
 167         conn
 168         |> put_status(403)
 169         |> json(%{error: "Can't delete this post"})
 170     end
 171   end
 172
 173   def reblog_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 174     with {:ok, _announce, %{data: %{"id" => id}}} = CommonAPI.repeat(ap_id_or_id, user),
 175          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 176       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 177     end
 178   end
 179
 180   def fav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 181     with {:ok, _fav, %{data: %{"id" => id}}} = CommonAPI.favorite(ap_id_or_id, user),
 182          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 183       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 184     end
 185   end
 186
 187   def unfav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 188     with {:ok, %{data: %{"id" => id}}} = CommonAPI.unfavorite(ap_id_or_id, user),
 189          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 190       render conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity}
 191     end
 192   end
 193
 194   def notifications(%{assigns: %{user: user}} = conn, params) do
 195     notifications = Notification.for_user(user, params)
 196     result = Enum.map(notifications, fn (%{id: id, activity: activity, inserted_at: created_at}) ->
 197       actor = User.get_cached_by_ap_id(activity.data["actor"])
 198       created_at = NaiveDateTime.to_iso8601(created_at)
 199       |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
 200       case activity.data["type"] do
 201         "Create" ->
 202           %{id: id, type: "mention", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: activity, for: user})}
 203         "Like" ->
 204           liked_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 205           %{id: id, type: "favourite", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: liked_activity, for: user})}
 206         "Announce" ->
 207           announced_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
 208           %{id: id, type: "reblog", created_at: created_at, account: AccountView.render("account.json", %{user: actor}), status: StatusView.render("status.json", %{activity: announced_activity, for: user})}
 209         "Follow" ->
 210           %{id: id, type: "follow", created_at: created_at, account: AccountView.render("account.json", %{user: actor})}
 211         _ -> nil
 212       end
 213     end)
 214     |> Enum.filter(&(&1))
 215
 216     conn
 217     |> add_link_headers(:notifications, notifications)
 218     |> json(result)
 219   end
 220
 221   def relationships(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 222     id = List.wrap(id)
 223     q = from u in User,
 224       where: u.id in ^id
 225     targets = Repo.all(q)
 226     render conn, AccountView, "relationships.json", %{user: user, targets: targets}
 227   end
 228
 229   def upload(%{assigns: %{user: user}} = conn, %{"file" => file}) do
 230     with {:ok, object} <- ActivityPub.upload(file) do
 231       data = object.data
 232       |> Map.put("id", object.id)
 233
 234       render conn, StatusView, "attachment.json", %{attachment: data}
 235     end
 236   end
 237
 238   def favourited_by(conn, %{"id" => id}) do
 239     with %Activity{data: %{"object" => %{"likes" => likes} = data}} <- Repo.get(Activity, id) do
 240       q = from u in User,
 241         where: u.ap_id in ^likes
 242       users = Repo.all(q)
 243       render conn, AccountView, "accounts.json", %{users: users, as: :user}
 244     else
 245       _ -> json(conn, [])
 246     end
 247   end
 248
 249   def reblogged_by(conn, %{"id" => id}) do
 250     with %Activity{data: %{"object" => %{"announcements" => announces}}} <- Repo.get(Activity, id) do
 251       q = from u in User,
 252         where: u.ap_id in ^announces
 253       users = Repo.all(q)
 254       render conn, AccountView, "accounts.json", %{users: users, as: :user}
 255     else
 256       _ -> json(conn, [])
 257     end
 258   end
 259
 260   # TODO: Link headers
 261   def hashtag_timeline(%{assigns: %{user: user}} = conn, params) do
 262     params = params
 263     |> Map.put("type", "Create")
 264     |> Map.put("local_only", !!params["local"])
 265     |> Map.put("blocking_user", user)
 266
 267     activities = ActivityPub.fetch_public_activities(params)
 268     |> Enum.reverse
 269
 270     conn
 271     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 272   end
 273
 274   # TODO: Pagination
 275   def followers(conn, %{"id" => id}) do
 276     with %User{} = user <- Repo.get(User, id),
 277          {:ok, followers} <- User.get_followers(user) do
 278       render conn, AccountView, "accounts.json", %{users: followers, as: :user}
 279     end
 280   end
 281
 282   def following(conn, %{"id" => id}) do
 283     with %User{} = user <- Repo.get(User, id),
 284          {:ok, followers} <- User.get_friends(user) do
 285       render conn, AccountView, "accounts.json", %{users: followers, as: :user}
 286     end
 287   end
 288
 289   def follow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 290     with %User{} = followed <- Repo.get(User, id),
 291          {:ok, follower} <- User.follow(follower, followed),
 292          {:ok, activity} <- ActivityPub.follow(follower, followed) do
 293       render conn, AccountView, "relationship.json", %{user: follower, target: followed}
 294     else
 295       {:error, message} = err ->
 296         conn
 297         |> put_resp_content_type("application/json")
 298         |> send_resp(403, Poison.encode!(%{"error" => message}))
 299     end
 300   end
 301
 302   def follow(%{assigns: %{user: follower}} = conn, %{"uri" => uri}) do
 303     with %User{} = followed <- Repo.get_by(User, nickname: uri),
 304          {:ok, follower} <- User.follow(follower, followed),
 305          {:ok, activity} <- ActivityPub.follow(follower, followed) do
 306       render conn, AccountView, "account.json", %{user: followed}
 307     else
 308       {:error, message} = err ->
 309         conn
 310         |> put_resp_content_type("application/json")
 311         |> send_resp(403, Poison.encode!(%{"error" => message}))
 312     end
 313   end
 314
 315   # TODO: Clean up and unify
 316   def unfollow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 317     with %User{} = followed <- Repo.get(User, id),
 318          { :ok, follower, follow_activity } <- User.unfollow(follower, followed),
 319          { :ok, _activity } <- ActivityPub.insert(%{
 320            "type" => "Undo",
 321            "actor" => follower.ap_id,
 322            "object" => follow_activity.data["id"] # get latest Follow for these users
 323          }) do
 324       render conn, AccountView, "relationship.json", %{user: follower, target: followed}
 325     end
 326   end
 327
 328   def block(%{assigns: %{user: blocker}} = conn, %{"id" => id}) do
 329     with %User{} = blocked <- Repo.get(User, id),
 330          {:ok, blocker} <- User.block(blocker, blocked) do
 331       render conn, AccountView, "relationship.json", %{user: blocker, target: blocked}
 332     else
 333       {:error, message} = err ->
 334         conn
 335         |> put_resp_content_type("application/json")
 336         |> send_resp(403, Poison.encode!(%{"error" => message}))
 337     end
 338   end
 339
 340   def unblock(%{assigns: %{user: blocker}} = conn, %{"id" => id}) do
 341     with %User{} = blocked <- Repo.get(User, id),
 342          {:ok, blocker} <- User.unblock(blocker, blocked) do
 343       render conn, AccountView, "relationship.json", %{user: blocker, target: blocked}
 344     else
 345       {:error, message} = err ->
 346         conn
 347         |> put_resp_content_type("application/json")
 348         |> send_resp(403, Poison.encode!(%{"error" => message}))
 349     end
 350   end
 351
 352   # TODO: Use proper query
 353   def blocks(%{assigns: %{user: user}} = conn, _) do
 354     with blocked_users <- user.info["blocks"] || [],
 355          accounts <- Enum.map(blocked_users, fn (ap_id) -> User.get_cached_by_ap_id(ap_id) end) do
 356       res = AccountView.render("accounts.json", users: accounts, for: user, as: :user)
 357       json(conn, res)
 358     end
 359   end
 360
 361   def search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 362     accounts = User.search(query, params["resolve"] == "true")
 363
 364     q = from a in Activity,
 365       where: fragment("?->>'type' = 'Create'", a.data),
 366       where: fragment("to_tsvector('english', ?->'object'->>'content') @@ plainto_tsquery('english', ?)", a.data, ^query),
 367       limit: 20
 368     statuses = Repo.all(q)
 369
 370     res = %{
 371       "accounts" => AccountView.render("accounts.json", users: accounts, for: user, as: :user),
 372       "statuses" => StatusView.render("index.json", activities: statuses, for: user, as: :activity),
 373       "hashtags" => []
 374     }
 375
 376     json(conn, res)
 377   end
 378
 379   def account_search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 380     accounts = User.search(query, params["resolve"] == "true")
 381
 382     res = AccountView.render("accounts.json", users: accounts, for: user, as: :user)
 383
 384     json(conn, res)
 385   end
 386
 387   def favourites(%{assigns: %{user: user}} = conn, params) do
 388     params = conn
 389     |> Map.put("type", "Create")
 390     |> Map.put("favorited_by", user.ap_id)
 391     |> Map.put("blocking_user", user)
 392
 393     activities = ActivityPub.fetch_activities([], params)
 394     |> Enum.reverse
 395
 396     conn
 397     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 398   end
 399
 400   def relationship_noop(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 401     Logger.debug("Unimplemented, returning unmodified relationship")
 402     with %User{} = target <- Repo.get(User, id) do
 403       render conn, AccountView, "relationship.json", %{user: user, target: target}
 404     end
 405   end
 406
 407   def empty_array(conn, _) do
 408     Logger.debug("Unimplemented, returning an empty array")
 409     json(conn, [])
 410   end
 411 end