user.ex 38 KB
Newer Older
1
# Pleroma: A lightweight social networking server
kaniini's avatar
kaniini committed
2
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
3 4
# SPDX-License-Identifier: AGPL-3.0-only

lain's avatar
lain committed
5 6
defmodule Pleroma.User do
  use Ecto.Schema
7

Haelwenn's avatar
Haelwenn committed
8 9 10
  import Ecto.Changeset
  import Ecto.Query

11 12
  alias Comeonin.Pbkdf2
  alias Pleroma.Activity
13
  alias Pleroma.Bookmark
14 15 16
  alias Pleroma.Formatter
  alias Pleroma.Notification
  alias Pleroma.Object
17
  alias Pleroma.Registration
Haelwenn's avatar
Haelwenn committed
18 19 20
  alias Pleroma.Repo
  alias Pleroma.User
  alias Pleroma.Web
21 22
  alias Pleroma.Web.ActivityPub.ActivityPub
  alias Pleroma.Web.ActivityPub.Utils
Maxim Filippov's avatar
Maxim Filippov committed
23
  alias Pleroma.Web.CommonAPI.Utils, as: CommonUtils
Haelwenn's avatar
Haelwenn committed
24
  alias Pleroma.Web.OAuth
25
  alias Pleroma.Web.OStatus
26
  alias Pleroma.Web.RelMe
27
  alias Pleroma.Web.Websub
lain's avatar
lain committed
28

29 30
  require Logger

Maksim's avatar
Maksim committed
31 32
  @type t :: %__MODULE__{}

href's avatar
href committed
33 34
  @primary_key {:id, Pleroma.FlakeId, autogenerate: true}

35
  # credo:disable-for-next-line Credo.Check.Readability.MaxLineLength
