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(:background, :string, default: nil)
   8     field(:source_data, :map, default: %{})
   9     field(:note_count, :integer, default: 0)
  10     field(:follower_count, :integer, default: 0)
  11     field(:locked, :boolean, default: false)
  12     field(:default_scope, :string, default: "public")
  13     field(:blocks, {:array, :string}, default: [])
  14     field(:domain_blocks, {:array, :string}, default: [])
  15     field(:deactivated, :boolean, default: false)
  16     field(:no_rich_text, :boolean, default: false)
  17     field(:ap_enabled, :boolean, default: false)
  18     field(:is_moderator, :boolean, default: false)
  19     field(:keys, :string, default: nil)
  20   end
  21
  22   def set_activation_status(info, deactivated) do
  23     params = %{deactivated: deactivated}
  24
  25     info
  26     |> cast(params, [:deactivated])
  27     |> validate_required([:deactivated])
  28   end
  29
  30   def add_to_note_count(info, number) do
  31     set_note_count(info, info.note_count + number)
  32   end
  33
  34   def set_note_count(info, number) do
  35     params = %{note_count: Enum.max([0, number])}
  36
  37     info
  38     |> cast(params, [:note_count])
  39     |> validate_required([:note_count])
  40   end
  41
  42   def set_follower_count(info, number) do
  43     params = %{follower_count: Enum.max([0, number])}
  44
  45     info
  46     |> cast(params, [:follower_count])
  47     |> validate_required([:follower_count])
  48   end
  49
  50   def set_blocks(info, blocks) do
  51     params = %{blocks: blocks}
  52
  53     info
  54     |> cast(params, [:blocks])
  55     |> validate_required([:blocks])
  56   end
  57
  58   def add_to_block(info, blocked) do
  59     set_blocks(info, Enum.uniq([blocked | info.blocks]))
  60   end
  61
  62   def remove_from_block(info, blocked) do
  63     set_blocks(info, List.delete(info.blocks, blocked))
  64   end
  65
  66   def set_domain_blocks(info, domain_blocks) do
  67     params = %{domain_blocks: domain_blocks}
  68
  69     info
  70     |> cast(params, [:domain_blocks])
  71     |> validate_required([:domain_blocks])
  72   end
  73
  74   def add_to_domain_block(info, domain_blocked) do
  75     set_domain_blocks(info, Enum.uniq([domain_blocked | info.domain_blocks]))
  76   end
  77
  78   def remove_from_domain_block(info, domain_blocked) do
  79     set_domain_blocks(info, List.delete(info.domain_blocks, domain_blocked))
  80   end
  81
  82   def set_keys(info, keys) do
  83     params = %{keys: keys}
  84
  85     info
  86     |> cast(params, [:keys])
  87     |> validate_required([:keys])
  88   end
  89
  90   def remote_user_creation(info, params) do
  91     info
  92     |> cast(params, [:source_data])
  93   end
  94
  95   # Receives data from user_data_from_user_object
  96   def user_upgrade(info, params) do
  97     info
  98     |> cast(params, [:ap_enabled, :source_data, :banner, :locked])
  99   end
 100 end