Support Postgres cidr column type

Shopify · May 21, 2024 · e12b986 · e12b986
1 parent 12a8fb7
commit e12b986
Show file tree

Hide file tree

Showing 2 changed files with 4 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
@@ -101,6 +101,9 @@ def type_for_activerecord_value(column_type)
           when defined?(ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Uuid) &&
             ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Uuid
             "::String"
+          when defined?(ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Cidr) &&
+            ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Cidr
+            "::IPAddr"
           when defined?(ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Hstore) &&
             ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Hstore
             "T::Hash[::String, ::String]"

diff --git a/sorbet/rbi/shims/activerecord_postgresql.rbi b/sorbet/rbi/shims/activerecord_postgresql.rbi
@@ -4,3 +4,4 @@
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Uuid; end
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Array; end
 module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Hstore; end
+module ActiveRecord::ConnectionAdapters::PostgreSQL::OID::Cidr; end