summaryrefslogtreecommitdiff
path: root/lib/gitlab/wiki_file_finder.rb
blob: f97278f05cd17f1f4bd5cfb77d3a2db45de03c8b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
module Gitlab
  class WikiFileFinder < FileFinder
    attr_reader :repository

    def initialize(project, ref)
      @project = project
      @ref = ref
      @repository = project.wiki.repository
    end

    private

    def search_filenames(query, except)
      safe_query = Regexp.escape(query.tr(' ', '-'))
      safe_query = Regexp.new(safe_query, Regexp::IGNORECASE)
      filenames = repository.ls_files(ref)

      filenames.delete_if { |filename| except.include?(filename) } unless except.empty?

      filenames.grep(safe_query).first(BATCH_SIZE)
    end
  end
end