git.squeep.com Git - akkoma/blob - priv/repo/migrations/20191009154608_copy_users_info_fields_to_users.exs

   1 defmodule Pleroma.Repo.Migrations.CopyUsersInfoaddsToUsers do
   2   use Ecto.Migration
   3
   4   @info_fields [
   5     :banner,
   6     :background,
   7     :source_data,
   8     :note_count,
   9     :follower_count,
  10     :following_count,
  11     :locked,
  12     :confirmation_pending,
  13     :password_reset_pending,
  14     :confirmation_token,
  15     :default_scope,
  16     :blocks,
  17     :domain_blocks,
  18     :mutes,
  19     :muted_reblogs,
  20     :muted_notifications,
  21     :subscribers,
  22     :deactivated,
  23     :no_rich_text,
  24     :ap_enabled,
  25     :is_moderator,
  26     :is_admin,
  27     :show_role,
  28     :settings,
  29     :magic_key,
  30     :uri,
  31     :topic,
  32     :hub,
  33     :salmon,
  34     :hide_followers_count,
  35     :hide_follows_count,
  36     :hide_followers,
  37     :hide_follows,
  38     :hide_favorites,
  39     :unread_conversation_count,
  40     :pinned_activities,
  41     :email_notifications,
  42     :mascot,
  43     :emoji,
  44     :pleroma_settings_store,
  45     :fields,
  46     :raw_fields,
  47     :discoverable,
  48     :skip_thread_containment,
  49     :notification_settings
  50   ]
  51
  52   @jsonb_fields [
  53     :banner,
  54     :background,
  55     :source_data,
  56     :settings,
  57     :email_notifications,
  58     :mascot,
  59     :pleroma_settings_store,
  60     :notification_settings
  61   ]
  62
  63   @array_jsonb_fields [:emoji, :fields, :raw_fields]
  64
  65   @int_fields [:note_count, :follower_count, :following_count, :unread_conversation_count]
  66
  67   @boolean_fields [
  68     :locked,
  69     :confirmation_pending,
  70     :password_reset_pending,
  71     :deactivated,
  72     :no_rich_text,
  73     :ap_enabled,
  74     :is_moderator,
  75     :is_admin,
  76     :show_role,
  77     :hide_followers_count,
  78     :hide_follows_count,
  79     :hide_followers,
  80     :hide_follows,
  81     :hide_favorites,
  82     :discoverable,
  83     :skip_thread_containment
  84   ]
  85
  86   @array_text_fields [
  87     :blocks,
  88     :domain_blocks,
  89     :mutes,
  90     :muted_reblogs,
  91     :muted_notifications,
  92     :subscribers,
  93     :pinned_activities
  94   ]
  95
  96   def change do
  97     alter table(:users) do
  98       add(:banner, :map, default: %{})
  99       add(:background, :map, default: %{})
 100       add(:source_data, :map, default: %{})
 101       add(:note_count, :integer, default: 0)
 102       add(:follower_count, :integer, default: 0)
 103       add(:following_count, :integer, default: nil)
 104       add(:locked, :boolean, default: false, null: false)
 105       add(:confirmation_pending, :boolean, default: false, null: false)
 106       add(:password_reset_pending, :boolean, default: false, null: false)
 107       add(:confirmation_token, :text, default: nil)
 108       add(:default_scope, :string, default: "public")
 109       add(:blocks, {:array, :text}, default: [])
 110       add(:domain_blocks, {:array, :text}, default: [])
 111       add(:mutes, {:array, :text}, default: [])
 112       add(:muted_reblogs, {:array, :text}, default: [])
 113       add(:muted_notifications, {:array, :text}, default: [])
 114       add(:subscribers, {:array, :text}, default: [])
 115       add(:deactivated, :boolean, default: false, null: false)
 116       add(:no_rich_text, :boolean, default: false, null: false)
 117       add(:ap_enabled, :boolean, default: false, null: false)
 118       add(:is_moderator, :boolean, default: false, null: false)
 119       add(:is_admin, :boolean, default: false, null: false)
 120       add(:show_role, :boolean, default: true, null: false)
 121       add(:settings, :map, default: nil)
 122       add(:magic_key, :text, default: nil)
 123       add(:uri, :text, default: nil)
 124       add(:topic, :text, default: nil)
 125       add(:hub, :text, default: nil)
 126       add(:salmon, :text, default: nil)
 127       add(:hide_followers_count, :boolean, default: false, null: false)
 128       add(:hide_follows_count, :boolean, default: false, null: false)
 129       add(:hide_followers, :boolean, default: false, null: false)
 130       add(:hide_follows, :boolean, default: false, null: false)
 131       add(:hide_favorites, :boolean, default: true, null: false)
 132       add(:unread_conversation_count, :integer, default: 0)
 133       add(:pinned_activities, {:array, :text}, default: [])
 134       add(:email_notifications, :map, default: %{"digest" => false})
 135       add(:mascot, :map, default: nil)
 136       add(:emoji, {:array, :map}, default: [])
 137       add(:pleroma_settings_store, :map, default: %{})
 138       add(:fields, {:array, :map}, default: nil)
 139       add(:raw_fields, {:array, :map}, default: [])
 140       add(:discoverable, :boolean, default: false, null: false)
 141       add(:notification_settings, :map, default: %{})
 142       add(:skip_thread_containment, :boolean, default: false, null: false)
 143     end
 144
 145     if direction == :up do
 146       for f <- @info_fields do
 147         set_field = "update users set #{f} ="
 148
 149         cond do
 150           f in @jsonb_fields ->
 151             execute("#{set_field} info->'#{f}'")
 152
 153           f in @array_jsonb_fields ->
 154             execute("#{set_field} ARRAY(SELECT jsonb_array_elements(info->'#{f}'))")
 155
 156           f in @int_fields ->
 157             execute("#{set_field} (info->>'#{f}')::int")
 158
 159           f in @boolean_fields ->
 160             execute("#{set_field} coalesce((info->>'#{f}')::boolean, false)")
 161
 162           f in @array_text_fields ->
 163             execute("#{set_field} ARRAY(SELECT jsonb_array_elements_text(info->'#{f}'))")
 164
 165           true ->
 166             execute("#{set_field} info->>'#{f}'")
 167         end
 168       end
 169
 170       for index_name <- [
 171             :users_deactivated_index,
 172             :users_is_moderator_index,
 173             :users_is_admin_index,
 174             :users_subscribers_index
 175           ] do
 176         drop_if_exists(index(:users, [], name: index_name))
 177       end
 178     end
 179
 180     create_if_not_exists(index(:users, [:deactivated]))
 181     create_if_not_exists(index(:users, [:is_moderator]))
 182     create_if_not_exists(index(:users, [:is_admin]))
 183     create_if_not_exists(index(:users, [:subscribers]))
 184   end
 185 end