RubyGems - vagrant-zones - Versions diffs - 0.0.1 - Mend

vagrant-zones 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (67) hide show

checksums.yaml +7 -0
data/.github/ISSUE_TEMPLATE/bug_report.md +27 -0
data/.github/ISSUE_TEMPLATE/feature_request.md +20 -0
data/.github/dependabot.yml +6 -0
data/.github/workflows/codeql-analysis.yml +72 -0
data/.github/workflows/lint-release-and-publish.yml +70 -0
data/.github/workflows/ruby-lint.yml +35 -0
data/.gitignore +35 -0
data/.rspec +2 -0
data/.rubocop.yml +143 -0
data/CHANGELOG.md +0 -0
data/CODE_OF_CONDUCT.md +128 -0
data/CONTRIBUTING.md +96 -0
data/Gemfile +14 -0
data/LICENSE +651 -0
data/PULL_REQUEST_TEMPLATE.md +39 -0
data/README.md +81 -0
data/RELEASE.md +15 -0
data/Rakefile +32 -0
data/SECURITY.md +19 -0
data/docs/CNAME +1 -0
data/docs/_config.yml +1 -0
data/docs/css/main.css +55 -0
data/docs/css/styles.css +8678 -0
data/docs/index.html +127 -0
data/lib/vagrant-zones/action/create.rb +29 -0
data/lib/vagrant-zones/action/destroy.rb +27 -0
data/lib/vagrant-zones/action/halt.rb +24 -0
data/lib/vagrant-zones/action/import.rb +112 -0
data/lib/vagrant-zones/action/is_created.rb +22 -0
data/lib/vagrant-zones/action/network.rb +26 -0
data/lib/vagrant-zones/action/not_created.rb +20 -0
data/lib/vagrant-zones/action/package.rb +134 -0
data/lib/vagrant-zones/action/prepare_nfs_valid_ids.rb +24 -0
data/lib/vagrant-zones/action/restart.rb +53 -0
data/lib/vagrant-zones/action/setup.rb +26 -0
data/lib/vagrant-zones/action/shutdown.rb +47 -0
data/lib/vagrant-zones/action/start.rb +25 -0
data/lib/vagrant-zones/action/wait_till_boot.rb +59 -0
data/lib/vagrant-zones/action/wait_till_up.rb +65 -0
data/lib/vagrant-zones/action.rb +204 -0
data/lib/vagrant-zones/command/configure_snapshots.rb +49 -0
data/lib/vagrant-zones/command/console.rb +63 -0
data/lib/vagrant-zones/command/create_snapshots.rb +46 -0
data/lib/vagrant-zones/command/delete_snapshots.rb +38 -0
data/lib/vagrant-zones/command/guest_power_controls.rb +58 -0
data/lib/vagrant-zones/command/list_snapshots.rb +44 -0
data/lib/vagrant-zones/command/restart_guest.rb +29 -0
data/lib/vagrant-zones/command/shutdown_guest.rb +29 -0
data/lib/vagrant-zones/command/vnc_console.rb +48 -0
data/lib/vagrant-zones/command/webvnc_console.rb +49 -0
data/lib/vagrant-zones/command/zfssnapshot.rb +67 -0
data/lib/vagrant-zones/command/zlogin_console.rb +40 -0
data/lib/vagrant-zones/command/zone.rb +73 -0
data/lib/vagrant-zones/config.rb +78 -0
data/lib/vagrant-zones/driver.rb +1710 -0
data/lib/vagrant-zones/errors.rb +61 -0
data/lib/vagrant-zones/executor.rb +38 -0
data/lib/vagrant-zones/plugin.rb +79 -0
data/lib/vagrant-zones/provider.rb +83 -0
data/lib/vagrant-zones/util/subprocess.rb +31 -0
data/lib/vagrant-zones/util/timer.rb +19 -0
data/lib/vagrant-zones/version.rb +7 -0
data/lib/vagrant-zones.rb +29 -0
data/locales/en.yml +326 -0
data/vagrant-zones.gemspec +51 -0
metadata +412 -0

data/docs/index.html ADDED Viewed

