Contents

module ActiveRecordNestedScope
  class Node
    attr_accessor :klass, :name, :parent

    def initialize(klass, name, parent = nil)
      @klass = klass
      @name = name
      @parent = parent
    end

    def has_options?
      options = @klass.nested_scope_options
      options && options[@name]
    end

    def options(key)
      options = @klass.nested_scope_options.to_h
      options.dig(@name, key)
    end

    def reflection
      @klass.reflect_on_association(options(:through))
    end

    def leaf?
      options(:through).blank?
    end

    def has_many?
      reflection.class.name.in?(['ActiveRecord::Reflection::HasManyReflection', 'ActiveRecord::Reflection::HasOneReflection'])
    end

    def belongs_to?
      reflection.class.name == 'ActiveRecord::Reflection::BelongsToReflection' && !reflection.polymorphic?
    end

    def polymorphic_belongs_to?
      reflection.class.name == 'ActiveRecord::Reflection::BelongsToReflection' && reflection.polymorphic?
    end

    def children
      @children ||= search_children.select(&:valid?)
    end

    def search_children
      if leaf?
        []
      elsif polymorphic_belongs_to?
        polymorphic_klasses.map { |klass| Node.new(klass, @name, self) }
      else
        [Node.new(reflection.klass, @name, self)]
      end
    end

    def valid?
      return false unless has_options?

      if options(:through) && !reflection
        STDERR.puts "can't find reflection for #{options(:through)} in #{klass}"
        return false
      end

      return true
    end

    def has_scope?
      @parent && @parent.reflection.scope.present?
    end

    def scope
      @klass.all.instance_eval(&@parent.reflection.scope) if has_scope?
    end

    private

    def polymorphic_klasses
      types = @klass.unscoped.group(reflection.foreign_type).pluck(reflection.foreign_type).compact
      types.map { |type| type.safe_constantize }.compact
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
activerecord_nested_scope-1.1.1	lib/activerecord_nested_scope/node.rb
activerecord_nested_scope-1.1.0	lib/activerecord_nested_scope/node.rb