user.ex 39.4 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.Keys
14 15
  alias Pleroma.Notification
  alias Pleroma.Object
16
  alias Pleroma.Registration
Haelwenn's avatar
Haelwenn committed
17 18 19
  alias Pleroma.Repo
  alias Pleroma.User
  alias Pleroma.Web
20 21
  alias Pleroma.Web.ActivityPub.ActivityPub
  alias Pleroma.Web.ActivityPub.Utils
Maxim Filippov's avatar
Maxim Filippov committed
22
  alias Pleroma.Web.CommonAPI.Utils, as: CommonUtils
Haelwenn's avatar
Haelwenn committed
23
  alias Pleroma.Web.OAuth
24
  alias Pleroma.Web.OStatus
25
  alias Pleroma.Web.RelMe
26
  alias Pleroma.Web.Websub
lain's avatar
lain committed
27

28 29
  require Logger

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

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

34
  # credo:disable-for-next-line Credo.Check.Readability.MaxLineLength
href's avatar
href committed
35 36 37
  @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
38
  @extended_local_nickname_regex ~r/^[a-zA-Z\d_-]+$/
href's avatar
href committed
39

lain's avatar
lain committed
40
  schema "users" do
lain's avatar
lain committed
41 42 43 44 45 46 47 48 49 50 51 52
    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)
53
    field(:search_rank, :float, virtual: true)
54
    field(:search_type, :integer, virtual: true)
55
    field(:tags, {:array, :string}, default: [])
rinpatch's avatar
rinpatch committed
56
    field(:last_refreshed_at, :naive_datetime_usec)
lain's avatar
lain committed
57
    has_many(:notifications, Notification)
58
    has_many(:registrations, Registration)
59
    embeds_one(:info, User.Info)
lain's avatar
lain committed
60 61 62

    timestamps()
  end
lain's avatar
lain committed
63

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

67
  def auth_active?(%User{}), do: true
68

69 70 71 72 73
  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
74
    auth_active?(user) || superuser?(for_user)
75 76
  end

77 78
  def visible_for?(_, _), do: false

79 80
  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
81
  def superuser?(_), do: false
82

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

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

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

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

105 106
  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
107

lain's avatar
lain committed
108 109
  def user_info(%User{} = user) do
    %{
110
      following_count: following_count(user),
lain's avatar
lain committed
111 112 113
      note_count: user.info.note_count,
      follower_count: user.info.follower_count,
      locked: user.info.locked,
Ivan Tashkinov's avatar
Ivan Tashkinov committed
114
      confirmation_pending: user.info.confirmation_pending,
lain's avatar
lain committed
115
      default_scope: user.info.default_scope
lain's avatar
lain committed
116 117 118
    }
  end

119
  def restrict_deactivated(query) do
120
    from(u in query,
121
      where: not fragment("? \\? 'deactivated' AND ?->'deactivated' @> 'true'", u.info, u.info)
122 123 124 125 126
    )
  end

  def following_count(%User{following: []}), do: 0

127 128 129
  def following_count(%User{} = user) do
    user
    |> get_friends_query()
130 131 132
    |> Repo.aggregate(:count, :id)
  end

lain's avatar
lain committed
133
  def remote_user_creation(params) do
lain's avatar
lain committed
134 135 136
    params =
      params
      |> Map.put(:info, params[:info] || %{})
lain's avatar
lain committed
137 138 139

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

lain's avatar
lain committed
140
    changes =
lain's avatar
lain committed
141
      %User{}
lain's avatar
lain committed
142
      |> cast(params, [:bio, :name, :ap_id, :nickname, :avatar])
143
      |> validate_required([:name, :ap_id])
lain's avatar
lain committed
144 145 146 147 148
      |> 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
149
      |> put_embed(:info, info_cng)
lain's avatar
lain committed
150

151
    if changes.valid? do
lain's avatar
lain committed
152
      case info_cng.changes[:source_data] do
