summaryrefslogtreecommitdiff
path: root/app/models/concerns/chronic_duration_attribute.rb
blob: ad8934f58d8b1733dd57b7713f7c3b1707844c5d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
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)
    end

    def chronic_duration_attr_reader(virtual_attribute, source_attribute)
      define_method(virtual_attribute) do
        value = self.send(source_attribute) # rubocop:disable GitlabSecurity/PublicSend

        return '' if value.nil?

        ChronicDuration.output(value, format: :short)
      end
    end

    def chronic_duration_attr_writer(virtual_attribute, source_attribute)
      virtual_attribute_validator = "#{virtual_attribute}_validator".to_sym
      validation_error = "#{virtual_attribute}_error".to_sym

      validate virtual_attribute_validator
      attr_accessor validation_error

      define_method("#{virtual_attribute}=") do |value|
        begin
          self.send("#{validation_error}=", '') # rubocop:disable GitlabSecurity/PublicSend

          new_value =
            if value.blank?
              nil
            else
              ChronicDuration.parse(value).to_i
            end

          self.send("#{source_attribute}=", new_value) # rubocop:disable GitlabSecurity/PublicSend
        rescue ChronicDuration::DurationParseError => ex
          self.send("#{validation_error}=", ex.message) # rubocop:disable GitlabSecurity/PublicSend
        end
      end

      define_method(virtual_attribute_validator) do
        error = self.send(validation_error) # rubocop:disable GitlabSecurity/PublicSend
        self.send('errors').add(source_attribute, error) unless error.blank? # rubocop:disable GitlabSecurity/PublicSend
      end
    end
  end
end