Chris@1494: # Redmine - project management software
Chris@1494: # Copyright (C) 2006-2014  Jean-Philippe Lang
Chris@1494: #
Chris@1494: # This program is free software; you can redistribute it and/or
Chris@1494: # modify it under the terms of the GNU General Public License
Chris@1494: # as published by the Free Software Foundation; either version 2
Chris@1494: # of the License, or (at your option) any later version.
Chris@1494: #
Chris@1494: # This program is distributed in the hope that it will be useful,
Chris@1494: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1494: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1494: # GNU General Public License for more details.
Chris@1494: #
Chris@1494: # You should have received a copy of the GNU General Public License
Chris@1494: # along with this program; if not, write to the Free Software
Chris@1494: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1494: 
Chris@1494: class MemberRole < ActiveRecord::Base
Chris@1494:   belongs_to :member
Chris@1494:   belongs_to :role
Chris@1494: 
Chris@1494:   after_destroy :remove_member_if_empty
Chris@1494: 
Chris@1494:   after_create :add_role_to_group_users, :add_role_to_subprojects
Chris@1494:   after_destroy :remove_inherited_roles
Chris@1494: 
Chris@1494:   validates_presence_of :role
Chris@1494:   validate :validate_role_member
Chris@1494: 
Chris@1494:   def validate_role_member
Chris@1494:     errors.add :role_id, :invalid if role && !role.member?
Chris@1494:   end
Chris@1494: 
Chris@1494:   def inherited?
Chris@1494:     !inherited_from.nil?
Chris@1494:   end
Chris@1494: 
Chris@1494:   private
Chris@1494: 
Chris@1494:   def remove_member_if_empty
Chris@1494:     if member.roles.empty?
Chris@1494:       member.destroy
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def add_role_to_group_users
Chris@1494:     if member.principal.is_a?(Group) && !inherited?
Chris@1494:       member.principal.users.each do |user|
Chris@1494:         user_member = Member.find_or_new(member.project_id, user.id)
Chris@1494:         user_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
Chris@1494:         user_member.save!
Chris@1494:       end
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def add_role_to_subprojects
Chris@1494:     member.project.children.each do |subproject|
Chris@1494:       if subproject.inherit_members?
Chris@1494:         child_member = Member.find_or_new(subproject.id, member.user_id)
Chris@1494:         child_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
Chris@1494:         child_member.save!
Chris@1494:       end
Chris@1494:     end
Chris@1494:   end
Chris@1494: 
Chris@1494:   def remove_inherited_roles
Chris@1494:     MemberRole.where(:inherited_from => id).all.group_by(&:member).each do |member, member_roles|
Chris@1494:       member_roles.each(&:destroy)
Chris@1494:     end
Chris@1494:   end
Chris@1494: end