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}
   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.3.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 search2(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 658     accounts = User.search(query, params["resolve"] == "true")
 659
 660     fetched =
 661       if Regex.match?(~r/https?:/, query) do
 662         with {:ok, object} <- ActivityPub.fetch_object_from_id(query) do
 663           [Activity.get_create_activity_by_object_ap_id(object.data["id"])]
 664         else
 665           _e -> []
 666         end
 667       end || []
 668
 669     q =
 670       from(
 671         a in Activity,
 672         where: fragment("?->>'type' = 'Create'", a.data),
 673         where: "https://www.w3.org/ns/activitystreams#Public" in a.recipients,
 674         where:
 675           fragment(
 676             "to_tsvector('english', ?->'object'->>'content') @@ plainto_tsquery('english', ?)",
 677             a.data,
 678             ^query
 679           ),
 680         limit: 20,
 681         order_by: [desc: :id]
 682       )
 683
 684     statuses = Repo.all(q) ++ fetched
 685
 686     tags_path = Web.base_url() <> "/tag/"
 687
 688     tags =
 689       String.split(query)
 690       |> Enum.uniq()
 691       |> Enum.filter(fn tag -> String.starts_with?(tag, "#") end)
 692       |> Enum.map(fn tag -> String.slice(tag, 1..-1) end)
 693       |> Enum.map(fn tag -> %{name: tag, url: tags_path <> tag} end)
 694
 695     res = %{
 696       "accounts" => AccountView.render("accounts.json", users: accounts, for: user, as: :user),
 697       "statuses" =>
 698         StatusView.render("index.json", activities: statuses, for: user, as: :activity),
 699       "hashtags" => tags
 700     }
 701
 702     json(conn, res)
 703   end
 704
 705   def search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 706     accounts = User.search(query, params["resolve"] == "true")
 707
 708     fetched =
 709       if Regex.match?(~r/https?:/, query) do
 710         with {:ok, object} <- ActivityPub.fetch_object_from_id(query) do
 711           [Activity.get_create_activity_by_object_ap_id(object.data["id"])]
 712         else
 713           _e -> []
 714         end
 715       end || []
 716
 717     q =
 718       from(
 719         a in Activity,
 720         where: fragment("?->>'type' = 'Create'", a.data),
 721         where: "https://www.w3.org/ns/activitystreams#Public" in a.recipients,
 722         where:
 723           fragment(
 724             "to_tsvector('english', ?->'object'->>'content') @@ plainto_tsquery('english', ?)",
 725             a.data,
 726             ^query
 727           ),
 728         limit: 20,
 729         order_by: [desc: :id]
 730       )
 731
 732     statuses = Repo.all(q) ++ fetched
 733
 734     tags =
 735       String.split(query)
 736       |> Enum.uniq()
 737       |> Enum.filter(fn tag -> String.starts_with?(tag, "#") end)
 738       |> Enum.map(fn tag -> String.slice(tag, 1..-1) end)
 739
 740     res = %{
 741       "accounts" => AccountView.render("accounts.json", users: accounts, for: user, as: :user),
 742       "statuses" =>
 743         StatusView.render("index.json", activities: statuses, for: user, as: :activity),
 744       "hashtags" => tags
 745     }
 746
 747     json(conn, res)
 748   end
 749
 750   def account_search(%{assigns: %{user: user}} = conn, %{"q" => query} = params) do
 751     accounts = User.search(query, params["resolve"] == "true")
 752
 753     res = AccountView.render("accounts.json", users: accounts, for: user, as: :user)
 754
 755     json(conn, res)
 756   end
 757
 758   def favourites(%{assigns: %{user: user}} = conn, _) do
 759     params =
 760       %{}
 761       |> Map.put("type", "Create")
 762       |> Map.put("favorited_by", user.ap_id)
 763       |> Map.put("blocking_user", user)
 764
 765     activities =
 766       ActivityPub.fetch_public_activities(params)
 767       |> Enum.reverse()
 768
 769     conn
 770     |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 771   end
 772
 773   def get_lists(%{assigns: %{user: user}} = conn, opts) do
 774     lists = Pleroma.List.for_user(user, opts)
 775     res = ListView.render("lists.json", lists: lists)
 776     json(conn, res)
 777   end
 778
 779   def get_list(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 780     with %Pleroma.List{} = list <- Pleroma.List.get(id, user) do
 781       res = ListView.render("list.json", list: list)
 782       json(conn, res)
 783     else
 784       _e -> json(conn, "error")
 785     end
 786   end
 787
 788   def delete_list(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 789     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 790          {:ok, _list} <- Pleroma.List.delete(list) do
 791       json(conn, %{})
 792     else
 793       _e ->
 794         json(conn, "error")
 795     end
 796   end
 797
 798   def create_list(%{assigns: %{user: user}} = conn, %{"title" => title}) do
 799     with {:ok, %Pleroma.List{} = list} <- Pleroma.List.create(title, user) do
 800       res = ListView.render("list.json", list: list)
 801       json(conn, res)
 802     end
 803   end
 804
 805   def add_to_list(%{assigns: %{user: user}} = conn, %{"id" => id, "account_ids" => accounts}) do
 806     accounts
 807     |> Enum.each(fn account_id ->
 808       with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 809            %User{} = followed <- Repo.get(User, account_id) do
 810         Pleroma.List.follow(list, followed)
 811       end
 812     end)
 813
 814     json(conn, %{})
 815   end
 816
 817   def remove_from_list(%{assigns: %{user: user}} = conn, %{"id" => id, "account_ids" => accounts}) do
 818     accounts
 819     |> Enum.each(fn account_id ->
 820       with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 821            %User{} = followed <- Repo.get(Pleroma.User, account_id) do
 822         Pleroma.List.unfollow(list, followed)
 823       end
 824     end)
 825
 826     json(conn, %{})
 827   end
 828
 829   def list_accounts(%{assigns: %{user: user}} = conn, %{"id" => id}) do
 830     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 831          {:ok, users} = Pleroma.List.get_following(list) do
 832       render(conn, AccountView, "accounts.json", %{users: users, as: :user})
 833     end
 834   end
 835
 836   def rename_list(%{assigns: %{user: user}} = conn, %{"id" => id, "title" => title}) do
 837     with %Pleroma.List{} = list <- Pleroma.List.get(id, user),
 838          {:ok, list} <- Pleroma.List.rename(list, title) do
 839       res = ListView.render("list.json", list: list)
 840       json(conn, res)
 841     else
 842       _e ->
 843         json(conn, "error")
 844     end
 845   end
 846
 847   def list_timeline(%{assigns: %{user: user}} = conn, %{"list_id" => id} = params) do
 848     with %Pleroma.List{title: title, following: following} <- Pleroma.List.get(id, user) do
 849       params =
 850         params
 851         |> Map.put("type", "Create")
 852         |> Map.put("blocking_user", user)
 853
 854       # adding title is a hack to not make empty lists function like a public timeline
 855       activities =
 856         ActivityPub.fetch_activities([title | following], params)
 857         |> Enum.reverse()
 858
 859       conn
 860       |> render(StatusView, "index.json", %{activities: activities, for: user, as: :activity})
 861     else
 862       _e ->
 863         conn
 864         |> put_status(403)
 865         |> json(%{error: "Error."})
 866     end
 867   end
 868
 869   def index(%{assigns: %{user: user}} = conn, _params) do
 870     token =
 871       conn
 872       |> get_session(:oauth_token)
 873
 874     if user && token do
 875       mastodon_emoji = mastodonized_emoji()
 876       accounts = Map.put(%{}, user.id, AccountView.render("account.json", %{user: user}))
 877
 878       initial_state =
 879         %{
 880           meta: %{
 881             streaming_api_base_url:
 882               String.replace(Pleroma.Web.Endpoint.static_url(), "http", "ws"),
 883             access_token: token,
 884             locale: "en",
 885             domain: Pleroma.Web.Endpoint.host(),
 886             admin: "1",
 887             me: "#{user.id}",
 888             unfollow_modal: false,
 889             boost_modal: false,
 890             delete_modal: true,
 891             auto_play_gif: false,
 892             display_sensitive_media: false,
 893             reduce_motion: false,
 894             max_toot_chars: Keyword.get(@instance, :limit)
 895           },
 896           rights: %{
 897             delete_others_notice: !!user.info["is_moderator"]
 898           },
 899           compose: %{
 900             me: "#{user.id}",
 901             default_privacy: user.info["default_scope"] || "public",
 902             default_sensitive: false
 903           },
 904           media_attachments: %{
 905             accept_content_types: [
 906               ".jpg",
 907               ".jpeg",
 908               ".png",
 909               ".gif",
 910               ".webm",
 911               ".mp4",
 912               ".m4v",
 913               "image\/jpeg",
 914               "image\/png",
 915               "image\/gif",
 916               "video\/webm",
 917               "video\/mp4"
 918             ]
 919           },
 920           settings:
 921             Map.get(user.info, "settings") ||
 922               %{
 923                 onboarded: true,
 924                 home: %{
 925                   shows: %{
 926                     reblog: true,
 927                     reply: true
 928                   }
 929                 },
 930                 notifications: %{
 931                   alerts: %{
 932                     follow: true,
 933                     favourite: true,
 934                     reblog: true,
 935                     mention: true
 936                   },
 937                   shows: %{
 938                     follow: true,
 939                     favourite: true,
 940                     reblog: true,
 941                     mention: true
 942                   },
 943                   sounds: %{
 944                     follow: true,
 945                     favourite: true,
 946                     reblog: true,
 947                     mention: true
 948                   }
 949                 }
 950               },
 951           push_subscription: nil,
 952           accounts: accounts,
 953           custom_emojis: mastodon_emoji,
 954           char_limit: Keyword.get(@instance, :limit)
 955         }
 956         |> Jason.encode!()
 957
 958       conn
 959       |> put_layout(false)
 960       |> render(MastodonView, "index.html", %{initial_state: initial_state})
 961     else
 962       conn
 963       |> redirect(to: "/web/login")
 964     end
 965   end
 966
 967   def put_settings(%{assigns: %{user: user}} = conn, %{"data" => settings} = _params) do
 968     with new_info <- Map.put(user.info, "settings", settings),
 969          change <- User.info_changeset(user, %{info: new_info}),
 970          {:ok, _user} <- User.update_and_set_cache(change) do
 971       conn
 972       |> json(%{})
 973     else
 974       e ->
 975         conn
 976         |> json(%{error: inspect(e)})
 977     end
 978   end
 979
 980   def login(conn, _) do
 981     conn
 982     |> render(MastodonView, "login.html", %{error: false})
 983   end
 984
 985   defp get_or_make_app() do
 986     with %App{} = app <- Repo.get_by(App, client_name: "Mastodon-Local") do
 987       {:ok, app}
 988     else
 989       _e ->
 990         cs =
 991           App.register_changeset(%App{}, %{
 992             client_name: "Mastodon-Local",
 993             redirect_uris: ".",
 994             scopes: "read,write,follow"
 995           })
 996
 997         Repo.insert(cs)
 998     end
 999   end
1000
1001   def login_post(conn, %{"authorization" => %{"name" => name, "password" => password}}) do
1002     with %User{} = user <- User.get_by_nickname_or_email(name),
1003          true <- Pbkdf2.checkpw(password, user.password_hash),
1004          {:ok, app} <- get_or_make_app(),
1005          {:ok, auth} <- Authorization.create_authorization(app, user),
1006          {:ok, token} <- Token.exchange_token(app, auth) do
1007       conn
1008       |> put_session(:oauth_token, token.token)
1009       |> redirect(to: "/web/getting-started")
1010     else
1011       _e ->
1012         conn
1013         |> render(MastodonView, "login.html", %{error: "Wrong username or password"})
1014     end
1015   end
1016
1017   def logout(conn, _) do
1018     conn
1019     |> clear_session
1020     |> redirect(to: "/")
1021   end
1022
1023   def relationship_noop(%{assigns: %{user: user}} = conn, %{"id" => id}) do
1024     Logger.debug("Unimplemented, returning unmodified relationship")
1025
1026     with %User{} = target <- Repo.get(User, id) do
1027       render(conn, AccountView, "relationship.json", %{user: user, target: target})
1028     end
1029   end
1030
1031   def empty_array(conn, _) do
1032     Logger.debug("Unimplemented, returning an empty array")
1033     json(conn, [])
1034   end
1035
1036   def empty_object(conn, _) do
1037     Logger.debug("Unimplemented, returning an empty object")
1038     json(conn, %{})
1039   end
1040
1041   def render_notification(user, %{id: id, activity: activity, inserted_at: created_at} = _params) do
1042     actor = User.get_cached_by_ap_id(activity.data["actor"])
1043
1044     created_at =
1045       NaiveDateTime.to_iso8601(created_at)
1046       |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
1047
1048     case activity.data["type"] do
1049       "Create" ->
1050         %{
1051           id: id,
1052           type: "mention",
1053           created_at: created_at,
1054           account: AccountView.render("account.json", %{user: actor}),
1055           status: StatusView.render("status.json", %{activity: activity, for: user})
1056         }
1057
1058       "Like" ->
1059         liked_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
1060
1061         %{
1062           id: id,
1063           type: "favourite",
1064           created_at: created_at,
1065           account: AccountView.render("account.json", %{user: actor}),
1066           status: StatusView.render("status.json", %{activity: liked_activity, for: user})
1067         }
1068
1069       "Announce" ->
1070         announced_activity = Activity.get_create_activity_by_object_ap_id(activity.data["object"])
1071
1072         %{
1073           id: id,
1074           type: "reblog",
1075           created_at: created_at,
1076           account: AccountView.render("account.json", %{user: actor}),
1077           status: StatusView.render("status.json", %{activity: announced_activity, for: user})
1078         }
1079
1080       "Follow" ->
1081         %{
1082           id: id,
1083           type: "follow",
1084           created_at: created_at,
1085           account: AccountView.render("account.json", %{user: actor})
1086         }
1087
1088       _ ->
1089         nil
1090     end
1091   end
1092
1093   def errors(conn, _) do
1094     conn
1095     |> put_status(500)
1096     |> json("Something went wrong")
1097   end
1098
1099   @suggestions Application.get_env(:pleroma, :suggestions)
1100
1101   def suggestions(%{assigns: %{user: user}} = conn, _) do
1102     if Keyword.get(@suggestions, :enabled, false) do
1103       api = Keyword.get(@suggestions, :third_party_engine, "")
1104       timeout = Keyword.get(@suggestions, :timeout, 5000)
1105
1106       host =
1107         Application.get_env(:pleroma, Pleroma.Web.Endpoint)
1108         |> Keyword.get(:url)
1109         |> Keyword.get(:host)
1110
1111       user = user.nickname
1112       url = String.replace(api, "{{host}}", host) |> String.replace("{{user}}", user)
1113
1114       with {:ok, %{status_code: 200, body: body}} <-
1115              @httpoison.get(url, [], timeout: timeout, recv_timeout: timeout),
1116            {:ok, data} <- Jason.decode(body) do
1117         data2 =
1118           Enum.slice(data, 0, 40)
1119           |> Enum.map(fn x ->
1120             Map.put(
1121               x,
1122               "id",
1123               case User.get_or_fetch(x["acct"]) do
1124                 %{id: id} -> id
1125                 _ -> 0
1126               end
1127             )
1128           end)
1129           |> Enum.map(fn x ->
1130             Map.put(x, "avatar", MediaProxy.url(x["avatar"]))
1131           end)
1132           |> Enum.map(fn x ->
1133             Map.put(x, "avatar_static", MediaProxy.url(x["avatar_static"]))
1134           end)
1135
1136         conn
1137         |> json(data2)
1138       else
1139         e -> Logger.error("Could not retrieve suggestions at fetch #{url}, #{inspect(e)}")
1140       end
1141     else
1142       json(conn, [])
1143     end
1144   end
1145 end