[#483] User.get_by_nickname/1: allowed retrieving user by fully-qualified local nickname
[akkoma] / lib / pleroma / user.ex
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.User do
6 use Ecto.Schema
7
8 import Ecto.{Changeset, Query}
9 alias Pleroma.{Repo, User, Object, Web, Activity, Notification}
10 alias Comeonin.Pbkdf2
11 alias Pleroma.Formatter
12 alias Pleroma.Web.CommonAPI.Utils, as: CommonUtils
13 alias Pleroma.Web.{OStatus, Websub, OAuth}
14 alias Pleroma.Web.ActivityPub.{Utils, ActivityPub}
15
16 require Logger
17
18 @type t :: %__MODULE__{}
19
20 @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])?)*$/
21
22 @strict_local_nickname_regex ~r/^[a-zA-Z\d]+$/
23 @extended_local_nickname_regex ~r/^[a-zA-Z\d_-]+$/
24
25 schema "users" do
26 field(:bio, :string)
27 field(:email, :string)
28 field(:name, :string)
29 field(:nickname, :string)
30 field(:password_hash, :string)
31 field(:password, :string, virtual: true)
32 field(:password_confirmation, :string, virtual: true)
33 field(:following, {:array, :string}, default: [])
34 field(:ap_id, :string)
35 field(:avatar, :map)
36 field(:local, :boolean, default: true)
37 field(:follower_address, :string)
38 field(:search_distance, :float, virtual: true)
39 field(:tags, {:array, :string}, default: [])
40 field(:last_refreshed_at, :naive_datetime)
41 has_many(:notifications, Notification)
42 embeds_one(:info, Pleroma.User.Info)
43
44 timestamps()
45 end
46
47 def auth_active?(%User{} = user) do
48 (user.info && !user.info.confirmation_pending) ||
49 !Pleroma.Config.get([:instance, :account_activation_required])
50 end
51
52 def superuser?(%User{} = user), do: user.info && User.Info.superuser?(user.info)
53
54 def avatar_url(user) do
55 case user.avatar do
56 %{"url" => [%{"href" => href} | _]} -> href
57 _ -> "#{Web.base_url()}/images/avi.png"
58 end
59 end
60
61 def banner_url(user) do
62 case user.info.banner do
63 %{"url" => [%{"href" => href} | _]} -> href
64 _ -> "#{Web.base_url()}/images/banner.png"
65 end
66 end
67
68 def profile_url(%User{info: %{source_data: %{"url" => url}}}), do: url
69 def profile_url(%User{ap_id: ap_id}), do: ap_id
70 def profile_url(_), do: nil
71
72 def ap_id(%User{nickname: nickname}) do
73 "#{Web.base_url()}/users/#{nickname}"
74 end
75
76 def ap_followers(%User{} = user) do
77 "#{ap_id(user)}/followers"
78 end
79
80 def follow_changeset(struct, params \\ %{}) do
81 struct
82 |> cast(params, [:following])
83 |> validate_required([:following])
84 end
85
86 def user_info(%User{} = user) do
87 oneself = if user.local, do: 1, else: 0
88
89 %{
90 following_count: length(user.following) - oneself,
91 note_count: user.info.note_count,
92 follower_count: user.info.follower_count,
93 locked: user.info.locked,
94 confirmation_pending: user.info.confirmation_pending,
95 default_scope: user.info.default_scope
96 }
97 end
98
99 def remote_user_creation(params) do
100 params =
101 params
102 |> Map.put(:info, params[:info] || %{})
103
104 info_cng = User.Info.remote_user_creation(%User.Info{}, params[:info])
105
106 changes =
107 %User{}
108 |> cast(params, [:bio, :name, :ap_id, :nickname, :avatar])
109 |> validate_required([:name, :ap_id])
110 |> unique_constraint(:nickname)
111 |> validate_format(:nickname, @email_regex)
112 |> validate_length(:bio, max: 5000)
113 |> validate_length(:name, max: 100)
114 |> put_change(:local, false)
115 |> put_embed(:info, info_cng)
116
117 if changes.valid? do
118 case info_cng.changes[:source_data] do
119 %{"followers" => followers} ->
120 changes
121 |> put_change(:follower_address, followers)
122
123 _ ->
124 followers = User.ap_followers(%User{nickname: changes.changes[:nickname]})
125
126 changes
127 |> put_change(:follower_address, followers)
128 end
129 else
130 changes
131 end
132 end
133
134 def update_changeset(struct, params \\ %{}) do
135 struct
136 |> cast(params, [:bio, :name, :avatar])
137 |> unique_constraint(:nickname)
138 |> validate_format(:nickname, local_nickname_regex())
139 |> validate_length(:bio, max: 5000)
140 |> validate_length(:name, min: 1, max: 100)
141 end
142
143 def upgrade_changeset(struct, params \\ %{}) do
144 params =
145 params
146 |> Map.put(:last_refreshed_at, NaiveDateTime.utc_now())
147
148 info_cng =
149 struct.info
150 |> User.Info.user_upgrade(params[:info])
151
152 struct
153 |> cast(params, [:bio, :name, :follower_address, :avatar, :last_refreshed_at])
154 |> unique_constraint(:nickname)
155 |> validate_format(:nickname, local_nickname_regex())
156 |> validate_length(:bio, max: 5000)
157 |> validate_length(:name, max: 100)
158 |> put_embed(:info, info_cng)
159 end
160
161 def password_update_changeset(struct, params) do
162 changeset =
163 struct
164 |> cast(params, [:password, :password_confirmation])
165 |> validate_required([:password, :password_confirmation])
166 |> validate_confirmation(:password)
167
168 OAuth.Token.delete_user_tokens(struct)
169 OAuth.Authorization.delete_user_authorizations(struct)
170
171 if changeset.valid? do
172 hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
173
174 changeset
175 |> put_change(:password_hash, hashed)
176 else
177 changeset
178 end
179 end
180
181 def reset_password(user, data) do
182 update_and_set_cache(password_update_changeset(user, data))
183 end
184
185 def register_changeset(struct, params \\ %{}, opts \\ []) do
186 confirmation_status =
187 if opts[:confirmed] || !Pleroma.Config.get([:instance, :account_activation_required]) do
188 :confirmed
189 else
190 :unconfirmed
191 end
192
193 info_change = User.Info.confirmation_changeset(%User.Info{}, confirmation_status)
194
195 changeset =
196 struct
197 |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation])
198 |> validate_required([:email, :name, :nickname, :password, :password_confirmation])
199 |> validate_confirmation(:password)
200 |> unique_constraint(:email)
201 |> unique_constraint(:nickname)
202 |> validate_exclusion(:nickname, Pleroma.Config.get([Pleroma.User, :restricted_nicknames]))
203 |> validate_format(:nickname, local_nickname_regex())
204 |> validate_format(:email, @email_regex)
205 |> validate_length(:bio, max: 1000)
206 |> validate_length(:name, min: 1, max: 100)
207 |> put_change(:info, info_change)
208
209 if changeset.valid? do
210 hashed = Pbkdf2.hashpwsalt(changeset.changes[:password])
211 ap_id = User.ap_id(%User{nickname: changeset.changes[:nickname]})
212 followers = User.ap_followers(%User{nickname: changeset.changes[:nickname]})
213
214 changeset
215 |> put_change(:password_hash, hashed)
216 |> put_change(:ap_id, ap_id)
217 |> put_change(:following, [followers])
218 |> put_change(:follower_address, followers)
219 else
220 changeset
221 end
222 end
223
224 @doc "Inserts provided changeset, performs post-registration actions (confirmation email sending etc.)"
225 def register(%Ecto.Changeset{} = changeset) do
226 with {:ok, user} <- Repo.insert(changeset),
227 {:ok, _} = try_send_confirmation_email(user) do
228 {:ok, user}
229 end
230 end
231
232 def try_send_confirmation_email(%User{} = user) do
233 if user.info.confirmation_pending &&
234 Pleroma.Config.get([:instance, :account_activation_required]) do
235 user
236 |> Pleroma.UserEmail.account_confirmation_email()
237 |> Pleroma.Mailer.deliver()
238 else
239 {:ok, :noop}
240 end
241 end
242
243 def needs_update?(%User{local: true}), do: false
244
245 def needs_update?(%User{local: false, last_refreshed_at: nil}), do: true
246
247 def needs_update?(%User{local: false} = user) do
248 NaiveDateTime.diff(NaiveDateTime.utc_now(), user.last_refreshed_at) >= 86400
249 end
250
251 def needs_update?(_), do: true
252
253 def maybe_direct_follow(%User{} = follower, %User{local: true, info: %{locked: true}}) do
254 {:ok, follower}
255 end
256
257 def maybe_direct_follow(%User{} = follower, %User{local: true} = followed) do
258 follow(follower, followed)
259 end
260
261 def maybe_direct_follow(%User{} = follower, %User{} = followed) do
262 if not User.ap_enabled?(followed) do
263 follow(follower, followed)
264 else
265 {:ok, follower}
266 end
267 end
268
269 def maybe_follow(%User{} = follower, %User{info: _info} = followed) do
270 if not following?(follower, followed) do
271 follow(follower, followed)
272 else
273 {:ok, follower}
274 end
275 end
276
277 def follow(%User{} = follower, %User{info: info} = followed) do
278 user_config = Application.get_env(:pleroma, :user)
279 deny_follow_blocked = Keyword.get(user_config, :deny_follow_blocked)
280
281 ap_followers = followed.follower_address
282
283 cond do
284 following?(follower, followed) or info.deactivated ->
285 {:error, "Could not follow user: #{followed.nickname} is already on your list."}
286
287 deny_follow_blocked and blocks?(followed, follower) ->
288 {:error, "Could not follow user: #{followed.nickname} blocked you."}
289
290 true ->
291 if !followed.local && follower.local && !ap_enabled?(followed) do
292 Websub.subscribe(follower, followed)
293 end
294
295 following =
296 [ap_followers | follower.following]
297 |> Enum.uniq()
298
299 follower =
300 follower
301 |> follow_changeset(%{following: following})
302 |> update_and_set_cache
303
304 {:ok, _} = update_follower_count(followed)
305
306 follower
307 end
308 end
309
310 def unfollow(%User{} = follower, %User{} = followed) do
311 ap_followers = followed.follower_address
312
313 if following?(follower, followed) and follower.ap_id != followed.ap_id do
314 following =
315 follower.following
316 |> List.delete(ap_followers)
317
318 {:ok, follower} =
319 follower
320 |> follow_changeset(%{following: following})
321 |> update_and_set_cache
322
323 {:ok, followed} = update_follower_count(followed)
324
325 {:ok, follower, Utils.fetch_latest_follow(follower, followed)}
326 else
327 {:error, "Not subscribed!"}
328 end
329 end
330
331 @spec following?(User.t(), User.t()) :: boolean
332 def following?(%User{} = follower, %User{} = followed) do
333 Enum.member?(follower.following, followed.follower_address)
334 end
335
336 def follow_import(%User{} = follower, followed_identifiers)
337 when is_list(followed_identifiers) do
338 Enum.map(
339 followed_identifiers,
340 fn followed_identifier ->
341 with %User{} = followed <- get_or_fetch(followed_identifier),
342 {:ok, follower} <- maybe_direct_follow(follower, followed),
343 {:ok, _} <- ActivityPub.follow(follower, followed) do
344 followed
345 else
346 err ->
347 Logger.debug("follow_import failed for #{followed_identifier} with: #{inspect(err)}")
348 err
349 end
350 end
351 )
352 end
353
354 def locked?(%User{} = user) do
355 user.info.locked || false
356 end
357
358 def get_by_ap_id(ap_id) do
359 Repo.get_by(User, ap_id: ap_id)
360 end
361
362 def update_and_set_cache(changeset) do
363 with {:ok, user} <- Repo.update(changeset) do
364 Cachex.put(:user_cache, "ap_id:#{user.ap_id}", user)
365 Cachex.put(:user_cache, "nickname:#{user.nickname}", user)
366 Cachex.put(:user_cache, "user_info:#{user.id}", user_info(user))
367 {:ok, user}
368 else
369 e -> e
370 end
371 end
372
373 def invalidate_cache(user) do
374 Cachex.del(:user_cache, "ap_id:#{user.ap_id}")
375 Cachex.del(:user_cache, "nickname:#{user.nickname}")
376 Cachex.del(:user_cache, "user_info:#{user.id}")
377 end
378
379 def get_cached_by_ap_id(ap_id) do
380 key = "ap_id:#{ap_id}"
381 Cachex.fetch!(:user_cache, key, fn _ -> get_by_ap_id(ap_id) end)
382 end
383
384 def get_cached_by_nickname(nickname) do
385 key = "nickname:#{nickname}"
386 Cachex.fetch!(:user_cache, key, fn _ -> get_or_fetch_by_nickname(nickname) end)
387 end
388
389 def get_by_nickname(nickname) do
390 Repo.get_by(User, nickname: nickname) ||
391 if String.ends_with?(nickname, "@" <> Pleroma.Web.Endpoint.host()) do
392 [local_nickname, _] = String.split(nickname, "@")
393 Repo.get_by(User, nickname: local_nickname)
394 end
395 end
396
397 def get_by_nickname_or_email(nickname_or_email) do
398 case user = Repo.get_by(User, nickname: nickname_or_email) do
399 %User{} -> user
400 nil -> Repo.get_by(User, email: nickname_or_email)
401 end
402 end
403
404 def get_cached_user_info(user) do
405 key = "user_info:#{user.id}"
406 Cachex.fetch!(:user_cache, key, fn _ -> user_info(user) end)
407 end
408
409 def fetch_by_nickname(nickname) do
410 ap_try = ActivityPub.make_user_from_nickname(nickname)
411
412 case ap_try do
413 {:ok, user} -> {:ok, user}
414 _ -> OStatus.make_user(nickname)
415 end
416 end
417
418 def get_or_fetch_by_nickname(nickname) do
419 with %User{} = user <- get_by_nickname(nickname) do
420 user
421 else
422 _e ->
423 with [_nick, _domain] <- String.split(nickname, "@"),
424 {:ok, user} <- fetch_by_nickname(nickname) do
425 user
426 else
427 _e -> nil
428 end
429 end
430 end
431
432 def get_followers_query(%User{id: id, follower_address: follower_address}) do
433 from(
434 u in User,
435 where: fragment("? <@ ?", ^[follower_address], u.following),
436 where: u.id != ^id
437 )
438 end
439
440 def get_followers(user) do
441 q = get_followers_query(user)
442
443 {:ok, Repo.all(q)}
444 end
445
446 def get_friends_query(%User{id: id, following: following}) do
447 from(
448 u in User,
449 where: u.follower_address in ^following,
450 where: u.id != ^id
451 )
452 end
453
454 def get_friends(user) do
455 q = get_friends_query(user)
456
457 {:ok, Repo.all(q)}
458 end
459
460 def get_follow_requests_query(%User{} = user) do
461 from(
462 a in Activity,
463 where:
464 fragment(
465 "? ->> 'type' = 'Follow'",
466 a.data
467 ),
468 where:
469 fragment(
470 "? ->> 'state' = 'pending'",
471 a.data
472 ),
473 where:
474 fragment(
475 "? @> ?",
476 a.data,
477 ^%{"object" => user.ap_id}
478 )
479 )
480 end
481
482 def get_follow_requests(%User{} = user) do
483 q = get_follow_requests_query(user)
484 reqs = Repo.all(q)
485
486 users =
487 Enum.map(reqs, fn req -> req.actor end)
488 |> Enum.uniq()
489 |> Enum.map(fn ap_id -> get_by_ap_id(ap_id) end)
490 |> Enum.filter(fn u -> !following?(u, user) end)
491
492 {:ok, users}
493 end
494
495 def increase_note_count(%User{} = user) do
496 info_cng = User.Info.add_to_note_count(user.info, 1)
497
498 cng =
499 change(user)
500 |> put_embed(:info, info_cng)
501
502 update_and_set_cache(cng)
503 end
504
505 def decrease_note_count(%User{} = user) do
506 info_cng = User.Info.add_to_note_count(user.info, -1)
507
508 cng =
509 change(user)
510 |> put_embed(:info, info_cng)
511
512 update_and_set_cache(cng)
513 end
514
515 def update_note_count(%User{} = user) do
516 note_count_query =
517 from(
518 a in Object,
519 where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data),
520 select: count(a.id)
521 )
522
523 note_count = Repo.one(note_count_query)
524
525 info_cng = User.Info.set_note_count(user.info, note_count)
526
527 cng =
528 change(user)
529 |> put_embed(:info, info_cng)
530
531 update_and_set_cache(cng)
532 end
533
534 def update_follower_count(%User{} = user) do
535 follower_count_query =
536 from(
537 u in User,
538 where: ^user.follower_address in u.following,
539 where: u.id != ^user.id,
540 select: count(u.id)
541 )
542
543 follower_count = Repo.one(follower_count_query)
544
545 info_cng =
546 user.info
547 |> User.Info.set_follower_count(follower_count)
548
549 cng =
550 change(user)
551 |> put_embed(:info, info_cng)
552
553 update_and_set_cache(cng)
554 end
555
556 def get_users_from_set_query(ap_ids, false) do
557 from(
558 u in User,
559 where: u.ap_id in ^ap_ids
560 )
561 end
562
563 def get_users_from_set_query(ap_ids, true) do
564 query = get_users_from_set_query(ap_ids, false)
565
566 from(
567 u in query,
568 where: u.local == true
569 )
570 end
571
572 def get_users_from_set(ap_ids, local_only \\ true) do
573 get_users_from_set_query(ap_ids, local_only)
574 |> Repo.all()
575 end
576
577 def get_recipients_from_activity(%Activity{recipients: to}) do
578 query =
579 from(
580 u in User,
581 where: u.ap_id in ^to,
582 or_where: fragment("? && ?", u.following, ^to)
583 )
584
585 query = from(u in query, where: u.local == true)
586
587 Repo.all(query)
588 end
589
590 def search(query, resolve \\ false) do
591 # strip the beginning @ off if there is a query
592 query = String.trim_leading(query, "@")
593
594 if resolve do
595 User.get_or_fetch_by_nickname(query)
596 end
597
598 inner =
599 from(
600 u in User,
601 select_merge: %{
602 search_distance:
603 fragment(
604 "? <-> (? || ?)",
605 ^query,
606 u.nickname,
607 u.name
608 )
609 },
610 where: not is_nil(u.nickname)
611 )
612
613 q =
614 from(
615 s in subquery(inner),
616 order_by: s.search_distance,
617 limit: 20
618 )
619
620 Repo.all(q)
621 end
622
623 def blocks_import(%User{} = blocker, blocked_identifiers) when is_list(blocked_identifiers) do
624 Enum.map(
625 blocked_identifiers,
626 fn blocked_identifier ->
627 with %User{} = blocked <- get_or_fetch(blocked_identifier),
628 {:ok, blocker} <- block(blocker, blocked),
629 {:ok, _} <- ActivityPub.block(blocker, blocked) do
630 blocked
631 else
632 err ->
633 Logger.debug("blocks_import failed for #{blocked_identifier} with: #{inspect(err)}")
634 err
635 end
636 end
637 )
638 end
639
640 def block(blocker, %User{ap_id: ap_id} = blocked) do
641 # sever any follow relationships to prevent leaks per activitypub (Pleroma issue #213)
642 blocker =
643 if following?(blocker, blocked) do
644 {:ok, blocker, _} = unfollow(blocker, blocked)
645 blocker
646 else
647 blocker
648 end
649
650 if following?(blocked, blocker) do
651 unfollow(blocked, blocker)
652 end
653
654 info_cng =
655 blocker.info
656 |> User.Info.add_to_block(ap_id)
657
658 cng =
659 change(blocker)
660 |> put_embed(:info, info_cng)
661
662 update_and_set_cache(cng)
663 end
664
665 # helper to handle the block given only an actor's AP id
666 def block(blocker, %{ap_id: ap_id}) do
667 block(blocker, User.get_by_ap_id(ap_id))
668 end
669
670 def unblock(blocker, %{ap_id: ap_id}) do
671 info_cng =
672 blocker.info
673 |> User.Info.remove_from_block(ap_id)
674
675 cng =
676 change(blocker)
677 |> put_embed(:info, info_cng)
678
679 update_and_set_cache(cng)
680 end
681
682 def blocks?(user, %{ap_id: ap_id}) do
683 blocks = user.info.blocks
684 domain_blocks = user.info.domain_blocks
685 %{host: host} = URI.parse(ap_id)
686
687 Enum.member?(blocks, ap_id) ||
688 Enum.any?(domain_blocks, fn domain ->
689 host == domain
690 end)
691 end
692
693 def blocked_users(user),
694 do: Repo.all(from(u in User, where: u.ap_id in ^user.info.blocks))
695
696 def block_domain(user, domain) do
697 info_cng =
698 user.info
699 |> User.Info.add_to_domain_block(domain)
700
701 cng =
702 change(user)
703 |> put_embed(:info, info_cng)
704
705 update_and_set_cache(cng)
706 end
707
708 def unblock_domain(user, domain) do
709 info_cng =
710 user.info
711 |> User.Info.remove_from_domain_block(domain)
712
713 cng =
714 change(user)
715 |> put_embed(:info, info_cng)
716
717 update_and_set_cache(cng)
718 end
719
720 def local_user_query() do
721 from(
722 u in User,
723 where: u.local == true,
724 where: not is_nil(u.nickname)
725 )
726 end
727
728 def moderator_user_query() do
729 from(
730 u in User,
731 where: u.local == true,
732 where: fragment("?->'is_moderator' @> 'true'", u.info)
733 )
734 end
735
736 def deactivate(%User{} = user, status \\ true) do
737 info_cng = User.Info.set_activation_status(user.info, status)
738
739 cng =
740 change(user)
741 |> put_embed(:info, info_cng)
742
743 update_and_set_cache(cng)
744 end
745
746 def delete(%User{} = user) do
747 {:ok, user} = User.deactivate(user)
748
749 # Remove all relationships
750 {:ok, followers} = User.get_followers(user)
751
752 followers
753 |> Enum.each(fn follower -> User.unfollow(follower, user) end)
754
755 {:ok, friends} = User.get_friends(user)
756
757 friends
758 |> Enum.each(fn followed -> User.unfollow(user, followed) end)
759
760 query = from(a in Activity, where: a.actor == ^user.ap_id)
761
762 Repo.all(query)
763 |> Enum.each(fn activity ->
764 case activity.data["type"] do
765 "Create" ->
766 ActivityPub.delete(Object.normalize(activity.data["object"]))
767
768 # TODO: Do something with likes, follows, repeats.
769 _ ->
770 "Doing nothing"
771 end
772 end)
773
774 {:ok, user}
775 end
776
777 def html_filter_policy(%User{info: %{no_rich_text: true}}) do
778 Pleroma.HTML.Scrubber.TwitterText
779 end
780
781 def html_filter_policy(_), do: nil
782
783 def get_or_fetch_by_ap_id(ap_id) do
784 user = get_by_ap_id(ap_id)
785
786 if !is_nil(user) and !User.needs_update?(user) do
787 user
788 else
789 ap_try = ActivityPub.make_user_from_ap_id(ap_id)
790
791 case ap_try do
792 {:ok, user} ->
793 user
794
795 _ ->
796 case OStatus.make_user(ap_id) do
797 {:ok, user} -> user
798 _ -> {:error, "Could not fetch by AP id"}
799 end
800 end
801 end
802 end
803
804 def get_or_create_instance_user do
805 relay_uri = "#{Pleroma.Web.Endpoint.url()}/relay"
806
807 if user = get_by_ap_id(relay_uri) do
808 user
809 else
810 changes =
811 %User{info: %User.Info{}}
812 |> cast(%{}, [:ap_id, :nickname, :local])
813 |> put_change(:ap_id, relay_uri)
814 |> put_change(:nickname, nil)
815 |> put_change(:local, true)
816 |> put_change(:follower_address, relay_uri <> "/followers")
817
818 {:ok, user} = Repo.insert(changes)
819 user
820 end
821 end
822
823 # AP style
824 def public_key_from_info(%{
825 source_data: %{"publicKey" => %{"publicKeyPem" => public_key_pem}}
826 }) do
827 key =
828 public_key_pem
829 |> :public_key.pem_decode()
830 |> hd()
831 |> :public_key.pem_entry_decode()
832
833 {:ok, key}
834 end
835
836 # OStatus Magic Key
837 def public_key_from_info(%{magic_key: magic_key}) do
838 {:ok, Pleroma.Web.Salmon.decode_key(magic_key)}
839 end
840
841 def get_public_key_for_ap_id(ap_id) do
842 with %User{} = user <- get_or_fetch_by_ap_id(ap_id),
843 {:ok, public_key} <- public_key_from_info(user.info) do
844 {:ok, public_key}
845 else
846 _ -> :error
847 end
848 end
849
850 defp blank?(""), do: nil
851 defp blank?(n), do: n
852
853 def insert_or_update_user(data) do
854 data =
855 data
856 |> Map.put(:name, blank?(data[:name]) || data[:nickname])
857
858 cs = User.remote_user_creation(data)
859
860 Repo.insert(cs, on_conflict: :replace_all, conflict_target: :nickname)
861 end
862
863 def ap_enabled?(%User{local: true}), do: true
864 def ap_enabled?(%User{info: info}), do: info.ap_enabled
865 def ap_enabled?(_), do: false
866
867 @doc "Gets or fetch a user by uri or nickname."
868 @spec get_or_fetch(String.t()) :: User.t()
869 def get_or_fetch("http" <> _host = uri), do: get_or_fetch_by_ap_id(uri)
870 def get_or_fetch(nickname), do: get_or_fetch_by_nickname(nickname)
871
872 # wait a period of time and return newest version of the User structs
873 # this is because we have synchronous follow APIs and need to simulate them
874 # with an async handshake
875 def wait_and_refresh(_, %User{local: true} = a, %User{local: true} = b) do
876 with %User{} = a <- Repo.get(User, a.id),
877 %User{} = b <- Repo.get(User, b.id) do
878 {:ok, a, b}
879 else
880 _e ->
881 :error
882 end
883 end
884
885 def wait_and_refresh(timeout, %User{} = a, %User{} = b) do
886 with :ok <- :timer.sleep(timeout),
887 %User{} = a <- Repo.get(User, a.id),
888 %User{} = b <- Repo.get(User, b.id) do
889 {:ok, a, b}
890 else
891 _e ->
892 :error
893 end
894 end
895
896 def parse_bio(bio, user \\ %User{info: %{source_data: %{}}})
897 def parse_bio(nil, _user), do: ""
898 def parse_bio(bio, _user) when bio == "", do: bio
899
900 def parse_bio(bio, user) do
901 mentions = Formatter.parse_mentions(bio)
902 tags = Formatter.parse_tags(bio)
903
904 emoji =
905 (user.info.source_data["tag"] || [])
906 |> Enum.filter(fn %{"type" => t} -> t == "Emoji" end)
907 |> Enum.map(fn %{"icon" => %{"url" => url}, "name" => name} ->
908 {String.trim(name, ":"), url}
909 end)
910
911 bio
912 |> CommonUtils.format_input(mentions, tags, "text/plain")
913 |> Formatter.emojify(emoji)
914 end
915
916 def tag(user_identifiers, tags) when is_list(user_identifiers) do
917 Repo.transaction(fn ->
918 for user_identifier <- user_identifiers, do: tag(user_identifier, tags)
919 end)
920 end
921
922 def tag(nickname, tags) when is_binary(nickname),
923 do: tag(User.get_by_nickname(nickname), tags)
924
925 def tag(%User{} = user, tags),
926 do: update_tags(user, Enum.uniq((user.tags || []) ++ normalize_tags(tags)))
927
928 def untag(user_identifiers, tags) when is_list(user_identifiers) do
929 Repo.transaction(fn ->
930 for user_identifier <- user_identifiers, do: untag(user_identifier, tags)
931 end)
932 end
933
934 def untag(nickname, tags) when is_binary(nickname),
935 do: untag(User.get_by_nickname(nickname), tags)
936
937 def untag(%User{} = user, tags),
938 do: update_tags(user, (user.tags || []) -- normalize_tags(tags))
939
940 defp update_tags(%User{} = user, new_tags) do
941 {:ok, updated_user} =
942 user
943 |> change(%{tags: new_tags})
944 |> Repo.update()
945
946 updated_user
947 end
948
949 defp normalize_tags(tags) do
950 [tags]
951 |> List.flatten()
952 |> Enum.map(&String.downcase(&1))
953 end
954
955 defp local_nickname_regex() do
956 if Pleroma.Config.get([:instance, :extended_nickname_format]) do
957 @extended_local_nickname_regex
958 else
959 @strict_local_nickname_regex
960 end
961 end
962 end