search.rb 4.4 KB

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