ManageIQ/manageiq

class YamlPermittedClasses
  DEFAULT_PERMITTED_CLASSES = [
    ActiveSupport::Duration,
    ActiveSupport::HashWithIndifferentAccess,
    ActiveSupport::TimeWithZone,
    ActiveSupport::TimeZone,
    Date,
    DateTime,
    Object,
    Range,
    Regexp,
    Symbol,
    Time
  ].freeze
  def self.app_yaml_permitted_classes
    @app_yaml_permitted_classes ||= DEFAULT_PERMITTED_CLASSES + [MiqExpression]
  end

  def self.app_yaml_permitted_classes=(classes)
    @app_yaml_permitted_classes = Array(classes)
  end

  def self.default_permitted_classes
    @default_permitted_classes ||= DEFAULT_PERMITTED_CLASSES
  end

  def self.permitted_classes
    initialized? ? app_yaml_permitted_classes : default_permitted_classes
  end

  def self.initialize_app_yaml_permitted_classes
    @initialize_app_yaml_permitted_classes ||= begin
      ActiveRecord.yaml_column_permitted_classes = YamlPermittedClasses.app_yaml_permitted_classes
      true
    end
  end

  def self.initialized?
    !!@initialize_app_yaml_permitted_classes
  end
end