user.ex 37.8 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),
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
284
285
      user
      |> 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
426
427
428
429
430
431
432
433
434
435
436
437
438
  def follow_import(%User{} = follower, followed_identifiers)
      when is_list(followed_identifiers) do
    Enum.map(
      followed_identifiers,
      fn followed_identifier ->
        with %User{} = followed <- get_or_fetch(followed_identifier),
             {: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}"
Thog's avatar
Thog committed
510
    Cachex.fetch!(:user_cache, key, fn _ -> get_or_fetch_by_nickname(nickname) end)
lain's avatar
lain committed
511
  end
lain's avatar
lain committed
512

513
  def get_cached_by_nickname_or_id(nickname_or_id) do
514
    get_cached_by_id(nickname_or_id) || get_cached_by_nickname(nickname_or_id)
515
516
  end

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

524
525
  def get_by_email(email), do: Repo.get_by(User, email: email)

526
  def get_by_nickname_or_email(nickname_or_email) do
527
    get_by_nickname(nickname_or_email) || get_by_email(nickname_or_email)
528
529
  end

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

lain's avatar
lain committed
535
536
537
538
539
540
541
542
543
  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
544
  def get_or_fetch_by_nickname(nickname) do
lain's avatar
lain committed
545
    with %User{} = user <- get_by_nickname(nickname) do
lain's avatar
lain committed
546
      user
lain's avatar
lain committed
547
548
549
550
    else
      _e ->
        with [_nick, _domain] <- String.split(nickname, "@"),
             {:ok, user} <- fetch_by_nickname(nickname) do
551
          if Pleroma.Config.get([:fetch_initial_posts, :enabled]) do
minibikini's avatar
minibikini committed
552
            # TODO turn into job
553
554
555
            {:ok, _} = Task.start(__MODULE__, :fetch_initial_posts, [user])
          end

lain's avatar
lain committed
556
557
558
559
          user
        else
          _e -> nil
        end
lain's avatar
lain committed
560
    end
lain's avatar
lain committed
561
  end
lain's avatar
lain committed
562

563
564
565
566
567
568
569
570
571
572
573
  @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

574
  def get_followers_query(%User{id: id, follower_address: follower_address}, nil) do
575
576
577
578
579
580
581
    from(
      u in User,
      where: fragment("? <@ ?", ^[follower_address], u.following),
      where: u.id != ^id
    )
  end

582
  def get_followers_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
583
584
    from(u in get_followers_query(user, nil))
    |> paginate(page, 20)
585
586
587
588
589
590
  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
591
592
593
594

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

595
596
597
598
599
600
  def get_followers_ids(user, page \\ nil) do
    q = get_followers_query(user, page)

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

601
  def get_friends_query(%User{id: id, following: following}, nil) do
602
603
604
605
606
607
608
    from(
      u in User,
      where: u.follower_address in ^following,
      where: u.id != ^id
    )
  end

609
  def get_friends_query(user, page) do
Maxim Filippov's avatar
Maxim Filippov committed
610
611
    from(u in get_friends_query(user, nil))
    |> paginate(page, 20)
612
613
614
615
616
617
  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
618
619
620

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

622
623
624
625
626
627
  def get_friends_ids(user, page \\ nil) do
    q = get_friends_query(user, page)

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

628
629
630
  def get_follow_requests_query(%User{} = user) do
    from(
      a in Activity,
kaniini's avatar
kaniini committed
631
632
633
634
635
636
637
638
639
640
641
642
      where:
        fragment(
          "? ->> 'type' = 'Follow'",
          a.data
        ),
      where:
        fragment(
          "? ->> 'state' = 'pending'",
          a.data
        ),
      where:
        fragment(
643
          "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
kaniini's avatar
kaniini committed
644
          a.data,
645
646
          a.data,
          ^user.ap_id
kaniini's avatar
kaniini committed
647
        )
648
649
650
651
652
    )
  end

  def get_follow_requests(%User{} = user) do
    users =
653
654
      user
      |> User.get_follow_requests_query()
rinpatch's avatar
rinpatch committed
655
      |> join(:inner, [a], u in User, on: a.actor == u.ap_id)
656
657
658
659
      |> where([a, u], not fragment("? @> ?", u.following, ^[user.follower_address]))
      |> group_by([a, u], u.id)
      |> select([a, u], u)
      |> Repo.all()
660
661
662
663

    {:ok, users}
  end

664
  def increase_note_count(%User{} = user) do
665
666
667
668
669
670
671
672
673
674
675
676
    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
677
678
    |> select([u], u)
    |> Repo.update_all([])
679
680
681
682
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
683
684
  end

685
  def decrease_note_count(%User{} = user) do
686
687
688
689
690
691
692
693
694
695
696
697
    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
698
699
    |> select([u], u)
    |> Repo.update_all([])
700
701
702
703
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
704
705
  end

706
  def update_note_count(%User{} = user) do
lain's avatar
lain committed
707
708
709
710
711
712
    note_count_query =
      from(
        a in Object,
        where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data),
        select: count(a.id)
      )
