acts_as_searchable.rb
1.95 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
module ActsAsSearchable
  module ClassMethods
    def acts_as_searchable(options = {})
      if Noosfero::MultiTenancy.on? and ActiveRecord::Base.postgresql?
        options[:additional_fields] ||= {}
        options[:additional_fields] = Hash[*options[:additional_fields].collect{ |v| [v, {}] }.flatten] if options[:additional_fields].is_a?(Array)
        options[:additional_fields].merge!(:schema_name => { :index => :untokenized })
      end
      acts_as_ferret({ :remote => true }.merge(options))
      extend FindByContents
      send :include, InstanceMethods
    end
    module InstanceMethods
      def schema_name
        ActiveRecord::Base.connection.schema_search_path
      end
    end
    module FindByContents
      def schema_name
        ActiveRecord::Base.connection.schema_search_path
      end
      def find_by_contents(query, ferret_options = {}, db_options = {})
        pg_options = {}
        if ferret_options[:page]
          pg_options[:page] = ferret_options.delete(:page)
        end
        if ferret_options[:per_page]
          pg_options[:per_page] = ferret_options.delete(:per_page)
        end
        ferret_options[:limit] = :all
        ferret_query = (Noosfero::MultiTenancy.on? and ActiveRecord::Base.postgresql?) ? "+schema_name:\"#{schema_name}\" AND #{query}" : query
        # FIXME this is a HORRIBLE HACK
        ids = find_ids_with_ferret(ferret_query, ferret_options)[1][0..8000].map{|r|r[:id].to_i}
        if ids.empty?
          ids << -1
        end
        if db_options[:conditions]
          db_options[:conditions] = sanitize_sql_for_conditions(db_options[:conditions]) + " and #{table_name}.id in (#{ids.join(', ')})"
        else
          db_options[:conditions] = "#{table_name}.id in (#{ids.join(', ')})"
        end
        pg_options[:page] ||= 1
        result = find(:all, db_options)
        result.paginate(pg_options)
      end
    end
  end
end
ActiveRecord::Base.send(:extend, ActsAsSearchable::ClassMethods)