search.rb 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174
  1. # Copyright (C) 2012-2016 Zammad Foundation, http://zammad-foundation.org/
  2. class User
  3. module Search
  4. extend ActiveSupport::Concern
  5. # methods defined here are going to extend the class, not the instance of it
  6. class_methods do
  7. =begin
  8. search user preferences
  9. result = User.search_preferences(user_model)
  10. returns if user has permissions to search
  11. result = {
  12. prio: 1000,
  13. direct_search_index: true
  14. }
  15. returns if user has no permissions to search
  16. result = false
  17. =end
  18. def search_preferences(current_user)
  19. return false if !current_user.permissions?('ticket.agent') && !current_user.permissions?('admin.user')
  20. {
  21. prio: 2000,
  22. direct_search_index: true,
  23. }
  24. end
  25. =begin
  26. search user
  27. result = User.search(
  28. query: 'some search term',
  29. limit: 15,
  30. offset: 100,
  31. current_user: user_model,
  32. )
  33. or with certain role_ids | permissions
  34. result = User.search(
  35. query: 'some search term',
  36. limit: 15,
  37. offset: 100,
  38. current_user: user_model,
  39. role_ids: [1,2,3],
  40. group_ids: [1,2,3],
  41. permissions: ['ticket.agent']
  42. # sort single column
  43. sort_by: 'created_at',
  44. order_by: 'asc',
  45. # sort multiple columns
  46. sort_by: [ 'created_at', 'updated_at' ],
  47. order_by: [ 'asc', 'desc' ],
  48. )
  49. returns
  50. result = [user_model1, user_model2, ...]
  51. =end
  52. def search(params)
  53. # get params
  54. query = params[:query]
  55. limit = params[:limit] || 10
  56. offset = params[:offset] || 0
  57. current_user = params[:current_user]
  58. sql_helper = ::SqlHelper.new(object: self)
  59. # check sort - positions related to order by
  60. sort_by = sql_helper.get_sort_by(params, %w[active updated_at])
  61. # check order - positions related to sort by
  62. order_by = sql_helper.get_order_by(params, %w[desc desc])
  63. # enable search only for agents and admins
  64. return [] if !search_preferences(current_user)
  65. # lookup for roles of permission
  66. if params[:permissions].present?
  67. params[:role_ids] ||= []
  68. role_ids = Role.with_permissions(params[:permissions]).pluck(:id)
  69. params[:role_ids].concat(role_ids)
  70. end
  71. # try search index backend
  72. if SearchIndexBackend.enabled?
  73. query_extension = {}
  74. if params[:role_ids].present?
  75. query_extension['bool'] ||= {}
  76. query_extension['bool']['must'] ||= []
  77. if !params[:role_ids].is_a?(Array)
  78. params[:role_ids] = [params[:role_ids]]
  79. end
  80. access_condition = {
  81. 'query_string' => { 'default_field' => 'role_ids', 'query' => "\"#{params[:role_ids].join('" OR "')}\"" }
  82. }
  83. query_extension['bool']['must'].push access_condition
  84. end
  85. if params[:group_ids].present?
  86. query_extension['bool'] ||= {}
  87. query_extension['bool']['must'] ||= []
  88. user_ids = []
  89. params[:group_ids].each do |group_id, access|
  90. user_ids |= User.group_access(group_id.to_i, access).pluck(:id)
  91. end
  92. return [] if user_ids.blank?
  93. query_extension['bool']['must'].push({ 'terms' => { '_id' => user_ids } })
  94. end
  95. items = SearchIndexBackend.search(query, 'User', limit: limit,
  96. query_extension: query_extension,
  97. from: offset,
  98. sort_by: sort_by,
  99. order_by: order_by)
  100. users = []
  101. items.each do |item|
  102. user = User.lookup(id: item[:id])
  103. next if !user
  104. users.push user
  105. end
  106. return users
  107. end
  108. order_sql = sql_helper.get_order(sort_by, order_by, 'users.updated_at DESC')
  109. # fallback do sql query
  110. # - stip out * we already search for *query* -
  111. query.delete! '*'
  112. statement = User
  113. if params[:role_ids]
  114. statement = statement.joins(:roles).where('roles.id' => params[:role_ids])
  115. end
  116. if params[:group_ids]
  117. user_ids = []
  118. params[:group_ids].each do |group_id, access|
  119. user_ids |= User.group_access(group_id.to_i, access).pluck(:id)
  120. end
  121. statement = if user_ids.present?
  122. statement.where(id: user_ids)
  123. else
  124. statement.none
  125. end
  126. end
  127. statement.where(
  128. '(users.firstname LIKE ? OR users.lastname LIKE ? OR users.email LIKE ? OR users.login LIKE ?) AND users.id != 1', "%#{query}%", "%#{query}%", "%#{query}%", "%#{query}%"
  129. )
  130. .order(Arel.sql(order_sql))
  131. .offset(offset)
  132. .limit(limit)
  133. end
  134. end
  135. end
  136. end