Chris@1295: # Redmine - project management software
Chris@1295: # Copyright (C) 2006-2012  Jean-Philippe Lang
Chris@1295: #
Chris@1295: # This program is free software; you can redistribute it and/or
Chris@1295: # modify it under the terms of the GNU General Public License
Chris@1295: # as published by the Free Software Foundation; either version 2
Chris@1295: # of the License, or (at your option) any later version.
Chris@1295: #
Chris@1295: # This program is distributed in the hope that it will be useful,
Chris@1295: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1295: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1295: # GNU General Public License for more details.
Chris@1295: #
Chris@1295: # You should have received a copy of the GNU General Public License
Chris@1295: # along with this program; if not, write to the Free Software
Chris@1295: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1295: 
Chris@1295: module Redmine
Chris@1295:   module Acts
Chris@1295:     module Customizable
Chris@1295:       def self.included(base)
Chris@1295:         base.extend ClassMethods
Chris@1295:       end
Chris@1295: 
Chris@1295:       module ClassMethods
Chris@1295:         def acts_as_customizable(options = {})
Chris@1295:           return if self.included_modules.include?(Redmine::Acts::Customizable::InstanceMethods)
Chris@1295:           cattr_accessor :customizable_options
Chris@1295:           self.customizable_options = options
Chris@1295:           has_many :custom_values, :as => :customized,
Chris@1295:                                    :include => :custom_field,
Chris@1295:                                    :order => "#{CustomField.table_name}.position",
Chris@1295:                                    :dependent => :delete_all,
Chris@1295:                                    :validate => false
Chris@1295:           send :include, Redmine::Acts::Customizable::InstanceMethods
Chris@1295:           validate :validate_custom_field_values
Chris@1295:           after_save :save_custom_field_values
Chris@1295:         end
Chris@1295:       end
Chris@1295: 
Chris@1295:       module InstanceMethods
Chris@1295:         def self.included(base)
Chris@1295:           base.extend ClassMethods
Chris@1295:         end
Chris@1295: 
Chris@1295:         def available_custom_fields
Chris@1295:           CustomField.find(:all, :conditions => "type = '#{self.class.name}CustomField'",
Chris@1295:                                  :order => 'position')
Chris@1295:         end
Chris@1295: 
Chris@1295:         # Sets the values of the object's custom fields
Chris@1295:         # values is an array like [{'id' => 1, 'value' => 'foo'}, {'id' => 2, 'value' => 'bar'}]
Chris@1295:         def custom_fields=(values)
Chris@1295:           values_to_hash = values.inject({}) do |hash, v|
Chris@1295:             v = v.stringify_keys
Chris@1295:             if v['id'] && v.has_key?('value')
Chris@1295:               hash[v['id']] = v['value']
Chris@1295:             end
Chris@1295:             hash
Chris@1295:           end
Chris@1295:           self.custom_field_values = values_to_hash
Chris@1295:         end
Chris@1295: 
Chris@1295:         # Sets the values of the object's custom fields
Chris@1295:         # values is a hash like {'1' => 'foo', 2 => 'bar'}
Chris@1295:         def custom_field_values=(values)
Chris@1295:           values = values.stringify_keys
Chris@1295: 
Chris@1295:           custom_field_values.each do |custom_field_value|
Chris@1295:             key = custom_field_value.custom_field_id.to_s
Chris@1295:             if values.has_key?(key)
Chris@1295:               value = values[key]
Chris@1295:               if value.is_a?(Array)
Chris@1295:                 value = value.reject(&:blank?).uniq
Chris@1295:                 if value.empty?
Chris@1295:                   value << ''
Chris@1295:                 end
Chris@1295:               end
Chris@1295:               custom_field_value.value = value
Chris@1295:             end
Chris@1295:           end
Chris@1295:           @custom_field_values_changed = true
Chris@1295:         end
Chris@1295: 
Chris@1295:         def custom_field_values
Chris@1295:           @custom_field_values ||= available_custom_fields.collect do |field|
Chris@1295:             x = CustomFieldValue.new
Chris@1295:             x.custom_field = field
Chris@1295:             x.customized = self
Chris@1295:             if field.multiple?
Chris@1295:               values = custom_values.select { |v| v.custom_field == field }
Chris@1295:               if values.empty?
Chris@1295:                 values << custom_values.build(:customized => self, :custom_field => field, :value => nil)
Chris@1295:               end
Chris@1295:               x.value = values.map(&:value)
Chris@1295:             else
Chris@1295:               cv = custom_values.detect { |v| v.custom_field == field }
Chris@1295:               cv ||= custom_values.build(:customized => self, :custom_field => field, :value => nil)
Chris@1295:               x.value = cv.value
Chris@1295:             end
Chris@1295:             x
Chris@1295:           end
Chris@1295:         end
Chris@1295: 
Chris@1295:         def visible_custom_field_values
Chris@1295:           custom_field_values.select(&:visible?)
Chris@1295:         end
Chris@1295: 
Chris@1295:         def custom_field_values_changed?
Chris@1295:           @custom_field_values_changed == true
Chris@1295:         end
Chris@1295: 
Chris@1295:         def custom_value_for(c)
Chris@1295:           field_id = (c.is_a?(CustomField) ? c.id : c.to_i)
Chris@1295:           custom_values.detect {|v| v.custom_field_id == field_id }
Chris@1295:         end
Chris@1295: 
Chris@1295:         def custom_field_value(c)
Chris@1295:           field_id = (c.is_a?(CustomField) ? c.id : c.to_i)
Chris@1295:           custom_field_values.detect {|v| v.custom_field_id == field_id }.try(:value)
Chris@1295:         end
Chris@1295: 
Chris@1295:         def validate_custom_field_values
Chris@1295:           if new_record? || custom_field_values_changed?
Chris@1295:             custom_field_values.each(&:validate_value)
Chris@1295:           end
Chris@1295:         end
Chris@1295: 
Chris@1295:         def save_custom_field_values
Chris@1295:           target_custom_values = []
Chris@1295:           custom_field_values.each do |custom_field_value|
Chris@1295:             if custom_field_value.value.is_a?(Array)
Chris@1295:               custom_field_value.value.each do |v|
Chris@1295:                 target = custom_values.detect {|cv| cv.custom_field == custom_field_value.custom_field && cv.value == v}
Chris@1295:                 target ||= custom_values.build(:customized => self, :custom_field => custom_field_value.custom_field, :value => v)
Chris@1295:                 target_custom_values << target
Chris@1295:               end
Chris@1295:             else
Chris@1295:               target = custom_values.detect {|cv| cv.custom_field == custom_field_value.custom_field}
Chris@1295:               target ||= custom_values.build(:customized => self, :custom_field => custom_field_value.custom_field)
Chris@1295:               target.value = custom_field_value.value
Chris@1295:               target_custom_values << target
Chris@1295:             end
Chris@1295:           end
Chris@1295:           self.custom_values = target_custom_values
Chris@1295:           custom_values.each(&:save)
Chris@1295:           @custom_field_values_changed = false
Chris@1295:           true
Chris@1295:         end
Chris@1295: 
Chris@1295:         def reset_custom_values!
Chris@1295:           @custom_field_values = nil
Chris@1295:           @custom_field_values_changed = true
Chris@1295:         end
Chris@1295: 
Chris@1295:         module ClassMethods
Chris@1295:         end
Chris@1295:       end
Chris@1295:     end
Chris@1295:   end
Chris@1295: end