Contents

require 'roles_active_record/strategy/shared'

module Roles::ActiveRecord
  module Strategy
    module Multi
      include Shared
                 
      # assign multiple roles
      def roles=(*role_names)
        extracted_roles = extract_roles(role_names)
        return nil if extracted_roles.empty?
        set_roles(select_valid_roles extracted_roles)
      end      
      
      def add_roles *roles
        new_roles = select_valid_roles(roles)
        if !new_roles.empty?
          self.roles = self.roles + new_roles
          self.save
        end
      end      
      
      # should remove the current single role (set = nil) 
      # only if it is contained in the list of roles to be removed
      def remove_roles *role_names
        role_names = role_names.flat_uniq
        set_empty_roles and return if roles_diff(role_names).empty?
        roles_to_remove = select_valid_roles(role_names)
        set_roles roles_diff(role_names)
        true
      end

      # query assigned roles
      def roles
        get_roles.map do |role|
          role.respond_to?(:sym) ? role.to_sym : role
        end
      end

      def roles_list
        my_roles = [roles].flat_uniq
        return [] if my_roles.empty?
        has_role_class? ? my_roles.map{|r| r.name.to_sym } : my_roles          
      end
      
      protected
      
      def set_roles *roles                      
        self.send("#{role_attribute}=", new_roles(roles))
      end

      def get_roles
        self.send(role_attribute) || []
      end      
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
roles_active_record-0.5.0	lib/roles_active_record/strategy/multi.rb
roles_active_record-0.4.9	lib/roles_active_record/strategy/multi.rb
roles_active_record-0.4.8	lib/roles_active_record/strategy/multi.rb
roles_active_record-0.4.7	lib/roles_active_record/strategy/multi.rb
roles_active_record-0.4.6.4	lib/roles_active_record/strategy/multi.rb