108 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			108 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| class Authorization
 | |
|   include Mongoid::Document
 | |
|   include Mongoid::Timestamps
 | |
| 
 | |
|   field :title
 | |
| 
 | |
|   belongs_to :module_app
 | |
| 
 | |
|   has_and_belongs_to_many :roles
 | |
|   has_and_belongs_to_many :sub_roles
 | |
| 
 | |
|   delegate :update_auth_approval_users, :update_auth_manager_users, :update_auth_sub_manager_users, to: :module_app, prefix: true, allow_nil: true
 | |
| 
 | |
|   after_save :update_module_app
 | |
| 
 | |
| 
 | |
|   def add_roles(roles)
 | |
|   	users = []
 | |
|   	roles = Array(roles)
 | |
|     add_operation(:roles, roles)
 | |
|     sub_roles = [] 
 | |
|     roles.each{|role| role.sub_roles.each{|sub_role| sub_roles << sub_role.id}}
 | |
|     add_operation(:sub_roles, sub_roles)
 | |
|     roles.each{|role| role.users.where(admin: false).each{|user| users << user}}
 | |
| 		add_users(users, false)
 | |
|   end
 | |
|   
 | |
|   def add_sub_roles(sub_roles)
 | |
|   	users = []
 | |
|   	sub_roles = Array(sub_roles)
 | |
|     add_operation(:sub_roles, sub_roles)
 | |
|     sub_roles.each do |sub_role| 
 | |
| 			self.roles << sub_role.role unless self.roles.include?(sub_role.role)
 | |
|     	sub_role.users.where(admin: false).each{|user| users << user}
 | |
|     end
 | |
| 		add_users(users, false)
 | |
|   end
 | |
|   
 | |
|   def add_users(users, with_parents = true)
 | |
|   	users = Array(users)
 | |
|     add_operation(:authorized_users, users)
 | |
|     users.each do |user|
 | |
|     	user.roles.each do |role|
 | |
| 	    	self.roles << role unless self.roles.include?(role)
 | |
|     	end
 | |
|     	user.sub_roles.each do |sub_role|
 | |
| 	    	self.sub_roles << sub_role unless self.sub_roles.include?(sub_role)
 | |
|     	end
 | |
|     end if with_parents
 | |
|    	self.save
 | |
|   end
 | |
| 
 | |
|   def remove_roles(roles)
 | |
|   	users = []
 | |
|   	sub_roles = []
 | |
|   	roles = Array(roles)
 | |
|   	remove_operation(:roles, roles)
 | |
|  		roles.each do |role|
 | |
|  			role.sub_roles.each{|sub_role| sub_roles << sub_role}
 | |
|  			role.users.where(admin: false).each{|user| users << user}
 | |
|  		end
 | |
| 		remove_operation(:sub_role_ids, sub_roles)
 | |
| 		remove_operation(:authorized_user_ids, users)
 | |
| 		add_roles(self.roles)
 | |
|   end
 | |
| 
 | |
|   def remove_sub_roles(sub_roles)
 | |
|   	users = []
 | |
|   	sub_roles = Array(sub_roles)
 | |
|   	remove_operation(:sub_roles, sub_roles)
 | |
|   	sub_roles.each do |sub_role|
 | |
|   		users << sub_role.users.where(admin: false)
 | |
|   	end
 | |
| 		remove_operation(:authorized_user_ids, users)
 | |
| 		add_roles(self.roles)
 | |
|   end
 | |
| 
 | |
|   def remove_users(users)
 | |
|   	users = Array(users)
 | |
|   	remove_operation(:authorized_user_ids, users)
 | |
|    	self.save
 | |
|   end
 | |
| 
 | |
|   protected
 | |
| 
 | |
|   def add_operation(db_field, objs)
 | |
|    	objs.each do |obj|
 | |
| 	   	self.send(db_field) << obj unless self.send(db_field).include?(obj)
 | |
|    	end
 | |
|   end  
 | |
| 
 | |
|   def remove_operation(db_field, obj)
 | |
|     self.write_attribute(db_field, self.send(db_field) - obj.map{|y| y.id})
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def update_module_app
 | |
|   	case self._type
 | |
|   	when "AuthApproval"
 | |
|   		self.module_app_update_auth_approval_users
 | |
|   	when "AuthManager"
 | |
|   		self.module_app_update_auth_manager_users
 | |
|   	when "AuthSubManager"
 | |
|   		self.module_app_update_auth_sub_manager_users
 | |
|   	end
 | |
|   end
 | |
| end |