pleroma

commit: 825541b27c71d3a9b6f597a9934151831c52f6e5
parent 38e6166d93b26b34d2efe38cc65d0217ad994527
Author: lain <lain@soykaf.club>
Date:   Mon, 17 Jun 2024 13:09:11 +0000

Merge branch 'async-user-refresh' into 'develop'

Async user refreshing

See merge request pleroma/pleroma!4148
Diffstat:
A
changelog.d/user-refresh.change
1
+
M
config/test.exs
2
++
M
lib/pleroma/user.ex
25
++++++++++++++-----------
M
test/pleroma/user_test.exs
7
+++++--

4 files changed, 22 insertions(+), 13 deletions(-)
diff --git a/changelog.d/user-refresh.change b/changelog.d/user-refresh.change
@@ -0,0 +1 @@
+User profile refreshes are now asynchronous
diff --git a/config/test.exs b/config/test.exs
@@ -183,6 +183,8 @@ config :pleroma, Pleroma.Emoji.Loader, test_emoji: true
 
 config :pleroma, Pleroma.Web.RichMedia.Backfill, provider: Pleroma.Web.RichMedia.Backfill
 
+config :pleroma, Pleroma.User, sync_refreshing: true
+
 if File.exists?("./config/test.secret.exs") do
   import_config "test.secret.exs"
 else
diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
@@ -2154,20 +2154,23 @@ defmodule Pleroma.User do
 
   def fetch_by_ap_id(ap_id), do: ActivityPub.make_user_from_ap_id(ap_id)
 
+  @spec get_or_fetch_by_ap_id(String.t()) :: {:ok, User.t()} | {:error, any()}
   def get_or_fetch_by_ap_id(ap_id) do
-    cached_user = get_cached_by_ap_id(ap_id)
-
-    maybe_fetched_user = needs_update?(cached_user) && fetch_by_ap_id(ap_id)
-
-    case {cached_user, maybe_fetched_user} do
-      {_, {:ok, %User{} = user}} ->
-        {:ok, user}
+    with cached_user = %User{} <- get_cached_by_ap_id(ap_id),
+         _ <- maybe_refresh(cached_user) do
+      {:ok, cached_user}
+    else
+      _ -> fetch_by_ap_id(ap_id)
+    end
+  end
 
-      {%User{} = user, _} ->
-        {:ok, user}
+  defp maybe_refresh(user) do
+    fun = fn -> needs_update?(user) && fetch_by_ap_id(user.ap_id) end
 
-      _ ->
-        {:error, :not_found}
+    if Config.get([__MODULE__, :sync_refreshing], false) do
+      fun.()
+    else
+      Task.start(fun)
     end
   end
 
diff --git a/test/pleroma/user_test.exs b/test/pleroma/user_test.exs
@@ -953,9 +953,12 @@ defmodule Pleroma.UserTest do
 
       {:ok, user} = User.get_or_fetch_by_ap_id("http://mastodon.example.org/users/admin")
 
-      assert user.inbox
+      # User was updated async, fetch from cache now
+      updated_user = User.get_cached_by_ap_id(user.ap_id)
+
+      assert updated_user.inbox
 
-      refute user.last_refreshed_at == orig_user.last_refreshed_at
+      refute updated_user.last_refreshed_at == orig_user.last_refreshed_at
     end
 
     test "if nicknames clash, the old user gets a prefix with the old id to the nickname" do