vagrant-hivemind 0.1.0

data/lib/vagrant/hivemind/commands/kill.rb

@@ -0,0 +1,77 @@
+require "optparse"
+
+require "vagrant/hivemind/constants"
+require "vagrant/hivemind/util"
+require "vagrant/hivemind/host"
+
+module Vagrant
+  module Hivemind
+    module Command
+      class Kill < Vagrant.plugin("2", :command)
+          include Vagrant::Hivemind::Constants
+          include Vagrant::Hivemind::Util
+
+        def self.synopsis
+          "Removes a Drone in the Hive"
+        end
+
+        def execute
+          options = {}
+
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant hivemind kill [options]"
+            o.separator ""
+            o.separator "Options:"
+            o.separator ""
+
+            o.on("-n", "--hostname HOSTNAME", "The hostname of the Drone (REQUIRED)") do |n|
+              options[:hostname] = n
+            end
+
+            o.on("-d", "--directory DIRECTORY", "Specify the directory where '#{HIVE_FILE}' is located (default: current directory)") do |d|
+              options[:directory] = []
+              options[:directory] << d
+            end
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          unless options[:hostname]
+            @env.ui.info opts.help
+            return 0
+          end
+
+          root_path = Path.get_root_path_from_options options
+
+          unless HiveFile.exist? root_path
+            @env.ui.error "There is no Hive file in the working directory."
+            return 1
+          end
+
+          hosts = HiveFile.read_from root_path
+
+          unless hosts.values.map(&:hostname).include? options[:hostname]
+            @env.ui.error "The specified hostname does not exist!"
+            return 1
+          end
+
+          host = hosts[options[:hostname]]
+
+          Vagrant::Hivemind::Util::Vagrantfile.generate_hivemind_vagrantfile @env, host, root_path
+
+          with_target_vms(options[:hostname]) do |vm|
+            action_env = vm.action(:destroy, force_confirm_destroy: true)
+          end
+
+          hosts.delete options[:hostname]
+          HiveFile.write_to hosts, root_path
+          @env.ui.info "Killed the Drone with hostname '#{options[:hostname]}'"
+
+          0
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant/hivemind/commands/list.rb

@@ -0,0 +1,112 @@
+require "optparse"
+
+require "vagrant/hivemind/constants"
+require "vagrant/hivemind/util"
+require "vagrant/hivemind/host"
+
+module Vagrant
+  module Hivemind
+    module Command
+      class List < Vagrant.plugin("2", :command)
+          include Vagrant::Hivemind::Constants
+          include Vagrant::Hivemind::Util
+
+        def self.synopsis
+          "Lists all the Drones in the Hive"
+        end
+
+        def execute
+          options = {}
+
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant hivemind list [options]"
+            o.separator ""
+            o.separator "Options:"
+            o.separator ""
+
+            o.on("-n", "--hostname", "List the Drones ordered by the hostname") do |n|
+              options[:hostname] = n
+            end
+
+            o.on("-a", "--ip-address", "List the Drones ordered by the IP address") do |a|
+              options[:ip_address] = a
+            end
+
+            o.on("-c", "--control", "List the Drones, Control Machines first") do |c|
+              options[:control] = c
+            end
+
+            o.on("-s", "--size", "List the Drones ordered by the Box Size") do |s|
+              options[:size] = s
+            end
+
+            o.on("-d", "--directory DIRECTORY", "Specify the directory where '#{HIVE_FILE}' is located (default: current directory)") do |d|
+              options[:directory] = []
+              options[:directory] << d
+            end
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          root_path = Path.get_root_path_from_options options
+
+          unless HiveFile.exist? root_path
+            @env.ui.error "There is no Hive file in the working directory."
+            return 1
+          end
+
+          hosts = HiveFile.read_from root_path
+
+          sorted_hosts = sort_hosts hosts, options
+
+          @env.ui.info "+----------------------+----------------+---+--------------+------------+---+"
+          @env.ui.info "| Hostname             | IP Address     | C | Size         | Box Type   | G |"
+          @env.ui.info "+----------------------+----------------+---+--------------+------------+---+"
+          sorted_hosts.values.each do |host|
+            hostname       = host.hostname
+            ip_address     = host.ip_address
+            is_control_y_n = host.is_control ? 'Y' : 'N'
+            box_size       = BOX_SIZES[host.box_size.to_sym][:name]
+            box_type       = BOX_TYPES[host.box_type.to_sym][:name]
+            is_gui_y_n     = BOX_TYPES[host.box_type.to_sym][:is_gui] ? 'Y' : 'N'
+            @env.ui.info "| #{'%-20.20s' % hostname} | #{'%-14.14s' % ip_address} | #{is_control_y_n} | #{'%-12.12s' % box_size} | #{'%-10.10s' % box_type} | #{is_gui_y_n} |"
+          end
+          @env.ui.info "+----------------------+----------------+---+--------------+------------+---+"
+          @env.ui.info ""
+
+          0
+        end
+
+        private
+          def sort_hosts(hosts, options = {})
+            sorted_hosts = hosts
+            options.keys.each do |key|
+              sorted_hosts = sort_hosts_by_hostname(hosts)   if key == :hostname
+              sorted_hosts = sort_hosts_by_ip_address(hosts) if key == :ip_address
+              sorted_hosts = sort_hosts_by_control(hosts)    if key == :control
+              sorted_hosts = sort_hosts_by_box_size(hosts)   if key == :size
+            end
+            sorted_hosts
+          end
+
+          def sort_hosts_by_hostname(hosts)
+            (hosts.sort_by { |k,v| v.hostname }).to_h
+          end
+
+          def sort_hosts_by_ip_address(hosts)
+            (hosts.sort_by { |k,v| v.ip_address }).to_h
+          end
+
+          def sort_hosts_by_control(hosts)
+            (hosts.sort_by { |k,v| v.is_control.to_s }.reverse).to_h
+          end
+
+          def sort_hosts_by_box_size(hosts)
+            (hosts.sort_by { |k,v| BOX_SIZES[v.box_size.to_sym][:memory_in_mb] }).to_h
+          end
+
+      end
+    end
+  end
+end