lain's avatar
lain committed
153 154 155
        %{"followers" => followers} ->
          changes
          |> put_change(:follower_address, followers)
lain's avatar
lain committed
156

lain's avatar
lain committed
157 158
        _ ->
          followers = User.ap_followers(%User{nickname: changes.changes[:nickname]})
lain's avatar
lain committed
159

lain's avatar
lain committed
160 161 162
          changes
          |> put_change(:follower_address, followers)
      end
163 164 165
    else
      changes
    end
lain's avatar
lain committed
166 167
  end

lain's avatar
lain committed
168
  def update_changeset(struct, params \\ %{}) do
Thog's avatar
Thog committed
169
    struct
170
    |> cast(params, [:bio, :name, :avatar, :following])
lain's avatar
lain committed
171
    |> unique_constraint(:nickname)
href's avatar
href committed
172
    |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
173
    |> validate_length(:bio, max: 5000)
lain's avatar
lain committed
174 175 176
    |> validate_length(:name, min: 1, max: 100)
  end

lain's avatar
lain committed
177
  def upgrade_changeset(struct, params \\ %{}) do
178 179 180 181
    params =
      params
      |> Map.put(:last_refreshed_at, NaiveDateTime.utc_now())

lain's avatar
lain committed
182 183 184 185
    info_cng =
      struct.info
      |> User.Info.user_upgrade(params[:info])

lain's avatar
lain committed
186
    struct
lain's avatar
lain committed
187
    |> cast(params, [:bio, :name, :follower_address, :avatar, :last_refreshed_at])
lain's avatar
lain committed
188
    |> unique_constraint(:nickname)
href's avatar
href committed
189
    |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
190 191
    |> validate_length(:bio, max: 5000)
    |> validate_length(:name, max: 100)
lain's avatar
lain committed
192
    |> put_embed(:info, info_cng)
lain's avatar
lain committed
193 194
  end

Roger Braun's avatar
Roger Braun committed
195
  def password_update_changeset(struct, params) do
lain's avatar
lain committed
196 197 198 199 200
    changeset =
      struct
      |> cast(params, [:password, :password_confirmation])
      |> validate_required([:password, :password_confirmation])
      |> validate_confirmation(:password)
Roger Braun's avatar
Roger Braun committed
201

202 203 204
    OAuth.Token.delete_user_tokens(struct)
    OAuth.Authorization.delete_user_authorizations(struct)

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

Roger Braun's avatar
Roger Braun committed
208 209 210 211 212 213 214 215
      changeset
      |> put_change(:password_hash, hashed)
    else
      changeset
    end
  end

  def reset_password(user, data) do
lain's avatar
lain committed
216
    update_and_set_cache(password_update_changeset(user, data))
Roger Braun's avatar
Roger Braun committed
217 218
  end

219
  def register_changeset(struct, params \\ %{}, opts \\ []) do
220 221 222
    need_confirmation? =
      if is_nil(opts[:need_confirmation]) do
        Pleroma.Config.get([:instance, :account_activation_required])
223
      else
224
        opts[:need_confirmation]
225 226
      end

227 228
    info_change =
      User.Info.confirmation_changeset(%User.Info{}, need_confirmation: need_confirmation?)
Ivan Tashkinov's avatar
Ivan Tashkinov committed
229

lain's avatar
lain committed
230 231 232
    changeset =
      struct
      |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation])
233
      |> validate_required([:name, :nickname, :password, :password_confirmation])
lain's avatar
lain committed
234 235 236
      |> validate_confirmation(:password)
      |> unique_constraint(:email)
      |> unique_constraint(:nickname)
237
      |> validate_exclusion(:nickname, Pleroma.Config.get([User, :restricted_nicknames]))
href's avatar
href committed
238
      |> validate_format(:nickname, local_nickname_regex())
lain's avatar
lain committed
239 240 241
      |> validate_format(:email, @email_regex)
      |> validate_length(:bio, max: 1000)
      |> validate_length(:name, min: 1, max: 100)
