module PolyBelongsTo::SortedReflectionDecorator

Public Class Methods

included(base) click to toggle source
# File lib/poly_belongs_to/sorted_reflection_decorator.rb, line 3
def self.included(base)
  base.module_exec do
    original_method = instance_method(:reflect_on_all_associations)
    define_method(:reflect_on_all_associations) do |*args, &block|
      original_method.bind(self).call(*args, &block).sort_by {|a| a.polymorphic? ? 0 : 1 }
    end
  end
end