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, Object, Activity, User, Notification, Stats}
   4   alias Pleroma.Web
   5   alias Pleroma.Web.MastodonAPI.{StatusView, AccountView, MastodonView, ListView, FilterView}
   6   alias Pleroma.Web.ActivityPub.ActivityPub
   7   alias Pleroma.Web.ActivityPub.Utils
   8   alias Pleroma.Web.CommonAPI
   9   alias Pleroma.Web.OAuth.{Authorization, Token, App}
  10   alias Pleroma.Web.MediaProxy
  11   alias Comeonin.Pbkdf2
  12   import Ecto.Query
  13   require Logger
  14
  15   @httpoison Application.get_env(:pleroma, :httpoison)
  16
  17   action_fallback(:errors)
  18
  19   def create_app(conn, params) do
  20     with cs <- App.register_changeset(%App{}, params) |> IO.inspect(),
  21          {:ok, app} <- Repo.insert(cs) |> IO.inspect() do
  22       res = %{
  23         id: app.id |> to_string,
  24         name: app.client_name,
  25         client_id: app.client_id,
  26         client_secret: app.client_secret,
  27         redirect_uri: app.redirect_uris,
  28         website: app.website
  29       }
  30
  31       json(conn, res)
  32     end
  33   end
  34
  35   def update_credentials(%{assigns: %{user: user}} = conn, params) do
  36     original_user = user
  37
  38     params =
  39       if bio = params["note"] do
  40         Map.put(params, "bio", bio)
  41       else
  42         params
  43       end
  44
  45     params =
  46       if name = params["display_name"] do
  47         Map.put(params, "name", name)
  48       else
  49         params
  50       end
  51
  52     user =
  53       if avatar = params["avatar"] do
  54         with %Plug.Upload{} <- avatar,
  55              {:ok, object} <- ActivityPub.upload(avatar),
  56              change = Ecto.Changeset.change(user, %{avatar: object.data}),
  57              {:ok, user} = User.update_and_set_cache(change) do
  58           user
  59         else
  60           _e -> user
  61         end
  62       else
  63         user
  64       end
  65
  66     user =
  67       if banner = params["header"] do
  68         with %Plug.Upload{} <- banner,
  69              {:ok, object} <- ActivityPub.upload(banner),
  70              new_info <- Map.put(user.info, "banner", object.data),
  71              change <- User.info_changeset(user, %{info: new_info}),
  72              {:ok, user} <- User.update_and_set_cache(change) do
  73           user
  74         else
  75           _e -> user
  76         end
  77       else
  78         user
  79       end
  80
  81     user =
  82       if locked = params["locked"] do
  83         with locked <- locked == "true",
  84              new_info <- Map.put(user.info, "locked", locked),
  85              change <- User.info_changeset(user, %{info: new_info}),
  86              {:ok, user} <- User.update_and_set_cache(change) do
  87           user
  88         else
  89           _e -> user
  90         end
  91       else
  92         user
  93       end
  94
  95     with changeset <- User.update_changeset(user, params),
  96          {:ok, user} <- User.update_and_set_cache(changeset) do
  97       if original_user != user do
  98         CommonAPI.update(user)
  99       end
 100
 101       json(conn, AccountView.render("account.json", %{user: user}))
 102     else
 103       _e ->
 104         conn
 105         |> put_status(403)
 106         |> json(%{error: "Invalid request"})
 107     end
 108   end
 109
 110   def verify_credentials(%{assigns: %{user: user}} = conn, _) do
 111     account = AccountView.render("account.json", %{user: user})
 112     json(conn, account)
 113   end
 114
 115   def user(conn, %{"id" => id}) do
 116     with %User{} = user <- Repo.get(User, id) do
 117       account = AccountView.render("account.json", %{user: user})
 118       json(conn, account)
 119     else
 120       _e ->
 121         conn
 122         |> put_status(404)
 123         |> json(%{error: "Can't find user"})
 124     end
 125   end
 126
 127   @instance Application.get_env(:pleroma, :instance)
 128   @mastodon_api_level "2.4.3"
 129
 130   def masto_instance(conn, _params) do
 131     response = %{
 132       uri: Web.base_url(),
 133       title: Keyword.get(@instance, :name),
 134       description: Keyword.get(@instance, :description),
 135       version: "#{@mastodon_api_level} (compatible; #{Keyword.get(@instance, :version)})",
 136       email: Keyword.get(@instance, :email),
 137       urls: %{
 138         streaming_api: String.replace(Pleroma.Web.Endpoint.static_url(), "http", "ws")
 139       },
 140       stats: Stats.get_stats(),
 141       thumbnail: Web.base_url() <> "/instance/thumbnail.jpeg",
 142       max_toot_chars: Keyword.get(@instance, :limit)
 143     }
 144
 145     json(conn, response)
 146   end
 147
 148   def peers(conn, _params) do
 149     json(conn, Stats.get_peers())
 150   end
 151
 152   defp mastodonized_emoji do
 153     Pleroma.Formatter.get_custom_emoji()
 154     |> Enum.map(fn {shortcode, relative_url} ->
 155       url = to_string(URI.merge(Web.base_url(), relative_url))
 156
 157       %{
 158         "shortcode" => shortcode,
 159         "static_url" => url,
 160         "visible_in_picker" => true,
 161         "url" => url
 162       }
 163     end)
 164   end
 165
 166   def custom_emojis(conn, _params) do
 167     mastodon_emoji = mastodonized_emoji()
 168     json(conn, mastodon_emoji)
 169   end
 170
 171   defp add_link_headers(conn, method, activities, param \\ nil, params \\ %{}) do
 172     last = List.last(activities)
 173     first = List.first(activities)
 174
 175     if last do
 176       min = last.id
 177       max = first.id
 178
 179       {next_url, prev_url} =
 180         if param do
 181           {
 182             mastodon_api_url(
 183               Pleroma.Web.Endpoint,
 184               method,
 185               param,
 186               Map.merge(params, %{max_id: min})
 187             ),
 188             mastodon_api_url(
 189               Pleroma.Web.Endpoint,
 190               method,
 191               param,
 192               Map.merge(params, %{since_id: max})
 193             )
 194           }
 195         else
 196           {
 197             mastodon_api_url(
 198               Pleroma.Web.Endpoint,
 199               method,
 200               Map.merge(params, %{max_id: min})
 201             ),
 202             mastodon_api_url(
 203               Pleroma.Web.Endpoint,
 204               method,
 205               Map.merge(params, %{since_id: max})
 206             )
 207           }
 208         end
 209
 210       conn
 211       |> put_resp_header("link", "<#{next_url}>; rel=\"next\", <#{prev_url}>; rel=\"prev\"")
 212     else
 213       conn
 214     end
 215   end
 216
 217   def home_timeline(%{assigns: %{user: user}} = conn, params) do
 218     params =
 219       params
 220       |> Map.put("type", ["Create", "Announce"])
 221       |> Map.put("blocking_user", user)
 222       |> Map.put("user", user)
 223
 224     activities =
 225       ActivityPub.fetch_activities([user.ap_id | user.following], params)
 226       |> Enum.reverse()
 227
 228     conn
 229     |> add_link_headers(:home_timeline, activities)
 230     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 231   end
 232
 233   def public_timeline(%{assigns: %{user: user}} = conn, params) do
 234     local_only = params["local"] in [true, "True", "true", "1"]
 235
 236     params =
 237       params
 238       |> Map.put("type", ["Create", "Announce"])
 239       |> Map.put("local_only", local_only)
 240       |> Map.put("blocking_user", user)
 241
 242     activities =
 243       ActivityPub.fetch_public_activities(params)
 244       |> Enum.reverse()
 245
 246     conn
 247     |> add_link_headers(:public_timeline, activities, false, %{"local" => local_only})
 248     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 249   end
 250
 251   def user_statuses(%{assigns: %{user: reading_user}} = conn, params) do
 252     with %User{} = user <- Repo.get(User, params["id"]) do
 253       # Since Pleroma has no "pinned" posts feature, we'll just set an empty list here
 254       activities =
 255         if params["pinned"] == "true" do
 256           []
 257         else
 258           ActivityPub.fetch_user_activities(user, reading_user, params)
 259         end
 260
 261       conn
 262       |> add_link_headers(:user_statuses, activities, params["id"])
 263       |> render(StatusView, "index.json", %{
 264         activities: activities,
 265         for: reading_user,
 266         as: :activity
 267       })
 268     end
 269   end
 270
 271   def dm_timeline(%{assigns: %{user: user}} = conn, _params) do
 272     query =
 273       ActivityPub.fetch_activities_query([user.ap_id], %{"type" => "Create", visibility: "direct"})
 274
 275     activities = Repo.all(query)
 276
 277     conn
 278     |> add_link_headers(:dm_timeline, activities)
 279     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 280   end
 281
 282   def get_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 283     with %Activity{} = activity <- Repo.get(Activity, id),
 284          true <- ActivityPub.visible_for_user?(activity, user) do
 285       render(conn, StatusView, "status.json", %{activity: activity, for: user})
 286     end
 287   end
 288
 289   def get_context(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 290     with %Activity{} = activity <- Repo.get(Activity, id),
 291          activities <-
 292            ActivityPub.fetch_activities_for_context(activity.data["context"], %{
 293              "blocking_user" => user,
 294              "user" => user
 295            }),
 296          activities <-
 297            activities |> Enum.filter(fn %{id: aid} -> to_string(aid) != to_string(id) end),
 298          activities <-
 299            activities |> Enum.filter(fn %{data: %{"type" => type}} -> type == "Create" end),
 300          grouped_activities <- Enum.group_by(activities, fn %{id: id} -> id < activity.id end) do
 301       result = %{
 302         ancestors:
 303           StatusView.render(
 304             "index.json",
 305             for: user,
 306             activities: grouped_activities[true] || [],
 307             as: :activity
 308           )
 309           |> Enum.reverse(),
 310         descendants:
 311           StatusView.render(
 312             "index.json",
 313             for: user,
 314             activities: grouped_activities[false] || [],
 315             as: :activity
 316           )
 317           |> Enum.reverse()
 318       }
 319
 320       json(conn, result)
 321     end
 322   end
 323
 324   def post_status(conn, %{"status" => "", "media_ids" => media_ids} = params)
 325       when length(media_ids) > 0 do
 326     params =
 327       params
 328       |> Map.put("status", ".")
 329
 330     post_status(conn, params)
 331   end
 332
 333   def post_status(%{assigns: %{user: user}} = conn, %{"status" => _} = params) do
 334     params =
 335       params
 336       |> Map.put("in_reply_to_status_id", params["in_reply_to_id"])
 337       |> Map.put("no_attachment_links", true)
 338
 339     idempotency_key =
 340       case get_req_header(conn, "idempotency-key") do
 341         [key] -> key
 342         _ -> Ecto.UUID.generate()
 343       end
 344
 345     {:ok, activity} =
 346       Cachex.fetch!(:idempotency_cache, idempotency_key, fn _ -> CommonAPI.post(user, params) end)
 347
 348     render(conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity})
 349   end
 350
 351   def delete_status(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 352     with {:ok, %Activity{}} <- CommonAPI.delete(id, user) do
 353       json(conn, %{})
 354     else
 355       _e ->
 356         conn
 357         |> put_status(403)
 358         |> json(%{error: "Can't delete this post"})
 359     end
 360   end
 361
 362   def reblog_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 363     with {:ok, announce, _activity} <- CommonAPI.repeat(ap_id_or_id, user) do
 364       render(conn, StatusView, "status.json", %{activity: announce, for: user, as: :activity})
 365     end
 366   end
 367
 368   def unreblog_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 369     with {:ok, _unannounce, %{data: %{"id" => id}}} <- CommonAPI.unrepeat(ap_id_or_id, user),
 370          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 371       render(conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity})
 372     end
 373   end
 374
 375   def fav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 376     with {:ok, _fav, %{data: %{"id" => id}}} <- CommonAPI.favorite(ap_id_or_id, user),
 377          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 378       render(conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity})
 379     end
 380   end
 381
 382   def unfav_status(%{assigns: %{user: user}} = conn, %{"id" => ap_id_or_id}) do
 383     with {:ok, _, _, %{data: %{"id" => id}}} <- CommonAPI.unfavorite(ap_id_or_id, user),
 384          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id) do
 385       render(conn, StatusView, "status.json", %{activity: activity, for: user, as: :activity})
 386     end
 387   end
 388
 389   def notifications(%{assigns: %{user: user}} = conn, params) do
 390     notifications = Notification.for_user(user, params)
 391
 392     result =
 393       Enum.map(notifications, fn x ->
 394         render_notification(user, x)
 395       end)
 396       |> Enum.filter(& &1)
 397
 398     conn
 399     |> add_link_headers(:notifications, notifications)
 400     |> json(result)
 401   end
 402
 403   def get_notification(%{assigns: %{user: user}} = conn, %{"id" => id} = _params) do
 404     with {:ok, notification} <- Notification.get(user, id) do
 405       json(conn, render_notification(user, notification))
 406     else
 407       {:error, reason} ->
 408         conn
 409         |> put_resp_content_type("application/json")
 410         |> send_resp(403, Jason.encode!(%{"error" => reason}))
 411     end
 412   end
 413
 414   def clear_notifications(%{assigns: %{user: user}} = conn, _params) do
 415     Notification.clear(user)
 416     json(conn, %{})
 417   end
 418
 419   def dismiss_notification(%{assigns: %{user: user}} = conn, %{"id" => id} = _params) do
 420     with {:ok, _notif} <- Notification.dismiss(user, id) do
 421       json(conn, %{})
 422     else
 423       {:error, reason} ->
 424         conn
 425         |> put_resp_content_type("application/json")
 426         |> send_resp(403, Jason.encode!(%{"error" => reason}))
 427     end
 428   end
 429
 430   def relationships(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 431     id = List.wrap(id)
 432     q = from(u in User, where: u.id in ^id)
 433     targets = Repo.all(q)
 434     render(conn, AccountView, "relationships.json", %{user: user, targets: targets})
 435   end
 436
 437   def update_media(%{assigns: %{user: _}} = conn, data) do
 438     with %Object{} = object <- Repo.get(Object, data["id"]),
 439          true <- is_binary(data["description"]),
 440          description <- data["description"] do
 441       new_data = %{object.data | "name" => description}
 442
 443       change = Object.change(object, %{data: new_data})
 444       {:ok, media_obj} = Repo.update(change)
 445
 446       data =
 447         new_data
 448         |> Map.put("id", object.id)
 449
 450       render(conn, StatusView, "attachment.json", %{attachment: data})
 451     end
 452   end
 453
 454   def upload(%{assigns: %{user: _}} = conn, %{"file" => file} = data) do
 455     with {:ok, object} <- ActivityPub.upload(file) do
 456       objdata =
 457         if Map.has_key?(data, "description") do
 458           Map.put(object.data, "name", data["description"])
 459         else
 460           object.data
 461         end
 462
 463       change = Object.change(object, %{data: objdata})
 464       {:ok, object} = Repo.update(change)
 465
 466       objdata =
 467         objdata
 468         |> Map.put("id", object.id)
 469
 470       render(conn, StatusView, "attachment.json", %{attachment: objdata})
 471     end
 472   end
 473
 474   def favourited_by(conn, %{"id" => id}) do
 475     with %Activity{data: %{"object" => %{"likes" => likes}}} <- Repo.get(Activity, id) do
 476       q = from(u in User, where: u.ap_id in ^likes)
 477       users = Repo.all(q)
 478       render(conn, AccountView, "accounts.json", %{users: users, as: :user})
 479     else
 480       _ -> json(conn, [])
 481     end
 482   end
 483
 484   def reblogged_by(conn, %{"id" => id}) do
 485     with %Activity{data: %{"object" => %{"announcements" => announces}}} <- Repo.get(Activity, id) do
 486       q = from(u in User, where: u.ap_id in ^announces)
 487       users = Repo.all(q)
 488       render(conn, AccountView, "accounts.json", %{users: users, as: :user})
 489     else
 490       _ -> json(conn, [])
 491     end
 492   end
 493
 494   def hashtag_timeline(%{assigns: %{user: user}} = conn, params) do
 495     local_only = params["local"] in [true, "True", "true", "1"]
 496
 497     params =
 498       params
 499       |> Map.put("type", "Create")
 500       |> Map.put("local_only", local_only)
 501       |> Map.put("blocking_user", user)
 502
 503     activities =
 504       ActivityPub.fetch_public_activities(params)
 505       |> Enum.reverse()
 506
 507     conn
 508     |> add_link_headers(:hashtag_timeline, activities, params["tag"], %{"local" => local_only})
 509     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 510   end
 511
 512   # TODO: Pagination
 513   def followers(conn, %{"id" => id}) do
 514     with %User{} = user <- Repo.get(User, id),
 515          {:ok, followers} <- User.get_followers(user) do
 516       render(conn, AccountView, "accounts.json", %{users: followers, as: :user})
 517     end
 518   end
 519
 520   def following(conn, %{"id" => id}) do
 521     with %User{} = user <- Repo.get(User, id),
 522          {:ok, followers} <- User.get_friends(user) do
 523       render(conn, AccountView, "accounts.json", %{users: followers, as: :user})
 524     end
 525   end
 526
 527   def follow_requests(%{assigns: %{user: followed}} = conn, _params) do
 528     with {:ok, follow_requests} <- User.get_follow_requests(followed) do
 529       render(conn, AccountView, "accounts.json", %{users: follow_requests, as: :user})
 530     end
 531   end
 532
 533   def authorize_follow_request(%{assigns: %{user: followed}} = conn, %{"id" => id}) do
 534     with %User{} = follower <- Repo.get(User, id),
 535          {:ok, follower} <- User.maybe_follow(follower, followed),
 536          %Activity{} = follow_activity <- Utils.fetch_latest_follow(follower, followed),
 537          {:ok, follow_activity} <- Utils.update_follow_state(follow_activity, "accept"),
 538          {:ok, _activity} <-
 539            ActivityPub.accept(%{
 540              to: [follower.ap_id],
 541              actor: followed.ap_id,
 542              object: follow_activity.data["id"],
 543              type: "Accept"
 544            }) do
 545       render(conn, AccountView, "relationship.json", %{user: followed, target: follower})
 546     else
 547       {:error, message} ->
 548         conn
 549         |> put_resp_content_type("application/json")
 550         |> send_resp(403, Jason.encode!(%{"error" => message}))
 551     end
 552   end
 553
 554   def reject_follow_request(%{assigns: %{user: followed}} = conn, %{"id" => id}) do
 555     with %User{} = follower <- Repo.get(User, id),
 556          %Activity{} = follow_activity <- Utils.fetch_latest_follow(follower, followed),
 557          {:ok, follow_activity} <- Utils.update_follow_state(follow_activity, "reject"),
 558          {:ok, _activity} <-
 559            ActivityPub.reject(%{
 560              to: [follower.ap_id],
 561              actor: followed.ap_id,
 562              object: follow_activity.data["id"],
 563              type: "Reject"
 564            }) do
 565       render(conn, AccountView, "relationship.json", %{user: followed, target: follower})
 566     else
 567       {:error, message} ->
 568         conn
 569         |> put_resp_content_type("application/json")
 570         |> send_resp(403, Jason.encode!(%{"error" => message}))
 571     end
 572   end
 573
 574   def follow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 575     with %User{} = followed <- Repo.get(User, id),
 576          {:ok, follower} <- User.maybe_direct_follow(follower, followed),
 577          {:ok, _activity} <- ActivityPub.follow(follower, followed) do
 578       render(conn, AccountView, "relationship.json", %{user: follower, target: followed})
 579     else
 580       {:error, message} ->
 581         conn
 582         |> put_resp_content_type("application/json")
 583         |> send_resp(403, Jason.encode!(%{"error" => message}))
 584     end
 585   end
 586
 587   def follow(%{assigns: %{user: follower}} = conn, %{"uri" => uri}) do
 588     with %User{} = followed <- Repo.get_by(User, nickname: uri),
 589          {:ok, follower} <- User.maybe_direct_follow(follower, followed),
 590          {:ok, _activity} <- ActivityPub.follow(follower, followed) do
 591       render(conn, AccountView, "account.json", %{user: followed})
 592     else
 593       {:error, message} ->
 594         conn
 595         |> put_resp_content_type("application/json")
 596         |> send_resp(403, Jason.encode!(%{"error" => message}))
 597     end
 598   end
 599
 600   def unfollow(%{assigns: %{user: follower}} = conn, %{"id" => id}) do
 601     with %User{} = followed <- Repo.get(User, id),
 602          {:ok, _activity} <- ActivityPub.unfollow(follower, followed),
 603          {:ok, follower, _} <- User.unfollow(follower, followed) do
 604       render(conn, AccountView, "relationship.json", %{user: follower, target: followed})
 605     end
 606   end
 607
 608   def block(%{assigns: %{user: blocker}} = conn, %{"id" => id}) do
 609     with %User{} = blocked <- Repo.get(User, id),
 610          {:ok, blocker} <- User.block(blocker, blocked),
 611          {:ok, _activity} <- ActivityPub.block(blocker, blocked) do
 612       render(conn, AccountView, "relationship.json", %{user: blocker, target: blocked})
 613     else
 614       {:error, message} ->
 615         conn
 616         |> put_resp_content_type("application/json")
 617         |> send_resp(403, Jason.encode!(%{"error" => message}))
 618     end
 619   end
 620
 621   def unblock(%{assigns: %{user: blocker}} = conn, %{"id" => id}) do
 622     with %User{} = blocked <- Repo.get(User, id),
 623          {:ok, blocker} <- User.unblock(blocker, blocked),
 624          {:ok, _activity} <- ActivityPub.unblock(blocker, blocked) do
 625       render(conn, AccountView, "relationship.json", %{user: blocker, target: blocked})
 626     else
 627       {:error, message} ->
 628         conn
 629         |> put_resp_content_type("application/json")
 630         |> send_resp(403, Jason.encode!(%{"error" => message}))
 631     end
 632   end
 633
 634   # TODO: Use proper query
 635   def blocks(%{assigns: %{user: user}} = conn, _) do
 636     with blocked_users <- user.info["blocks"] || [],
 637          accounts <- Enum.map(blocked_users, fn ap_id -> User.get_cached_by_ap_id(ap_id) end) do
 638       res = AccountView.render("accounts.json", users: accounts, for: user, as: :user)
 639       json(conn, res)
 640     end
 641   end
 642
 643   def domain_blocks(%{assigns: %{user: %{info: info}}} = conn, _) do
 644     json(conn, info["domain_blocks"] || [])
 645   end
 646
 647   def block_domain(%{assigns: %{user: blocker}} = conn, %{"domain" => domain}) do
 648     User.block_domain(blocker, domain)
 649     json(conn, %{})
 650   end
 651
 652   def unblock_domain(%{assigns: %{user: blocker}} = conn, %{"domain" => domain}) do
 653     User.unblock_domain(blocker, domain)
 654     json(conn, %{})
 655   end
 656
 657   def status_search(query) do
 658     fetched =
 659       if Regex.match?(~r/https?:/, query) do
 660         with {:ok, object} <- ActivityPub.fetch_object_from_id(query) do
 661           [Activity.get_create_activity_by_object_ap_id(object.data["id"])]
 662         else
 663           _e -> []
 664         end
 665       end || []
 666
 667     q =
 668       from(
 669         a in Activity,
 670         where: fragment("?->>'type' = 'Create'", a.data),
 671         where: "https://www.w3.org/ns/activitystreams#Public" in a.recipients,
 672         where:
 673           fragment(
 674             "to_tsvector('english', ?->'object'->>'content') @@ plainto_tsquery('english', ?)",
 675             a.data,
 676             ^query
 677           ),
 678         limit: 20,
 679         order_by: [desc: :id]
 680       )
 681
 682     Repo.all(q) ++ fetched
 683   end
 684
 685   def search2(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 686     accounts = User.search(query, params["resolve"] == "true")
 687
 688     statuses = status_search(query)
 689
 690     tags_path = Web.base_url() <> "/tag/"
 691
 692     tags =
 693       String.split(query)
 694       |> Enum.uniq()
 695       |> Enum.filter(fn tag -> String.starts_with?(tag, "#") end)
 696       |> Enum.map(fn tag -> String.slice(tag, 1..-1) end)
 697       |> Enum.map(fn tag -> %{name: tag, url: tags_path <> tag} end)
 698
 699     res = %{
 700       "accounts" => AccountView.render("accounts.json", users: accounts, for: user, as: :user),
 701       "statuses" =>
 702         StatusView.render("index.json", activities: statuses, for: user, as: :activity),
 703       "hashtags" => tags
 704     }
 705
 706     json(conn, res)
 707   end
 708
 709   def search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 710     accounts = User.search(query, params["resolve"] == "true")
 711
 712     statuses = status_search(query)
 713
 714     tags =
 715       String.split(query)
 716       |> Enum.uniq()
 717       |> Enum.filter(fn tag -> String.starts_with?(tag, "#") end)
 718       |> Enum.map(fn tag -> String.slice(tag, 1..-1) end)
 719
 720     res = %{
 721       "accounts" => AccountView.render("accounts.json", users: accounts, for: user, as: :user),
 722       "statuses" =>
 723         StatusView.render("index.json", activities: statuses, for: user, as: :activity),
 724       "hashtags" => tags
 725     }
 726
 727     json(conn, res)
 728   end
 729
 730   def account_search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 731     accounts = User.search(query, params["resolve"] == "true")
 732
 733     res = AccountView.render("accounts.json", users: accounts, for: user, as: :user)
 734
 735     json(conn, res)
 736   end
 737
 738   def favourites(%{assigns: %{user: user}} = conn, _) do
 739     params =
 740       %{}
 741       |> Map.put("type", "Create")
 742       |> Map.put("favorited_by", user.ap_id)
 743       |> Map.put("blocking_user", user)
 744
 745     activities =
 746       ActivityPub.fetch_public_activities(params)
 747       |> Enum.reverse()
 748
 749     conn
 750     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 751   end
 752
 753   def get_lists(%{assigns: %{user: user}} = conn, opts) do
 754     lists = Pleroma.List.for_user(user, opts)
 755     res = ListView.render("lists.json", lists: lists)
 756     json(conn, res)
 757   end
 758
 759   def get_list(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 760     with %Pleroma.List{} = list <- Pleroma.List.get(id, user) do
 761       res = ListView.render("list.json", list: list)
 762       json(conn, res)
 763     else
 764       _e -> json(conn, "error")
 765     end
 766   end
 767
 768   def delete_list(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 769     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 770          {:ok, _list} <- Pleroma.List.delete(list) do
 771       json(conn, %{})
 772     else
 773       _e ->
 774         json(conn, "error")
 775     end
 776   end
 777
 778   def create_list(%{assigns: %{user: user}} = conn, %{"title" => title}) do
 779     with {:ok, %Pleroma.List{} = list} <- Pleroma.List.create(title, user) do
 780       res = ListView.render("list.json", list: list)
 781       json(conn, res)
 782     end
 783   end
 784
 785   def add_to_list(%{assigns: %{user: user}} = conn, %{"id" => id, "account_ids" => accounts}) do
 786     accounts
 787     |> Enum.each(fn account_id ->
 788       with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 789            %User{} = followed <- Repo.get(User, account_id) do
 790         Pleroma.List.follow(list, followed)
 791       end
 792     end)
 793
 794     json(conn, %{})
 795   end
 796
 797   def remove_from_list(%{assigns: %{user: user}} = conn, %{"id" => id, "account_ids" => accounts}) do
 798     accounts
 799     |> Enum.each(fn account_id ->
 800       with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 801            %User{} = followed <- Repo.get(Pleroma.User, account_id) do
 802         Pleroma.List.unfollow(list, followed)
 803       end
 804     end)
 805
 806     json(conn, %{})
 807   end
 808
 809   def list_accounts(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 810     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 811          {:ok, users} = Pleroma.List.get_following(list) do
 812       render(conn, AccountView, "accounts.json", %{users: users, as: :user})
 813     end
 814   end
 815
 816   def rename_list(%{assigns: %{user: user}} = conn, %{"id" => id, "title" => title}) do
 817     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 818          {:ok, list} <- Pleroma.List.rename(list, title) do
 819       res = ListView.render("list.json", list: list)
 820       json(conn, res)
 821     else
 822       _e ->
 823         json(conn, "error")
 824     end
 825   end
 826
 827   def list_timeline(%{assigns: %{user: user}} = conn, %{"list_id" => id} = params) do
 828     with %Pleroma.List{title: title, following: following} <- Pleroma.List.get(id, user) do
 829       params =
 830         params
 831         |> Map.put("type", "Create")
 832         |> Map.put("blocking_user", user)
 833
 834       # we must filter the following list for the user to avoid leaking statuses the user
 835       # does not actually have permission to see (for more info, peruse security issue #270).
 836       following_to =
 837         following
 838         |> Enum.filter(fn x -> x in user.following end)
 839
 840       activities =
 841         ActivityPub.fetch_activities_bounded(following_to, following, params)
 842         |> Enum.reverse()
 843
 844       conn
 845       |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 846     else
 847       _e ->
 848         conn
 849         |> put_status(403)
 850         |> json(%{error: "Error."})
 851     end
 852   end
 853
 854   def index(%{assigns: %{user: user}} = conn, _params) do
 855     token =
 856       conn
 857       |> get_session(:oauth_token)
 858
 859     if user && token do
 860       mastodon_emoji = mastodonized_emoji()
 861       accounts = Map.put(%{}, user.id, AccountView.render("account.json", %{user: user}))
 862
 863       initial_state =
 864         %{
 865           meta: %{
 866             streaming_api_base_url:
 867               String.replace(Pleroma.Web.Endpoint.static_url(), "http", "ws"),
 868             access_token: token,
 869             locale: "en",
 870             domain: Pleroma.Web.Endpoint.host(),
 871             admin: "1",
 872             me: "#{user.id}",
 873             unfollow_modal: false,
 874             boost_modal: false,
 875             delete_modal: true,
 876             auto_play_gif: false,
 877             display_sensitive_media: false,
 878             reduce_motion: false,
 879             max_toot_chars: Keyword.get(@instance, :limit)
 880           },
 881           rights: %{
 882             delete_others_notice: !!user.info["is_moderator"]
 883           },
 884           compose: %{
 885             me: "#{user.id}",
 886             default_privacy: user.info["default_scope"] || "public",
 887             default_sensitive: false
 888           },
 889           media_attachments: %{
 890             accept_content_types: [
 891               ".jpg",
 892               ".jpeg",
 893               ".png",
 894               ".gif",
 895               ".webm",
 896               ".mp4",
 897               ".m4v",
 898               "image\/jpeg",
 899               "image\/png",
 900               "image\/gif",
 901               "video\/webm",
 902               "video\/mp4"
 903             ]
 904           },
 905           settings:
 906             Map.get(user.info, "settings") ||
 907               %{
 908                 onboarded: true,
 909                 home: %{
 910                   shows: %{
 911                     reblog: true,
 912                     reply: true
 913                   }
 914                 },
 915                 notifications: %{
 916                   alerts: %{
 917                     follow: true,
 918                     favourite: true,
 919                     reblog: true,
 920                     mention: true
 921                   },
 922                   shows: %{
 923                     follow: true,
 924                     favourite: true,
 925                     reblog: true,
 926                     mention: true
 927                   },
 928                   sounds: %{
 929                     follow: true,
 930                     favourite: true,
 931                     reblog: true,
 932                     mention: true
 933                   }
 934                 }
 935               },
 936           push_subscription: nil,
 937           accounts: accounts,
 938           custom_emojis: mastodon_emoji,
 939           char_limit: Keyword.get(@instance, :limit)
 940         }
 941         |> Jason.encode!()
 942
 943       conn
 944       |> put_layout(false)
 945       |> render(MastodonView, "index.html", %{initial_state: initial_state})
 946     else
 947       conn
 948       |> redirect(to: "/web/login")
 949     end
 950   end
 951
 952   def put_settings(%{assigns: %{user: user}} = conn, %{"data" => settings} = _params) do
 953     with new_info <- Map.put(user.info, "settings", settings),
 954          change <- User.info_changeset(user, %{info: new_info}),
 955          {:ok, _user} <- User.update_and_set_cache(change) do
 956       conn
 957       |> json(%{})
 958     else
 959       e ->
 960         conn
 961         |> json(%{error: inspect(e)})
 962     end
 963   end
 964
 965   def login(conn, _) do
 966     conn
 967     |> render(MastodonView, "login.html", %{error: false})
 968   end
 969
 970   defp get_or_make_app() do
 971     with %App{} = app <- Repo.get_by(App, client_name: "Mastodon-Local") do
 972       {:ok, app}
 973     else
 974       _e ->
 975         cs =
 976           App.register_changeset(%App{}, %{
 977             client_name: "Mastodon-Local",
 978             redirect_uris: ".",
 979             scopes: "read,write,follow"
 980           })
 981
 982         Repo.insert(cs)
 983     end
 984   end
 985
 986   def login_post(conn, %{"authorization" => %{"name" => name, "password" => password}}) do
 987     with %User{} = user <- User.get_by_nickname_or_email(name),
 988          true <- Pbkdf2.checkpw(password, user.password_hash),
 989          {:ok, app} <- get_or_make_app(),
 990          {:ok, auth} <- Authorization.create_authorization(app, user),
 991          {:ok, token} <- Token.exchange_token(app, auth) do
 992       conn
 993       |> put_session(:oauth_token, token.token)
 994       |> redirect(to: "/web/getting-started")
 995     else
 996       _e ->
 997         conn
 998         |> render(MastodonView, "login.html", %{error: "Wrong username or password"})
 999     end
1000   end
1001
1002   def logout(conn, _) do
1003     conn
1004     |> clear_session
1005     |> redirect(to: "/")
1006   end
1007
1008   def relationship_noop(%{assigns: %{user: user}} = conn, %{"id" => id}) do
1009     Logger.debug("Unimplemented, returning unmodified relationship")
1010
1011     with %User{} = target <- Repo.get(User, id) do
1012       render(conn, AccountView, "relationship.json", %{user: user, target: target})
1013     end
1014   end
1015
1016   def empty_array(conn, _) do
1017     Logger.debug("Unimplemented, returning an empty array")
1018     json(conn, [])
1019   end
1020
1021   def empty_object(conn, _) do
1022     Logger.debug("Unimplemented, returning an empty object")
1023     json(conn, %{})
1024   end
1025
1026   def render_notification(user, %{id: id, activity: activity, inserted_at: created_at} = _params) do
1027     actor = User.get_cached_by_ap_id(activity.data["actor"])
1028
1029     created_at =
1030       NaiveDateTime.to_iso8601(created_at)
1031       |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
1032
1033     id = id |> to_string
1034
1035     case activity.data["type"] do
1036       "Create" ->
1037         %{
1038           id: id,
1039           type: "mention",
1040           created_at: created_at,
1041           account: AccountView.render("account.json", %{user: actor}),
1042           status: StatusView.render("status.json", %{activity: activity, for: user})
1043         }
1044
1045       "Like" ->
1046         liked_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
1047
1048         %{
1049           id: id,
1050           type: "favourite",
1051           created_at: created_at,
1052           account: AccountView.render("account.json", %{user: actor}),
1053           status: StatusView.render("status.json", %{activity: liked_activity, for: user})
1054         }
1055
1056       "Announce" ->
1057         announced_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
1058
1059         %{
1060           id: id,
1061           type: "reblog",
1062           created_at: created_at,
1063           account: AccountView.render("account.json", %{user: actor}),
1064           status: StatusView.render("status.json", %{activity: announced_activity, for: user})
1065         }
1066
1067       "Follow" ->
1068         %{
1069           id: id,
1070           type: "follow",
1071           created_at: created_at,
1072           account: AccountView.render("account.json", %{user: actor})
1073         }
1074
1075       _ ->
1076         nil
1077     end
1078   end
1079
1080   def get_filters(%{assigns: %{user: user}} = conn, params) do
1081     filters = Pleroma.Filter.get_filters(user)
1082     res = FilterView.render("filters.json", filters: filters)
1083     json(conn, res)
1084   end
1085
1086   def create_filter(
1087         %{assigns: %{user: user}} = conn,
1088         %{"phrase" => phrase, "context" => context} = params
1089       ) do
1090     query = %Pleroma.Filter{
1091       user_id: user.id,
1092       phrase: phrase,
1093       context: context,
1094       hide: Map.get(params, "irreversible", nil),
1095       whole_word: Map.get(params, "boolean", true)
1096       # expires_at
1097     }
1098
1099     {:ok, response} = Pleroma.Filter.create(query)
1100     res = FilterView.render("filter.json", filter: response)
1101     json(conn, res)
1102   end
1103
1104   def get_filter(%{assigns: %{user: user}} = conn, %{"id" => filter_id} = params) do
1105     filter = Pleroma.Filter.get(filter_id, user)
1106     res = FilterView.render("filter.json", filter: filter)
1107     json(conn, res)
1108   end
1109
1110   def update_filter(
1111         %{assigns: %{user: user}} = conn,
1112         %{"phrase" => phrase, "context" => context, "id" => filter_id} = params
1113       ) do
1114     query = %Pleroma.Filter{
1115       user_id: user.id,
1116       filter_id: filter_id,
1117       phrase: phrase,
1118       context: context,
1119       hide: Map.get(params, "irreversible", nil),
1120       whole_word: Map.get(params, "boolean", true)
1121       # expires_at
1122     }
1123
1124     {:ok, response} = Pleroma.Filter.update(query)
1125     res = FilterView.render("filter.json", filter: response)
1126     json(conn, res)
1127   end
1128
1129   def delete_filter(%{assigns: %{user: user}} = conn, %{"id" => filter_id} = params) do
1130     query = %Pleroma.Filter{
1131       user_id: user.id,
1132       filter_id: filter_id
1133     }
1134
1135     {:ok, response} = Pleroma.Filter.delete(query)
1136     json(conn, %{})
1137   end
1138
1139   def errors(conn, _) do
1140     conn
1141     |> put_status(500)
1142     |> json("Something went wrong")
1143   end
1144
1145   @suggestions Application.get_env(:pleroma, :suggestions)
1146
1147   def suggestions(%{assigns: %{user: user}} = conn, _) do
1148     if Keyword.get(@suggestions, :enabled, false) do
1149       api = Keyword.get(@suggestions, :third_party_engine, "")
1150       timeout = Keyword.get(@suggestions, :timeout, 5000)
1151
1152       host =
1153         Application.get_env(:pleroma, Pleroma.Web.Endpoint)
1154         |> Keyword.get(:url)
1155         |> Keyword.get(:host)
1156
1157       user = user.nickname
1158       url = String.replace(api, "{{host}}", host) |> String.replace("{{user}}", user)
1159
1160       with {:ok, %{status_code: 200, body: body}} <-
1161              @httpoison.get(url, [], timeout: timeout, recv_timeout: timeout),
1162            {:ok, data} <- Jason.decode(body) do
1163         data2 =
1164           Enum.slice(data, 0, 40)
1165           |> Enum.map(fn x ->
1166             Map.put(
1167               x,
1168               "id",
1169               case User.get_or_fetch(x["acct"]) do
1170                 %{id: id} -> id
1171                 _ -> 0
1172               end
1173             )
1174           end)
1175           |> Enum.map(fn x ->
1176             Map.put(x, "avatar", MediaProxy.url(x["avatar"]))
1177           end)
1178           |> Enum.map(fn x ->
1179             Map.put(x, "avatar_static", MediaProxy.url(x["avatar_static"]))
1180           end)
1181
1182         conn
1183         |> json(data2)
1184       else
1185         e -> Logger.error("Could not retrieve suggestions at fetch #{url}, #{inspect(e)}")
1186       end
1187     else
1188       json(conn, [])
1189     end
1190   end
1191
1192   def filters(conn, _) do
1193     json(conn, [])
1194   end
1195 end