mongoid_roles_zhd 0.1

data/lib/mongoid_roles/strategy/shared.rb

@@ -0,0 +1,29 @@
+module Mongoid::Roles
+  module Strategy
+    module Shared
+      def set_role role
+        raise ArgumentError, "#set_role only takes a single role as argument, not #{role}" if role.kind_of?(Array)
+        self.send("#{role_attribute}=", role)
+      end
+      
+      def set_roles *roles
+        roles = roles.flat_uniq
+        roles = roles.first if self.class.role_strategy.multiplicity == :single        
+        self.send("#{role_attribute}=", roles)
+      end
+            
+      def get_role
+        r = self.send(role_attribute) 
+      end
+
+      def get_roles
+        r = self.send(role_attribute)
+      end
+
+      def select_valid_roles *roles
+        roles.flat_uniq.select{|role| valid_role? role }
+        has_role_class? ? role_class.find_roles(roles).to_a.flat_uniq : roles.flat_uniq
+      end           
+    end
+  end
+end

data/lib/mongoid_roles/strategy/single.rb

@@ -0,0 +1,34 @@
+require 'mongoid_roles/strategy/shared'
+
+class Symbol
+  def admin?
+    self == :admin
+  end
+end
+
+module Mongoid::Roles
+  module Strategy
+    module Single
+      include Shared            
+      # assigns first valid role from list of roles
+      def add_roles *role_names
+        new_roles = select_valid_roles(role_names) 
+        new_role = new_roles.first if !new_roles.empty?
+        set_role new_role
+      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
+        roles = role_names.flat_uniq
+        set_empty_role if roles_diff(roles).empty?
+        true
+      end 
+      
+      def roles_list
+        raise 'the method #roles should be present' if !respond_to? :roles
+        present_roles(roles)
+      end            
+    end
+  end
+end

data/lib/mongoid_roles/strategy/single/admin_flag.rb

@@ -0,0 +1,55 @@
+require 'mongoid_roles/strategy/single'
+
+module RoleStrategy::Mongoid
+  module AdminFlag    
+    def self.default_role_attribute
+      :admin_flag
+    end
+
+    def self.included base
+      base.extend ClassMethods
+    end
+
+    module ClassMethods 
+      def role_attribute
+        strategy_class.roles_attribute_name.to_sym
+      end
+           
+      def in_role(role_name) 
+        case role_name.downcase.to_sym
+        when :admin
+          where(role_attribute => true)
+        else
+          where(role_attribute => false)
+        end          
+      end
+    end
+
+    module Implementation
+      include Mongoid::Roles::Strategy::Single
+
+      def new_role role
+        role = role.kind_of?(Array) ? role.flatten.first : role
+        role.admin?
+      end
+      
+      def get_role
+        self.send(role_attribute) ? strategy_class.admin_role_key : strategy_class.default_role_key
+      end 
+      
+      def present_roles *roles
+        roles = roles.flat_uniq
+        roles.map{|role| role ? :admin : :guest}
+      end   
+      
+      def set_empty_role
+        self.send("#{role_attribute}=", false)
+      end      
+
+
+    end # Implementation
+    
+    extend Roles::Generic::User::Configuration
+    configure :num => :single
+  end   
+end

data/lib/mongoid_roles/strategy/single/embed_one_role.rb

@@ -0,0 +1,83 @@
+require 'mongoid_roles/strategy/single'
+
+module RoleStrategy::Mongoid
+  module EmbedOneRole
+    def self.default_role_attribute
+      :one_role
+    end
+
+    def self.included base
+      base.extend Roles::Generic::Role::ClassMethods
+      base.extend ClassMethods
+    end
+
+    module ClassMethods 
+      def role_attribute
+        strategy_class.roles_attribute_name.to_sym
+      end       
+
+      def role_attribute_id
+        "#{role_attribute}_id"
+      end       
+         
+      def in_role(role_name)      
+        in_any_role(role_name)
+      end
+
+      def in_any_role(*role_names)
+        begin          
+          any_in("one_role.name" => role_names)
+        rescue 
+          return []
+        end
+      end  
+    end
+
+    module Implementation  
+      include Mongoid::Roles::Strategy::Single 
+
+      def set_role role
+        raise ArgumentError, "#set_role only takes a single role as argument, not #{role}" if role.kind_of?(Array)
+        self.send("create_#{role_attribute}", :name => role)
+      end      
+
+      def set_roles *roles
+        roles = roles.flat_uniq
+        self.send("create_#{role_attribute}", :name => roles.first)
+      end
+
+      def exchange_roles *role_names
+        options = last_option role_names
+        raise ArgumentError, "Must take an options hash as last argument with a :with option signifying which role(s) to replace with" if !options || !options.kind_of?(Hash)
+        common = role_names.to_symbols & roles_list
+        if !common.empty?
+          with_roles = options[:with]
+          set_role with_roles
+        end
+      end
+      
+      def new_role role
+        role_class.find_role(extract_role role)
+      end  
+      
+      def new_roles *roles
+        new_role roles.flatten.first
+      end     
+      
+      def present_roles *roles
+        roles.map{|role| extract_role role}
+      end                 
+      
+      def set_empty_role
+        self.send("#{role_attribute}=", nil)
+      end
+
+      def select_valid_roles *role_names
+        role_names = role_names.flat_uniq.select{|role| valid_role? role }        
+      end      
+    end
+
+    extend Roles::Generic::User::Configuration
+    configure :num => :single, :type => :role_class    
+  end  
+end

