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