@@ -0,0 +1,127 @@
+<!DOCTYPE html>
+<html lang="en">
+    <head>
+        <meta charset="UTF-8" />
+        <meta http-equiv="X-UA-Compatible" content="IE=edge" />
+        <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+        <title>Vagrant Zones</title>
+        <link rel="stylesheet" href="css/styles.css" />
+        <link rel="stylesheet" href="css/main.css" />
+    </head>
+    <body>
+        <div class="header">
+            <div class="container">
+                <div class="row d-flex">
+                    <img
+                        src="assets/logo-big.jpg"
+                        class="logo-big img-fluid mx-auto mt-4"
+                        alt="..."
+                    />
+                </div>
+                <div class="row justify-content-md-center">
+                    <div class="menu-link col-md-auto fs-3 p-3 text-center">
+                        <a href="index.html">Home</a>
+                    </div>
+                    <div class="menu-link col-md-auto fs-3 p-3 text-center">
+                        <a href="own-your-data.html">Create your Automation</a>
+                    </div>
+                    <div class="menu-link col-md-auto fs-3 p-3 text-center">
+                        <a href="contact-us.html">Issue Tracking</a>
+                    </div>
+                </div>
+            </div>
+        </div>
+        <div class="container">
+            <div class="row my-4">
+                <div class="col">
+                    <div class="card shadow">
+                        <div class="card-body">
+                            <p class="card-text fs-3 text-center">
+                                Vagrant-Zone is a plugin for Vagrant which
+                                will allow you to sping up bhyve, kvm,
+                                LX, illumos, and other zone types as
+                                you would a VirtualBox vagrant machine.
+                            </p>
+                        </div>
+                    </div>
+                </div>
+            </div>
+        </div>
+        <div class="container">
+            <div class="row">
+                <!-- Card 1 -->
+                <div class="col-md-6 col-lg-3 my-4">
+                    <div class="card shadow">
+                        <div class="card-body">
+                            <h5 class="card-title mb-4">Manage with Confidence</h5>
+                            <p class="card-text">
+                                Versioning, and automated deployment controls
+                                allow you to build your Virtual Machines
+                                with confidence the process will be transparent.
+                            </p>
+                        </div>
+                    </div>
+                </div>
+                <!-- Card 2 -->
+                <div class="col-md-6 col-lg-3 my-4">
+                    <div class="card shadow">
+                        <div class="card-body">
+                            <h5 class="card-title mb-4">Create your Automation</h5>
+                            <p class="card-text">
+                                The ability to quickly spin up zones of any type
+                                in a repeatable fashion allows, portable, versioned
+                                and automatable packages ensuring your projects environments
+                                are uniform and debuggable.
+                            </p>
+                        </div>
+                    </div>
+                </div>
+                <!-- Card 3 -->
+                <div class="col-md-6 col-lg-3 my-4">
+                    <div class="card shadow">
+                        <div class="card-body">
+                            <h5 class="card-title mb-4">Share with Others</h5>
+                            <p class="card-text">
+                                This tool will allow you to package and share boxes
+                                with other users, helping you develope on a wide
+                                array of Hypervisors and Operating Systems.
+                            </p>
+                        </div>
+                    </div>
+                </div>
+                <!-- Card 4 -->
+                <div class="col-md-6 col-lg-3 my-4">
+                    <div class="card shadow">
+                        <div class="card-body">
+                            <h5 class="card-title mb-4">Feature Requests</h5>
+                            <p class="card-text">
+                                Reach out to use in a GitHub issue and let us know what
+                                your projects goals are. If you can commit to testing
+                                we can work on getting your feature commited.
+                            </p>
+                        </div>
+                    </div>
+                </div>
+            </div>
+        </div>
+        <div class="container">
+            <div class="card shadow footer mt-5">
+                <img
+                    src="assets/logo-small.png"
+                    class="logo-small mx-auto my-4"
+                    alt="..."
+                />
+                <p class="copyright text-center pb-4 mb-0">
+                    ©2022 STARTcloud, Inc.
+                </p>
+            </div>
+        </div>
+    </body>
+</html>

