33 lines
864 B
Ruby
33 lines
864 B
Ruby
|
# This class finds files in a repository by name and content
|
||
|
|
# the result is joined and sorted by file name
|
||
|
|
module Gitlab
|
||
|
|
class FileFinder
|
||
|
|
BATCH_SIZE = 100
|
||
|
|
|
||
|
|
attr_reader :project, :ref
|
||
|
|
|
||
|
|
def initialize(project, ref)
|
||
|
|
@project = project
|
||
|
|
@ref = ref
|
||
|
|
end
|
||
|
|
|
||
|
|
def find(query)
|
||
|
|
blobs = project.repository.search_files_by_content(query, ref).first(BATCH_SIZE)
|
||
|
|
found_file_names = Set.new
|
||
|
|
|
||
|
|
results = blobs.map do |blob|
|
||
|
|
blob = Gitlab::ProjectSearchResults.parse_search_result(blob)
|
||
|
|
found_file_names << blob.filename
|
||
|
|
|
||
|
|
[blob.filename, blob]
|
||
|
|
end
|
||
|
|
|
||
|
|
project.repository.search_files_by_name(query, ref).first(BATCH_SIZE).each do |filename|
|
||
|
|
results << [filename, OpenStruct.new(ref: ref)] unless found_file_names.include?(filename)
|
||
|
|
end
|
||
|
|
|
||
|
|
results.sort_by(&:first)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|