data/lib/vagrant/hivemind/commands/morph.rb

@@ -0,0 +1,166 @@
+require "optparse"
+
+require "vagrant/hivemind/constants"
+require "vagrant/hivemind/util"
+require "vagrant/hivemind/host"
+
+module Vagrant
+  module Hivemind
+    module Command
+      class Morph < Vagrant.plugin("2", :command)
+          include Vagrant::Hivemind::Constants
+          include Vagrant::Hivemind::Util
+
+        def self.synopsis
+          "Changes the settings of a Drone in the Hive"
+        end
+
+        def execute
+          options = {}
+
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant hivemind morph [options]"
+            o.separator ""
+            o.separator "Options:"
+            o.separator ""
+
+            o.on("-n", "--hostname HOSTNAME", "The hostname of the Drone (REQUIRED)") do |n|
+              options[:hostname] = n
+            end
+
+            o.on("-a", "--ip-address IPADDRESS", "The new IP address of the Drone") do |a|
+              options[:ip_address] = a
+            end
+
+            o.on("-c", "--control", "Promote the Drone to be a Control Machine") do |c|
+              options[:control] = c
+            end
+
+            o.on("-s", "--size SIZE", BOX_SIZES.keys.map(&:to_s), "The new Box Size of the Drone [:extra_small, :small, :medium, :large, :extra_large] (default: :small)") do |s|
+              options[:size] = s
+            end
+
+            o.on("-p", "--forwarded-port PORTPAIR", "Forward GUESTPORT to HOSTPORT (the format is GUESTPORT:HOSTPORT)") do |p|
+              options[:forwarded_port] = p
+            end
+
+            o.on("-d", "--directory DIRECTORY", "Specify the directory where '#{HIVE_FILE}' is located (default: current directory)") do |d|
+              options[:directory] = []
+              options[:directory] << d
+            end
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          root_path = Path.get_root_path_from_options options
+
+          unless options[:hostname]
+            @env.ui.info opts.help
+            return 0
+          end
+
+          unless HiveFile.exist? root_path
+            @env.ui.error "There is no Hive file in the working directory."
+            return 1
+          end
+
+          hosts = HiveFile.read_from root_path
+
+          unless hosts.values.map(&:hostname).include? options[:hostname]
+            @env.ui.error "The specified hostname does not exist!"
+            return 1
+          end
+
+          if options[:ip_address]
+            validation_error = is_valid_ip_address?(options[:ip_address], hosts)
+            if validation_error
+              @env.ui.error validation_error
+              return 1
+            end
+          end
+
+          if options[:forwarded_port]
+            validation_error = is_valid_forwarded_port?(options[:forwarded_port], hosts)
+            if validation_error
+              @env.ui.error validation_error
+              return 1
+            end
+          end
+
+          host = hosts[options[:hostname]]
+
+          if options[:ip_address]
+            host.ip_address = options[:ip_address]
+          end
+
+          if options[:control]
+            host.is_control = true
+          end
+
+          if options[:size]
+            host.box_size = options[:size]
+          end
+
+          if options[:forwarded_port]
+            guest_port, host_port = Network.port_pair_to_i(options[:forwarded_port])
+            host.forwarded_ports ||= []
+
+            port_pair = Network.get_host_port_pair_with_guest_port(guest_port, host)
+            if port_pair
+              port_pair["host_port"] = host_port
+            else
+              port_pair = {
+                "guest_port" => guest_port,
+                "host_port"  => host_port
+              }
+              host.forwarded_ports << port_pair
+            end
+          end
+
+          hosts.delete options[:hostname]
+          drone = {
+            options[:hostname] => host
+          }
+          HiveFile.write_to hosts.merge(drone), root_path
+          @env.ui.info "Morphed the Drone with hostname '#{options[:hostname]}'"
+
+          0
+        end
+
+        private
+          def is_valid_ip_address?(ip_address, hosts)
+            if !Network.is_valid_ip_address? ip_address
+              return "Invalid IP address format!"
+            end
+            if Network.get_network(ip_address) != Network.get_network(PRIVATE_NETWORK)
+              return "The specified IP address does not belong to the private network of the Hive!"
+            end
+            if hosts.values.map(&:ip_address).include? ip_address
+              return "The specified IP address is already used!"
+            end
+            return nil
+          end
+
+          def is_valid_forwarded_port?(port_pair, hosts)
+            if !Network.is_valid_port_pair? port_pair
+              return "Invalid port pair format!"
+            end
+
+            guest_port, host_port = Network.port_pair_to_i(port_pair)
+            if !Network.is_valid_port_value? guest_port
+              return "Guest port is out of range!"
+            end
+            if !Network.is_valid_port_value? host_port
+              return "Host port is out of range!"
+            end
+
+            if Network.get_host_keys_using_host_port(host_port, hosts).size > 0
+              return "Host port is already used!"
+            end
+          end
+
+      end
+    end
+  end
+end

