File: attr_required.rb

package info (click to toggle)
ruby-attr-required 1.0.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 164 kB
  • sloc: ruby: 290; makefile: 2
file content (74 lines) | stat: -rw-r--r-- 1,375 bytes parent folder | download | duplicates (3)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
module AttrRequired

  class AttrMissing < StandardError; end

  def self.included(klass)
    klass.send :extend, ClassMethods
  end

  module ClassMethods

    def inherited(klass)
      super
      unless required_attributes.empty?
        klass.attr_required *required_attributes
      end
    end

    def attr_required(*keys)
      if defined? undef_optional_attributes
        undef_optional_attributes *keys
      end
      required_attributes.concat keys
      attr_accessor *keys
    end

    def attr_required?(key)
      required_attributes.include? key
    end

    def required_attributes
      @required_attributes ||= []
    end

    def undef_required_attributes(*keys)
      keys.each do |key|
        if attr_required?(key)
          undef_method key, :"#{key}="
          required_attributes.delete key
        end
      end
    end

  end

  def required_attributes
    self.class.required_attributes
  end

  def attr_required?(key)
    self.class.attr_required? key
  end

  def attr_missing?
    !attr_missing.empty?
  end

  def attr_missing!
    if attr_missing?
      raise AttrMissing.new("'#{attr_missing.join('\', \'')}' required.")
    end
  end

  def attr_missing
    required_attributes.select do |key|
      value = send(key)
      if value.respond_to?(:empty?)
        value.empty?
      else
        value.nil?
      end
    end
  end

end