First
[anni] / lib / pleroma / thread_mute.ex
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.ThreadMute do
6   use Ecto.Schema
7
8   alias Pleroma.Repo
9   alias Pleroma.ThreadMute
10   alias Pleroma.User
11
12   import Ecto.Changeset
13   import Ecto.Query
14
15   schema "thread_mutes" do
16     belongs_to(:user, User, type: FlakeId.Ecto.CompatType)
17     field(:context, :string)
18   end
19
20   def changeset(mute, params \\ %{}) do
21     mute
22     |> cast(params, [:user_id, :context])
23     |> foreign_key_constraint(:user_id)
24     |> unique_constraint(:user_id, name: :unique_index)
25   end
26
27   def query(user_id, context) do
28     user_binary_id = User.binary_id(user_id)
29
30     ThreadMute
31     |> where(user_id: ^user_binary_id)
32     |> where(context: ^context)
33   end
34
35   def muters_query(context) do
36     ThreadMute
37     |> join(:inner, [tm], u in assoc(tm, :user))
38     |> where([tm], tm.context == ^context)
39     |> select([tm, u], u.ap_id)
40   end
41
42   def muter_ap_ids(context, ap_ids \\ nil)
43
44   # Note: applies to fake activities (ActivityPub.Utils.get_notified_from_object/1 etc.)
45   def muter_ap_ids(context, _ap_ids) when is_nil(context), do: []
46
47   def muter_ap_ids(context, ap_ids) do
48     context
49     |> muters_query()
50     |> maybe_filter_on_ap_id(ap_ids)
51     |> Repo.all()
52   end
53
54   defp maybe_filter_on_ap_id(query, ap_ids) when is_list(ap_ids) do
55     where(query, [tm, u], u.ap_id in ^ap_ids)
56   end
57
58   defp maybe_filter_on_ap_id(query, _ap_ids), do: query
59
60   def add_mute(user_id, context) do
61     %ThreadMute{}
62     |> changeset(%{user_id: user_id, context: context})
63     |> Repo.insert()
64   end
65
66   def remove_mute(user_id, context) do
67     query(user_id, context)
68     |> Repo.delete_all()
69   end
70
71   def exists?(user_id, context) do
72     query(user_id, context)
73     |> Repo.exists?()
74   end
75 end