solr_plugin.rb
2.68 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
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 solr_search? empty_query, klass
not empty_query or 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 and 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