discourse/db/migrate/20190529002752_add_unique_constraint_to_shadow_accounts.rb

# frozen_string_literal: true

class AddUniqueConstraintToShadowAccounts < ActiveRecord::Migration[5.2]
  def up
    create_table :anonymous_users do |t|
      t.integer :user_id, null: false
      t.integer :master_user_id, null: false
      t.boolean :active, null: false
      t.timestamps

      t.index [:user_id], unique: true
      t.index [:master_user_id], unique: true, where: "active"
    end

    rows = DB.exec <<~SQL
      DELETE FROM user_custom_fields
      WHERE name = 'shadow_id' AND value in (
        SELECT value
        FROM user_custom_fields
        WHERE name = 'shadow_id'
        GROUP BY value
        HAVING COUNT(*) > 1
      )
    SQL

    STDERR.puts "Removed #{rows} duplicate shadow users" if rows > 0

    rows = DB.exec <<~SQL
      INSERT INTO anonymous_users(user_id, master_user_id, created_at, updated_at, active)
      SELECT value::int, user_id, created_at, updated_at, 't'
      FROM user_custom_fields
      WHERE name = 'shadow_id'
    SQL

    rows += DB.exec <<~SQL
      INSERT INTO anonymous_users(user_id, master_user_id, created_at, updated_at, active)
      SELECT f.user_id, value::int, f.created_at, f.updated_at, 'f'
      FROM user_custom_fields f
      LEFT JOIN anonymous_users a on a.user_id = f.user_id
      WHERE name = 'master_id' AND a.user_id IS NULL
    SQL

    STDERR.puts "Migrated #{rows} anon users to new structure" if rows > 0

    DB.exec <<~SQL
      DELETE FROM user_custom_fields
      WHERE name in ('shadow_id', 'master_id')
    SQL
  end

  def down
    raise ActiveRecord::IrreversibleMigration
  end
end