href's avatar
href committed
36 37 38
  @email_regex ~r/^[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$/

  @strict_local_nickname_regex ~r/^[a-zA-Z\d]+$/
href's avatar
href committed
39
  @extended_local_nickname_regex ~r/^[a-zA-Z\d_-]+$/
href's avatar
href committed
40

lain's avatar
lain committed
41
  schema "users" do
lain's avatar
lain committed
42 43 44 45 46 47 48 49 50 51 52 53
    field(:bio, :string)
    field(:email, :string)
    field(:name, :string)
    field(:nickname, :string)
    field(:password_hash, :string)
    field(:password, :string, virtual: true)
    field(:password_confirmation, :string, virtual: true)
    field(:following, {:array, :string}, default: [])
    field(:ap_id, :string)
    field(:avatar, :map)
    field(:local, :boolean, default: true)
    field(:follower_address, :string)
54
    field(:search_rank, :float, virtual: true)
55
    field(:search_type, :integer, virtual: true)
56
    field(:tags, {:array, :string}, default: [])
rinpatch's avatar
rinpatch committed
57
    field(:last_refreshed_at, :naive_datetime_usec)
58
    has_many(:bookmarks, Bookmark)
lain's avatar
lain committed
59
    has_many(:notifications, Notification)
60
    has_many(:registrations, Registration)
lain's avatar
lain committed
61
    embeds_one(:info, Pleroma.User.Info)
lain's avatar
lain committed
62 63 64

    timestamps()
  end
lain's avatar
lain committed
65

66 67
  def auth_active?(%User{info: %User.Info{confirmation_pending: true}}),
    do: !Pleroma.Config.get([:instance, :account_activation_required])
68

69
  def auth_active?(%User{}), do: true
70

71 72 73 74 75
  def visible_for?(user, for_user \\ nil)

  def visible_for?(%User{id: user_id}, %User{id: for_id}) when user_id == for_id, do: true

  def visible_for?(%User{} = user, for_user) do
76
    auth_active?(user) || superuser?(for_user)
77 78
  end

79 80
  def visible_for?(_, _), do: false

81 82
  def superuser?(%User{local: true, info: %User.Info{is_admin: true}}), do: true
  def superuser?(%User{local: true, info: %User.Info{is_moderator: true}}), do: true
83
  def superuser?(_), do: false
84

85
  def avatar_url(user, options \\ []) do
lain's avatar
lain committed
86 87
    case user.avatar do
      %{"url" => [%{"href" => href} | _]} -> href
88
      _ -> !options[:no_default] && "#{Web.base_url()}/images/avi.png"
lain's avatar
lain committed
89 90 91
    end
  end

92
  def banner_url(user, options \\ []) do
lain's avatar
lain committed
93
    case user.info.banner do
lain's avatar
lain committed
94
      %{"url" => [%{"href" => href} | _]} -> href
95
      _ -> !options[:no_default] && "#{Web.base_url()}/images/banner.png"
lain's avatar
lain committed
96 97 98
    end
  end

lain's avatar
lain committed
99
  def profile_url(%User{info: %{source_data: %{"url" => url}}}), do: url
100 101 102
  def profile_url(%User{ap_id: ap_id}), do: ap_id
  def profile_url(_), do: nil

lain's avatar
lain committed
103
  def ap_id(%User{nickname: nickname}) do
lain's avatar
lain committed
104
    "#{Web.base_url()}/users/#{nickname}"
lain's avatar
lain committed
105 106
  end

107 108
  def ap_followers(%User{follower_address: fa}) when is_binary(fa), do: fa
  def ap_followers(%User{} = user), do: "#{ap_id(user)}/followers"
lain's avatar
lain committed
109

lain's avatar
lain committed
110
  def user_info(%User{} = user) do
111
    oneself = if user.local, do: 1, else: 0
lain's avatar
lain committed
112

lain's avatar
lain committed
113
    %{
114
      following_count: length(user.following) - oneself,
lain's avatar
lain committed
115 116 117
      note_count: user.info.note_count,
      follower_count: user.info.follower_count,
      locked: user.info.locked,
Ivan Tashkinov's avatar
Ivan Tashkinov committed
118
      confirmation_pending: user.info.confirmation_pending,
lain's avatar
lain committed
119
      default_scope: user.info.default_scope
lain's avatar
lain committed
120 121 122
    }
  end

lain's avatar
lain committed
123
  def remote_user_creation(params) do
lain's avatar
lain committed
124 125 126
    params =
      params
      |> Map.put(:info, params[:info] || %{})
lain's avatar
lain committed
127 128 129

    info_cng = User.Info.remote_user_creation(%User.Info{}, params[:info])

lain's avatar
lain committed
130
    changes =
lain's avatar
lain committed
131
      %User{}
lain's avatar
lain committed
132
      |> cast(params, [:bio, :name, :ap_id, :nickname, :avatar])
133
      |> validate_required([:name, :ap_id])
lain's avatar
lain committed
134 135 136 137 138
      |> unique_constraint(:nickname)
      |> validate_format(:nickname, @email_regex)
      |> validate_length(:bio, max: 5000)
      |> validate_length(:name, max: 100)
      |> put_change(:local, false)
lain's avatar
lain committed
139
      |> put_embed(:info, info_cng)
lain's avatar
lain committed
140

141
    if changes.valid? do
lain's avatar
lain committed
142
      case info_cng.changes[:source_data] do
lain's avatar
lain committed
143 144 145
        %{"followers" => followers} ->
          changes
          |> put_change(:follower_address, followers)
lain's avatar
lain committed
146

lain's avatar
lain committed
147 148
        _ ->
          followers = User.ap_followers(%User{nickname: changes.changes[:nickname]})
lain's avatar
lain committed
149

lain's avatar
lain committed
150 151 152
          changes
          |> put_change(:follower_address, followers)
      end
153 154 155
    else
      changes
    end
lain's avatar
lain committed
156 157
  end

lain's avatar
lain committed
158
  def update_changeset(struct, params \\ %{}) do
Thog's avatar
Thog committed
159
    struct
lain's avatar
lain committed
160
    |> cast(params, [:bio, :name, :avatar])
lain's avatar
lain committed
161
    |> unique_constraint(:nickname)
href's avatar
href committed
162
    |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
163
    |> validate_length(:bio, max: 5000)
lain's avatar
lain committed
164 165 166
    |> validate_length(:name, min: 1, max: 100)
  end

lain's avatar
lain committed
167
  def upgrade_changeset(struct, params \\ %{}) do
168 169 170 171
    params =
      params
      |> Map.put(:last_refreshed_at, NaiveDateTime.utc_now())

lain's avatar
lain committed
172 173 174 175
    info_cng =
      struct.info
      |> User.Info.user_upgrade(params[:info])

lain's avatar
lain committed
176
    struct
lain's avatar
lain committed
177
    |> cast(params, [:bio, :name, :follower_address, :avatar, :last_refreshed_at])
lain's avatar
lain committed
178
    |> unique_constraint(:nickname)
href's avatar
href committed
179
    |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
180 181
    |> validate_length(:bio, max: 5000)
    |> validate_length(:name, max: 100)
lain's avatar
lain committed
182
    |> put_embed(:info, info_cng)
lain's avatar
lain committed
183 184
  end

Roger Braun's avatar
Roger Braun committed
185
  def password_update_changeset(struct, params) do
lain's avatar
lain committed
186 187 188 189 190
    changeset =
      struct
      |> cast(params, [:password, :password_confirmation])
      |> validate_required([:password, :password_confirmation])
      |> validate_confirmation(:password)
Roger Braun's avatar
Roger Braun committed
191

192 193 194
    OAuth.Token.delete_user_tokens(struct)
    OAuth.Authorization.delete_user_authorizations(struct)

Roger Braun's avatar
Roger Braun committed
195 196
    if changeset.valid? do
      hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
lain's avatar
lain committed
197

Roger Braun's avatar
Roger Braun committed
198 199 200 201 202 203 204 205
      changeset
      |> put_change(:password_hash, hashed)
    else
      changeset
    end
  end

  def reset_password(user, data) do
lain's avatar
lain committed
206
    update_and_set_cache(password_update_changeset(user, data))
Roger Braun's avatar
Roger Braun committed
207 208
  end

209 210 211 212 213 214 215 216
  def register_changeset(struct, params \\ %{}, opts \\ []) do
    confirmation_status =
      if opts[:confirmed] || !Pleroma.Config.get([:instance, :account_activation_required]) do
        :confirmed
      else
        :unconfirmed
      end

Ivan Tashkinov's avatar
Ivan Tashkinov committed
217 218
    info_change = User.Info.confirmation_changeset(%User.Info{}, confirmation_status)

lain's avatar
lain committed
219 220 221
    changeset =
      struct
      |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation])
