vagrant-zones 0.0.1

data/lib/vagrant-zones/command/shutdown_guest.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used to shutdown the guest from inside the guest
+      class ShutdownGuest < Vagrant.plugin('2', :command)
+        def execute
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant zone control shutdown [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          unless argv.empty?
+            @env.ui.info(opts.help)
+            return
+          end
+
+          ## Wait for VM up
+          with_target_vms(argv, provider: :zone) do |machine|
+            machine.action('shutdown')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/command/vnc_console.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used to start a VNC console to the guest
+      class VNCConsole < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant zone console vnc [options]'
+            o.on('--ip <host_ip>', 'Specify host IP to listen on') do |p|
+              options[:ip] = p
+            end
+            o.on('--port <port>', 'Specify port to listen on') do |p|
+              options[:port] = p
+            end
+            o.on('--detach <yes/no>', 'Run console server in background') do |p|
+              options[:detach] = p
+            end
+            o.on('--kill <yes/no>', 'Kill the previous background console session') do |p|
+              options[:kill] = p
+            end
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          unless argv.length <= 4
+            @env.ui.info(opts.help)
+            return
+          end
+
+          options[:port] = nil unless options[:port] =~ /\d/
+          with_target_vms(argv, provider: :zone) do |machine|
+            driver = machine.provider.driver
+            detach = 'yes'
+            detach = 'no' unless options[:detach] == 'yes'
+            kill = 'yes'
+            kill = 'no' unless options[:kill] == 'yes'
+            exit = { detach: detach, kill: kill }
+            driver.console(@env.ui, 'vnc', options[:ip], options[:port], exit)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/command/webvnc_console.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require 'resolv'
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used to start a WebVNC console to the guest
+      class WebVNCConsole < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant zone console webvnc [options]'
+            o.on('--ip <host_ip>', 'Specify host IP to listen on') do |p|
+              options[:ip] = p
+            end
+            o.on('--port <port>', 'Specify port to listen on') do |p|
+              options[:port] = p
+            end
+            o.on('--detach <yes/no>', 'Run console server in background') do |p|
+              options[:detach] = p
+            end
+            o.on('--kill <yes/no>', 'Kill the previous background console session') do |p|
+              options[:kill] = p
+            end
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          unless argv.length <= 4
+            @env.ui.info(opts.help)
+            return
+          end
+
+          options[:port] = nil unless options[:port] =~ /\d/
+          with_target_vms(argv, provider: :zone) do |machine|
+            driver = machine.provider.driver
+            detach = 'yes'
+            detach = 'no' unless options[:detach] == 'yes'
+            kill = 'yes'
+            kill = 'no' unless options[:kill] == 'yes'
+            exit = { detach: detach, kill: kill }
+            driver.console(@env.ui, 'webvnc', options[:ip], options[:port], exit)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/command/zfssnapshot.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used to manage ZFS snapshtos for the zone
+      class ZFSSnapshot < Vagrant.plugin('2', :command)
+        def initialize(argv, env)
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+          @subcommands.register(:list) do
+            require File.expand_path('list_snapshots', __dir__)
+            ListSnapshots
+          end
+          @subcommands.register(:create) do
+            require File.expand_path('create_snapshots', __dir__)
+            CreateSnapshots
+          end
+          @subcommands.register(:delete) do
+            require File.expand_path('delete_snapshots', __dir__)
+            DeleteSnapshots
+          end
+          @subcommands.register(:configure) do
+            require File.expand_path('configure_snapshots', __dir__)
+            ConfigureSnapshots
+          end
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?('-h') || @main_args.include?('--help')
+            # Print the help for all the vagrant-zones commands.
+            return help
+          end
+
+          command_class = @subcommands.get(:create) if @sub_command.nil?
+          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+
+          subargs = @sub_args unless @sub_args.nil?
+          @logger.debug("Invoking command class: #{command_class} #{subargs.inspect}")
+
+          # Initialize and execute the command class
+          command_class.new(subargs, @env).execute
+        end
+
+        def help
+          opts = OptionParser.new do |subopts|
+            subopts.banner = 'Usage: vagrant zone zfssnapshot <subcommand> [<args>]'
+            subopts.separator ''
+            subopts.separator 'Available subcommands:'
+            # Add the available subcommands as separators in order to print them
+            # out as well.
+            keys = []
+            @subcommands.each { |key, _value| keys << key.to_s }
+            keys.sort.each do |key|
+              subopts.separator "     #{key}"
+            end
+            subopts.separator ''
+            subopts.separator 'For help on any individual subcommand run `vagrant zone zfssnapshot <subcommand> -h`'
+          end
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/command/zlogin_console.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used to acces the zone via console, zlogin
+      class ZloginConsole < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant zone console zlogin [options]'
+            o.on('--ip <host_ip>', 'Specify host IP to listen on') do |p|
+              options[:ip] = p
+            end
+            o.on('--port <port>', 'Specify port to listen on') do |p|
+              options[:port] = p
+            end
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          unless argv.length <= 4
+            @env.ui.info(opts.help)
+            return
+          end
+
+          options[:port] = nil unless options[:port] =~ /\d/
+          with_target_vms(argv, provider: :zone) do |machine|
+            driver = machine.provider.driver
+            detach = 'no'
+            kill = 'no'
+            exit = { detach: detach, kill: kill }
+            driver.console(@env.ui, 'zlogin', options[:ip], options[:port], exit)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/command/zone.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+require 'vagrant-zones/action'
+
+module VagrantPlugins
+  module ProviderZone
+    module Command
+      # This is used manage the zone where vagrant cannot
+      class Zone < Vagrant.plugin('2', :command)
+        def self.synopsis
+          'Manage zones and query zone information'
+        end
+
+        def initialize(argv, env)
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+
+          @subcommands.register(:zfssnapshot) do
+            require File.expand_path('zfssnapshot', __dir__)
+            ZFSSnapshot
+          end
+          @subcommands.register(:control) do
+            require File.expand_path('guest_power_controls', __dir__)
+            GuestPowerControls
+          end
+          @subcommands.register(:console) do
+            require File.expand_path('console', __dir__)
+            Console
+          end
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?('-h') || @main_args.include?('--help')
+            # Print the help for all the vagrant-zones commands.
+            return help
+          end
+
+          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+          return help if !command_class || !@sub_command
+
+          @logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")
+
+          # Initialize and execute the command class
+          command_class.new(@sub_args, @env).execute
+        end
+
+        def help
+          opts = OptionParser.new do |subopts|
+            subopts.banner = 'Usage: vagrant zone <subcommand> [<args>]'
+            subopts.separator ''
+            subopts.separator 'Available subcommands:'
+
+            # Add the available subcommands as separators in order to print them
+            # out as well.
+            keys = []
+            @subcommands.each { |key, _value| keys << key.to_s }
+
+            keys.sort.each do |key|
+              subopts.separator "     #{key}"
+            end
+
+            subopts.separator ''
+            subopts.separator 'For help on any individual subcommand run `vagrant zone <subcommand> -h`'
+          end
+
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-zones/config.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require 'vagrant'
+## Do not Modify this File! Modify the Hosts.yml, Hosts.rb, or Vagrantfile!
+module VagrantPlugins
+  module ProviderZone
+    # This is used define the variables for the project
+    class Config < Vagrant.plugin('2', :config)
+      # rubocop:disable Layout/LineLength
+      attr_accessor :brand, :autoboot, :setup_method, :safe_restart, :allowed_address, :safe_shutdown, :boxshortname, :kernel, :debug, :debug_boot, :private_network, :winalcheck, :winlcheck, :lcheck, :alcheck, :snapshot_script, :diskif, :netif, :cdroms, :disk1path, :disk1size, :cpus, :cpu_configuration, :boot, :complex_cpu_conf, :memory, :vagrant_user, :vagrant_user_private_key_path, :setup_wait, :clean_shutdown_time, :dhcp, :vagrant_user_pass, :firmware_type, :vm_type, :partition_id, :shared_disk_enabled, :shared_dir, :acpi, :os_type, :console, :consolehost, :consoleport, :console_onboot, :hostbridge, :sshport, :rdpport, :override, :additional_disks, :cloud_init_resolvers, :cloud_init_enabled, :cloud_init_dnsdomain, :cloud_init_password, :cloud_init_sshkey, :cloud_init_conf, :dns, :box, :vagrant_cloud_creator, :winbooted_string, :booted_string, :zunlockbootkey, :zunlockboot, :xhci_enabled
+
+      # rubocop:enable Layout/LineLength
+
+      def initialize
+        super
+        @brand = 'bhyve'
+        @additional_disks = UNSET_VALUE
+        @autoboot = true
+        @kernel = UNSET_VALUE
+        @boxshortname = UNSET_VALUE
+        @cdroms = nil
+        @shared_dir = nil
+        @os_type = 'generic'
+        @lcheck = UNSET_VALUE
+        @booted_string = UNSET_VALUE
+        @winbooted_string = UNSET_VALUE
+        @allowed_address = true
+        @alcheck = 'login: '
+        @winalcheck = 'EVENT: The CMD command is now available.'
+        @winlcheck = 'EVENT: The CMD command is now available.'
+        @zunlockbootkey = ''
+        @zunlockboot = false
+        @safe_restart = nil
+        @safe_shutdown = nil
+        @debug_boot = nil
+        @debug = nil
+        @shared_disk_enabled = true
+        @consoleport = nil
+        @consolehost = '0.0.0.0'
+        @console_onboot = 'false'
+        @console = 'webvnc'
+        @memory = '4G'
+        @diskif = 'virtio-blk'
+        @netif = 'virtio-net-viona'
+        @cpus = 2
+        @cpu_configuration = 'simple'
+        @complex_cpu_conf = UNSET_VALUE
+        @boot = UNSET_VALUE
+        @hostbridge = 'i440fx'
+        @acpi = 'on'
+        @setup_wait = 90
+        @box = UNSET_VALUE
+        @clean_shutdown_time = 300
+        @dns = [{ 'nameserver' => '1.1.1.1' }, { 'nameserver' => '1.0.0.1' }]
+        @vmtype = 'production'
+        @partition_id = '0000'
+        @sshport = '22'
+        @rdpport = '3389'
+        @vagrant_user = 'vagrant'
+        @vagrant_user_pass = 'vagrant'
+        @vagrant_user_private_key_path = './id_rsa'
+        @xhci_enabled = 'off'
+        @override = false
+        @cloud_init_enabled = false
+        @cloud_init_conf = 'on'
+        @cloud_init_dnsdomain = nil
+        @cloud_init_password = nil
+        @cloud_init_resolvers = nil
+        @cloud_init_sshkey = nil
+        @private_network = nil
+        @firmware_type = 'compatability'
+        @vm_type = 'production'
+        @setup_method = nil
+        @snapshot_script = '/opt/vagrant/bin/Snapshooter.sh'
+      end
+    end
+  end
+end