data/lib/vagrant/hivemind/commands/root.rb

@@ -0,0 +1,90 @@
+require "optparse"
+
+module Vagrant
+  module Hivemind
+    module Command
+      class Root < Vagrant.plugin("2", :command)
+        def self.synopsis
+          "Hivemind operations: init list desc spawn kill morph up halt"
+        end
+
+        def initialize(argv, env)
+          super
+
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+
+          @subcommands.register(:init) do
+            require_relative "init"
+            Init
+          end
+
+          @subcommands.register(:list) do
+            require_relative "list"
+            List
+          end
+
+          @subcommands.register(:desc) do
+            require_relative "desc"
+            Desc
+          end
+
+          @subcommands.register(:spawn) do
+            require_relative "spawn"
+            Spawn
+          end
+
+          @subcommands.register(:kill) do
+            require_relative "kill"
+            Kill
+          end
+
+          @subcommands.register(:morph) do
+            require_relative "morph"
+            Morph
+          end
+
+          @subcommands.register(:up) do
+            require_relative "up"
+            Up
+          end
+
+          @subcommands.register(:halt) do
+            require_relative "halt"
+            Halt
+          end
+
+        end
+
+        def execute
+          if @main_args.include?("-h") || @main_args.include?("--help")
+            return help
+          end
+
+          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+          return help if !command_class || !@sub_command
+
+          command_class.new(@sub_args, @env).execute
+        end
+
+        def help
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant hivemind <command> [<args>]"
+            o.separator ""
+            o.separator "Available commands:"
+
+            @subcommands.each do |key, data|
+              o.separator "#{'%8.6s' % key.to_s}  #{'%.60s' % data.synopsis}"
+            end
+
+            o.separator ""
+            o.separator "For help on any individual command run `vagrant hivemind <command> -h`"
+          end
+
+          @env.ui.info(opts.help, prefix: false)
+        end
+      end
+    end
+  end
+end