222
      |> validate_required([:name, :nickname, :password, :password_confirmation])
lain's avatar
lain committed
223 224 225
      |> validate_confirmation(:password)
      |> unique_constraint(:email)
      |> unique_constraint(:nickname)
lain's avatar
lain committed
226
      |> validate_exclusion(:nickname, Pleroma.Config.get([Pleroma.User, :restricted_nicknames]))
href's avatar
href committed
227
      |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
228 229 230
      |> validate_format(:email, @email_regex)
      |> validate_length(:bio, max: 1000)
      |> validate_length(:name, min: 1, max: 100)
Ivan Tashkinov's avatar
Ivan Tashkinov committed
231
      |> put_change(:info, info_change)
lain's avatar
lain committed
232

233 234 235 236 237 238 239
    changeset =
      if opts[:external] do
        changeset
      else
        validate_required(changeset, [:email])
      end

lain's avatar
lain committed
240
    if changeset.valid? do
241
      hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
lain's avatar
lain committed
242 243
      ap_id = User.ap_id(%User{nickname: changeset.changes[:nickname]})
      followers = User.ap_followers(%User{nickname: changeset.changes[:nickname]})
lain's avatar
lain committed
244

lain's avatar
lain committed
245 246 247
      changeset
      |> put_change(:password_hash, hashed)
      |> put_change(:ap_id, ap_id)
rinpatch's avatar
rinpatch committed
248
      |> unique_constraint(:ap_id)
lain's avatar
lain committed
249
      |> put_change(:following, [followers])
250
      |> put_change(:follower_address, followers)
lain's avatar
lain committed
251 252 253 254 255
    else
      changeset
    end
  end

256 257 258 259 260 261 262 263 264 265
  defp autofollow_users(user) do
    candidates = Pleroma.Config.get([:instance, :autofollowed_nicknames])

    autofollowed_users =
      from(u in User,
        where: u.local == true,
        where: u.nickname in ^candidates
      )
      |> Repo.all()

lain's avatar
lain committed
266
    follow_all(user, autofollowed_users)
267 268
  end

269 270
  @doc "Inserts provided changeset, performs post-registration actions (confirmation email sending etc.)"
  def register(%Ecto.Changeset{} = changeset) do
Ivan Tashkinov's avatar
Ivan Tashkinov committed
271
    with {:ok, user} <- Repo.insert(changeset),
lain's avatar
lain committed
272
         {:ok, user} <- autofollow_users(user),
