Support Postgres Enum

Shopify · Jan 7, 2025 · 6e2aa38 · 6e2aa38
1 parent bce1224
commit 6e2aa38
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/tapioca/dsl/helpers/active_record_column_type_helper.rb b/lib/tapioca/dsl/helpers/active_record_column_type_helper.rb
@@ -216,6 +216,11 @@ def type_for_activerecord_value(column_type, column_nullability:)
                    ActiveRecord::Locking::LockingType === type
                }
             as_non_nilable_if_persisted_and_not_nullable("::Integer", column_nullability:)
+          when ->(type) {
+                 defined?(ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Enum) &&
+                   ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Enum === type
+               }
+            "::String"
           else
             as_non_nilable_if_persisted_and_not_nullable(
               ActiveModelTypeHelper.type_for(column_type),

diff --git a/sorbet/rbi/shims/activerecord_postgresql.rbi b/sorbet/rbi/shims/activerecord_postgresql.rbi
@@ -9,3 +9,4 @@ module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Interval; end
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Range; end
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Bit; end
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::BitVarying; end
+module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Enum; end