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 Member < ActiveRecord::Base Chris@1494: belongs_to :user Chris@1494: belongs_to :principal, :foreign_key => 'user_id' Chris@1494: has_many :member_roles, :dependent => :destroy Chris@1494: has_many :roles, :through => :member_roles Chris@1494: belongs_to :project Chris@1494: Chris@1494: validates_presence_of :principal, :project Chris@1494: validates_uniqueness_of :user_id, :scope => :project_id Chris@1494: validate :validate_role Chris@1494: Chris@1494: before_destroy :set_issue_category_nil Chris@1494: Chris@1494: def role Chris@1494: end Chris@1494: Chris@1494: def role= Chris@1494: end Chris@1494: Chris@1494: def name Chris@1494: self.user.name Chris@1494: end Chris@1494: Chris@1494: alias :base_role_ids= :role_ids= Chris@1494: def role_ids=(arg) Chris@1494: ids = (arg || []).collect(&:to_i) - [0] Chris@1494: # Keep inherited roles Chris@1494: ids += member_roles.select {|mr| !mr.inherited_from.nil?}.collect(&:role_id) Chris@1494: Chris@1494: new_role_ids = ids - role_ids Chris@1494: # Add new roles Chris@1494: new_role_ids.each {|id| member_roles << MemberRole.new(:role_id => id) } Chris@1494: # Remove roles (Rails' #role_ids= will not trigger MemberRole#on_destroy) Chris@1494: member_roles_to_destroy = member_roles.select {|mr| !ids.include?(mr.role_id)} Chris@1494: if member_roles_to_destroy.any? Chris@1494: member_roles_to_destroy.each(&:destroy) Chris@1494: end Chris@1494: end Chris@1494: Chris@1494: def <=>(member) Chris@1494: a, b = roles.sort.first, member.roles.sort.first Chris@1494: if a == b Chris@1494: if principal Chris@1494: principal <=> member.principal Chris@1494: else Chris@1494: 1 Chris@1494: end Chris@1494: elsif a Chris@1494: a <=> b Chris@1494: else Chris@1494: 1 Chris@1494: end Chris@1494: end Chris@1494: Chris@1494: def deletable? Chris@1494: member_roles.detect {|mr| mr.inherited_from}.nil? Chris@1494: end Chris@1494: Chris@1494: def include?(user) Chris@1494: if principal.is_a?(Group) Chris@1494: !user.nil? && user.groups.include?(principal) Chris@1494: else Chris@1494: self.user == user Chris@1494: end Chris@1494: end Chris@1494: Chris@1494: def set_issue_category_nil Chris@1494: if user Chris@1494: # remove category based auto assignments for this member Chris@1494: IssueCategory.update_all "assigned_to_id = NULL", ["project_id = ? AND assigned_to_id = ?", project.id, user.id] Chris@1494: end Chris@1494: end Chris@1494: Chris@1494: # Find or initilize a Member with an id, attributes, and for a Principal Chris@1494: def self.edit_membership(id, new_attributes, principal=nil) Chris@1494: @membership = id.present? ? Member.find(id) : Member.new(:principal => principal) Chris@1494: @membership.attributes = new_attributes Chris@1494: @membership Chris@1494: end Chris@1494: Chris@1494: # Finds or initilizes a Member for the given project and principal Chris@1494: def self.find_or_new(project, principal) Chris@1494: project_id = project.is_a?(Project) ? project.id : project Chris@1494: principal_id = principal.is_a?(Principal) ? principal.id : principal Chris@1494: Chris@1494: member = Member.find_by_project_id_and_user_id(project_id, principal_id) Chris@1494: member ||= Member.new(:project_id => project_id, :user_id => principal_id) Chris@1494: member Chris@1494: end Chris@1494: Chris@1494: protected Chris@1494: Chris@1494: def validate_role Chris@1494: errors.add_on_empty :role if member_roles.empty? && roles.empty? Chris@1494: end Chris@1494: end