minibikini's avatar
minibikini committed
273
         {:ok, user} <- set_cache(user),
lain's avatar
lain committed
274
         {:ok, _} <- Pleroma.User.WelcomeMessage.post_welcome_message_to_user(user),
lain's avatar
lain committed
275
         {:ok, _} <- try_send_confirmation_email(user) do
276 277 278 279
      {:ok, user}
    end
  end

280
  def try_send_confirmation_email(%User{} = user) do
281 282
    if user.info.confirmation_pending &&
         Pleroma.Config.get([:instance, :account_activation_required]) do
283
      user
284 285
      |> Pleroma.Emails.UserEmail.account_confirmation_email()
      |> Pleroma.Emails.Mailer.deliver_async()
286 287

      {:ok, :enqueued}
288 289 290 291 292
    else
      {:ok, :noop}
    end
  end

293 294 295 296 297
  def needs_update?(%User{local: true}), do: false

  def needs_update?(%User{local: false, last_refreshed_at: nil}), do: true

  def needs_update?(%User{local: false} = user) do
298
    NaiveDateTime.diff(NaiveDateTime.utc_now(), user.last_refreshed_at) >= 86_400
299 300 301 302
  end

  def needs_update?(_), do: true

lain's avatar
lain committed
303
  def maybe_direct_follow(%User{} = follower, %User{local: true, info: %{locked: true}}) do
304 305 306 307 308 309 310 311
    {:ok, follower}
  end

  def maybe_direct_follow(%User{} = follower, %User{local: true} = followed) do
    follow(follower, followed)
  end

  def maybe_direct_follow(%User{} = follower, %User{} = followed) do
Maksim's avatar
Maksim committed
312
    if not User.ap_enabled?(followed) do
313
      follow(follower, followed)
314 315 316 317 318
    else
      {:ok, follower}
    end
  end

Maksim's avatar
Maksim committed
319
  def maybe_follow(%User{} = follower, %User{info: _info} = followed) do
320 321
    if not following?(follower, followed) do
      follow(follower, followed)
322
    else
323
      {:ok, follower}
324 325 326
    end
  end

327
  @doc "A mass follow for local users. Respects blocks in both directions but does not create activities."
lain's avatar
lain committed
328 329
  @spec follow_all(User.t(), list(User.t())) :: {atom(), User.t()}
  def follow_all(follower, followeds) do
lain's avatar
lain committed
330 331
    followed_addresses =
      followeds
332
      |> Enum.reject(fn followed -> blocks?(follower, followed) || blocks?(followed, follower) end)
lain's avatar
lain committed
333
      |> Enum.map(fn %{follower_address: fa} -> fa end)
lain's avatar
lain committed
334

lain's avatar
lain committed
335 336 337
    q =
      from(u in User,
        where: u.id == ^follower.id,
338 339 340 341 342 343 344 345 346
        update: [
          set: [
            following:
              fragment(
                "array(select distinct unnest (array_cat(?, ?)))",
                u.following,
                ^followed_addresses
              )
          ]
rinpatch's avatar
rinpatch committed
347 348
        ],
        select: u
lain's avatar
lain committed
349 350
      )

rinpatch's avatar
rinpatch committed
351
    {1, [follower]} = Repo.update_all(q, [])
lain's avatar
lain committed
352 353 354 355 356

    Enum.each(followeds, fn followed ->
      update_follower_count(followed)
    end)

lain's avatar
lain committed
357
    set_cache(follower)
lain's avatar
lain committed
358 359
  end

lain's avatar
lain committed
360
  def follow(%User{} = follower, %User{info: info} = followed) do
361 362
    user_config = Application.get_env(:pleroma, :user)
    deny_follow_blocked = Keyword.get(user_config, :deny_follow_blocked)
363

364
    ap_followers = followed.follower_address
365

366
    cond do
lain's avatar
lain committed
367
      following?(follower, followed) or info.deactivated ->
368
        {:error, "Could not follow user: #{followed.nickname} is already on your list."}
lain's avatar
lain committed
369

370
      deny_follow_blocked and blocks?(followed, follower) ->
371
        {:error, "Could not follow user: #{followed.nickname} blocked you."}
lain's avatar
lain committed
372

373 374 375 376 377
      true ->
        if !followed.local && follower.local && !ap_enabled?(followed) do
          Websub.subscribe(follower, followed)
        end

