Sha256: 00b9d031950b558a6bb6fc359687b934e6175790c928f19c88f05f2cdaf64f60

Contents?: true

Size: 1.63 KB

Versions: 3

Compression:

Stored size: 1.63 KB

Contents

module Specinfra
  module Backend
    class Docker < Exec
      def initialize
        begin
          require 'docker' unless defined?(::Docker)
        rescue LoadError
          fail "Docker client library is not available. Try installing `docker-api' gem."
        end

        @images = []
        ::Docker.url = Specinfra.configuration.docker_url
      end

      def run_command(cmd, opts={})
        cmd = build_command(cmd)
        cmd = add_pre_command(cmd)
        docker_run!(cmd)
      end

      def build_command(cmd)
        cmd
      end

      def add_pre_command(cmd)
        cmd
      end

      def copy_file(from, to)
        @images << current_image.insert_local('localPath' => from, 'outputPath' => to)
      end

      private

      def base_image
        @base_image ||= ::Docker::Image.get(Specinfra.configuration.docker_image)
      end

      def current_image
        @images.last || base_image
      end

      def docker_run!(cmd, opts={})
        opts = {
          'Image' => current_image.id,
          'Cmd' => %W{/bin/sh -c #{cmd}},
        }.merge(opts)

        if path = Specinfra::configuration::path
          (opts['Env'] ||= {})['PATH'] = path
        end

        container = ::Docker::Container.create(opts)
        begin
          container.start
          begin
            stdout, stderr = container.attach
            result = container.wait
            return CommandResult.new :stdout => stdout.join, :stderr => stderr.join,
              :exit_status => result['StatusCode']
          rescue
            container.kill
          end
        ensure
          container.delete
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version Path
specinfra-2.0.0.beta31 lib/specinfra/backend/docker.rb
specinfra-2.0.0.beta30 lib/specinfra/backend/docker.rb
specinfra-2.0.0.beta29 lib/specinfra/backend/docker.rb