Support Psych 4.

Kaligo · Oct 20, 2023 · 3b9798d · 3b9798d
1 parent c115ed3
commit 3b9798d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lib/rails_admin.rb b/lib/rails_admin.rb
@@ -45,7 +45,14 @@ def self.yaml_load(yaml)
   rescue LoadError
     if YAML.respond_to?(:safe_load)
       def self.yaml_load(yaml)
-        YAML.safe_load(yaml)
+        # https://github.com/rails/rails/blob/v6.1.7.6/activerecord/lib/active_record/coders/yaml_column.rb#L50-L56
+        if ActiveRecord::Base.use_yaml_unsafe_load
+          YAML.unsafe_load(yaml)
+        elsif YAML.method(:safe_load).parameters.include?([:key, :permitted_classes])
+          YAML.safe_load(yaml, permitted_classes: ActiveRecord::Base.yaml_column_permitted_classes, aliases: true)
+        else
+          YAML.safe_load(yaml, ActiveRecord::Base.yaml_column_permitted_classes, [], true)
+        end
       end
     else
       raise LoadError.new "Safe-loading of YAML is not available. Please install 'safe_yaml' or install Psych 2.0+"