data/lib/mongoid_roles/strategy/single/one_role.rb

@@ -0,0 +1,66 @@
+require 'mongoid_roles/strategy/single'
+
+module RoleStrategy::Mongoid
+  module OneRole
+    def self.default_role_attribute
+      :one_role
+    end
+
+    def self.included base
+      base.extend Roles::Generic::Role::ClassMethods
+      base.extend ClassMethods
+    end
+
+    module ClassMethods 
+      def role_attribute
+        strategy_class.roles_attribute_name.to_sym
+      end       
+
+      def role_attribute_id
+        "#{role_attribute}_id"
+      end       
+         
+      def in_role(role_name)      
+        in_any_role(role_name)
+      end
+
+      def in_any_role(*role_names)
+        begin                       
+          the_roles = Role.where(:name.in => role_names.to_strings).to_a 
+          role_ids = the_roles.map(&:_id).map(&:to_s)
+          where(:"#{role_attribute_id}".in => role_ids)
+        rescue 
+          return []
+        end
+      end  
+    end
+
+    module Implementation  
+      include Mongoid::Roles::Strategy::Single 
+      
+      def new_role role
+        role_class.find_role(extract_role role)        
+      end  
+      
+      def new_roles *roles
+        new_role roles.flatten.first
+      end     
+      
+      def present_roles *roles
+        roles.map{|role| extract_role role}
+      end                 
+      
+      def set_empty_role
+        self.send("#{role_attribute}=", nil)
+      end
+
+      def select_valid_roles *role_names
+        role_names = role_names.flat_uniq.select{|role| valid_role? role }
+        has_role_class? ? role_class.find_roles(role_names).to_a : role_names
+      end      
+    end
+
+    extend Roles::Generic::User::Configuration
+    configure :num => :single, :type => :role_class    
+  end  
+end

data/lib/mongoid_roles/strategy/single/role_string.rb

@@ -0,0 +1,61 @@
+require 'mongoid_roles/strategy/single'
+
+module RoleStrategy::Mongoid
+  module RoleString    
+    def self.default_role_attribute
+      :role_string
+    end
+
+    def self.included base
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def role_attribute
+        strategy_class.roles_attribute_name.to_sym
+      end       
+         
+      def in_role(role_name)      
+        in_any_role(role_name)
+      end
+
+      def in_any_role(*role_names)
+        begin                       
+          where(:"#{role_attribute}".in => role_names)
+        rescue 
+          return []
+        end
+      end  
+    end
+
+    
+    module Implementation   
+      include Mongoid::Roles::Strategy::Single
+
+      def new_role role
+        role = role.kind_of?(Array) ? role.first : role
+        role.to_s
+      end
+
+      def new_roles *roles
+        new_role roles.flatten.first
+      end
+      
+      def present_role role
+        return role if role.kind_of? Array
+        role.split(',').map(&:to_sym)
+      end
+
+      def set_empty_role
+        self.send("#{role_attribute}=", "")
+      end
+
+      alias_method :present_roles, :present_role      
+    end
+
+    extend Roles::Generic::User::Configuration
+    configure :num => :single
+  end
+end
+
+

data/mongoid_roles.gemspec

