53 lines
1.9 KiB
Ruby
53 lines
1.9 KiB
Ruby
# 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
|