Module: HasBitField

Defined in:
lib/has-bit-field.rb

Instance Method Summary collapse

Instance Method Details

#has_bit_field(bit_field_attribute, *args) ⇒ Object

The first arguement bit_field_attribute should be a symbol, the name of attribute that will hold the actual bit field all following arguments should also be symbols, which will be the name of each flag in the bit field



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
 
# File 'lib/has-bit-field.rb', line 6

def has_bit_field(bit_field_attribute, *args)
  unless table_exists?
    Rails.logger.error("[has_bit_field] table undefined #{table_name}") if defined?(Rails) && Rails.respond_to?(:logger)
    return
  end
  if columns_hash[bit_field_attribute.to_s].blank?
    Rails.logger.error("[has_bit_field] column undefined #{bit_field_attribute}") if defined?(Rails) && Rails.respond_to?(:logger)
    return
  end
  args.each_with_index do |field,i|
    class_eval %{
      class << self
        def #{field}_bit
          (1 << #{i})
        end
      end

      def #{field}
        (#{bit_field_attribute} & self.class.#{field}_bit) != 0
      end

      alias #{field}? #{field}

      def #{field}=(v)
        if v.to_s == "true" || v.to_s == "1"
          self.#{bit_field_attribute} = (#{bit_field_attribute} || 0) | self.class.#{field}_bit
        else
          self.#{bit_field_attribute} = (#{bit_field_attribute} || 0) & ~self.class.#{field}_bit
        end
      end

      def #{field}_was
        (#{bit_field_attribute}_was & self.class.#{field}_bit) != 0
      end

      def #{field}_changed?
        #{field} != #{field}_was
      end
    }

    if columns_hash[bit_field_attribute.to_s].null
      scope field, lambda {
        where(arel_table[bit_field_attribute].not_eq(nil).
              and(Arel::Nodes::InfixOperation.new(:&, arel_table[bit_field_attribute], 1<<i).not_eq(0)))
      }
      scope "not_#{field}", lambda {
        where(arel_table[bit_field_attribute].eq(nil).
              or(Arel::Nodes::InfixOperation.new(:&, arel_table[bit_field_attribute], 1<<i).eq(0)))
      }
    else
      scope field, lambda {
        where(Arel::Nodes::InfixOperation.new(:&, arel_table[bit_field_attribute], 1<<i).not_eq(0))
      }
      scope "not_#{field}", lambda {
        where(Arel::Nodes::InfixOperation.new(:&, arel_table[bit_field_attribute], 1<<i).eq(0))
      }
    end

  end
end