# frozen_string_literal: true

require Rails.root.join('lib', 'mastodon', 'migration_helpers')

class AddMasterSettingsToAccounts < ActiveRecord::Migration[7.1]
  include Mastodon::MigrationHelpers

  disable_ddl_transaction!

  class Account < ApplicationRecord; end

  def up
    safety_assured do
      add_column :accounts, :master_settings, :jsonb

      if Rails.env.test?
        Account.transaction do
          Account.find_in_batches do |accounts|
            accounts.each do |account|
              account.update(master_settings: { 'subscription_policy' => account.dissubscribable ? 'block' : 'allow' })
            end
          end
        end
      else
        Account.where(dissubscribable: true).update_all(master_settings: { 'subscription_policy' => 'block' }) # rubocop:disable Rails/SkipsModelValidations
        Account.where(dissubscribable: false).update_all(master_settings: { 'subscription_policy' => 'allow' }) # rubocop:disable Rails/SkipsModelValidations
      end

      remove_column :accounts, :dissubscribable
    end
  end

  def down
    safety_assured do
      add_column_with_default :accounts, :dissubscribable, :boolean, default: false, allow_null: false

      if Rails.env.test?
        Account.transaction do
          Account.find_in_batches do |accounts|
            accounts.each do |account|
              account.update(dissubscribable: account.master_settings.present? && account.master_settings['subscription_policy'] != 'allow')
            end
          end
        end
      else
        Account.where(master_settings: { subscription_policy: 'block' }).update_all(dissubscribable: true) # rubocop:disable Rails/SkipsModelValidations
        Account.where(master_settings: { subscription_policy: 'allow' }).update_all(dissubscribable: false) # rubocop:disable Rails/SkipsModelValidations
      end

      remove_column :accounts, :master_settings
    end
  end
end