ManageIQ/manageiq

View on GitHub
lib/extensions/ar_to_model_hash.rb

Summary

Maintainability
B
5 hrs
Test Coverage
B
87%
module ToModelHash
  extend ActiveSupport::Concern

  module ClassMethods
    def to_model_hash_options
      fname = "#{table_name}.yaml"
      r = MiqReport.find_by(:filename => fname, :template_type => "compare").try(:attributes)
      if r.nil?
        fname = Rails.root.join("product", "compare", fname)
        r = YAML.load_file(fname) if File.exist?(fname)
      end
      r ||= {}
      to_model_hash_options_fixup(r)
    end

    protected

    # Fixes up inconsistencies in the report format, if found
    def to_model_hash_options_fixup(options)
      return nil if options.nil?

      ret = {}

      cols  = (options["cols"] || options["columns"] || [])
      cols += (options["key"] || []).compact
      ret[:columns] = cols.uniq.sort.collect(&:to_sym) unless cols.blank?

      includes = options["include"]
      if includes
        if includes.key?("categories")
          includes.delete("categories")
          includes[:tags] = nil
        end

        ret[:include] = includes.each_with_object({}) do |(k, v), h|
          sub_options = to_model_hash_options_fixup(v)
          h[k.to_sym] = sub_options.blank? ? nil : sub_options
        end
      end

      ret
    end
  end

  def to_model_hash(options = nil)
    options ||= self.class.to_model_hash_options
    MiqPreloader.preload(self, to_model_hash_build_preload(options))
    to_model_hash_recursive(options)
  end

  def to_model_yaml(options = nil)
    to_model_hash(options).to_yaml
  end

  protected

  def to_model_hash_attrs(options)
    columns  = ((options && options[:columns]) || self.class.column_names_symbols.dup)
    columns << :id

    columns.each_with_object({:class => self.class.name}) do |c, h|
      next unless self.respond_to?(c)
      value = send(c)
      h[c.to_sym] = value unless value.nil?
    end
  end

  def to_model_hash_recursive(options, result = nil)
    result ||= to_model_hash_attrs(options)

    spec = (options && options[:include])

    case spec
    when Symbol, String
      if self.respond_to?(spec)
        recs = send(spec)
        if recs.kind_of?(ActiveRecord::Base) || (recs.kind_of?(Array) && recs.first.kind_of?(ActiveRecord::Base))
          single_rec = !recs.kind_of?(Array)
          recs = Array.wrap(recs).collect { |v| v.to_model_hash_attrs(spec) }
          recs = recs.first if single_rec
        end
        result[spec] = recs unless recs.nil?
      end
    when Array
      spec.each { |s| to_model_hash_recursive(s, result) }
    when Hash
      spec.each do |k, v|
        next unless self.respond_to?(k)
        if k == :tags
          recs = tags.collect { |t| Classification.tag_to_model_hash(t) }
        else
          recs = send(k)
          single_rec = !iterable?(self.class, k)
          recs = Array.wrap(recs).collect { |c| c.to_model_hash_recursive(v) }
          recs = recs.first if single_rec
        end
        result[k] = recs unless recs.nil?
      end
    end

    result
  end

  def iterable?(klass, association)
    reflection = klass.reflection_with_virtual(association)
    reflection && reflection.collection?
  end

  def to_model_hash_build_preload(options, parent_class = self.class)
    columns  = ((options && options[:columns]) || [])
    includes = ((options && options[:include]) || {})

    result = columns.select { |c| parent_class.virtual_attribute?(c) }

    result + includes.collect do |k, v|
      association_class = parent_class.reflections_with_virtual[k.to_sym].klass
      sub_result        = to_model_hash_build_preload(v, association_class)
      sub_result.blank? ? k : {k => sub_result}
    end
  end
end