foobara-postgresql-crud-driver 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6846188257d6cad52ea962972ddd82f82d72162ea0a46df18983a92bb3c2b444
-  data.tar.gz: 349cb1d585f814335f083ed9cb5105a42bb9e9b10a8a8e5f9fa6444b3670dcfd
+  metadata.gz: c6ce8966a4872d4a3e81a1bcfec564df6be8903cd20e7bf8b998b9fc6ba423b8
+  data.tar.gz: ead889be3b61d1293dd8c78ada43f3b2f3c68cbaa11af81ef6fbc3ce975fdc5f
 SHA512:
-  metadata.gz: 516fede04188fb07fdb5b62be6400cdaf507c3d7b1ffc961ec5045591f9257c1bbeca2967dd21a12a6e96484690b30344422509e7a4ea5dc5724afab426223d0
-  data.tar.gz: 4cab91509e1450dceafe8bcd022f25f48af4d52906a0d9b5c4a86d75cb1520f5347bf96b4b3f76c09bbb9cdb053e9fd03d8009f77d931831db9e1cfb5c291efa
+  metadata.gz: 3d7bdb1286bc1a27ad62217f278155225722e88cb01c8faa193c80a932f1b07d441ef78bc0dac56ade1ce7e29c51d85e63f28e4b77c7913184ccd695a077caca
+  data.tar.gz: a4509b915954600c4c7001e0e078856a8ef38063298877ff076cdfd94bf76c1b37b08a2488453e54084c7458189189a19881762406b4019a8d186e4de2f032b9

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## [0.0.4] - 2025-11-09
+
+- Support a few more column types
+- Make sure an entity -> entity association is cast properly
+
 ## [0.0.3] - 2025-10-21
 
 - Pass through options to EntityAttributesCrudDriver instead of disallowing them

data/src/postgresql_crud_driver.rb

@@ -4,6 +4,18 @@ module Foobara
     class NoSuchColumnOrTableError < StandardError; end
 
     class UnsupportedPgColumnTypeError < StandardError
+      attr_accessor :pg_type, :attribute_name, :entity_class
+
+      def initialize(pg_type, entity_class)
+        # :nocov:
+        super("Unsupported column type #{pg_type} on #{entity_class.entity_name}")
+        # :nocov:
+      end
+    end
+
+    class UnsupportedPgColumnTypeForAttributeError < StandardError
+      attr_accessor :pg_type, :attribute_name, :entity_class
+
       def initialize(pg_type, attribute_name, entity_class)
         # :nocov:
         super("Unsupported column type #{pg_type} for attribute #{attribute_name} on #{entity_class.entity_name}")
@@ -16,7 +28,7 @@ module Foobara
         @get_transaction_number ||= 0
         @get_transaction_number += 1
         if @get_transaction_number > 65_535
-          # TODO: test this codepath somehow
+          # TODO: test this code path somehow
           # :nocov:
           @get_transaction_number = 1
           # :nocov:
@@ -267,9 +279,11 @@ module Foobara
           info = column_info[attribute_name.to_s]
 
           value = case info[:type]
-                  when "integer", "text", "timestamp without time zone"
+                  when "integer", "bigint",
+                       "text", "character varying",
+                       "timestamp without time zone"
                     value
-                  when "jsonb"
+                  when "jsonb", "json"
                     if value.nil?
                       unless info[:is_nullable]
                         # :nocov:
@@ -306,79 +320,19 @@ module Foobara
           # :nocov:
         end
 
-        pg_type = info[:type]
         foobara_type = entity_class.model_type.element_types.element_types[attribute_name]
 
