forked from cybrespace/mastodon
		
	* Clean up SQL output in Tag and Account search methods * Add basic coverage for Tag.search_for * Add coverage for Account.search_for * Add coverage for Account.advanced_search_for
		
			
				
	
	
		
			33 lines
		
	
	
	
		
			814 B
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			33 lines
		
	
	
	
		
			814 B
		
	
	
	
		
			Ruby
		
	
	
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
class Tag < ApplicationRecord
 | 
						|
  has_and_belongs_to_many :statuses
 | 
						|
 | 
						|
  HASHTAG_RE = /(?:^|[^\/\)\w])#([[:word:]_]*[[:alpha:]_][[:word:]_]*)/i
 | 
						|
 | 
						|
  validates :name, presence: true, uniqueness: true
 | 
						|
 | 
						|
  def to_param
 | 
						|
    name
 | 
						|
  end
 | 
						|
 | 
						|
  class << self
 | 
						|
    def search_for(terms, limit = 5)
 | 
						|
      terms      = Arel.sql(connection.quote(terms.gsub(/['?\\:]/, ' ')))
 | 
						|
      textsearch = 'to_tsvector(\'simple\', tags.name)'
 | 
						|
      query      = 'to_tsquery(\'simple\', \'\'\' \' || ' + terms + ' || \' \'\'\' || \':*\')'
 | 
						|
 | 
						|
      sql = <<-SQL.squish
 | 
						|
        SELECT
 | 
						|
          tags.*,
 | 
						|
          ts_rank_cd(#{textsearch}, #{query}) AS rank
 | 
						|
        FROM tags
 | 
						|
        WHERE #{query} @@ #{textsearch}
 | 
						|
        ORDER BY rank DESC
 | 
						|
        LIMIT ?
 | 
						|
      SQL
 | 
						|
 | 
						|
      Tag.find_by_sql([sql, limit])
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |