git.squeep.com Git - akkoma/blob - lib/pleroma/user_relationship.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.UserRelationship do
   6   use Ecto.Schema
   7
   8   import Ecto.Changeset
   9   import Ecto.Query
  10
  11   alias Pleroma.Repo
  12   alias Pleroma.User
  13   alias Pleroma.UserRelationship
  14
  15   schema "user_relationships" do
  16     belongs_to(:source, User, type: FlakeId.Ecto.CompatType)
  17     belongs_to(:target, User, type: FlakeId.Ecto.CompatType)
  18     field(:relationship_type, UserRelationshipTypeEnum)
  19
  20     timestamps(updated_at: false)
  21   end
  22
  23   def changeset(%UserRelationship{} = user_relationship, params \\ %{}) do
  24     user_relationship
  25     |> cast(params, [:relationship_type, :source_id, :target_id])
  26     |> validate_required([:relationship_type, :source_id, :target_id])
  27     |> unique_constraint(:relationship_type,
  28       name: :user_relationships_source_id_relationship_type_target_id_index
  29     )
  30     |> validate_not_self_relationship()
  31   end
  32
  33   def exists?(relationship_type, %User{} = source, %User{} = target) do
  34     UserRelationship
  35     |> where(relationship_type: ^relationship_type, source_id: ^source.id, target_id: ^target.id)
  36     |> Repo.exists?()
  37   end
  38
  39   def block_exists?(%User{} = blocker, %User{} = blockee), do: exists?(:block, blocker, blockee)
  40
  41   def mute_exists?(%User{} = muter, %User{} = mutee), do: exists?(:mute, muter, mutee)
  42
  43   def create(relationship_type, %User{} = source, %User{} = target) do
  44     %UserRelationship{}
  45     |> changeset(%{
  46       relationship_type: relationship_type,
  47       source_id: source.id,
  48       target_id: target.id
  49     })
  50     |> Repo.insert(
  51       on_conflict: :replace_all_except_primary_key,
  52       conflict_target: [:source_id, :relationship_type, :target_id]
  53     )
  54   end
  55
  56   def create_block(%User{} = blocker, %User{} = blockee), do: create(:block, blocker, blockee)
  57
  58   def create_mute(%User{} = muter, %User{} = mutee), do: create(:mute, muter, mutee)
  59
  60   def delete(relationship_type, %User{} = source, %User{} = target) do
  61     attrs = %{relationship_type: relationship_type, source_id: source.id, target_id: target.id}
  62
  63     case Repo.get_by(UserRelationship, attrs) do
  64       %UserRelationship{} = existing_record -> Repo.delete(existing_record)
  65       nil -> {:ok, nil}
  66     end
  67   end
  68
  69   def delete_block(%User{} = blocker, %User{} = blockee), do: delete(:block, blocker, blockee)
  70
  71   def delete_mute(%User{} = muter, %User{} = mutee), do: delete(:mute, muter, mutee)
  72
  73   defp validate_not_self_relationship(%Ecto.Changeset{} = changeset) do
  74     changeset
  75     |> validate_change(:target_id, fn _, target_id ->
  76       if target_id == get_field(changeset, :source_id) do
  77         [target_id: "can't be equal to source_id"]
  78       else
  79         []
  80       end
  81     end)
  82     |> validate_change(:source_id, fn _, source_id ->
  83       if source_id == get_field(changeset, :target_id) do
  84         [source_id: "can't be equal to target_id"]
  85       else
  86         []
  87       end
  88     end)
  89   end
  90 end