require 'lib/migrate'

class CreateUserRoles < ActiveRecord::Migration
  def self.up
    create_enumeration :user_role_enum, ["administrator", "moderator"]

    create_table :user_roles do |t|
      t.column :user_id, :bigint, :null => false

      t.timestamps
    end
    add_column :user_roles, :role, :user_role_enum, :null => false

    User.all(:conditions => ['administrator = ?', true]).each do |user|
      UserRole.create(:user_id => user.id, :role => "administrator")
    end
    remove_column :users, :administrator

    add_foreign_key :user_roles, [:user_id], :users, [:id]
  end

  def self.down
    add_column :users, :administrator, :boolean, :default => false, :null => false
    UserRole.all(:conditions => ['role = ?', "administrator"]).each do |role|
      user = User.find(role.user_id)
      user.administrator = true
      user.save!
    end
    drop_table :user_roles
    drop_enumeration :user_role_enum
  end
end