vagrant-local 0.0.1

data/lib/vagrant-local/action/wait_till_up.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require 'log4r'
+require 'vagrant-local/util/timer'
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module ProviderLocal
+    module Action
+      # This is used wait till the instance is booted
+      class WaitTillUp
+        include Vagrant::Util::Retryable
+
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_local::action::import')
+          @app = app
+        end
+
+        def terminate(env)
+          return unless env[:machine].state.id != :not_created
+
+          # If we're not supposed to destroy on error then just return
+          return unless env[:destroy_on_error]
+
+          if env[:halt_on_error]
+            halt_env = env.dup
+            halt_env.delete(:interrupted)
+            halt_env[:config_validate] = false
+            env[:action_runner].run(Action.action_halt, halt_env)
+          else
+            destroy_env = env.dup
+            destroy_env.delete(:interrupted)
+            destroy_env[:config_validate] = false
+            destroy_env[:force_confirm_destroy] = true
+            env[:action_runner].run(Action.action_destroy, destroy_env)
+          end
+        end
+
+        def call(env)
+          @machine = env[:machine]
+          @driver  = @machine.provider.driver
+          ui = env[:ui]
+          # Initialize metrics if they haven't been
+          env[:metrics] ||= {}
+          env[:metrics]['instance_ssh_time'] = Util::Timer.time do
+            retryable(on: Errors::TimeoutError, tries: 60) do
+              # If we're interrupted don't worry about waiting
+              next if env[:interrupted]
+
+              loop do
+                break if env[:interrupted]
+                break if env[:machine].communicate.ready?
+              end
+            end
+          end
+          # if interrupted above, just terminate immediately
+          return terminate(env) if env[:interrupted]
+
+          ui.info(I18n.t('vagrant_local.ssh_ready') + " in #{env[:metrics]['instance_ssh_time']} Seconds")
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-local/action.rb

@@ -0,0 +1,187 @@
+# frozen_string_literal: true
+
+require 'vagrant/action/builder'
+require 'log4r'
+
+module VagrantPlugins
+  module ProviderLocal
+    # Run actions against the machine
+    module Action
+      include Vagrant::Action::Builtin
+      @logger = Log4r::Logger.new('vagrant_local::action')
+
+      # This action is called to bring the box up from nothing.
+      def self.action_up
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |env, b2|
+            if env[:result]
+              env[:halt_on_error] = true
+              b2.use action_start
+            elsif !env[:result]
+              b2.use Create
+              b2.use Start
+              b2.use WaitTillBoot
+              b2.use Setup
+              b2.use WaitTillUp
+              b2.use Provision
+              b2.use SetHostname
+              b2.use SyncedFolders
+              b2.use SyncedFolderCleanup
+            end
+          end
+        end
+      end
+
+      # Assuming VM is created, just start it. This action is not called directly by any subcommand.
+      def self.action_start
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsState, :running do |env, b1|
+            if env[:result]
+              b1.use Message, I18n.t('vagrant_local.states.is_running')
+              next
+            end
+            b1.use Call, IsState, :uncleaned do |env2, b2|
+              b2.use Cleanup if env2[:result]
+            end
+            b1.use Start
+            b1.use WaitTillUp
+          end
+        end
+      end
+
+      def self.action_restart
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |_env, b2|
+            b2.use Call, IsState, :stopped do |env2, b3|
+              unless env2[:result]
+                b3.use WaitTillUp
+                b3.use Restart
+              end
+            end
+          end
+        end
+      end
+
+      def self.action_shutdown
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |_env, b2|
+            b2.use Call, IsState, :stopped do |env2, b3|
+              unless env2[:result]
+                b3.use WaitTillUp
+                b3.use Shutdown
+              end
+            end
+          end
+        end
+      end
+
+      # This is the action that is primarily responsible for halting the virtual machine.
+      def self.action_halt
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |env, b2|
+            unless env[:result]
+              b2.use NotCreated
+              next
+            end
+            b2.use Halt if env[:result]
+          end
+        end
+      end
+
+      # This action is called to SSH into the machine.
+      def self.action_ssh
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use SSHExec
+        end
+      end
+
+      def self.action_ssh_run
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use SSHRun
+        end
+      end
+
+      # This action is called when you try to package an existing virtual machine to an box image.
+      def self.action_package
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Package
+        end
+      end
+
+      # This is the action that is primarily responsible for completely freeing the resources of the underlying virtual machine.
+      def self.action_destroy
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |_env, b2|
+            # unless env[:result]
+            #  b2.use NotCreated
+            #  next
+            # end
+            b2.use Halt
+            b2.use Destroy
+          end
+        end
+      end
+
+      # This action is called when `vagrant provision` is called.
+      def self.action_provision
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |_env, b2|
+            b2.use Call, IsState, :running do |_env2, b3|
+              b3.use Provision
+            end
+          end
+        end
+      end
+
+      # This is the action implements the reload command It uses the halt and start actions
+      def self.action_reload
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use Call, IsCreated do |env, b2|
+            unless env[:result]
+              b2.use NotCreated
+              next
+            end
+            b2.use action_halt
+            b2.use action_start
+          end
+        end
+      end
+
+      def self.action_box_outdated
+        Builder.new.tap do |b|
+          b.use Builtin::BoxCheckOutdated
+        end
+      end
+
+      # This is the action that will remove a box given a name (and optionally
+      # a provider). This middleware sequence is built-in to Vagrant. Plugins
+      # can hook into this like any other middleware sequence.
+      def self.action_box_remove
+        Builder.new.tap do |b|
+          b.use Builtin::BoxRemove
+        end
+      end
+
+      action_root = Pathname.new(File.expand_path('action', __dir__))
+      autoload :Import, action_root.join('import')
+      autoload :Create, action_root.join('create')
+      autoload :Network, action_root.join('network')
+      autoload :Setup, action_root.join('setup')
+      autoload :Start, action_root.join('start')
+      autoload :NetworkingCleanup, action_root.join('network_cleanup')
+      autoload :IsCreated, action_root.join('is_created')
+      autoload :NotCreated, action_root.join('not_created')
+      autoload :CreateSnapshots, action_root.join('create_zfs_snapshots')
+      autoload :DeleteSnapshots, action_root.join('delete_zfs_snapshots')
+      autoload :ConfigureSnapshots, action_root.join('configure_zfs_snapshots')
+      autoload :Halt, action_root.join('halt')
+      autoload :Destroy, action_root.join('destroy')
+      autoload :WaitTillBoot, action_root.join('wait_till_boot')
+      autoload :WaitTillUp, action_root.join('wait_till_up')
+      autoload :Restart, action_root.join('restart')
+      autoload :Shutdown, action_root.join('shutdown')
+      autoload :PrepareNFSValidIds, action_root.join('prepare_nfs_valid_ids.rb')
+      autoload :Package, action_root.join('package.rb')
+    end
+  end
+end