Ivan Tashkinov's avatar
Ivan Tashkinov committed
242
      |> put_change(:info, info_change)
lain's avatar
lain committed
243

244 245 246 247 248 249 250
    changeset =
      if opts[:external] do
        changeset
      else
        validate_required(changeset, [:email])
      end

lain's avatar
lain committed
251
    if changeset.valid? do
252
      hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
lain's avatar
lain committed
253 254
      ap_id = User.ap_id(%User{nickname: changeset.changes[:nickname]})
      followers = User.ap_followers(%User{nickname: changeset.changes[:nickname]})
lain's avatar
lain committed
255

lain's avatar
lain committed
256 257 258
      changeset
      |> put_change(:password_hash, hashed)
      |> put_change(:ap_id, ap_id)
rinpatch's avatar
rinpatch committed
259
      |> unique_constraint(:ap_id)
lain's avatar
lain committed
260
      |> put_change(:following, [followers])
261
      |> put_change(:follower_address, followers)
lain's avatar
lain committed
262 263 264 265 266
    else
      changeset
    end
  end

267 268 269 270
  defp autofollow_users(user) do
    candidates = Pleroma.Config.get([:instance, :autofollowed_nicknames])

    autofollowed_users =
271
      User.Query.build(%{nickname: candidates, local: true, deactivated: false})
272 273
      |> Repo.all()

lain's avatar
lain committed
274
    follow_all(user, autofollowed_users)
275 276
  end

277 278
  @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
279
    with {:ok, user} <- Repo.insert(changeset),
lain's avatar
lain committed
280
         {:ok, user} <- autofollow_users(user),
minibikini's avatar
minibikini committed
281
         {:ok, user} <- set_cache(user),
282
         {:ok, _} <- User.WelcomeMessage.post_welcome_message_to_user(user),
lain's avatar
lain committed
283
         {:ok, _} <- try_send_confirmation_email(user) do
284 285 286 287
      {:ok, user}
    end
  end

288
  def try_send_confirmation_email(%User{} = user) do
289 290
    if user.info.confirmation_pending &&
         Pleroma.Config.get([:instance, :account_activation_required]) do
291
      user
292 293
      |> Pleroma.Emails.UserEmail.account_confirmation_email()
      |> Pleroma.Emails.Mailer.deliver_async()
294 295

      {:ok, :enqueued}
296 297 298 299 300
    else
      {:ok, :noop}
    end
  end

301 302 303 304 305
  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
306
    NaiveDateTime.diff(NaiveDateTime.utc_now(), user.last_refreshed_at) >= 86_400
307 308 309 310
  end

  def needs_update?(_), do: true

lain's avatar
lain committed
311
  def maybe_direct_follow(%User{} = follower, %User{local: true, info: %{locked: true}}) do
