gitlab-org--gitlab-foss/app/models/concerns/chronic_duration_attribute.rb

32 lines
1023 B
Ruby
Raw Normal View History

2018-02-20 19:09:59 -05:00
module ChronicDurationAttribute
extend ActiveSupport::Concern
class_methods do
def chronic_duration_attr(virtual_attribute, source_attribute)
chronic_duration_attr_reader(virtual_attribute, source_attribute)
chronic_duration_attr_writer(virtual_attribute, source_attribute)
2018-02-20 19:09:59 -05:00
end
def chronic_duration_attr_reader(virtual_attribute, source_attribute)
2018-02-20 19:09:59 -05:00
define_method(virtual_attribute) do
value = self.send(source_attribute) # rubocop:disable GitlabSecurity/PublicSend
return '' if value.nil?
ChronicDuration.output(value, format: :short)
2018-02-20 19:09:59 -05:00
end
end
def chronic_duration_attr_writer(virtual_attribute, source_attribute)
2018-02-20 19:09:59 -05:00
define_method("#{virtual_attribute}=") do |value|
2018-03-05 14:22:00 -05:00
new_value = ChronicDuration.parse(value).to_i unless value.nil?
new_value = nil if !new_value.nil? && new_value <= 0
2018-02-20 19:09:59 -05:00
self.send("#{source_attribute}=", new_value) # rubocop:disable GitlabSecurity/PublicSend
2018-02-20 19:09:59 -05:00
new_value
end
end
end
end