713
714
715

    note_count = Repo.one(note_count_query)

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

lain's avatar
lain committed
718
719
720
    cng =
      change(user)
      |> put_embed(:info, info_cng)
721

lain's avatar
lain committed
722
    update_and_set_cache(cng)
723
724
725
  end

  def update_follower_count(%User{} = user) do
lain's avatar
lain committed
726
    follower_count_query =
727
728
729
730
      User
      |> where([u], ^user.follower_address in u.following)
      |> where([u], u.id != ^user.id)
      |> select([u], %{count: count(u.id)})
731

732
733
734
735
736
737
738
739
740
741
742
743
744
    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
745
746
    |> select([u], u)
    |> Repo.update_all([])
747
748
749
750
    |> case do
      {1, [user]} -> set_cache(user)
      _ -> {:error, user}
    end
751
  end
752

753
  def get_users_from_set_query(ap_ids, false) do
754
755
    from(
      u in User,
756
      where: u.ap_id in ^ap_ids
757
758
759
    )
  end

760
761
  def get_users_from_set_query(ap_ids, true) do
    query = get_users_from_set_query(ap_ids, false)
762
763
764

    from(
      u in query,
765
766
767
768
      where: u.local == true
    )
  end

769
770
771
772
773
  def get_users_from_set(ap_ids, local_only \\ true) do
    get_users_from_set_query(ap_ids, local_only)
    |> Repo.all()
  end

774
  def get_recipients_from_activity(%Activity{recipients: to}) do
lain's avatar
lain committed
775
776
777
778
779
780
    query =
      from(
        u in User,
        where: u.ap_id in ^to,
        or_where: fragment("? && ?", u.following, ^to)
      )
781

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

784
785
786
    Repo.all(query)
  end

787
  def search(query, resolve \\ false, for_user \\ nil) do
788
    # Strip the beginning @ off if there is a query
789
790
    query = String.trim_leading(query, "@")

791
    if resolve, do: get_or_fetch(query)
lain's avatar
lain committed
792

793
    {:ok, results} =
lain's avatar
lain committed
794
795
      Repo.transaction(fn ->
        Ecto.Adapters.SQL.query(Repo, "select set_limit(0.25)", [])
796
        Repo.all(search_query(query, for_user))
lain's avatar
lain committed
797
      end)
lain's avatar
lain committed
798

799
    results
800
  end
lain's avatar
lain committed
801

802
803
804
  def search_query(query, for_user) do
    fts_subquery = fts_search_subquery(query)
    trigram_subquery = trigram_search_subquery(query)
805
806
    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
807

808
809
810
811
812
    from(s in subquery(boost_search_rank_query(distinct_query, for_user)),
      order_by: [desc: s.search_rank],
      limit: 20
    )
  end
813