data/lib/vagrant-local/command/guest_power_controls.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderLocal
+    module Command
+      # This is used to manage the power controls for the instance
+      class GuestPowerControls < 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(:restart) do
+            require File.expand_path('restart_guest', __dir__)
+            RestartGuest
+          end
+          @subcommands.register(:shutdown) do
+            require File.expand_path('shutdown_guest', __dir__)
+            ShutdownGuest
+          end
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?('-h') || @main_args.include?('--help')
+            # Print the help for all the vagrant-local 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 local control <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 local control <subcommand> -h`'
+          end
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-local/command/local.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+require 'vagrant-local/action'
+
+module VagrantPlugins
+  module ProviderLocal
+    module Command
+      # This is used manage the host where vagrant cannot
+      class Local < Vagrant.plugin('2', :command)
+        def self.synopsis
+          'Manage host and query host information'
+        end
+
+        def initialize(argv, env)
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+
+          @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-local 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 local <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 local <subcommand> -h`'
+          end
+
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-local/command/restart_guest.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderLocal
+    module Command
+      # This is used to restart the guest from inside the guest
+      class RestartGuest < Vagrant.plugin('2', :command)
+        def execute
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant local control restart [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: :local) do |machine|
+            machine.action('restart')
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderLocal
+    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 local 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: :local) do |machine|
+            machine.action('shutdown')
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module ProviderLocal
+    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 local 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: :local) 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-local/command/webvnc_console.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require 'resolv'
+module VagrantPlugins
+  module ProviderLocal
+    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 local 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: :local) 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-local/config.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require 'vagrant'
+## Do not Modify this File! Modify the Hosts.yml, Hosts.rb, or Vagrantfile!
+module VagrantPlugins
+  module ProviderLocal
+    # 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, :post_provision_boot, :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, :on_demand_vnics, :clean_shutdown_time, :dhcp4, :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, :login_wait
+
+      # rubocop:enable Layout/LineLength
+
+      def initialize
+        super
+        @brand = 'bhyve'
+        @additional_disks = UNSET_VALUE
+        @autoboot = true
+        @post_provision_boot = false
+        @kernel = nil
+        @boxshortname = UNSET_VALUE
+        @cdroms = nil
+        @shared_dir = nil
+        @os_type = 'generic'
+        @debug_boot = nil
+        @debug = nil
+        @consoleport = nil
+        @consolehost = '0.0.0.0'
+        @console_onboot = 'false'
+        @console = 'webvnc'
+        @memory = '2G'
+        @diskif = 'virtio-blk'
+        @netif = 'virtio-net-viona'
+        @cpus = 1
+        @boot = UNSET_VALUE
+        @setup_wait = 90
+        @box = UNSET_VALUE
+        @clean_shutdown_time = 300
+        @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
+        @login_wait = 5
+        @private_network = nil
+        @vm_type = 'production'
+        @setup_method = nil
+      end
+    end
+  end
+end