mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-12-26 17:51:13 +08:00
298 lines
9.7 KiB
Ruby
298 lines
9.7 KiB
Ruby
class SearchController < ApplicationController
|
|
include IconsHelper
|
|
before_action :load_vars, only: :index
|
|
|
|
def index
|
|
redirect_to new_search_path unless @search_query
|
|
|
|
@search_id = params[:search_id] ? params[:search_id] : generate_search_id
|
|
|
|
count_search_results
|
|
|
|
search_projects if @search_category == :projects
|
|
search_experiments if @search_category == :experiments
|
|
search_modules if @search_category == :modules
|
|
search_results if @search_category == :results
|
|
search_tags if @search_category == :tags
|
|
search_reports if @search_category == :reports
|
|
search_protocols if @search_category == :protocols
|
|
search_steps if @search_category == :steps
|
|
search_checklists if @search_category == :checklists
|
|
if @search_category == :repositories && params[:repository]
|
|
search_repository
|
|
end
|
|
search_assets if @search_category == :assets
|
|
search_tables if @search_category == :tables
|
|
search_comments if @search_category == :comments
|
|
|
|
@search_pages = (@search_count.to_f / Constants::SEARCH_LIMIT.to_f).ceil
|
|
@start_page = @search_page - 2
|
|
@start_page = 1 if @start_page < 1
|
|
@end_page = @start_page + 4
|
|
|
|
if @end_page > @search_pages
|
|
@end_page = @search_pages
|
|
@start_page = @end_page - 4
|
|
@start_page = 1 if @start_page < 1
|
|
end
|
|
end
|
|
|
|
def new
|
|
end
|
|
|
|
private
|
|
|
|
def load_vars
|
|
query = (params.fetch(:q) { '' }).strip
|
|
@search_category = params[:category] || ''
|
|
@search_category = @search_category.to_sym
|
|
@search_page = params[:page].to_i || 1
|
|
@search_case = params[:match_case] == 'true'
|
|
@search_whole_word = params[:whole_word] == 'true'
|
|
@search_whole_phrase = params[:whole_phrase] == 'true'
|
|
@display_query = query
|
|
|
|
if @search_whole_phrase || query.count(' ').zero?
|
|
if query.length < Constants::NAME_MIN_LENGTH
|
|
flash[:error] = t('general.query.length_too_short',
|
|
min_length: Constants::NAME_MIN_LENGTH)
|
|
redirect_back(fallback_location: root_path)
|
|
elsif query.length > Constants::TEXT_MAX_LENGTH
|
|
flash[:error] = t('general.query.length_too_long',
|
|
max_length: Constants::TEXT_MAX_LENGTH)
|
|
redirect_back(fallback_location: root_path)
|
|
else
|
|
@search_query = query
|
|
end
|
|
else
|
|
# splits the search query to validate all entries
|
|
splited_query = query.split
|
|
@search_query = ''
|
|
splited_query.each_with_index do |w, i|
|
|
if w.length >= Constants::NAME_MIN_LENGTH &&
|
|
w.length <= Constants::TEXT_MAX_LENGTH
|
|
@search_query += "#{splited_query[i]} "
|
|
end
|
|
end
|
|
if @search_query.empty?
|
|
flash[:error] = t('general.query.wrong_query',
|
|
min_length: Constants::NAME_MIN_LENGTH,
|
|
max_length: Constants::TEXT_MAX_LENGTH)
|
|
redirect_back(fallback_location: root_path)
|
|
else
|
|
@search_query.strip!
|
|
end
|
|
end
|
|
@search_page = 1 if @search_page < 1
|
|
end
|
|
|
|
protected
|
|
|
|
def generate_search_id
|
|
SecureRandom.urlsafe_base64(32)
|
|
end
|
|
|
|
def search_by_name(model)
|
|
model.search(current_user,
|
|
true,
|
|
@search_query,
|
|
@search_page,
|
|
nil,
|
|
match_case: @search_case,
|
|
whole_word: @search_whole_word,
|
|
whole_phrase: @search_whole_phrase)
|
|
end
|
|
|
|
def count_by_name(model)
|
|
model.search(current_user,
|
|
true,
|
|
@search_query,
|
|
Constants::SEARCH_NO_LIMIT,
|
|
nil,
|
|
match_case: @search_case,
|
|
whole_word: @search_whole_word,
|
|
whole_phrase: @search_whole_phrase).size
|
|
end
|
|
|
|
def count_by_repository
|
|
@repository_search_count =
|
|
Rails.cache.fetch("#{@search_id}/repository_search_count",
|
|
expires_in: 5.minutes) do
|
|
search_count = {}
|
|
search_results = Repository.search(current_user,
|
|
@search_query,
|
|
Constants::SEARCH_NO_LIMIT,
|
|
nil,
|
|
match_case: @search_case,
|
|
whole_word: @search_whole_word,
|
|
whole_phrase: @search_whole_phrase)
|
|
|
|
current_user.teams.includes(:repositories).each do |team|
|
|
team_results = {}
|
|
team_results[:team] = team
|
|
team_results[:count] = 0
|
|
team_results[:repositories] = {}
|
|
Repository.accessible_by_teams(team).each do |repository|
|
|
repository_results = {}
|
|
repository_results[:id] = repository.id
|
|
repository_results[:repository] = repository
|
|
repository_results[:count] = 0
|
|
search_results.each do |result|
|
|
if repository.id == result.id
|
|
repository_results[:count] += result.counter
|
|
end
|
|
end
|
|
team_results[:repositories][repository.name] = repository_results
|
|
team_results[:count] += repository_results[:count]
|
|
end
|
|
search_count[team.name] = team_results
|
|
end
|
|
search_count
|
|
end
|
|
|
|
count_total = 0
|
|
@repository_search_count.each_value do |team_results|
|
|
count_total += team_results[:count]
|
|
end
|
|
count_total
|
|
end
|
|
|
|
def current_repository_search_count
|
|
@repository_search_count.each_value do |counter|
|
|
res = counter[:repositories].values.detect do |rep|
|
|
rep[:id] == @repository.id
|
|
end
|
|
return res[:count] if res && res[:count]
|
|
end
|
|
end
|
|
|
|
def count_search_results
|
|
@project_search_count = fetch_cached_count Project
|
|
@experiment_search_count = fetch_cached_count Experiment
|
|
@module_search_count = fetch_cached_count MyModule
|
|
@result_search_count = fetch_cached_count Result
|
|
@tag_search_count = fetch_cached_count Tag
|
|
@report_search_count = fetch_cached_count Report
|
|
@protocol_search_count = fetch_cached_count Protocol
|
|
@step_search_count = fetch_cached_count Step
|
|
@checklist_search_count = fetch_cached_count Checklist
|
|
@repository_search_count_total = count_by_repository
|
|
@asset_search_count = fetch_cached_count Asset
|
|
@table_search_count = fetch_cached_count Table
|
|
@comment_search_count = fetch_cached_count Comment
|
|
|
|
@search_results_count = @project_search_count
|
|
@search_results_count += @experiment_search_count
|
|
@search_results_count += @module_search_count
|
|
@search_results_count += @result_search_count
|
|
@search_results_count += @tag_search_count
|
|
@search_results_count += @report_search_count
|
|
@search_results_count += @protocol_search_count
|
|
@search_results_count += @step_search_count
|
|
@search_results_count += @checklist_search_count
|
|
@search_results_count += @repository_search_count_total
|
|
@search_results_count += @asset_search_count
|
|
@search_results_count += @table_search_count
|
|
@search_results_count += @comment_search_count
|
|
end
|
|
|
|
def fetch_cached_count(type)
|
|
exp = 5.minutes
|
|
Rails.cache.fetch(
|
|
"#{@search_id}/#{type.name.underscore}_search_count", expires_in: exp
|
|
) do
|
|
count_by_name type
|
|
end
|
|
end
|
|
|
|
def search_projects
|
|
@project_results = []
|
|
@project_results = search_by_name(Project) if @project_search_count > 0
|
|
@search_count = @project_search_count
|
|
end
|
|
|
|
def search_experiments
|
|
@experiment_results = []
|
|
if @experiment_search_count > 0
|
|
@experiment_results = search_by_name(Experiment)
|
|
end
|
|
@search_count = @experiment_search_count
|
|
end
|
|
|
|
def search_modules
|
|
@module_results = []
|
|
@module_results = search_by_name(MyModule) if @module_search_count > 0
|
|
@search_count = @module_search_count
|
|
end
|
|
|
|
def search_results
|
|
@result_results = []
|
|
@result_results = search_by_name(Result) if @result_search_count > 0
|
|
@search_count = @result_search_count
|
|
end
|
|
|
|
def search_tags
|
|
@tag_results = []
|
|
@tag_results = search_by_name(Tag) if @tag_search_count > 0
|
|
@search_count = @tag_search_count
|
|
end
|
|
|
|
def search_reports
|
|
@report_results = []
|
|
@report_results = search_by_name(Report) if @report_search_count > 0
|
|
@search_count = @report_search_count
|
|
end
|
|
|
|
def search_protocols
|
|
@protocol_results = []
|
|
@protocol_results = search_by_name(Protocol) if @protocol_search_count > 0
|
|
@search_count = @protocol_search_count
|
|
end
|
|
|
|
def search_steps
|
|
@step_results = []
|
|
@step_results = search_by_name(Step) if @step_search_count > 0
|
|
@search_count = @step_search_count
|
|
end
|
|
|
|
def search_checklists
|
|
@checklist_results = []
|
|
if @checklist_search_count > 0
|
|
@checklist_results = search_by_name(Checklist)
|
|
end
|
|
@search_count = @checklist_search_count
|
|
end
|
|
|
|
def search_repository
|
|
@repository = Repository.find_by_id(params[:repository])
|
|
render_403 unless can_read_repository?(@repository)
|
|
@repository_results = []
|
|
if @repository_search_count_total > 0
|
|
@repository_results =
|
|
Repository.search(current_user, @search_query, @search_page,
|
|
@repository,
|
|
match_case: @search_case,
|
|
whole_word: @search_whole_word,
|
|
whole_phrase: @search_whole_phrase)
|
|
end
|
|
@search_count = current_repository_search_count
|
|
end
|
|
|
|
def search_assets
|
|
@asset_results = []
|
|
@asset_results = search_by_name(Asset) if @asset_search_count > 0
|
|
@search_count = @asset_search_count
|
|
end
|
|
|
|
def search_tables
|
|
@table_results = []
|
|
@table_results = search_by_name(Table) if @table_search_count > 0
|
|
@search_count = @table_search_count
|
|
end
|
|
|
|
def search_comments
|
|
@comment_results = []
|
|
@comment_results = search_by_name(Comment) if @comment_search_count > 0
|
|
@search_count = @comment_search_count
|
|
end
|
|
end
|