378 379 380
        q =
          from(u in User,
            where: u.id == ^follower.id,
rinpatch's avatar
rinpatch committed
381 382
            update: [push: [following: ^ap_followers]],
            select: u
383
          )
384

rinpatch's avatar
rinpatch committed
385
        {1, [follower]} = Repo.update_all(q, [])
386

387 388
        {:ok, _} = update_follower_count(followed)

389
        set_cache(follower)
390
    end
lain's avatar
lain committed
391
  end
lain's avatar
lain committed
392 393

  def unfollow(%User{} = follower, %User{} = followed) do
394
    ap_followers = followed.follower_address
lain's avatar
lain committed
395

396
    if following?(follower, followed) and follower.ap_id != followed.ap_id do
397 398 399
      q =
        from(u in User,
          where: u.id == ^follower.id,
rinpatch's avatar
rinpatch committed
400 401
          update: [pull: [following: ^ap_followers]],
          select: u
402
        )
lain's avatar
lain committed
403

rinpatch's avatar
rinpatch committed
404
      {1, [follower]} = Repo.update_all(q, [])
405 406 407

      {:ok, followed} = update_follower_count(followed)

408 409
      set_cache(follower)

410
      {:ok, follower, Utils.fetch_latest_follow(follower, followed)}
411
    else
412
      {:error, "Not subscribed!"}
413
    end
lain's avatar
lain committed
414
  end
415

Maksim's avatar
Maksim committed
416
  @spec following?(User.t(), User.t()) :: boolean
417
  def following?(%User{} = follower, %User{} = followed) do
418
    Enum.member?(follower.following, followed.follower_address)
419
  end
lain's avatar
lain committed
420

421 422 423 424 425
  def follow_import(%User{} = follower, followed_identifiers)
      when is_list(followed_identifiers) do
    Enum.map(
      followed_identifiers,
      fn followed_identifier ->
426
        with {:ok, %User{} = followed} <- get_or_fetch(followed_identifier),
427 428 429 430 431 432 433 434 435 436 437 438
             {:ok, follower} <- maybe_direct_follow(follower, followed),
             {:ok, _} <- ActivityPub.follow(follower, followed) do
          followed
        else
          err ->
            Logger.debug("follow_import failed for #{followed_identifier} with: #{inspect(err)}")
            err
        end
      end
    )
  end

439
  def locked?(%User{} = user) do
440
    user.info.locked || false
441 442
  end

443 444 445 446
  def get_by_id(id) do
    Repo.get_by(User, id: id)
  end

lain's avatar
lain committed
447 448 449 450
  def get_by_ap_id(ap_id) do
    Repo.get_by(User, ap_id: ap_id)
  end

451 452
  # This is mostly an SPC migration fix. This guesses the user nickname by taking the last part
  # of the ap_id and the domain and tries to get that user
453 454 455 456 457
  def get_by_guessed_nickname(ap_id) do
    domain = URI.parse(ap_id).host
    name = List.last(String.split(ap_id, "/"))
    nickname = "#{name}@#{domain}"

minibikini's avatar
minibikini committed
458
    get_cached_by_nickname(nickname)
459 460
  end

minibikini's avatar
minibikini committed
461 462 463 464
  def set_cache({:ok, user}), do: set_cache(user)
  def set_cache({:error, err}), do: {:error, err}

  def set_cache(%User{} = user) do
465 466 467 468 469 470
    Cachex.put(:user_cache, "ap_id:#{user.ap_id}", user)
    Cachex.put(:user_cache, "nickname:#{user.nickname}", user)
    Cachex.put(:user_cache, "user_info:#{user.id}", user_info(user))
    {:ok, user}
  end

lain's avatar
lain committed
471 472
  def update_and_set_cache(changeset) do
    with {:ok, user} <- Repo.update(changeset) do
473
      set_cache(user)
lain's avatar
lain committed
474 475 476 477 478
    else
      e -> e
    end
  end

lain's avatar
lain committed
479 480 481
  def invalidate_cache(user) do
    Cachex.del(:user_cache, "ap_id:#{user.ap_id}")
    Cachex.del(:user_cache, "nickname:#{user.nickname}")
482
    Cachex.del(:user_cache, "user_info:#{user.id}")
lain's avatar
lain committed
483 484
  end

