Contents

require 'roles_active_record/strategy/multi'

module RoleStrategy::ActiveRecord
  module ManyRoles
    def self.default_role_attribute
      :many_roles
    end

    def self.included base
      base.extend Roles::Generic::Role::ClassMethods
      base.extend ClassMethods
    end

    module ClassMethods
      def in_role(role_name)
        in_any_role(role_name)
      end

      def in_any_role(*role_names)
        joins(:many_roles).where(role_arel_table[:name].in(role_names.flatten.to_strings))
      end
    end

    module Implementation
      include Roles::ActiveRecord::Strategy::Multi

      # assign multiple roles
      def roles=(*role_names)
        role_names = role_names.flat_uniq
        role_names = extract_roles(role_names)
        return nil if role_names.empty?
        valids = role_class.find_roles(role_names).to_a
        vrs = select_valid_roles role_names
        set_roles(vrs)
      end

      def new_roles *role_names
        role_class.find_roles(extract_roles role_names)
      end

      def present_roles roles_names
        roles_names.to_a.map{|role| role.name.to_s.to_sym}
      end

      def set_empty_roles
        self.send("#{role_attribute}=", [])
      end
      alias_method :set_empty_role, :set_empty_roles
    end

    extend Roles::Generic::User::Configuration
    configure :type => :role_class
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
roles_active_record-0.5.1	lib/roles_active_record/strategy/multi/many_roles.rb