@@ -0,0 +1,125 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{mongoid_roles_zhd}
+  s.version = "0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = [%q{ZhangHanDong}]
+  s.date = %q{2011-06-16}
+  s.description = %q{Makes it easy to set a role strategy on your User model in Mongoid}
+  s.email = %q{blackanger.z@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README.textile"
+  ]
+  s.files = [
+    ".bundle/config",
+    ".document",
+    ".rspec",
+    "Gemfile",
+    "LICENSE",
+    "README.textile",
+    "Rakefile",
+    "VERSION",
+    "lib/generators/mongoid/roles/core_ext.rb",
+    "lib/generators/mongoid/roles/roles_generator.rb",
+    "lib/generators/mongoid/roles/templates/role.rb",
+    "lib/mongoid_roles.rb",
+    "lib/mongoid_roles/base.rb",
+    "lib/mongoid_roles/base_role.rb",
+    "lib/mongoid_roles/embedded_role.rb",
+    "lib/mongoid_roles/namespaces.rb",
+    "lib/mongoid_roles/role.rb",
+    "lib/mongoid_roles/strategy.rb",
+    "lib/mongoid_roles/strategy/multi.rb",
+    "lib/mongoid_roles/strategy/multi/embed_many_roles.rb",
+    "lib/mongoid_roles/strategy/multi/many_roles.rb",
+    "lib/mongoid_roles/strategy/multi/role_strings.rb",
+    "lib/mongoid_roles/strategy/multi/roles_mask.rb",
+    "lib/mongoid_roles/strategy/shared.rb",
+    "lib/mongoid_roles/strategy/single.rb",
+    "lib/mongoid_roles/strategy/single/admin_flag.rb",
+    "lib/mongoid_roles/strategy/single/embed_one_role.rb",
+    "lib/mongoid_roles/strategy/single/one_role.rb",
+    "lib/mongoid_roles/strategy/single/role_string.rb",
+    "mongoid_roles.gemspec",
+    "sandbox/roles_assign.rb",
+    "sandbox/single_role.rb",
+    "sandbox/test.rb",
+    "sandbox/test_query.rb",
+    "spec/generator_spec_helper.rb",
+    "spec/mongoid_roles/generators/roles_generator_spec.rb",
+    "spec/mongoid_roles/strategy/api_examples.rb",
+    "spec/mongoid_roles/strategy/multi/embed_many_roles_spec.rb",
+    "spec/mongoid_roles/strategy/multi/many_roles_old_spec.rb",
+    "spec/mongoid_roles/strategy/multi/many_roles_spec.rb",
+    "spec/mongoid_roles/strategy/multi/role_strings_spec.rb",
+    "spec/mongoid_roles/strategy/multi/roles_mask_spec.rb",
+    "spec/mongoid_roles/strategy/single/admin_flag_spec.rb",
+    "spec/mongoid_roles/strategy/single/embed_one_role_spec.rb",
+    "spec/mongoid_roles/strategy/single/one_role_spec.rb",
+    "spec/mongoid_roles/strategy/single/role_string_spec.rb",
+    "spec/mongoid_roles/strategy/user_setup.rb",
+    "spec/spec_helper.rb",
+    "tmp/rails/app/models/user.rb",
+    "tmp/rails/config/routes.rb"
+  ]
+  s.homepage = %q{https://github.com/ZhangHanDong/roles_mongoid}
+  s.require_paths = [%q{lib}]
+  s.rubygems_version = %q{1.8.0}
+  s.summary = %q{Implementation of Roles generic API for Mongoid}
+  s.test_files = [
+    "spec/generator_spec_helper.rb",
+    "spec/mongoid_roles/generators/roles_generator_spec.rb",
+    "spec/mongoid_roles/strategy/api_examples.rb",
+    "spec/mongoid_roles/strategy/multi/embed_many_roles_spec.rb",
+    "spec/mongoid_roles/strategy/multi/many_roles_old_spec.rb",
+    "spec/mongoid_roles/strategy/multi/many_roles_spec.rb",
+    "spec/mongoid_roles/strategy/multi/role_strings_spec.rb",
+    "spec/mongoid_roles/strategy/multi/roles_mask_spec.rb",
+    "spec/mongoid_roles/strategy/single/admin_flag_spec.rb",
+    "spec/mongoid_roles/strategy/single/embed_one_role_spec.rb",
+    "spec/mongoid_roles/strategy/single/one_role_spec.rb",
+    "spec/mongoid_roles/strategy/single/role_string_spec.rb",
+    "spec/mongoid_roles/strategy/user_setup.rb",
+    "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, [">= 2.4.1"])
+      s.add_development_dependency(%q<generator-spec>, [">= 0.7.3"])
+      s.add_runtime_dependency(%q<mongoid>, [">= 2.0.2"])
+      s.add_runtime_dependency(%q<bson>, [">= 1.3.1"])
+      s.add_runtime_dependency(%q<sugar-high>, ["~> 0.4.0"])
+      s.add_runtime_dependency(%q<require_all>, ["~> 1.2.0"])
+      s.add_runtime_dependency(%q<roles_generic>, ["~> 0.3.9"])
+      s.add_runtime_dependency(%q<rails_artifactor>, [">= 0.3.3"])
+    else
+      s.add_dependency(%q<rspec>, [">= 2.4.1"])
+      s.add_dependency(%q<generator-spec>, [">= 0.7.3"])
+      s.add_dependency(%q<mongoid>, [">= 2.0.2"])
+      s.add_dependency(%q<bson>, [">= 1.3.1"])
+      s.add_dependency(%q<sugar-high>, ["~> 0.4.0"])
+      s.add_dependency(%q<require_all>, ["~> 1.2.0"])
+      s.add_dependency(%q<roles_generic>, ["~> 0.3.9"])
+      s.add_dependency(%q<rails_artifactor>, [">= 0.3.3"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 2.4.1"])
+    s.add_dependency(%q<generator-spec>, [">= 0.7.3"])
+    s.add_dependency(%q<mongoid>, [">= 2.0.2"])
+    s.add_dependency(%q<bson>, [">= 1.3.1"])
+    s.add_dependency(%q<sugar-high>, ["~> 0.4.0"])
+    s.add_dependency(%q<require_all>, ["~> 1.2.0"])
+    s.add_dependency(%q<roles_generic>, ["~> 0.3.9"])
+    s.add_dependency(%q<rails_artifactor>, [">= 0.3.3"])
+  end
+end
+