class Audited::Generators::UpgradeGenerator

def copy_templates
  migrations_to_be_applied do |m|
    migration_template "#{m}.rb", "db/migrate/#{m}.rb"
  end
end

def migrations_to_be_applied
  Audited::Audit.reset_column_information
  columns = Audited::Audit.columns.map(&:name)
  indexes = Audited::Audit.connection.indexes(Audited::Audit.table_name)
  yield :add_comment_to_audits unless columns.include?('comment')
  if columns.include?('changes')
    yield :rename_changes_to_audited_changes
  end
  unless columns.include?('remote_address')
    yield :add_remote_address_to_audits
  end
  unless columns.include?('request_uuid')
    yield :add_request_uuid_to_audits
  end
  unless columns.include?('association_id')
    if columns.include?('auditable_parent_id')
      yield :rename_parent_to_association
    else
      unless columns.include?('associated_id')
        yield :add_association_to_audits
      end
    end
  end
  if columns.include?('association_id')
    yield :rename_association_to_associated
  end
  if indexes.any? { |i| i.columns == %w[associated_id associated_type] }
    yield :revert_polymorphic_indexes_order
  end
end