312 313 314 315 316 317 318 319
    {: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
320
    if not User.ap_enabled?(followed) do
321
      follow(follower, followed)
322 323 324 325 326
    else
      {:ok, follower}
    end
  end

Maksim's avatar
Maksim committed
327
  def maybe_follow(%User{} = follower, %User{info: _info} = followed) do
328 329
    if not following?(follower, followed) do
      follow(follower, followed)
330
    else
331
      {:ok, follower}
332 333 334
    end
  end

335
  @doc "A mass follow for local users. Respects blocks in both directions but does not create activities."
lain's avatar
lain committed
336 337
  @spec follow_all(User.t(), list(User.t())) :: {atom(), User.t()}
  def follow_all(follower, followeds) do
lain's avatar
lain committed
338 339
    followed_addresses =
      followeds
340
      |> Enum.reject(fn followed -> blocks?(follower, followed) || blocks?(followed, follower) end)
lain's avatar
lain committed
341
      |> Enum.map(fn %{follower_address: fa} -> fa end)
lain's avatar
lain committed
342

lain's avatar
lain committed
343 344 345
    q =
      from(u in User,
        where: u.id == ^follower.id,
346 347 348 349 350 351 352 353 354
        update: [
          set: [
            following:
              fragment(
                "array(select distinct unnest (array_cat(?, ?)))",
                u.following,
                ^followed_addresses
              )
          ]
rinpatch's avatar
rinpatch committed
355 356
        ],
        select: u
lain's avatar
lain committed
357 358
      )

rinpatch's avatar
rinpatch committed
359
    {1, [follower]} = Repo.update_all(q, [])
lain's avatar
lain committed
360 361 362 363 364

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

lain's avatar
lain committed
365
    set_cache(follower)
lain's avatar
lain committed
366 367
  end

lain's avatar
lain committed
368
  def follow(%User{} = follower, %User{info: info} = followed) do
minibikini's avatar
minibikini committed
369
    deny_follow_blocked = Pleroma.Config.get([:user, :deny_follow_blocked])
370
    ap_followers = followed.follower_address
371

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

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

379 380 381 382 383
      true ->
        if !followed.local && follower.local && !ap_enabled?(followed) do
          Websub.subscribe(follower, followed)
        end

384 385 386
        q =
          from(u in User,
            where: u.id == ^follower.id,
rinpatch's avatar
rinpatch committed
387 388
            update: [push: [following: ^ap_followers]],
            select: u
389
          )
390

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

393 394
        {:ok, _} = update_follower_count(followed)

395
        set_cache(follower)
396
    end
lain's avatar
lain committed
397
  end
lain's avatar
lain committed
398 399

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

402
    if following?(follower, followed) and follower.ap_id != followed.ap_id do
403 404 405
      q =
        from(u in User,
          where: u.id == ^follower.id,
rinpatch's avatar
rinpatch committed
406 407
          update: [pull: [following: ^ap_followers]],
          select: u
408
        )
lain's avatar
lain committed
409

rinpatch's avatar
rinpatch committed
410
      {1, [follower]} = Repo.update_all(q, [])
411 412 413

      {:ok, followed} = update_follower_count(followed)

414 415
      set_cache(follower)

416
      {:ok, follower, Utils.fetch_latest_follow(follower, followed)}
417
    else
418
      {:error, "Not subscribed!"}
419
    end
lain's avatar
lain committed
420
  end
421

Maksim's avatar
Maksim committed
422
  @spec following?(User.t(), User.t()) :: boolean
423
  def following?(%User{} = follower, %User{} = followed) do
424
    Enum.member?(follower.following, followed.follower_address)
425
  end
lain's avatar
lain committed
426

427
  def locked?(%User{} = user) do
428
    user.info.locked || false
429 430
  end

431 432 433 434
  def get_by_id(id) do
    Repo.get_by(User, id: id)
  end

lain's avatar
lain committed
435 436 437 438
  def get_by_ap_id(ap_id) do
    Repo.get_by(User, ap_id: ap_id)
  end

439 440
  # 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
441 442 443 444 445
  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
446
    get_cached_by_nickname(nickname)
447 448
  end

minibikini's avatar
minibikini committed
449 450 451 452
  def set_cache({:ok, user}), do: set_cache(user)
  def set_cache({:error, err}), do: {:error, err}

  def set_cache(%User{} = user) do
453 454 455 456 457 458
    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
459 460
  def update_and_set_cache(changeset) do
    with {:ok, user} <- Repo.update(changeset) do
461
      set_cache(user)
lain's avatar
lain committed
462 463 464 465 466
    else
      e -> e
    end
  end

lain's avatar
lain committed
467 468 469
  def invalidate_cache(user) do
    Cachex.del(:user_cache, "ap_id:#{user.ap_id}")
    Cachex.del(:user_cache, "nickname:#{user.nickname}")
470
    Cachex.del(:user_cache, "user_info:#{user.id}")
lain's avatar
lain committed
471 472
  end

lain's avatar
lain committed
473
  def get_cached_by_ap_id(ap_id) do
474
    key = "ap_id:#{ap_id}"
Thog's avatar
Thog committed
475
    Cachex.fetch!(:user_cache, key, fn _ -> get_by_ap_id(ap_id) end)
lain's avatar
lain committed
476 477
  end

478 479
  def get_cached_by_id(id) do
    key = "id:#{id}"
480 481 482 483

    ap_id =
      Cachex.fetch!(:user_cache, key, fn _ ->
        user = get_by_id(id)
484 485 486 487 488 489 490

        if user do
          Cachex.put(:user_cache, "ap_id:#{user.ap_id}", user)
          {:commit, user.ap_id}
        else
          {:ignore, ""}
        end
491 492 493
      end)

    get_cached_by_ap_id(ap_id)
494 495
  end

lain's avatar
lain committed
496
  def get_cached_by_nickname(nickname) do
497
    key = "nickname:#{nickname}"
0x1C3B00DA's avatar
Run  
0x1C3B00DA committed
498

499 500 501 502 503
    Cachex.fetch!(:user_cache, key, fn ->
      user_result = get_or_fetch_by_nickname(nickname)

      case user_result do
        {:ok, user} -> {:commit, user}
Alexander Strizhakov's avatar
Alexander Strizhakov committed
504
        {:error, _error} -> {:ignore, nil}
505 506
      end
    end)
lain's avatar
lain committed
507
  end
lain's avatar
lain committed
508

509
  def get_cached_by_nickname_or_id(nickname_or_id) do
510
    get_cached_by_id(nickname_or_id) || get_cached_by_nickname(nickname_or_id)
511 512
  end

lain's avatar
lain committed
513
  def get_by_nickname(nickname) do
514
    Repo.get_by(User, nickname: nickname) ||
515
      if Regex.match?(~r(@#{Pleroma.Web.Endpoint.host()})i, nickname) do
516
        Repo.get_by(User, nickname: local_nickname(nickname))
517
      end
518 519
  end

520 521
  def get_by_email(email), do: Repo.get_by(User, email: email)

522
  def get_by_nickname_or_email(nickname_or_email) do
523
    get_by_nickname(nickname_or_email) || get_by_email(nickname_or_email)
524 525
  end

lain's avatar
lain committed
526 527
  def get_cached_user_info(user) do
    key = "user_info:#{user.id}"
Thog's avatar
Thog committed
528
    Cachex.fetch!(:user_cache, key, fn _ -> user_info(user) end)
lain's avatar
lain committed
529
  end
lain's avatar
lain committed
530

lain's avatar
lain committed
531 532 533 534 535 536 537 538 539
  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
540
  def get_or_fetch_by_nickname(nickname) do
lain's avatar
lain committed
541
    with %User{} = user <- get_by_nickname(nickname) do
542
      {:ok, user}
lain's avatar
lain committed
543 544 545 546
    else
      _e ->
        with [_nick, _domain] <- String.split(nickname, "@"),
             {:ok, user} <- fetch_by_nickname(nickname) do
547
          if Pleroma.Config.get([:fetch_initial_posts, :enabled]) do
548
            fetch_initial_posts(user)
549 550
          end

551
          {:ok, user}
lain's avatar
lain committed
552
        else
Alexander Strizhakov's avatar
Alexander Strizhakov committed
553
          _e -> {:error, "not found " <> nickname}
lain's avatar
lain committed
554
        end
lain's avatar
lain committed
555
    end
lain's avatar
lain committed
556
  end
lain's avatar
lain committed
557

558
  @doc "Fetch some posts when the user has just been federated with"
559 560
  def fetch_initial_posts(user),
    do: PleromaJobQueue.enqueue(:background, __MODULE__, [:fetch_initial_posts, user])
561

Alexander Strizhakov's avatar
Alexander Strizhakov committed
562 563
  @spec get_followers_query(User.t(), pos_integer() | nil) :: Ecto.Query.t()
  def get_followers_query(%User{} = user, nil) do
564
    User.Query.build(%{followers: user, deactivated: false})
565 566
  end

567
  def get_followers_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
568
    from(u in get_followers_query(user, nil))
Alexander Strizhakov's avatar
Alexander Strizhakov committed
569
    |> User.Query.paginate(page, 20)
570 571
  end

Alexander Strizhakov's avatar
Alexander Strizhakov committed
572
  @spec get_followers_query(User.t()) :: Ecto.Query.t()
573 574 575 576
  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
577 578 579 580

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

581 582 583 584 585 586
  def get_followers_ids(user, page \\ nil) do
    q = get_followers_query(user, page)

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

Alexander Strizhakov's avatar
Alexander Strizhakov committed
587 588
  @spec get_friends_query(User.t(), pos_integer() | nil) :: Ecto.Query.t()
  def get_friends_query(%User{} = user, nil) do
589
    User.Query.build(%{friends: user, deactivated: false})
590 591
  end

592
  def get_friends_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
593
    from(u in get_friends_query(user, nil))
Alexander Strizhakov's avatar
Alexander Strizhakov committed
594
    |> User.Query.paginate(page, 20)
595 596
  end

Alexander Strizhakov's avatar
Alexander Strizhakov committed
597
  @spec get_friends_query(User.t()) :: Ecto.Query.t()
598 599 600 601
  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
602 603 604

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

606 607 608 609 610 611
  def get_friends_ids(user, page \\ nil) do
    q = get_friends_query(user, page)

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

Alexander Strizhakov's avatar
Alexander Strizhakov committed
612
  @spec get_follow_requests(User.t()) :: {:ok, [User.t()]}
613 614
  def get_follow_requests(%User{} = user) do
    users =
Alexander Strizhakov's avatar
Alexander Strizhakov committed
615
      Activity.follow_requests_for_actor(user)
rinpatch's avatar
rinpatch committed
616
      |> join(:inner, [a], u in User, on: a.actor == u.ap_id)
617 618 619 620
      |> where([a, u], not fragment("? @> ?", u.following, ^[user.follower_address]))
      |> group_by([a, u], u.id)
      |> select([a, u], u)
      |> Repo.all()
621 622 623 624

    {:ok, users}
  end

625
  def increase_note_count(%User{} = user) do
626 627 628 629 630 631 632 633 634 635 636 637
    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
638 639
    |> select([u], u)
    |> Repo.update_all([])
640 641 642 643
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
644 645
  end

646
  def decrease_note_count(%User{} = user) do
647 648 649 650 651 652 653 654 655 656 657 658
    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
659 660
    |> select([u], u)
    |> Repo.update_all([])
661 662 663 664
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
665 666
  end

667
  def update_note_count(%User{} = user) do
lain's avatar
lain committed
668 669 670 671 672 673
    note_count_query =
      from(
        a in Object,
        where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data),
        select: count(a.id)
      )
674 675 676

    note_count = Repo.one(note_count_query)

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

679 680 681 682
    user
    |> change()
    |> put_embed(:info, info_cng)
    |> update_and_set_cache()
683 684 685
  end

  def update_follower_count(%User{} = user) do
lain's avatar
lain committed
686
    follower_count_query =
687
      User.Query.build(%{followers: user, deactivated: false})
688
      |> select([u], %{count: count(u.id)})
689

690 691 692 693 694 695 696 697 698 699 700 701 702
    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
703 704
    |> select([u], u)
    |> Repo.update_all([])
705 706 707 708
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
709
  end
710

711 712 713 714 715 716 717 718 719 720 721 722
  def remove_duplicated_following(%User{following: following} = user) do
    uniq_following = Enum.uniq(following)

    if length(following) == length(uniq_following) do
      {:ok, user}
    else
      user
      |> update_changeset(%{following: uniq_following})
      |> update_and_set_cache()
    end
  end

Alexander Strizhakov's avatar
Alexander Strizhakov committed
723
  @spec get_users_from_set([String.t()], boolean()) :: [User.t()]
724
  def get_users_from_set(ap_ids, local_only \\ true) do
725
    criteria = %{ap_id: ap_ids, deactivated: false}
Alexander Strizhakov's avatar
Alexander Strizhakov committed
726 727 728
    criteria = if local_only, do: Map.put(criteria, :local, true), else: criteria

    User.Query.build(criteria)
729 730 731
    |> Repo.all()
  end

Alexander Strizhakov's avatar
Alexander Strizhakov committed
732
  @spec get_recipients_from_activity(Activity.t()) :: [User.t()]
733
  def get_recipients_from_activity(%Activity{recipients: to}) do
734
    User.Query.build(%{recipients_from_activity: to, local: true, deactivated: false})
Alexander Strizhakov's avatar
Alexander Strizhakov committed
735
    |> Repo.all()
736 737
  end

738
  def search(query, resolve \\ false, for_user \\ nil) do
739
    # Strip the beginning @ off if there is a query
740 741
    query = String.trim_leading(query, "@")

742
    if resolve, do: get_or_fetch(query)
lain's avatar
lain committed
743

744
    {:ok, results} =
lain's avatar
lain committed
745 746
      Repo.transaction(fn ->
        Ecto.Adapters.SQL.query(Repo, "select set_limit(0.25)", [])
747
        Repo.all(search_query(query, for_user))
lain's avatar
lain committed
748
      end)
lain's avatar
lain committed
749

750
    results
751
  end
lain's avatar
lain committed
752

753 754 755
  def search_query(query, for_user) do
    fts_subquery = fts_search_subquery(query)
    trigram_subquery = trigram_search_subquery(query)
756 757
    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
758

759 760
    from(s in subquery(boost_search_rank_query(distinct_query, for_user)),
      order_by: [desc: s.search_rank],
761
      limit: 40
762 763
    )
  end
764

765 766 767 768 769
  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)
770

771 772 773 774 775
    from(u in subquery(query),
      select_merge: %{
        search_rank:
          fragment(
            """
776
             CASE WHEN (?) THEN (?) * 1.3
777 778 779 780 781 782 783 784 785 786 787 788 789 790
             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
          )
      }
    )
791
  end
792

Maxim Filippov's avatar
Maxim Filippov committed
793
  defp fts_search_subquery(term, query \\ User) do
794
    processed_query =
795
      term
796 797 798 799 800
      |> String.replace(~r/\W+/, " ")
      |> String.trim()
      |> String.split()
      |> Enum.map(&(&1 <> ":*"))
      |> Enum.join(" | ")
801

802
    from(
803
      u in query,
804
      select_merge: %{
805
        search_type: ^0,
806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
        search_rank:
          fragment(
            """
            ts_rank_cd(
              setweight(to_tsvector('simple', regexp_replace(?, '\\W', ' ', 'g')), 'A') ||
              setweight(to_tsvector('simple', regexp_replace(coalesce(?, ''), '\\W', ' ', 'g')), 'B'),
              to_tsquery('simple', ?),
              32
            )
            """,
            u.nickname,
            u.name,
            ^processed_query
          )
      },
lain's avatar
lain committed
821 822 823 824 825 826 827 828 829 830
      where:
        fragment(
          """
            (setweight(to_tsvector('simple', regexp_replace(?, '\\W', ' ', 'g')), 'A') ||
            setweight(to_tsvector('simple', regexp_replace(coalesce(?, ''), '\\W', ' ', 'g')), 'B')) @@ to_tsquery('simple', ?)
          """,
          u.nickname,
          u.name,
          ^processed_query
        )
831
    )
832
    |> restrict_deactivated()
833
  end
834

Maxim Filippov's avatar
Maxim Filippov committed
835
  defp trigram_search_subquery(term) do
836 837 838
    from(
      u in User,
      select_merge: %{
839 840
        # ^1 gives 'Postgrex expected a binary, got 1' for some weird reason
        search_type: fragment("?", 1),
841 842
        search_rank:
          fragment(
843
            "similarity(?, trim(? || ' ' || coalesce(?, '')))",
844
            ^term,
845 846 847 848