gistfile1.rb

# encoding: utf-8

class User < ActiveRecord::Base
  # Include default devise modules. Others available are:
  # :token_authenticatable, :encryptable, :confirmable, :lockable, :timeoutable and :omniauthable
  devise :database_authenticatable, :registerable,
         :recoverable, :rememberable, :trackable#, :validatable

  has_many :user_roles
  has_many :roles, :through => :user_roles
  has_many :orders

  after_create :assign_affiliate_role, :send_account_data_to_user

  # Role checking

  def has_role?(*role_names)
    self.roles.where(:name => role_names).present?
  end


  def client?
    self.has_role? 'client'
  end


  def affiliate?
    self.has_role? 'affiliate'
  end


  def admin?
    self.has_role? 'admin'
  end


  def master?
    self.has_role? 'master'
  end
  
  
  def pending_orders?
    if Order.where(:status => Order::PENDING, :user_id => self.id).first
      return true
    else
      return false
    end
  end


  private


  def assign_affiliate_role
    # If the user has no orders should be an affiliate
    if self.orders.empty?
      # Assign afiliate role
      role_asignation = UserRole.new
      role_asignation.role_id = 2
      role_asignation.user = self
      role_asignation.save
    end
  end

  def send_account_data_to_user
    if self.orders.empty?
      AffiliateMailer.access_data(self).deliver
    else
      ClientMailer.access_data(self).deliver
    end
  end

end