-        value = if value.nil?
-                  if info[:is_nullable]
-                    "NULL"
-                  else
-                    # :nocov:
-                    raise "Unexpected nil value for #{attribute_name}"
-                    # :nocov:
-                  end
-                elsif foobara_type.extends?(:number)
-                  case pg_type
-                  when "integer"
-                    value.to_i
-                  else
-                    # :nocov:
-                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                    # :nocov:
-                  end
-                elsif foobara_type.extends?(:string) || foobara_type.extends?(:symbol)
-                  case pg_type
-                  when "text"
-                    "'#{PG::Connection.escape(value.to_s)}'"
-                  else
-                    # :nocov:
-                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                    # :nocov:
-                  end
-                elsif foobara_type.extends?(:datetime)
-                  case pg_type
-                  when "timestamp without time zone"
-                    "'#{PG::Connection.escape(value.inspect)}'"
-                  else
-                    # :nocov:
-                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                    # :nocov:
-                  end
-                elsif foobara_type.extends?(:model) || foobara_type.extends?(:attributes)
-                  case pg_type
-                  when "jsonb"
-                    "'#{PG::Connection.escape(JSON.fast_generate(value))}'"
-                  else
-                    # :nocov:
-                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                    # :nocov:
-                  end
-                elsif foobara_type.extends?(:array)
-                  element_type = foobara_type.element_type
-
-                  if element_type.extends?(:detached_entity)
-                    case pg_type
-                    when "ARRAY"
-                      elements_type = ARRAY_ELEMENT_ENCODERS[info[:element_type]]
-                      array_string = PG::TextEncoder::Array.new(elements_type:).encode(value)
-                      escaped = PG::Connection.escape(array_string)
-
-                      "'#{escaped}'"
-                    else
-                      # :nocov:
-                      raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                      # :nocov:
-                    end
-                  else
-                    # :nocov:
-                    raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                    # :nocov:
-                  end
-                else
-                  # :nocov:
-                  raise UnsupportedPgColumnTypeError.new(pg_type, attribute_name, entity_class)
-                  # :nocov:
-                end
+        value = begin
+          pg_cast_value(value, foobara_type, info)
+        rescue UnsupportedPgColumnTypeError => e
+          # :nocov:
+          raise UnsupportedPgColumnTypeForAttributeError.new(
+            e.pg_type,
+            attribute_name,
+            entity_class
+          )
+          # :nocov:
+        end
 
         [PostgresqlCrudDriver.escape_identifier(attribute_name), value]
       end
@@ -406,6 +360,83 @@ module Foobara
           ]
         end
       end
+
+      def pg_cast_value(value, foobara_type, pg_info)
+        pg_type = pg_info[:type]
+
+        if value.nil?
+          if pg_info[:is_nullable]
+            "NULL"
+          else
+            # :nocov:
+            raise "Unexpected nil value for #{attribute_name}"
+            # :nocov:
+          end
+        elsif foobara_type.extends?(:number)
+          case pg_type
+          when "integer", "bigint"
+            value.to_i
+          else
+            # :nocov:
+            raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+            # :nocov:
+          end
+        elsif foobara_type.extends?(:string) || foobara_type.extends?(:symbol)
+          case pg_type
+          when "text", "character varying"
+            "'#{PG::Connection.escape(value.to_s)}'"
+          else
+            # :nocov:
+            raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+            # :nocov:
+          end
+        elsif foobara_type.extends?(:datetime)
+          case pg_type
+          when "timestamp without time zone"
+            "'#{PG::Connection.escape(value.inspect)}'"
+          else
+            # :nocov:
+            raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+            # :nocov:
+          end
+        elsif foobara_type.extends?(:detached_entity)
+          pg_cast_value(value, foobara_type.target_class.primary_key_type, pg_info)
+        elsif foobara_type.extends?(:model) || foobara_type.extends?(:attributes)
+          case pg_type
+          when "jsonb", "json"
+            "'#{PG::Connection.escape(JSON.fast_generate(value))}'"
+          else
+            # :nocov:
+            raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+            # :nocov:
+          end
+        elsif foobara_type.extends?(:array)
+          element_type = foobara_type.element_type
+
+          if element_type.extends?(:detached_entity)
+            case pg_type
+            when "ARRAY"
+              elements_type = ARRAY_ELEMENT_ENCODERS[pg_info[:element_type]]
+              array_string = PG::TextEncoder::Array.new(elements_type:).encode(value)
+              escaped = PG::Connection.escape(array_string)
+
+              "'#{escaped}'"
+            else
+              # :nocov:
+              raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+              # :nocov:
+            end
+          else
+            # :nocov:
+            raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+            # :nocov:
+          end
+        else
+          # :nocov:
+          raise UnsupportedPgColumnTypeError.new(pg_type, entity_class)
+          # :nocov:
+        end
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-postgresql-crud-driver
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Miles Georgi