Plugins » Docker

class ImageSearchController < ::ApplicationController

  before_filter :find_resource

  # this is incredibly odd. for some reason, rails sees the

  # requests ImageSearchControllerTest makes as requests from another host

  # thus, violating CSRF.

  protect_from_forgery :only => :nothing if Rails.env.test?

  def auto_complete_repository_name

    catch_network_errors do

      text = if use_hub?

               hub_image_exists?(params[:search])

             else

               registry_image_exists?(params[:search])

             end

      render :text => text.to_s

    end

  end

  def auto_complete_image_tag

    catch_network_errors do

      # This is the format jQuery UI autocomplete expects

      tags = if use_hub?

               hub_auto_complete_image_tags(params[:search])

             else

               registry_auto_complete_image_tags(params[:search])

             end

      respond_to do |format|

        format.js do

          tags.map! { |tag| { :label => CGI.escapeHTML(tag), :value => CGI.escapeHTML(tag) } }

          render :json => tags

        end

      end

    end

  end

  def search_repository

    catch_network_errors do

      repositories = if use_hub?

                       hub_search_image(params[:search])

                     else

                       registry_search_image(params[:search])

                     end

      respond_to do |format|

        format.js do

          render :partial => 'repository_search_results',

                 :locals  => { :repositories => repositories,

                               :use_hub => use_hub? }

        end

      end

    end

  end

  def catch_network_errors

    yield

  rescue Docker::Error::NotFoundError => e

    # not an error

    logger.debug "image not found: #{e.backtrace}"

    render :js, :nothing => true

  rescue Docker::Error::DockerError, Excon::Errors::Error, SystemCallError => e

    render :js => _("An error occured during repository search: '%s'") % e.message,

           :status => 500

  end

  def use_hub?

    @registry.nil?

  end

  def hub_image_exists?(terms)

    @compute_resource.exist?(terms)

  end

  def hub_auto_complete_image_tags(terms)

    @compute_resource.tags(terms)

  end

  def hub_search_image(terms)

    @compute_resource.search(terms).map do |item|

      # el7 returns -> "name" => "docker.io: docker.io/centos",

      # while f20 returns -> "name" => "centos"

      # we need repo name to be => "docker.io/centos" for el7 and "centos" for fedora

      # to ensure proper search with respect to the tags, image creation etc.

      new_item = item.clone

      new_item["name"] = item["name"].split.last

      new_item

    end

  end

  def registry_image_exists?(term)

    result = ::Service::RegistryApi.new(:url => @registry.url,

                                        :user => @registry.username,

                                        :password => @registry.password).search(term)

    registry_name = if term.split('/').size > 1

                      term

                    else

                      "library/#{term}"

                    end

    result['results'].any? { |r| r['name'] == registry_name }

  end

  def registry_auto_complete_image_tags(terms)

    ::Service::RegistryApi.new(:url => @registry.url,

                               :user => @registry.username,

                               :password => @registry.password).list_repository_tags(terms).keys

  end

  def registry_search_image(terms)

    r = ::Service::RegistryApi.new(:url => @registry.url,

                                   :user => @registry.username,

                                   :password => @registry.password).search(terms)

    r['results']

  end

  def action_permission

    case params[:action]

    when 'auto_complete_repository_name', 'auto_complete_image_tag', 'search_repository'

      :search_repository

    else

      super

    end

  end

  def find_resource

    if params[:registry_id].present?

      @registry = DockerRegistry.authorized(:view_registries).find(params[:registry_id])

    else

      @compute_resource = ComputeResource.authorized(:view_compute_resources).find(params[:id])

    end

  rescue ActiveRecord::RecordNotFound

    not_found

  end

end