814
815
816
817
818
819
820
821
822
823
824
  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)

    from(u in subquery(query),
      select_merge: %{
        search_rank:
          fragment(
            """
825
             CASE WHEN (?) THEN (?) * 1.3
826
827
828
829
830
831
832
833
834
835
836
837
838
839
             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
          )
      }
    )
840
  end
841

Maxim Filippov's avatar
Maxim Filippov committed
842
  defp fts_search_subquery(term, query \\ User) do
843
    processed_query =
844
      term
845
846
847
848
849
      |> String.replace(~r/\W+/, " ")
      |> String.trim()
      |> String.split()
      |> Enum.map(&(&1 <> ":*"))
      |> Enum.join(" | ")
850

851
    from(
852
      u in query,
853
      select_merge: %{
854
        search_type: ^0,
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
        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
870
871
872
873
874
875
876
877
878
879
      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
        )
880
881
    )
  end
882

Maxim Filippov's avatar
Maxim Filippov committed
883
  defp trigram_search_subquery(term) do
884
885
886
    from(
      u in User,
      select_merge: %{
887
888
        # ^1 gives 'Postgrex expected a binary, got 1' for some weird reason
        search_type: fragment("?", 1),
889
890
        search_rank:
          fragment(
891
            "similarity(?, trim(? || ' ' || coalesce(?, '')))",
892
            ^term,
893
894
895
896
            u.nickname,
            u.name
          )
      },
897
      where: fragment("trim(? || ' ' || coalesce(?, '')) % ?", u.nickname, u.name, ^term)
898
899
900
    )
  end

901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
  def blocks_import(%User{} = blocker, blocked_identifiers) when is_list(blocked_identifiers) do
    Enum.map(
      blocked_identifiers,
      fn blocked_identifier ->
        with %User{} = blocked <- get_or_fetch(blocked_identifier),
             {:ok, blocker} <- block(blocker, blocked),
             {:ok, _} <- ActivityPub.block(blocker, blocked) do
          blocked
        else
          err ->
            Logger.debug("blocks_import failed for #{blocked_identifier} with: #{inspect(err)}")
            err
        end
      end
    )
  end

918
  def mute(muter, %User{ap_id: ap_id}) do
919
920
921
922
923
924
925
    info_cng =
      muter.info
      |> User.Info.add_to_mutes(ap_id)

    cng =
      change(muter)
      |> put_embed(:info, info_cng)
926

927
    update_and_set_cache(cng)
928
929
  end

930
931
932
933
  def unmute(muter, %{ap_id: ap_id}) do
    info_cng =
      muter.info
      |> User.Info.remove_from_mutes(ap_id)
934

935
936
937
938
939
    cng =
      change(muter)
      |> put_embed(:info, info_cng)

    update_and_set_cache(cng)
940
941
  end

942
  def subscribe(subscriber, %{ap_id: ap_id}) do
943
    deny_follow_blocked = Pleroma.Config.get([:user, :deny_follow_blocked])
944

945
    with %User{} = subscribed <- get_cached_by_ap_id(ap_id) do
946
947
948
949
950
951
952
953
954
955
956
957
958
      blocked = blocks?(subscribed, subscriber) and deny_follow_blocked

      if blocked do
        {:error, "Could not subscribe: #{subscribed.nickname} is blocking you"}
      else
        info_cng =
          subscribed.info
          |> User.Info.add_to_subscribers(subscriber.ap_id)

        change(subscribed)
        |> put_embed(:info, info_cng)
        |> update_and_set_cache()
      end
959
    end
960
961
962
  end

  def unsubscribe(unsubscriber, %{ap_id: ap_id}) do
963
    with %User{} = user <- get_cached_by_ap_id(ap_id) do
964
965
966
      info_cng =
        user.info
        |> User.Info.remove_from_subscribers(unsubscriber.ap_id)
967

968
969
970
971
      change(user)
      |> put_embed(:info, info_cng)
      |> update_and_set_cache()
    end
972
973
  end

974
975
976
977
978
979
980
981
982
983
  def block(blocker, %User{ap_id: ap_id} = blocked) do
    # sever any follow relationships to prevent leaks per activitypub (Pleroma issue #213)
    blocker =
      if following?(blocker, blocked) do
        {:ok, blocker, _} = unfollow(blocker, blocked)
        blocker
      else
        blocker
      end

984
985
986
987
988
989
990
991
    blocker =
      if subscribed_to?(blocked, blocker) do
        {:ok, blocker} = unsubscribe(blocked, blocker)
        blocker
      else
        blocker
      end

992
993
994
995
    if following?(blocked, blocker) do
      unfollow(blocked, blocker)
    end

996
997
    {:ok, blocker} = update_follower_count(blocker)

lain's avatar
lain committed
998
999
1000
    info_cng =
      blocker.info
      |> User.Info.add_to_block(ap_id)
lain's avatar
lain committed
1001

lain's avatar
lain committed
1002
1003
1004
1005
1006
    cng =
      change(blocker)
      |> put_embed(:info, info_cng)

    update_and_set_cache(cng)
lain's avatar
lain committed
1007
1008
  end

1009
1010
  # helper to handle the block given only an actor's AP id
  def block(blocker, %{ap_id: ap_id}) do
minibikini's avatar
minibikini committed
1011
    block(blocker, get_cached_by_ap_id(ap_id))
1012
1013
  end