Plugins » Docker

class ImageSearchController < ::ApplicationController

  before_filter :find_resource

  def auto_complete_repository_name

    catch_network_errors do

      render :text => (use_hub? ? hub_image_exists?(params[:search]) :

          registry_image_exists?(params[:search])).to_s

    end

  end

  def auto_complete_image_tag

    catch_network_errors do

      # This is the format jQuery UI autocomplete expects

      tags = use_hub? ? hub_auto_complete_image_tags(params[:search]) :

          registry_auto_complete_image_tags(params[:search])

      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 = use_hub? ? hub_search_image(params[:search]) :

                                registry_search_image(params[:search])

      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)

  end

  def registry_image_exists?(term)

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

                                        :user => @registry.username,

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

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

        'library/' + term

    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