Add enum set value as Symbol too

pocke · Dec 6, 2024 · 1ba6df1 · 1ba6df1
1 parent 240074e
commit 1ba6df1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/rbs_rails/active_record.rb b/lib/rbs_rails/active_record.rb
@@ -563,7 +563,7 @@ def authenticate_#{attribute}: (String) -> (#{klass_name} | false)
         mod_sig = +"module GeneratedAttributeMethods\n"
         mod_sig << klass.columns.map do |col|
           class_name = if enum_definitions.any? { |hash| hash.key?(col.name) || hash.key?(col.name.to_sym) }
-                         '::String'
+                         '(::String | ::Symbol)'
                        else
                          sql_type_to_class(col.type)
                        end
@@ -572,6 +572,7 @@ def authenticate_#{attribute}: (String) -> (#{klass_name} | false)
 
           class_name_opt = optional(class_name)
           column_type = col.null && !col.type.in?([:jsonb, :json]) ? class_name_opt : class_name
+
           sig = <<~EOS
             def #{col.name}: () -> #{column_type}
             def #{col.name}=: (#{column_type}) -> #{column_type}