vagrant 1.5.0 → 2.4.4

data/plugins/provisioners/container/client.rb

@@ -0,0 +1,206 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require 'digest/sha1'
+
+module VagrantPlugins
+  module ContainerProvisioner
+    class Client
+      def initialize(machine, container_command)
+        @machine = machine
+        @container_command =  container_command
+      end
+
+      # Build an image given a path to a Dockerfile
+      #
+      # @param [Array<Array<String, Hash>>] - Path and options to the Dockerfile to pass to
+      #   container build command
+      def build_images(images)
+        @machine.communicate.tap do |comm|
+          images.each do |path, opts|
+            @machine.ui.info(I18n.t("vagrant.container_building_single", path: path))
+            comm.sudo("#{@container_command} build #{opts[:args]} #{path}") do |type, data|
+              handle_comm(type, data)
+            end
+          end
+        end
+      end
+
+      # Pull image given a list of images
+      #
+      # @param [String] - Image name
+      def pull_images(*images)
+        @machine.communicate.tap do |comm|
+          images.each do |image|
+            @machine.ui.info(I18n.t("vagrant.container_pulling_single", name: image))
+            comm.sudo("#{@container_command} pull #{image}") do |type, data|
+              handle_comm(type, data)
+            end
+          end
+        end
+      end
+
+      def run(containers)
+        containers.each do |name, config|
+          cids_dir = "/var/lib/vagrant/cids"
+          config[:cidfile] ||= "#{cids_dir}/#{Digest::SHA1.hexdigest name}"
+
+          @machine.ui.info(I18n.t("vagrant.container_running", name: name))
+          @machine.communicate.sudo("mkdir -p #{cids_dir}")
+          run_container({
+            name: name,
+            original_name: name,
+          }.merge(config))
+        end
+      end
+
+      # Run a OCI container. If the container does not exist it will be
+      # created. If the image is stale it will be recreated and restarted
+      def run_container(config)
+        raise "Container's cidfile was not provided!" if !config[:cidfile]
+
+        id = "$(cat #{config[:cidfile]})"
+
+        if container_exists?(id)
+          if container_args_changed?(config)
+            @machine.ui.info(I18n.t("vagrant.container_restarting_container_args",
+              name: config[:name],
+            ))
+            stop_container(id)
+            create_container(config)
+          elsif container_image_changed?(config)
+            @machine.ui.info(I18n.t("vagrant.container_restarting_container_image",
+              name: config[:name],
+            ))
+            stop_container(id)
+            create_container(config)
+          else
+            start_container(id)
+          end
+        else
+          create_container(config)
+        end
+      end
+
+      def container_exists?(id)
+        lookup_container(id, true)
+      end
+
+      # Start container
+      #
+      # @param String - Image id
+      def start_container(id)
+        if !container_running?(id)
+          @machine.communicate.sudo("#{@container_command} start #{id}")
+        end
+      end
+
+      # Stop and remove container
+      #
+      # @param String - Image id
+      def stop_container(id)
+        @machine.communicate.sudo %[
+          #{@container_command} stop #{id}
+          #{@container_command} rm #{id}
+        ]
+      end
+
+      def container_running?(id)
+        lookup_container(id)
+      end
+
+      def container_image_changed?(config)
+        # Returns true if there is a container running with the given :name,
+        # and the container is not using the latest :image.
+
+        # Here, "<cmd> inspect <container>" returns the id of the image
+        # that the container is using. We check that the latest image that
+        # has been built with that name (:image)  matches the one that the
+        # container is running.
+        cmd = ("#{@container_command} inspect --format='{{.Image}}' #{config[:name]} |" +
+               " grep $(#{@container_command} images -q #{config[:image]})")
+        return !@machine.communicate.test(cmd)
+      end
+
+      def container_args_changed?(config)
+        path = container_data_path(config)
+        return true if !path.exist?
+
+        args = container_run_args(config)
+        sha  = Digest::SHA1.hexdigest(args)
+        return true if path.read.chomp != sha
+
+        return false
+      end
+
+      def create_container(config)
+        args = container_run_args(config)
+
+        @machine.communicate.sudo %[rm -f "#{config[:cidfile]}"]
+        @machine.communicate.sudo %[#{@container_command} run #{args}]
+
+        sha  = Digest::SHA1.hexdigest(args)
+        container_data_path(config).open("w+") do |f|
+          f.write(sha)
+        end
+      end
+
+      # Looks up if a container with a given id exists using the
+      # `ps` command. Returns Boolean
+      #
+      # @param String - Image id
+      def lookup_container(id, list_all = false)
+        container_ps = "sudo #{@container_command} ps -q"
+        container_ps << " -a" if list_all
+        @machine.communicate.tap do |comm|
+          return comm.test("#{container_ps} --no-trunc | grep -wFq #{id}")
+        end
+      end
+
+      def container_name(config)
+        name = config[:name]
+
+        # If the name is the automatically assigned name, then
+        # replace the "/" with "-" because "/" is not a valid
+        # character for a container name.
+        name = name.gsub("/", "-").gsub(":", "-") if name == config[:original_name]
+        name
+      end
+
+      # Compiles run arguments to be appended to command string.
+      # Returns String
+      def container_run_args(config)
+        name = container_name(config)
+
+        args = "--cidfile=#{config[:cidfile]} "
+        args << "-d " if config[:daemonize]
+        args << "--name #{name} " if name && config[:auto_assign_name]
+        args << "--restart=#{config[:restart]}" if config[:restart]
+        args << " #{config[:args]}" if config[:args]
+
+        "#{args} #{config[:image]} #{config[:cmd]}".strip
+      end
+
+      def container_data_path(config)
+        name = container_name(config)
+        @machine.data_dir.join("#{@container_command}-#{name}")
+      end
+
+      protected
+
+      # This handles outputting the communication data back to the UI
+      def handle_comm(type, data)
+        if [:stderr, :stdout].include?(type)
+          # Clear out the newline since we add one
+          data = data.chomp
+          return if data.empty?
+
+          options = {}
+          #options[:color] = color if !config.keep_color
+
+          @machine.ui.info(data.chomp, **options)
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/container/config.rb

@@ -0,0 +1,86 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require 'set'
+
+module VagrantPlugins
+  module ContainerProvisioner
+    class Config < Vagrant.plugin("2", :config)
+      attr_reader :images
+      attr_accessor :post_install_provisioner
+
+      def initialize
+        @images = Set.new
+        @post_install_provisioner = nil
+
+        @__build_images   = []
+        @__containers     = Hash.new { |h, k| h[k] = {} }
+      end
+
+      # Accessor for internal state.
+      def build_images
+        @__build_images
+      end
+
+      # Accessor for the internal state.
+      def containers
+        @__containers
+      end
+
+      # Defines an image to build using `<cmd> build` within the machine.
+      #
+      # @param [String] path Path to the Dockerfile to pass to
+      #   container build command
+      def build_image(path, **opts)
+        @__build_images << [path, opts]
+      end
+
+      def images=(images)
+        @images = Set.new(images)
+      end
+
+      def pull_images(*images)
+        @images += images.map(&:to_s)
+      end
+
+      def post_install_provision(name, **options, &block)
+        proxy = VagrantPlugins::Kernel_V2::VMConfig.new
+        proxy.provision(name, **options, &block)
+        @post_install_provisioner = proxy.provisioners.first
+      end
+
+      def run(name, **options)
+        @__containers[name.to_s] = options.dup
+      end
+
+      def merge(other)
+        super.tap do |result|
+          result.pull_images(*(other.images + self.images))
+
+          build_images = @__build_images.dup
+          build_images += other.build_images
+          result.instance_variable_set(:@__build_images, build_images)
+
+          containers = {}
+          @__containers.each do |name, params|
+            containers[name] = params.dup
+          end
+          other.containers.each do |name, params|
+            containers[name] = @__containers[name].merge(params)
+          end
+
+          result.instance_variable_set(:@__containers, containers)
+        end
+      end
+
+      def finalize!
+        @__containers.each do |name, params|
+          params[:image] ||= name
+          params[:auto_assign_name] = true if !params.key?(:auto_assign_name)
+          params[:daemonize] = true if !params.key?(:daemonize)
+          params[:restart] = "always" if !params.key?(:restart)
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/container/installer.rb

@@ -0,0 +1,16 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module ContainerProvisioner
+    class Installer
+      def initialize(machine)
+        @machine = machine
+      end
+
+      def ensure_installed
+        # nothing to do
+      end
+    end
+  end
+end

data/plugins/provisioners/container/plugin.rb

@@ -0,0 +1,26 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require "vagrant"
+
+module VagrantPlugins
+  module ContainerProvisioner
+    class Plugin < Vagrant.plugin("2")
+      name "container"
+      description <<-DESC
+      Provides support for provisioning your virtual machines with
+      OCI images and containers.
+      DESC
+
+      config(:container, :provisioner) do
+        require_relative "config"
+        Config
+      end
+
+      provisioner(:container) do
+        require_relative "provisioner"
+        Provisioner
+      end
+    end
+  end
+end

data/plugins/provisioners/container/provisioner.rb

@@ -0,0 +1,31 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require_relative "client"
+require_relative "installer"
+
+module VagrantPlugins
+  module ContainerProvisioner
+    class Provisioner < Vagrant.plugin("2", :provisioner)
+      def initialize(machine, config, installer = nil, client = nil)
+        super(machine, config)
+
+        @installer = installer || Installer.new(@machine)
+        @client    = client    || Client.new(@machine, "")
+        @logger = Log4r::Logger.new("vagrant::provisioners::container")
+      end
+
+      def provision
+        # nothing to do
+      end
+
+      def run_provisioner(env)
+        klass  = Vagrant.plugin("2").manager.provisioners[env[:provisioner].type]
+        result = klass.new(env[:machine], env[:provisioner].config)
+        result.config.finalize!
+
+        result.provision
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/centos/docker_install.rb

@@ -0,0 +1,44 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Centos
+        module DockerInstall
+          def self.docker_install(machine)
+            machine.communicate.tap do |comm|
+              comm.sudo("yum -q -y update")
+              comm.sudo("yum -q -y remove docker-io* || true")
+              comm.sudo("yum install -y -q yum-utils")
+              comm.sudo("yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo")
+              comm.sudo("yum makecache")
+              comm.sudo("yum install -y -q docker-ce")
+            end
+
+            case machine.guest.capability("flavor")
+            when :centos
+              docker_enable_service(machine)
+            else
+              docker_enable_systemctl(machine)
+            end
+          end
+
+          def self.docker_enable_systemctl(machine)
+            machine.communicate.tap do |comm|
+              comm.sudo("systemctl start docker.service")
+              comm.sudo("systemctl enable docker.service")
+            end
+          end
+
+          def self.docker_enable_service(machine)
+            machine.communicate.tap do |comm|
+              comm.sudo("service docker start")
+              comm.sudo("chkconfig docker on")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/centos/docker_start_service.rb

@@ -0,0 +1,27 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Centos
+        module DockerStartService
+          def self.docker_start_service(machine)
+            case machine.guest.capability("flavor")
+            when :centos
+              machine.communicate.tap do |comm|
+                comm.sudo("service docker start")
+                comm.sudo("chkconfig docker on")
+              end
+            else
+              machine.communicate.tap do |comm|
+                comm.sudo("systemctl start docker.service")
+                comm.sudo("systemctl enable docker.service")
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/debian/docker_install.rb

@@ -0,0 +1,21 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Debian
+        module DockerInstall
+          def self.docker_install(machine)
+            machine.communicate.tap do |comm|
+              comm.sudo("apt-get update -qq -y")
+              comm.sudo("apt-get install -qq -y --force-yes curl apt-transport-https")
+              comm.sudo("apt-get purge -qq -y lxc-docker* || true")
+              comm.sudo("curl -sSL https://get.docker.com/ | sh")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/debian/docker_start_service.rb

@@ -0,0 +1,16 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Debian
+        module DockerStartService
+          def self.docker_start_service(machine)
+            machine.communicate.sudo("service docker start")
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/fedora/docker_install.rb

@@ -0,0 +1,30 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Fedora
+        module DockerInstall
+          def self.docker_install(machine)
+            machine.communicate.tap do |comm|
+              if dnf?(machine)
+                comm.sudo("dnf -y install docker")
+              else
+                comm.sudo("yum -y install docker")
+              end
+              comm.sudo("systemctl start docker.service")
+              comm.sudo("systemctl enable docker.service")
+            end
+          end
+
+          protected
+
+          def self.dnf?(machine)
+            machine.communicate.test("/usr/bin/which -s dnf")
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/linux/docker_configure_vagrant_user.rb

@@ -0,0 +1,23 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Linux
+        module DockerConfigureVagrantUser
+          def self.docker_configure_vagrant_user(machine)
+            ssh_info = machine.ssh_info
+
+            machine.communicate.tap do |comm|
+              if comm.test("getent group docker") && !comm.test("id -Gn | grep docker")
+                comm.sudo("usermod -a -G docker #{ssh_info[:username]}")
+                comm.reset!
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/linux/docker_daemon_running.rb

@@ -0,0 +1,16 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Linux
+        module DockerDaemonRunning
+          def self.docker_daemon_running(machine)
+            machine.communicate.test("test -f /var/run/docker.pid")
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/linux/docker_installed.rb

@@ -0,0 +1,29 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Linux
+        module DockerInstalled
+          def self.docker_installed(machine)
+            paths = [
+              "/bin/docker",
+              "/usr/bin/docker",
+              "/usr/local/bin/docker",
+              "/usr/sbin/docker",
+            ]
+
+            paths.each do |p|
+              if machine.communicate.test("test -f #{p}", sudo: true)
+                return true
+              end
+            end
+
+            return false
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/cap/windows/docker_daemon_running.rb

@@ -0,0 +1,16 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+module VagrantPlugins
+  module DockerProvisioner
+    module Cap
+      module Windows
+        module DockerDaemonRunning
+          def self.docker_daemon_running(machine)
+            machine.communicate.test("tasklist | find \"`\"dockerd`\"\"")
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/client.rb

@@ -0,0 +1,26 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require_relative "../container/client"
+
+module VagrantPlugins
+  module DockerProvisioner
+    class Client < VagrantPlugins::ContainerProvisioner::Client
+      def initialize(machine)
+        super(machine, "docker")
+        @container_command = "docker"
+      end
+
+      def start_service
+        if !daemon_running? && @machine.guest.capability?(:docker_start_service)
+          @machine.guest.capability(:docker_start_service)
+        end
+      end
+
+      def daemon_running?
+        @machine.guest.capability(:docker_daemon_running)
+      end
+
+    end
+  end
+end

data/plugins/provisioners/docker/config.rb

@@ -0,0 +1,19 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require_relative "../container/config"
+
+module VagrantPlugins
+  module DockerProvisioner
+    class Config < VagrantPlugins::ContainerProvisioner::Config
+      def post_install_provision(name, **options, &block)
+        # Abort
+        raise DockerError, :wrong_provisioner if options[:type] == "docker"
+
+        proxy = VagrantPlugins::Kernel_V2::VMConfig.new
+        proxy.provision(name, **options, &block)
+        @post_install_provisioner = proxy.provisioners.first
+      end
+    end
+  end
+end

data/plugins/provisioners/docker/installer.rb

@@ -0,0 +1,37 @@
+# Copyright (c) HashiCorp, Inc.
+# SPDX-License-Identifier: BUSL-1.1
+
+require_relative "../container/installer"
+
+module VagrantPlugins
+  module DockerProvisioner
+    class Installer < VagrantPlugins::ContainerProvisioner::Installer
+      # This handles verifying the Docker installation, installing it if it was
+      # requested, and so on. This method will raise exceptions if things are
+      # wrong.
+      # @return [Boolean] - false if docker cannot be detected on machine, else
+      #                     true if docker installs correctly or is installed
+      def ensure_installed
+        if !@machine.guest.capability?(:docker_installed)
+          @machine.ui.warn(I18n.t("vagrant.docker_cant_detect"))
+          return false
+        end
+
+        if !@machine.guest.capability(:docker_installed)
+          @machine.ui.detail(I18n.t("vagrant.docker_installing"))
+          @machine.guest.capability(:docker_install)
+        end
+
+        if !@machine.guest.capability(:docker_installed)
+          raise DockerError, :install_failed
+        end
+
+        if @machine.guest.capability?(:docker_configure_vagrant_user)
+          @machine.guest.capability(:docker_configure_vagrant_user)
+        end
+
+        true
+      end
+    end
+  end
+end