Noosfero/noosfero

class SolrPlugin < Noosfero::Plugin; end;

require_dependency "solr_plugin/search_helper"

class SolrPlugin < Noosfero::Plugin
  include SolrPlugin::SearchHelper

  def self.plugin_name
    "Solr"
  end

  def self.plugin_description
    _("Uses Solr as search engine.")
  end

  def stylesheet?
    true
  end

  def search_facets?
    true
  end

  def solr_search?(empty_query, klass)
    (not empty_query) || (klass == Product)
  end

  def find_by_contents(asset, scope, query, paginate_options = {}, options = {})
    klass = asset_class(asset)
    category = options[:category]
    empty_query = empty_query? query, category

    unless solr_search? empty_query, klass
      return options[:filter] ? { results: scope.send(options[:filter]).paginate(paginate_options) } : nil
    end

    solr_options = solr_options(class_asset(klass), category)
    solr_options[:filter_queries] ||= []
    solr_options[:filter_queries] += scopes_to_solr_options scope, klass, options
    solr_options.merge! products_options(user) if (asset == :products) && empty_query
    solr_options.merge! options.except(:category, :filter)

    scope.find_by_contents query, paginate_options, solr_options
  end

  protected

    def scopes_to_solr_options(scope, klass = nil, options = {})
      filter_queries = []
      klass ||= scope.base_class
      solr_fields = klass.configuration[:solr_fields].keys
      scopes_applied = scope.scopes_applied.dup rescue [] # rescue association and class direct filtering

      scope.scope_attributes.each do |attr, value|
        raise "Non-indexed attribute '#{attr}' speficied in scope_attributes" unless solr_fields.include? attr.to_sym

        # if the filter is present here, then prefer it
        scopes_applied.reject! { |name| name == attr.to_sym }

        filter_queries << "#{attr}:#{value}"
      end

      scopes_applied.each do |name|
        next if name.to_s == options[:filter].to_s

        has_value = name === Hash
        if has_value
          name, args = name.keys.first, name.values.first
          value = args.first
        end

        related_field = nil
        related_field = name if solr_fields.include? name
        related_field = "solr_plugin_#{name}" if solr_fields.include? :"solr_plugin_#{name}"

        if has_value
          if related_field
            filter_queries << "#{related_field}:#{value}"
          else
            filter_queries << klass.send("solr_filter_#{name}", *args)
          end
        else
          raise "Undeclared solr field for scope #{name}" if related_field.nil?

          filter_queries << "#{related_field}:true"
        end
      end

      filter_queries
    end

    def method_missing(method, *args, &block)
      if self.context.respond_to? method
        self.context.send method, *args, &block
      else
        super method, *args, &block
      end
    end
end