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

   1 defmodule Pleroma.User.Info do
   2   use Ecto.Schema
   3   import Ecto.Changeset
   4
   5   embedded_schema do
   6     field(:banner, :map, default: %{})
   7     field(:source_data, :map, default: %{})
   8     field(:note_count, :integer, default: 0)
   9     field(:follower_count, :integer, default: 0)
  10     field(:locked, :boolean, default: false)
  11     field(:default_scope, :string, default: "public")
  12     field(:blocks, {:array, :string}, default: [])
  13     field(:domain_blocks, {:array, :string}, default: [])
  14     field(:deactivated, :boolean, default: false)
  15     field(:no_rich_text, :boolean, default: false)
  16     field(:ap_enabled, :boolean, default: false)
  17     field(:is_moderator, :boolean, default: false)
  18     field(:keys, :map, default: %{})
  19   end
  20
  21   def set_activation_status(info, deactivated) do
  22     params = %{deactivated: deactivated}
  23
  24     info
  25     |> cast(params, [:deactivated])
  26     |> validate_required([:deactivated])
  27   end
  28
  29   def add_to_note_count(info, number) do
  30     set_note_count(info, info.note_count + number)
  31   end
  32
  33   def set_note_count(info, number) do
  34     params = %{note_count: Enum.max([0, number])}
  35
  36     info
  37     |> cast(params, [:note_count])
  38     |> validate_required([:note_count])
  39   end
  40
  41   def set_follower_count(info, number) do
  42     params = %{follower_count: Enum.max([0, number])}
  43
  44     info
  45     |> cast(params, [:follower_count])
  46     |> validate_required([:follower_count])
  47   end
  48
  49   def set_blocks(info, blocks) do
  50     params = %{blocks: blocks}
  51
  52     info
  53     |> cast(params, [:blocks])
  54     |> validate_required([:blocks])
  55   end
  56
  57   def add_to_block(info, blocked) do
  58     set_blocks(info, Enum.uniq([blocked | info.blocks]))
  59   end
  60
  61   def remove_from_block(info, blocked) do
  62     set_blocks(info, List.delete(info.blocks, blocked))
  63   end
  64 end