Merge branch 'develop' into 'reactions'

[akkoma] / lib / pleroma / user.ex

diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
index 494a67f2237729f22a5f17d11f35fbdb82de6bd4..2cfb13a8c0164b2b4f1989fae7bdc6a666202c0e 100644 (file)
--- a/lib/pleroma/user.ex
+++ b/lib/pleroma/user.ex
@@ -51,6 +51,7 @@ defmodule Pleroma.User do
     field(:password_hash, :string)
     field(:password, :string, virtual: true)
     field(:password_confirmation, :string, virtual: true)
+    field(:keys, :string)
     field(:following, {:array, :string}, default: [])
     field(:ap_id, :string)
     field(:avatar, :map)
@@ -847,7 +848,6 @@ defmodule Pleroma.User do
     unread_query = Participation.unread_conversation_count_for_user(user)
 
     User
-    |> where([u], u.id == ^user.id)
     |> join(:inner, [u], p in subquery(unread_query))
     |> update([u, p],
       set: [
@@ -859,10 +859,11 @@ defmodule Pleroma.User do
           )
       ]
     )
+    |> where([u], u.id == ^user.id)
     |> select([u], u)
     |> Repo.update_all([])
     |> case do
-      {1, [%{info: %User.Info{}} = user]} -> set_cache(user)
+      {1, [user]} -> set_cache(user)
       _ -> {:error, user}
     end
   end
@@ -880,7 +881,7 @@ defmodule Pleroma.User do
       set: [
         info:
           fragment(
-            "jsonb_set(?, '{unread_conversation_count}', ((?->>'unread_conversation_count')::int + 1)::varchar::jsonb, true)",
+            "jsonb_set(?, '{unread_conversation_count}', (coalesce((?->>'unread_conversation_count')::int, 0) + 1)::varchar::jsonb, true)",
             u.info,
             u.info
           )
@@ -891,7 +892,7 @@ defmodule Pleroma.User do
     |> select([u], u)
     |> Repo.update_all([])
     |> case do
-      {1, [%{info: %User.Info{}} = user]} -> set_cache(user)
+      {1, [user]} -> set_cache(user)
       _ -> {:error, user}
     end
   end
@@ -1554,11 +1555,14 @@ defmodule Pleroma.User do
     }
   end
 
-  def ensure_keys_present(%{info: %{keys: keys}} = user) when not is_nil(keys), do: {:ok, user}
+  def ensure_keys_present(%{keys: keys} = user) when not is_nil(keys), do: {:ok, user}
 
   def ensure_keys_present(%User{} = user) do
     with {:ok, pem} <- Keys.generate_rsa_pem() do
-      update_info(user, &User.Info.set_keys(&1, pem))
+      user
+      |> cast(%{keys: pem}, [:keys])
+      |> validate_required([:keys])
+      |> update_and_set_cache()
     end
   end