-
Notifications
You must be signed in to change notification settings - Fork 25
/
autocomplete.rb
88 lines (75 loc) · 2.17 KB
/
autocomplete.rb
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
module Queries
module Note
class Autocomplete < Queries::Query
# @param [Hash] args
def initialize(string, project_id: nil)
super
end
# @return [Scope]
def base_query
::Note.select('notes.*')
end
# @return [ActiveRecord::Relation]
def autocomplete_exact_note
a = table[:text].eq(query_string)
base_query.where(a.to_sql).limit(20)
end
# @return [ActiveRecord::Relation]
def autocomplete_wildcard_end
a = table[:text].matches(query_string + '%')
base_query.where(a.to_sql).limit(8)
end
# @return [ActiveRecord::Relation]
def autocomplete_wildcard_wrapped
a = table[:text].matches('%' + query_string + '%')
base_query.where(a.to_sql).limit(5)
end
# @return [Arel::Nodes::Matches]
def autocomplete_ordered_wildcard_pieces
a = table[:text].matches(wildcard_pieces)
base_query.where(a.to_sql).limit(10)
end
def autocomplete_wildcard_anywhere
b = fragments
return nil if b.empty?
a = table[:text].matches_all(b)
base_query.where(a.to_sql).limit(20)
end
def autocomplete_any_wildcard_anywhere
b = fragments
return nil if b.empty?
a = table[:text].matches_any(b)
base_query.where(a.to_sql).limit(20)
end
# @return [Array]
def autocomplete
queries = [
autocomplete_exact_note,
autocomplete_wildcard_end,
autocomplete_wildcard_wrapped,
autocomplete_ordered_wildcard_pieces,
autocomplete_wildcard_anywhere,
autocomplete_any_wildcard_anywhere
]
queries.compact!
updated_queries = []
queries.each_with_index do |q ,i|
a = q.where(with_project_id.to_sql) if project_id
a ||= q
updated_queries[i] = a
end
result = []
updated_queries.each do |q|
result += q.to_a
result.uniq!
break if result.count > 40
end
result[0..40]
end
# @return [Arel::Table]
def table
::Note.arel_table
end
end
end
end