lain's avatar
lain committed
485
  def get_cached_by_ap_id(ap_id) do
486
    key = "ap_id:#{ap_id}"
Thog's avatar
Thog committed
487
    Cachex.fetch!(:user_cache, key, fn _ -> get_by_ap_id(ap_id) end)
lain's avatar
lain committed
488 489
  end

490 491
  def get_cached_by_id(id) do
    key = "id:#{id}"
492 493 494 495

    ap_id =
      Cachex.fetch!(:user_cache, key, fn _ ->
        user = get_by_id(id)
496 497 498 499 500 501 502

        if user do
          Cachex.put(:user_cache, "ap_id:#{user.ap_id}", user)
          {:commit, user.ap_id}
        else
          {:ignore, ""}
        end
503 504 505
      end)

    get_cached_by_ap_id(ap_id)
506 507
  end

lain's avatar
lain committed
508
  def get_cached_by_nickname(nickname) do
509
    key = "nickname:#{nickname}"
0x1C3B00DA's avatar
Run  
0x1C3B00DA committed
510

511 512 513 514 515 516 517 518
    Cachex.fetch!(:user_cache, key, fn ->
      user_result = get_or_fetch_by_nickname(nickname)

      case user_result do
        {:ok, user} -> {:commit, user}
        {:error, error} -> {:ignore, error}
      end
    end)
lain's avatar
lain committed
519
  end
lain's avatar
lain committed
520

521
  def get_cached_by_nickname_or_id(nickname_or_id) do
522
    get_cached_by_id(nickname_or_id) || get_cached_by_nickname(nickname_or_id)
523 524
  end

lain's avatar
lain committed
525
  def get_by_nickname(nickname) do
526
    Repo.get_by(User, nickname: nickname) ||
527
      if Regex.match?(~r(@#{Pleroma.Web.Endpoint.host()})i, nickname) do
528
        Repo.get_by(User, nickname: local_nickname(nickname))
529
      end
530 531
  end

532 533
  def get_by_email(email), do: Repo.get_by(User, email: email)

534
  def get_by_nickname_or_email(nickname_or_email) do
535
    get_by_nickname(nickname_or_email) || get_by_email(nickname_or_email)
536 537
  end

lain's avatar
lain committed
538 539
  def get_cached_user_info(user) do
    key = "user_info:#{user.id}"
Thog's avatar
Thog committed
540
    Cachex.fetch!(:user_cache, key, fn _ -> user_info(user) end)
lain's avatar
lain committed
541
  end
lain's avatar
lain committed
542

lain's avatar
lain committed
543 544 545 546 547 548 549 550 551
  def fetch_by_nickname(nickname) do
    ap_try = ActivityPub.make_user_from_nickname(nickname)

    case ap_try do
      {:ok, user} -> {:ok, user}
      _ -> OStatus.make_user(nickname)
    end
  end

lain's avatar
lain committed
552
  def get_or_fetch_by_nickname(nickname) do
lain's avatar
lain committed
553
    with %User{} = user <- get_by_nickname(nickname) do
554
      {:ok, user}
lain's avatar
lain committed
555 556 557 558
    else
      _e ->
        with [_nick, _domain] <- String.split(nickname, "@"),
             {:ok, user} <- fetch_by_nickname(nickname) do
559
          if Pleroma.Config.get([:fetch_initial_posts, :enabled]) do
minibikini's avatar
minibikini committed
560
            # TODO turn into job
561 562 563
            {:ok, _} = Task.start(__MODULE__, :fetch_initial_posts, [user])
          end

564
          {:ok, user}
lain's avatar
lain committed
565
        else
0x1C3B00DA's avatar
0x1C3B00DA committed
566
          e -> {:error, e}
lain's avatar
lain committed
567
        end
lain's avatar
lain committed
568
    end
lain's avatar
lain committed
569
  end
lain's avatar
lain committed
570

571 572 573 574 575 576 577 578 579 580 581
  @doc "Fetch some posts when the user has just been federated with"
  def fetch_initial_posts(user) do
    pages = Pleroma.Config.get!([:fetch_initial_posts, :pages])

    Enum.each(
      # Insert all the posts in reverse order, so they're in the right order on the timeline
      Enum.reverse(Utils.fetch_ordered_collection(user.info.source_data["outbox"], pages)),
      &Pleroma.Web.Federator.incoming_ap_doc/1
    )
  end

582
  def get_followers_query(%User{id: id, follower_address: follower_address}, nil) do
583 584 585 586 587 588 589
    from(
      u in User,
      where: fragment("? <@ ?", ^[follower_address], u.following),
      where: u.id != ^id
    )
  end

590
  def get_followers_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
591 592
    from(u in get_followers_query(user, nil))
    |> paginate(page, 20)
593 594 595 596 597 598
  end

  def get_followers_query(user), do: get_followers_query(user, nil)

  def get_followers(user, page \\ nil) do
    q = get_followers_query(user, page)
lain's avatar
lain committed
599 600 601 602

    {:ok, Repo.all(q)}
  end

603 604 605 606 607 608
  def get_followers_ids(user, page \\ nil) do
    q = get_followers_query(user, page)

    Repo.all(from(u in q, select: u.id))
  end

609
  def get_friends_query(%User{id: id, following: following}, nil) do
610 611 612 613 614 615 616
    from(
      u in User,
      where: u.follower_address in ^following,
      where: u.id != ^id
    )
  end

617
  def get_friends_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
618 619
    from(u in get_friends_query(user, nil))
    |> paginate(page, 20)
620 621 622 623 624 625
  end

  def get_friends_query(user), do: get_friends_query(user, nil)

  def get_friends(user, page \\ nil) do
    q = get_friends_query(user, page)
lain's avatar
lain committed
626 627 628

    {:ok, Repo.all(q)}
  end
629

630 631 632 633 634 635
  def get_friends_ids(user, page \\ nil) do
    q = get_friends_query(user, page)

    Repo.all(from(u in q, select: u.id))
  end

636 637 638
  def get_follow_requests_query(%User{} = user) do
    from(
      a in Activity,
kaniini's avatar
kaniini committed
639 640 641 642 643 644 645 646 647 648 649 650
      where:
        fragment(
          "? ->> 'type' = 'Follow'",
          a.data
        ),
      where:
        fragment(
          "? ->> 'state' = 'pending'",
          a.data
        ),
      where:
        fragment(
651
          "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
kaniini's avatar
kaniini committed
652
          a.data,
653 654
          a.data,
          ^user.ap_id
kaniini's avatar
kaniini committed
655
        )
656 657 658 659 660
    )
  end

  def get_follow_requests(%User{} = user) do
    users =
661 662
      user
      |> User.get_follow_requests_query()
rinpatch's avatar
rinpatch committed
663
      |> join(:inner, [a], u in User, on: a.actor == u.ap_id)
664 665 666 667
      |> where([a, u], not fragment("? @> ?", u.following, ^[user.follower_address]))
      |> group_by([a, u], u.id)
      |> select([a, u], u)
      |> Repo.all()
668 669 670 671

    {:ok, users}
  end

672
  def increase_note_count(%User{} = user) do
673 674 675 676 677 678 679 680 681 682 683 684
    User
    |> where(id: ^user.id)
    |> update([u],
      set: [
        info:
          fragment(
            "jsonb_set(?, '{note_count}', ((?->>'note_count')::int + 1)::varchar::jsonb, true)",
            u.info,
            u.info
          )
      ]
    )
rinpatch's avatar
rinpatch committed
685 686
    |> select([u], u)
    |> Repo.update_all([])
687 688 689 690
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
691 692
  end

693
  def decrease_note_count(%User{} = user) do
694 695 696 697 698 699 700 701 702 703 704 705
    User
    |> where(id: ^user.id)
    |> update([u],
      set: [
        info:
          fragment(
            "jsonb_set(?, '{note_count}', (greatest(0, (?->>'note_count')::int - 1))::varchar::jsonb, true)",
            u.info,
            u.info
          )
      ]
    )
rinpatch's avatar
rinpatch committed
706 707
    |> select([u], u)
    |> Repo.update_all([])
708 709 710 711
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
712 713
  end

714
  def update_note_count(%User{} = user) do
lain's avatar
lain committed
715 716 717 718 719 720
    note_count_query =
      from(
        a in Object,
        where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data),
        select: count(a.id)
      )
721 722 723

    note_count = Repo.one(note_count_query)

lain's avatar
lain committed
724
    info_cng = User.Info.set_note_count(user.info, note_count)
725

lain's avatar
lain committed
726 727 728
    cng =
      change(user)
      |> put_embed(:info, info_cng)
729

lain's avatar
lain committed
730
    update_and_set_cache(cng)
731 732 733
  end

  def update_follower_count(%User{} = user) do
lain's avatar
lain committed
734
    follower_count_query =
735 736 737 738
      User
      |> where([u], ^user.follower_address in u.following)
      |> where([u], u.id != ^user.id)
      |> select([u], %{count: count(u.id)})
739

740 741 742 743 744 745 746 747 748 749 750 751 752
    User
    |> where(id: ^user.id)
    |> join(:inner, [u], s in subquery(follower_count_query))
    |> update([u, s],
      set: [
        info:
          fragment(
            "jsonb_set(?, '{follower_count}', ?::varchar::jsonb, true)",
            u.info,
            s.count
          )
      ]
    )
rinpatch's avatar
rinpatch committed
753 754
    |> select([u], u)
    |> Repo.update_all([])
755 756 757 758
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
759
  end
760

761
  def get_users_from_set_query(ap_ids, false) do
762 763
    from(
      u in User,
764
      where: u.ap_id in ^ap_ids
765 766 767
    )
  end

768 769
  def get_users_from_set_query(ap_ids, true) do
    query = get_users_from_set_query(ap_ids, false)
770 771 772

    from(
      u in query,
773 774 775 776
      where: u.local == true
    )
  end

777 778 779 780 781
  def get_users_from_set(ap_ids, local_only \\ true) do
    get_users_from_set_query(ap_ids, local_only)
    |> Repo.all()
  end

782
  def get_recipients_from_activity(%Activity{recipients: to}) do
lain's avatar
lain committed
783 784 785 786 787 788
    query =
      from(
        u in User,
        where: u.ap_id in ^to,
        or_where: fragment("? && ?", u.following, ^to)
      )
789

lain's avatar
lain committed
790
    query = from(u in query, where: u.local == true)
791

792 793 794
    Repo.all(query)
  end

795
  def search(query, resolve \\ false, for_user \\ nil) do
796
    # Strip the beginning @ off if there is a query
797 798
    query = String.trim_leading(query, "@")

799
    if resolve, do: get_or_fetch(query)
lain's avatar
lain committed
800

801
    {:ok, results} =
lain's avatar
lain committed
802 803
      Repo.transaction(fn ->
        Ecto.Adapters.SQL.query(Repo, "select set_limit(0.25)", [])
804
        Repo.all(search_query(query, for_user))
lain's avatar
lain committed
805
      end)
lain's avatar
lain committed
806

807
    results
808
  end
lain's avatar
lain committed
809

810 811 812
  def search_query(query, for_user) do
    fts_subquery = fts_search_subquery(query)
    trigram_subquery = trigram_search_subquery(query)
813 814
    union_query = from(s in trigram_subquery, union_all: ^fts_subquery)
    distinct_query = from(s in subquery(union_query), order_by: s.search_type, distinct: s.id)
lain's avatar
lain committed
815

816 817 818 819 820
    from(s in subquery(boost_search_rank_query(distinct_query, for_user)),
      order_by: [desc: s.search_rank],
      limit: 20
    )
  end
821

822 823 824 825 826
  defp boost_search_rank_query(query, nil), do: query

  defp boost_search_rank_query(query, for_user) do
    friends_ids = get_friends_ids(for_user)
    followers_ids = get_followers_ids(for_user)
827

828 829 830 831 832
    from(u in subquery(query),
      select_merge: %{
        search_rank:
          fragment(
            """
833
             CASE WHEN (?) THEN (?) * 1.3
834 835 836 837 838 839 840 841 842 843 844 845 846 847
             WHEN (?) THEN (?) * 1.2
             WHEN (?) THEN (?) * 1.1
             ELSE (?) END
            """,
            u.id in ^friends_ids and u.id in ^followers_ids,
            u.search_rank,
            u.id in ^friends_ids,
            u.search_rank,
            u.id in ^followers_ids,
            u.search_rank,
            u.search_rank
          )
      }
    )
848
  end
849

Maxim Filippov's avatar
Maxim Filippov committed
850
  defp fts_search_subquery(term, query \\ User) do
851
    processed_query =
852
      term
853 854 855 856 857
      |> String.replace(~r/\W+/, " ")
      |> String.trim()
      |> String.split()
      |> Enum.map(&(&1 <> ":*"))
      |> Enum.join(" | ")
858