data/lib/vagrant-zones/action/create.rb ADDED Viewed

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'securerandom'
+require 'digest/md5'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This will create the zone
+      class Create
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          @machine.id = SecureRandom.uuid
+          @driver.create_dataset(env[:ui])
+          @driver.zonecfg(env[:ui])
+          @driver.install(env[:ui])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/destroy.rb ADDED Viewed

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'securerandom'
+require 'digest/md5'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This will destroy the zone
+      class Destroy
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          @driver.halt(env[:ui])
+          @driver.destroy(env[:ui])
+          @driver.delete_dataset(env[:ui])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/halt.rb ADDED Viewed

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+require 'log4r'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This will halt the zone
+      class Halt
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          @driver.halt(env[:ui])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/import.rb ADDED Viewed

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+require 'net/http'
+require 'vagrant-zones/util/subprocess'
+require 'pp'
+require 'vagrant/box_metadata'
+require 'vagrant/util/downloader'
+require 'vagrant/util/platform'
+require 'vagrant/util/safe_chdir'
+require 'vagrant/util/subprocess'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This will import the zone boot image from the cloud, cache or file
+      class Import
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @joyent_images_url = 'https://images.joyent.com/images/'
+          @app = app
+        end
+        def validate_uuid_format(uuid)
+          uuid_regex = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/
+          return true if uuid_regex.match?(uuid.to_s.downcase)
+        end
+        def call(env)
+          @machine = env[:machine]
+          @executor = Executor::Exec.new
+          image = @machine.config.vm.box
+          curdir = Dir.pwd
+          datadir = @machine.data_dir
+          @driver = @machine.provider.driver
+          ui = env[:ui]
+          ui.info(I18n.t('vagrant_zones.meeting'))
+          ui.info(I18n.t('vagrant_zones.datadir'))
+          ui.info("  #{datadir}")
+          ui.info(I18n.t('vagrant_zones.detecting_box'))
+          # If image ends on '.zss' it's a local ZFS snapshot which should be used
+          if image[-4, 4] == '.zss'
+            if File.exist?("#{curdir}/#{image}")
+              FileUtils.cp("#{curdir}/#{image}", "#{datadir}/#{image}")
+              ui.info(I18n.t('vagrant_zones.zfs_snapshot_stream_detected'))
+            elsif !File.exist?("#{datadir}/#{image}")
+              raise Vagrant::Errors::BoxNotFound
+            end
+          ## If image looks like an UUID, download the ZFS snapshot from Joyent images server
+          elsif validate_uuid_format(image)
+            raise Vagrant::Errors::BoxNotFound unless check(image, ui)
+            uri = URI("#{@joyent_images_url}#{image}/file")
+            Net::HTTP.start(uri.host, uri.port, :use_ssl => uri.scheme == 'https') do |http|
+              request = Net::HTTP::Get.new uri
+              http.request request do |response|
+                file_size = response['content-length'].to_i
+                amount_downloaded = 0
+                ratelimit = 0
+                rate = 500
+                large_file = "#{datadir}/#{image}"
+                File.open(large_file, 'wb') do |io|
+                  response.read_body do |chunk|
+                    io.write chunk
+                    amount_downloaded += chunk.size
+                    ui.rewriting do |uiprogress|
+                      ratelimit += 1
+                      if ratelimit >= rate
+                        uiprogress.clear_line
+                        status = format('%.2f%%', (amount_downloaded.to_f / file_size * 100))
+                        uiprogress.info(I18n.t('vagrant_zones.importing_joyent_image') + "#{image} ==> ", new_line: false)
+                        uiprogress.report_progress(status, 100, false)
+                        ratelimit = 0
+                      end
+                    end
+                  end
+                  ui.clear_line
+                end
+              end
+            end
+            ui.info(I18n.t('vagrant_zones.joyent_image_uuid_detected') + image)
+          ## If it's a regular name (everything else), try to find it on Vagrant Cloud
+          else
+            # Support zss maybe zst? Same thing? format only for now, use other images and convert later
+            box_format = env[:machine].box.metadata['format'] unless env[:machine].box.nil?
+            if box_format == 'ovf'
+              ## Insert Future Code to try to convert existing box
+              ui.info(I18n.t('vagrant_zones.detected_ovf_format'))
+            end
+            ## No Local box template exists, Lets use Vagrant HandleBox to download the Box template
+            ui.info(I18n.t('vagrant_zones.vagrant_cloud_box_detected'))
+            ui.info("  #{image}")
+            ui.clear_line
+          end
+          @app.call(env)
+        end
+        def execute(*cmd, **opts, &block)
+          @executor.execute(*cmd, **opts, &block)
+        end
+        def check(uuid, env_ui)
+          execute(true, "curl --output /dev/null --silent  -r 0-0 --fail #{@joyent_images_url}/#{uuid}")
+          env_ui.info(I18n.t('vagrant_zones.joyent_image_uuid_verified') + @joyent_images_url + uuid)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/is_created.rb ADDED Viewed

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+require 'log4r'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This can be used with 'Call' built-in to check if the machine
+      # is created and branch in the middleware.
+      class IsCreated
+        def initialize(app, _env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_zones::action')
+        end
+        def call(env)
+          env[:result] = env[:machine].state.id != :not_created
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/network.rb ADDED Viewed

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'securerandom'
+require 'digest/md5'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is use to define the network
+      class Network
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          @driver.network(env[:ui], 'create')
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/not_created.rb ADDED Viewed

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+require 'log4r'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to determine if the VM is created
+      class NotCreated
+        def initialize(app, _env)
+          @app = app
+        end
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_zones.states.not_created'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/package.rb ADDED Viewed

@@ -0,0 +1,134 @@
+# frozen_string_literal: true
+require 'log4r'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to package the VM into a box
+      class Package
+        def initialize(app, env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+          @executor = Executor::Exec.new
+          @pfexec = if Process.uid.zero?
+                      ''
+                    elsif system('sudo -v')
+                      'sudo'
+                    else
+                      'pfexec'
+                    end
+          env['package.output'] ||= 'package.box'
+        end
+        def execute(*cmd, **opts, &block)
+          @executor.execute(*cmd, **opts, &block)
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver = @machine.provider.driver
+          name = @machine.name
+          boxname = env['package.output']
+          brand = @machine.provider_config.brand
+          kernel = @machine.provider_config.kernel
+          vcc = @machine.provider_config.vagrant_cloud_creator
+          boxshortname = @machine.provider_config.boxshortname
+          raise "#{boxname}: Already exists" if File.exist?(boxname)
+          ## Create Snapshot
+          tmp_dir = "#{Dir.pwd}/_tmp_package"
+          Dir.mkdir(tmp_dir)
+          datasetpath = "#{@machine.provider_config.boot['array']}/#{@machine.provider_config.boot['dataset']}/#{name}"
+          t = Time.new
+          d = '-'
+          c = ':'
+          datetime = %(#{t.year}#{d}#{t.month}#{d}#{t.day}#{d}#{t.hour}#{c}#{t.min}#{c}#{t.sec})
+          snapshot_create(datasetpath, datetime, env[:ui], @machine.provider_config)
+          snapshot_send(datasetpath, "#{tmp_dir}/box.zss", datetime, env[:ui], @machine.provider_config)
+          ## snapshot_delete(datasetpath, env[:ui], datetime)
+          # Package VM
+          extra = ''
+          ## Include User Extra Files
+          @tmp_include = "#{tmp_dir}/_include"
+          if env['package.include']
+            extra = './_include'
+            Dir.mkdir(@tmp_include)
+            env['package.include'].each do |f|
+              env[:ui].info("Including user file: #{f}")
+              FileUtils.cp(f, @tmp_include)
+            end
+          end
+          ## Include Vagrant file
+          if env['package.vagrantfile']
+            extra = './_include'
+            Dir.mkdir(@tmp_include) unless File.directory?(@tmp_include)
+            env[:ui].info('Including user Vagrantfile')
+            FileUtils.cp(env['package.vagrantfile'], "#{@tmp_include}/Vagrantfile")
+          end
+          ## Create the Metadata and Vagrantfile
+          Dir.chdir(tmp_dir)
+          File.write('./metadata.json', metadata_content(brand, kernel, vcc, boxshortname))
+          File.write('./Vagrantfile', vagrantfile_content(brand, kernel, datasetpath))
+          ## Create the Box file
+          assemble_box(boxname, extra)
+          FileUtils.mv("#{tmp_dir}/#{boxname}", "../#{boxname}")
+          Dir.chdir('../')
+          FileUtils.rm_rf(tmp_dir)
+          env[:ui].info("Box created, You can now add the box: 'vagrant box add #{boxname} --nameofnewbox'")
+          @app.call(env)
+        end
+        def snapshot_create(datasetpath, datetime, uii, config)
+          uii.info('Creating a Snapshot of the box.')
+          result = execute(true, "#{@pfexec} zfs snapshot -r #{datasetpath}/boot@vagrant_box#{datetime}")
+          uii.info("#{@pfexec} zfs snapshot -r #{datasetpath}/boot@vagrant_box#{datetime}") if result.zero? && config.debug
+        end
+        def snapshot_delete(datasetpath, uii, datetime)
+          result = execute(true, "#{@pfexec} zfs destroy -r -f #{datasetpath}/boot@vagrant_box#{datetime}")
+          uii.info("#{@pfexec} zfs destroy -r -f #{datasetpath}/boot@vagrant_box#{datetime}") if result.zero? && config.debug
+        end
+        def snapshot_send(datasetpath, destination, datetime, uii, config)
+          uii.info('Sending Snapshot to ZFS Send Stream image.')
+          result = execute(true, "#{@pfexec} zfs send #{datasetpath}/boot@vagrant_box#{datetime} > #{destination}")
+          puts "#{@pfexec} zfs send -r #{datasetpath}/boot@vagrant_box#{datetime} > #{destination}" if result.zero? && config.debug
+        end
+        def metadata_content(brand, _kernel, vcc, boxshortname)
+          <<-ZONEBOX
+          {
+            "provider": "zone",
+            "format": "zss",
+            "brand": "#{brand}",
+            "url": "https://app.vagrantup.com/#{vcc}/boxes/#{boxshortname}"
+          }
+          ZONEBOX
+        end
+        def vagrantfile_content(brand, _kernel, datasetpath)
+          <<-ZONEBOX
+          Vagrant.configure('2') do |config|
+            config.vm.provider :zone do |zone|
+              zone.brand = "#{brand}"
+              zone.datasetpath = "#{datasetpath}"
+            end
+          end
+          user_vagrantfile = File.expand_path('../_include/Vagrantfile', __FILE__)
+          load user_vagrantfile if File.exists?(user_vagrantfile)
+          ZONEBOX
+        end
+        def assemble_box(boxname, extra)
+          `tar -cvzEf "#{boxname}" ./metadata.json ./Vagrantfile ./box.zss #{extra}`
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/prepare_nfs_valid_ids.rb ADDED Viewed

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'securerandom'
+require 'digest/md5'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to prepare NFS ids for NFS Sharing
+      class PrepareNFSValidIds
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::prepare_nfs_valid_ids')
+          @app = app
+        end
+        def call(env)
+          env[:nfs_valid_ids] = [env[:machine].id]
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/restart.rb ADDED Viewed

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'vagrant-zones/util/timer'
+require 'vagrant/util/retryable'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to restart the zone
+      class Restart
+        include Vagrant::Util::Retryable
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::restart')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver = @machine.provider.driver
+          ui = env[:ui]
+          ui.info(I18n.t('vagrant_zones.graceful_restart'))
+          @driver.control(ui, 'restart')
+          env[:metrics] ||= {}
+          env[:metrics]['instance_ssh_time'] = Util::Timer.time do
+            retryable(on: Errors::TimeoutError, tries: 300) do
+              # If we're interrupted don't worry about waiting
+              next if env[:interrupted]
+              loop do
+                break if env[:interrupted]
+                break unless env[:machine].communicate.ready?
+              end
+            end
+          end
+          ui.info(I18n.t('vagrant_zones.zone_gracefully_stopped_waiting_for_boot'))
+          env[:metrics] ||= {}
+          env[:metrics]['instance_ssh_time'] = Util::Timer.time do
+            retryable(on: Errors::TimeoutError, tries: 300) do
+              # If we're interrupted don't worry about waiting
+              next if env[:interrupted]
+              break if env[:machine].communicate.ready?
+            end
+          end
+          ui.info(I18n.t('vagrant_zones.zone_gracefully_restarted'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/setup.rb ADDED Viewed

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'securerandom'
+require 'digest/md5'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to setup the zone
+      class Setup
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::import')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver = @machine.provider.driver
+          @driver.setup(env[:ui])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/action/shutdown.rb ADDED Viewed

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+require 'log4r'
+require 'vagrant-zones/util/timer'
+require 'vagrant/util/retryable'
+module VagrantPlugins
+  module ProviderZone
+    module Action
+      # This is used to shutdown the zone
+      class Shutdown
+        include Vagrant::Util::Retryable
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_zones::action::shutdown')
+          @app = app
+        end
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          ui = env[:ui]
+          ui.info(I18n.t('vagrant_zones.graceful_shutdown_started'))
+          @driver.control(ui, 'shutdown')
+          env[:metrics] ||= {}
+          env[:metrics]['instance_ssh_time'] = Util::Timer.time do
+            retryable(on: Errors::TimeoutError, tries: 300) do
+              # If we're interrupted don't worry about waiting
+              break if env[:interrupted]
+              break unless env[:machine].communicate.ready?
+            end
+          end
+          env[:metrics]['instance_ssh_time'] = Util::Timer.time do
+            300.times do
+              state_id = @driver.state(@machine)
+              ui.info(I18n.t('vagrant_zones.graceful_shutdown_complete')) unless state_id == :running
+              sleep 1 if state_id == :running
+              break unless state_id == :running
+              break if env[:interrupted]
+            end
+          end
+          @driver.halt(env[:ui])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end