module ThinkingSphinx
  module ActiveRecord
    module CollectionProxy
      def search(*args)
        options   = args.extract_options!
        options[:with] ||= {}
        options[:with].merge! default_filter

        args << options
        proxy_association.klass.search(*args)
      end

      private

      def attribute_for_foreign_key
        foreign_key = proxy_association.reflection.foreign_key
        stack = [proxy_association.reflection.options[:through]].compact

        proxy_association.klass.define_indexes
        (proxy_association.klass.sphinx_indexes || []).each do |index|
          attribute = index.attributes.detect { |attrib|
            attrib.columns.length == 1 &&
            attrib.columns.first.__name  == foreign_key.to_sym
          }
          return attribute unless attribute.nil?
        end

        raise "Missing Attribute for Foreign Key #{foreign_key}"
      end

      def default_filter
        {attribute_for_foreign_key.unique_name => proxy_association.owner.id}
      end
    end
  end
end

module ThinkingSphinx
  module ActiveRecord
    module CollectionProxy
      def search(*args)
        options   = args.extract_options!
        options[:with] ||= {}
        options[:with].merge! default_filter

        args << options
        proxy_association.klass.search(*args)
      end

      private

      def attribute_for_foreign_key
        if proxy_association.reflection.through_reflection
          foreign_key = proxy_association.reflection.through_reflection.foreign_key
        else
          foreign_key = proxy_association.reflection.foreign_key
        end

        proxy_association.klass.define_indexes
        (proxy_association.klass.sphinx_indexes || []).each do |index|
          attribute = index.attributes.detect { |attrib|
            attrib.columns.length == 1 &&
            attrib.columns.first.__name  == foreign_key.to_sym
          }
          return attribute unless attribute.nil?
        end

        raise "Missing Attribute for Foreign Key #{foreign_key}"
      end

      def default_filter
        {attribute_for_foreign_key.unique_name => proxy_association.owner.id}
      end
    end
  end
end

Handle through associations in model.association.search calls
Handle through associations in model.association.search calls
