Class: YiffSpace::Search::QueryDSL

Inherits:
Object
  • Object
show all
Defined in:
lib/yiffspace/search/query_dsl.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(relation) ⇒ QueryDSL

[param, field, type/proc, Proc(relation)]



9
10
11
12
13
14
15
 
# File 'lib/yiffspace/search/query_dsl.rb', line 9

def initialize(relation)
  @relation = relation
  @fields = []
  @associations = []
  @no_id = false
  @no_dates = false
end

Instance Attribute Details

#associations ⇒ Object

Returns the value of attribute associations.



6
7
8
 
# File 'lib/yiffspace/search/query_dsl.rb', line 6

def associations
  @associations
end

#fields ⇒ Object

Returns the value of attribute fields.



6
7
8
 
# File 'lib/yiffspace/search/query_dsl.rb', line 6

def fields
  @fields
end

#no_dates ⇒ Object

Returns the value of attribute no_dates.



6
7
8
 
# File 'lib/yiffspace/search/query_dsl.rb', line 6

def no_dates
  @no_dates
end

#no_id ⇒ Object

Returns the value of attribute no_id.



6
7
8
 
# File 'lib/yiffspace/search/query_dsl.rb', line 6

def no_id
  @no_id
end

#relation ⇒ Object

Returns the value of attribute relation.



6
7
8
 
# File 'lib/yiffspace/search/query_dsl.rb', line 6

def relation
  @relation
end

Instance Method Details

#association(*args, **kwargs) ⇒ Object

[attribute, class, dsl, join]



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
 
# File 'lib/yiffspace/search/query_dsl.rb', line 59

def association(*args, **kwargs)
  if args.any?
    attribute = args.first
    as = args.second || attribute
    ref = relation.reflect_on_association(attribute)
    raise("Association #{attribute} does not exist on #{relation.name}") if ref.nil?

    klass = ref.klass
    dsl = klass.query_dsl.build
    @associations << [as, klass, dsl, attribute]
    user_class = YiffSpace.config.user_class
    user(as, attribute) if klass == user_class && @fields.none? { |f| f.second == association_column(attribute) }
  elsif kwargs.any?
    attribute = kwargs.delete(:as)
    to_array = lambda { |h|
      k, v = h.first
      [k, v.is_a?(Hash) ? to_array.call(v) : v].flatten
    }
    through = to_array.call(kwargs)
    attribute ||= through.last
    klass = relation
    through.each { |k| klass = klass.reflect_on_association(k).klass }
    dsl = klass.query_dsl.build
    join = through.reverse.reduce { |acc, key| { key => acc } }
    @associations << [attribute, klass, dsl, join]
  else
    raise(ArgumentError, "Missing association")
  end
  self
end

#build ⇒ Object 



102
103
104
105
106
107
108
109
110
 
# File 'lib/yiffspace/search/query_dsl.rb', line 102

def build
  @fields << %i[id id integer] if @fields.none? { |field| field.first == :id }
  @fields << %i[created_at created_at datetime] if @fields.none? { |field| field.first == :created_at } && relation.attribute_names.include?("created_at")
  @fields << %i[updated_at updated_at datetime] if @fields.none? { |field| field.first == :updated_at } && relation.attribute_names.include?("updated_at")
  {
    fields:       fields,
    associations: associations,
  }
end

#custom(param, proc, not: false, multi: false, &block) ⇒ Object 



17
18
19
20
 
# File 'lib/yiffspace/search/query_dsl.rb', line 17

def custom(param, proc, not: false, multi: false, &block)
  @fields << [param, nil, proc, block, { not: binding.local_variable_get(:not), multi: multi }]
  self
end

#field(param, db_field = param, type = nil, not: false, multi: false, wildcard: false, like: false, ilike: false, normalize: nil, &block) ⇒ Object 



22
23
24
25
26
 
# File 'lib/yiffspace/search/query_dsl.rb', line 22

def field(param, db_field = param, type = nil, not: false, multi: false, wildcard: false, like: false, ilike: false, normalize: nil, &block)
  type ||= QueryHelper.get_column(db_field, relation.table_name).sql_type_metadata.type
  @fields << [param, db_field, type, block, { not: binding.local_variable_get(:not), multi: multi, wildcard: wildcard, like: like, ilike: ilike, normalize: normalize }]
  self
end

#no_dates! ⇒ Object 



96
97
98
99
100
 
# File 'lib/yiffspace/search/query_dsl.rb', line 96

def no_dates!
  @no_dates = true
  @fields.reject! { |field| %i[created_at updated_at].include?(field.first) }
  self
end

#no_id! ⇒ Object 



90
91
92
93
94
 
# File 'lib/yiffspace/search/query_dsl.rb', line 90

def no_id!
  @no_id = true
  @fields.reject! { |field| field.first == :id }
  self
end

#present(param, db_field = param) ⇒ Object 



54
55
56
 
# File 'lib/yiffspace/search/query_dsl.rb', line 54

def present(param, db_field = param, &)
  field(param, db_field, :present, &)
end

#user(param, association = nil, not: false) ⇒ Object 



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
 
# File 'lib/yiffspace/search/query_dsl.rb', line 28

def user(param, association = nil, not: false, &)
  if param.is_a?(Array)
    raise(ArgumentError, "You must specify an association when passing an array of parameters") if association.nil?
  else
    association ||= param
    param = [:"#{param}_id", :"#{param}_name"]
  end

  case association
  when Symbol, String
    association = association.to_sym
    associations = relation.reflect_on_all_associations(:belongs_to).map(&:name)
    if associations.include?(association)
      column = association_column(association)
    else
      column = association
    end
  when Array
    column = association.first
    association = association.second
  end
  field(param.first, column, not: binding.local_variable_get(:not), &) if param.first
  custom(param.second, ->(q, v) { q.user_name_matches(association, v) }, not: binding.local_variable_get(:not), &) if param.second
  self
end