Preserve type when dumping bit and bit varying columns

Commits -
  rails/rails@fcd5e56
  rails/rails@d5411fb

Author: kares

lib/arjdbc/postgresql/adapter.rb

@@ -241,6 +241,11 @@ def type_cast(value, column, array_member = false)
       :int8range => { :name => "int8range" },
     }) if AR4_COMPAT
 
+    NATIVE_DATABASE_TYPES.update(
+      :bit => { name: "bit" },
+      :bit_varying => { name: "bit varying" }
+    ) if ActiveRecord::VERSION.to_s >= '4.2'
+
     def native_database_types
       NATIVE_DATABASE_TYPES
     end
@@ -871,7 +876,22 @@ def quote(value, column = nil)
       else
         super
       end
-    end
+    end if ActiveRecord::VERSION.to_s < '4.2'
+
+    def quote(value, column = nil)
+      return super unless column
+
+      case value
+      when Float
+        if value.infinite? || value.nan?
+          "'#{value.to_s}'"
+        else
+          super
+        end
+      else
+        super
+      end
+    end if ActiveRecord::VERSION.to_s >= '4.2'
 
     # Quotes a string, escaping any ' (single quote) and \ (backslash) chars.
     # @return [String]
@@ -1414,6 +1434,14 @@ def uuid(name, options = {})
       def json(name, options = {})
         column(name, 'json', options)
       end
+
+      def bit(name, options)
+        column(name, 'bit', options)
+      end
+
+      def bit_varying(name, options)
+        column(name, 'bit varying', options)
+      end
     end
 
     class TableDefinition < ActiveRecord::ConnectionAdapters::TableDefinition

lib/arjdbc/postgresql/base/oid.rb

@@ -66,6 +66,57 @@ def type_cast(value)
           end
         end
 
+        class Bit < ActiveRecord::Type::Value
+          def type
+            :bit
+          end
+
+          def type_cast(value)
+            if ::String === value
+              case value
+              when /^0x/i
+                value[2..-1].hex.to_s(2) # Hexadecimal notation
+              else
+                value                    # Bit-string notation
+              end
+            else
+              value
+            end
+          end
+
+          def type_cast_for_database(value)
+            Data.new(super) if value
+          end
+
+          class Data
+            def initialize(value)
+              @value = value
+            end
+
+            def to_s
+              value
+            end
+
+            def binary?
+              /\A[01]*\Z/ === value
+            end
+
+            def hex?
+              /\A[0-9A-F]*\Z/i === value
+            end
+
+            protected
+
+            attr_reader :value
+          end
+        end if ActiveRecord::VERSION.to_s >= '4.2'
+
+        class BitVarying < Bit
+          def type
+            :bit_varying
+          end
+        end if ActiveRecord::VERSION.to_s >= '4.2'
+
         class Bytea < Type
           def type; :binary end