94 lines
2.2 KiB
Ruby
94 lines
2.2 KiB
Ruby
class Search
|
|
include ActiveModel::Validations
|
|
include ActiveModel::Conversion
|
|
include ActiveModel::AttributeMethods
|
|
extend ActiveModel::Naming
|
|
|
|
attr_accessor :q, :what, :order
|
|
attr_accessor :results, :page, :total_results, :per_page
|
|
|
|
validates_length_of :q, :minimum => 2
|
|
|
|
def initialize
|
|
@q = ""
|
|
@what = "all"
|
|
@order = "relevance"
|
|
|
|
@page = 1
|
|
@per_page = 20
|
|
|
|
@results = []
|
|
@total_results = 0
|
|
end
|
|
|
|
def persisted?
|
|
false
|
|
end
|
|
|
|
def to_url_params
|
|
[ :q, :what, :order ].map{|p| "#{p}=#{CGI.escape(self.send(p))}"
|
|
}.join("&")
|
|
end
|
|
|
|
def search_for_user!(user)
|
|
opts = {
|
|
:match_mode => :extended,
|
|
:rank_mode => :bm25,
|
|
:page => @page,
|
|
:per_page => @per_page,
|
|
}
|
|
|
|
if order == "newest"
|
|
opts[:order] = :created_at
|
|
opts[:sort_mode] = :desc
|
|
elsif order == "points"
|
|
opts[:order] = :score
|
|
opts[:sort_mode] = :desc
|
|
end
|
|
|
|
opts[:classes] = []
|
|
if what == "all"
|
|
opts[:classes] = [ Story, Comment ]
|
|
elsif what == "comments"
|
|
opts[:classes] = [ Comment ]
|
|
elsif what == "stories"
|
|
opts[:classes] = [ Story ]
|
|
end
|
|
|
|
opts[:include] = [ :story, :user ]
|
|
|
|
# sphinx seems to interpret slashes as a regex(?) so escape them since
|
|
# nobody is probably using them, but don't just use Riddle.escape because
|
|
# it removes boolean suport
|
|
query = self.q.gsub(/\//, "\\/")
|
|
|
|
# go go gadget search
|
|
@results = ThinkingSphinx.search query, opts
|
|
@total_results = @results.total_entries
|
|
|
|
# bind votes for both types
|
|
|
|
if opts[:classes].include?(Comment) && user
|
|
votes = Vote.comment_votes_by_user_for_comment_ids_hash(user.id,
|
|
@results.select{|r| r.class == Comment }.map{|c| c.id })
|
|
|
|
@results.each do |r|
|
|
if r.class == Comment && votes[r.id]
|
|
r.current_vote = votes[r.id]
|
|
end
|
|
end
|
|
end
|
|
|
|
if opts[:classes].include?(Story) && user
|
|
votes = Vote.story_votes_by_user_for_story_ids_hash(user.id,
|
|
@results.select{|r| r.class == Story }.map{|s| s.id })
|
|
|
|
@results.each do |r|
|
|
if r.class == Story && votes[r.id]
|
|
r.vote = votes[r.id][:vote]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|