Improve search method

lib/docker_registry2.rb

@@ -11,8 +11,8 @@ def self.connect(uri = 'https://registry.hub.docker.com', opts = {})
  @reg = DockerRegistry2::Registry.new(uri, opts)
  end
 
- def self.search(query = '')
- @reg.search(query)
+ def self.search(query = '', records = 100)
+ @reg.search(query, records)
  end
 
  def self.tags(repository)

lib/registry/registry.rb

@@ -56,13 +56,17 @@ def paginate_doget(url)
  end
  end
 
- def search(query = '')
+ def search(query = '', record_count = 100)
  all_repos = []
- paginate_doget('/v2/_catalog') do |response|
- repos = JSON.parse(response)['repositories']
- repos.select! { |repo| repo.match?(/#{query}/) } unless query.empty?
- all_repos += repos
+ regex = Regexp.new(Regexp.escape(query), Regexp::IGNORECASE) unless query.empty?
+
+ paginate_doget("/v2/_catalog?n=#{record_count}") do |response|
+ repos = JSON.parse(response).fetch('repositories', [])
+ repos.each do |repo|
+ all_repos << repo if query.empty? || regex.match?(repo)
+ end
  end
+
  all_repos
  end
 
@@ -246,7 +250,7 @@ def parse_link_header(header)
  links = {}
 
  # Parse each part into a named link
- parts.each do |part, _index|
+ parts.each_key do |part|
  section = part.split(';')
  url = section[0][/<(.*)>/, 1]
  name = section[1][/rel="?([^"]*)"?/, 1].to_sym