vagrant-invade 0.0.1 → 0.2.0

data/lib/vagrant-invade/builder/provision/puppet.rb

@@ -0,0 +1,47 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module Provision
+
+        require 'erubis'
+
+        class Puppet
+
+          attr_reader :result
+          attr_accessor :machine_name, :puppet_data
+
+          def initialize(machine_name, puppet_data, result: nil)
+            @machine_name = machine_name
+            @puppet_data  = puppet_data
+            @result   = result
+          end
+
+          def build
+            b = binding
+            template_file = "#{TEMPLATE_PATH}/provision/puppet.erb"
+
+            begin
+
+              # Get machine name
+              machine_name = @machine_name
+
+              # Values for provider section
+              name = @puppet_data['name']
+              type = @puppet_data['type']
+              cpus = @puppet_data['cores']
+              memory = @puppet_data['memory']
+              nicspeed = @puppet_data['nicspeed']
+              natdns = @puppet_data['natdns']
+
+              eruby = Erubis::Eruby.new(File.read(template_file))
+              @result = eruby.result b
+            rescue TypeError, SyntaxError, SystemCallError => e
+              raise(e)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/provision/shell.rb

@@ -0,0 +1,45 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module Provision
+
+        require 'erubis'
+
+        class Shell
+
+          attr_reader :result
+          attr_accessor :machine_name, :shell_data
+
+          def initialize(machine_name, shell_data, result: nil)
+            @machine_name = machine_name
+            @shell_data  = shell_data
+            @result   = result
+          end
+
+          def build
+            b = binding
+            template_file = "#{TEMPLATE_PATH}/provision/shell.erb"
+
+            begin
+
+              # Get machine name
+              machine_name = @machine_name
+
+              # Values for shell provision section
+              name = @shell_data['name']
+              path = @shell_data['path']
+              binary = @shell_data['binary']
+              privileged = @shell_data['privileged']
+
+              eruby = Erubis::Eruby.new(File.read(template_file))
+              @result = eruby.result b
+            rescue TypeError, SyntaxError, SystemCallError => e
+              raise(e)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/provision/shell_inline.rb

@@ -0,0 +1,45 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module Provision
+
+        require 'erubis'
+
+        class Shell
+
+          attr_reader :result
+          attr_accessor :machine_name, :shell_inline_data
+
+          def initialize(machine_name, shell_inline_data, result: nil)
+            @machine_name = machine_name
+            @shell_inline_data  = shell_inline_data
+            @result   = result
+          end
+
+          def build
+            b = binding
+            template_file = "#{TEMPLATE_PATH}/provision/shell_inline.erb"
+
+            begin
+
+              # Get machine name
+              machine_name = @machine_name
+
+              # Values for shell provision section
+              name = @shell_inline_data['name']
+              inline = @shell_inline_data['inline']
+              binary = @shell_inline_data['binary']
+              privileged = @shell_inline_data['privileged']
+
+              eruby = Erubis::Eruby.new(File.read(template_file))
+              @result = eruby.result b
+            rescue TypeError, SyntaxError, SystemCallError => e
+              raise(e)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/provision.rb

@@ -0,0 +1,12 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module Provision
+
+        autoload :Shell, 'vagrant-invade/builder/provision/shell.rb'
+        autoload :Puppet, 'vagrant-invade/builder/provision/puppet.rb'
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/synced_folder/nfs.rb

@@ -0,0 +1,48 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module SyncedFolder
+
+        require 'erubis'
+
+        class NFS
+
+          attr_reader :result
+          attr_accessor :machine_name, :nfs_data
+
+          def initialize(machine_name, nfs_data, result: nil)
+            @machine_name = machine_name
+            @nfs_data  = nfs_data
+            @result = result
+          end
+
+          def build
+            b = binding
+            template_file = "#{TEMPLATE_PATH}/synced_folder/nfs.erb"
+
+            begin
+
+              # Get machine name
+              machine_name = @machine_name
+
+              # Values for provider sections
+              enabled = !@nfs_data['enabled'] # negated because Vagrant asks if it is disabled
+              uid = @nfs_data['uid']
+              gid = @nfs_data['gid']
+              source = @nfs_data['source']
+              path = @nfs_data['path']
+              mount_options = @nfs_data['mount_options']
+
+              eruby = Erubis::Eruby.new(File.read(template_file))
+              @result = eruby.result b
+            rescue TypeError, SyntaxError, SystemCallError => e
+              raise(e)
+            end
+          end
+
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/synced_folder/virtualbox.rb

@@ -0,0 +1,48 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module SyncedFolder
+
+        require 'erubis'
+
+        class VirtualBox
+
+          attr_reader :result
+          attr_accessor :machine_name, :virtualbox_data
+
+          def initialize(machine_name, virtualbox_data, result: nil)
+            @machine_name = machine_name
+            @virtualbox_data  = virtualbox_data
+            @result   = result
+          end
+
+          def build
+            b = binding
+            template_file = "#{TEMPLATE_PATH}/synced_folder/virtualbox.erb"
+
+            begin
+
+              # Get machine name
+              machine_name = @machine_name
+
+              # Values for provider section
+              name = @virtualbox_data['name']
+              type = @virtualbox_data['type']
+              cpus = @virtualbox_data['cpus']
+              memory = @virtualbox_data['memory']
+              nicspeed = @virtualbox_data['nicspeed']
+              natdns = @virtualbox_data['natdns']
+
+              eruby = Erubis::Eruby.new(File.read(template_file))
+              @result = eruby.result b
+            rescue TypeError, SyntaxError, SystemCallError => e
+              raise(e)
+            end
+          end
+
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/synced_folder.rb

@@ -0,0 +1,12 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+      module SyncedFolder
+
+        autoload :VirtualBox, 'vagrant-invade/builder/synced_folder/virtualbox.rb'
+        autoload :NFS, 'vagrant-invade/builder/synced_folder/nfs.rb'
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/vagrantfile.rb

@@ -0,0 +1,35 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+
+      require 'erubis'
+
+      class Vagrantfile
+
+        attr_reader :result
+        attr_accessor :definitions
+
+        def initialize(definitions, result: nil)
+          @definitions  = definitions
+          @result   = result
+        end
+
+        def build
+          b = binding
+          template_file = "#{TEMPLATE_PATH}/v2.erb"
+
+          begin
+
+            # definitions for vagrantfile to build
+            definitions = @definitions
+
+            eruby = Erubis::Eruby.new(File.read(template_file))
+            @result = eruby.result b
+          rescue TypeError, SyntaxError, SystemCallError => e
+            raise(e)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder/vm.rb

@@ -0,0 +1,41 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+
+      require 'erubis'
+
+      class VM
+
+        attr_reader :result
+        attr_accessor :machine_name, :vm_data
+
+        def initialize(machine_name, vm_data, result: nil)
+          @machine_name = machine_name
+          @vm_data  = vm_data
+          @result   = result
+        end
+
+        def build
+          b = binding
+          template_file = "#{TEMPLATE_PATH}/vm/vm.erb"
+
+          begin
+
+            # Get machine name
+            machine_name = @machine_name
+
+            # Values for vm section
+            box  = @vm_data['box']
+            url   = @vm_data['url']
+            hostname = @vm_data['hostname']
+
+            eruby = Erubis::Eruby.new(File.read(template_file))
+            @result = eruby.result b
+          rescue TypeError, SyntaxError, SystemCallError => e
+            raise(e)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/builder.rb

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module Invade
+    module Builder
+
+      autoload :VM, 'vagrant-invade/builder/vm'
+      autoload :Network, 'vagrant-invade/builder/network'
+      autoload :Provider, 'vagrant-invade/builder/provider'
+      autoload :Provision, 'vagrant-invade/builder/provision'
+      autoload :SyncedFolder, 'vagrant-invade/builder/synced_folder'
+      autoload :Definition, 'vagrant-invade/builder/definition'
+      autoload :Vagrantfile, 'vagrant-invade/builder/vagrantfile'
+
+      TEMPLATE_PATH = File.join(File.dirname(__FILE__), 'template')
+
+    end
+  end
+end

data/lib/vagrant-invade/command/base.rb

@@ -0,0 +1,19 @@
+require "vagrant/plugin/state_file"
+
+module VagrantPlugins
+  module Invade
+    module Command
+      class Base < Vagrant.plugin('2', :command)
+        # This is a helper for executing an action sequence with the proper
+        # environment hash setup so that the plugin specific helpers are
+        # in.
+        #
+        # @param [Object] callable the Middleware callable
+        # @param [Hash] env Extra environment hash that is merged in.
+        def action(callable, env=nil)
+          @env.action_runner.run(callable, env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/command/build.rb

@@ -0,0 +1,37 @@
+require 'optparse'
+require_relative 'base'
+
+module VagrantPlugins
+  module Invade
+    module Command
+      class Build < Base
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant invade build [-f|--force] [-q|--quiet] [-h]"
+            o.separator ""
+            o.on("-f", "--force", "Overwrite existing Vagrantfile") do |f|
+              options[:force] = f
+            end
+            o.on("-q", "--quiet", "No verbose output.") do |q|
+              options[:quiet] = q
+            end
+          end
+
+          # Parse the options
+          argv = parse_options(opts)
+          return if !argv
+
+          # Validates InVaDE configuration
+          action(Action.build, {
+            :invade_command_force => options[:force],
+            :invade_command_quiet => options[:quiet]
+          })
+
+          # Success, exit status 0
+          0
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/command/root.rb

@@ -0,0 +1,73 @@
+require 'optparse'
+
+module VagrantPlugins
+  module Invade
+    module Command
+      class Root < Vagrant.plugin('2', :command)
+        def self.synopsis
+          'manages invade: validate, build, etc.'
+        end
+
+        def initialize(argv, env)
+          super
+
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+          @subcommands = Vagrant::Registry.new
+
+          # VALIDATE COMMAND ("vagrant invade validate")
+          @subcommands.register(:validate) do
+            require_relative "validate"
+            Validate
+          end
+
+          # BUILD COMMAND ("vagrant invade build")
+            @subcommands.register(:build) do
+            require_relative "build"
+            Build
+          end
+        end
+
+        def execute
+          if @main_args.include?("-h") || @main_args.include?("--help")
+            # Print the help for all the sub-commands.
+            return help
+          end
+
+        # If we reached this far then we must have a subcommand. If not,
+          # then we also just print the help and exit.
+          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
+
+        # Prints the help out for this command
+        def help
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant invade <command> [<args>]"
+            o.separator ""
+            o.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|
+              o.separator "     #{key}"
+            end
+
+            o.separator ""
+            o.separator "For help on any individual command run `vagrant invade COMMAND -h`"
+          end
+
+          @env.ui.info(opts.help, prefix: false)
+        end
+      end
+
+    end
+
+  end
+end

data/lib/vagrant-invade/command/validate.rb

@@ -0,0 +1,37 @@
+require 'optparse'
+require_relative 'base'
+
+module VagrantPlugins
+  module Invade
+    module Command
+      class Validate < Base
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant invade validate [-f|--force] [-q|--quiet] [-h]"
+            o.separator ""
+            o.on("-f", "--force", "Overwrite existing Vagrantfile") do |f|
+              options[:force] = f
+            end
+            o.on("-q", "--quiet", "No verbose output.") do |q|
+              options[:quiet] = q
+            end
+          end
+
+          # Parse the options
+          argv = parse_options(opts)
+          return if !argv
+
+          # Validates InVaDE configuration
+          action(Action.validate, {
+            :invade_command_force => options[:force],
+            :invade_command_quiet => options[:quiet]
+          })
+
+          # Success, exit status 0
+          0
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/generator/definition.rb

@@ -0,0 +1,25 @@
+module VagrantPlugins
+  module Invade
+    module Generator
+
+      class Definition
+
+        attr_accessor :definition_data
+
+        def initialize(machine_name, definition_data)
+            @machine_name = machine_name
+            @definition_data = definition_data
+        end
+
+        def generate
+            definition = Builder::Definition.new(@machine_name, @definition_data)
+            definition.build
+
+            definition.result
+        end
+
+      end
+
+    end
+  end
+end

data/lib/vagrant-invade/generator/section/network.rb

@@ -0,0 +1,38 @@
+module VagrantPlugins
+  module Invade
+    module Generator
+      module Section
+
+        class Network
+
+          attr_accessor :machine_name, :type, :network_data
+
+          def initialize(machine_name, type, network_data)
+            @machine_name = machine_name
+            @type = type
+            @network_data = network_data
+          end
+
+          def generate
+
+            case @type
+            when 'private', 'private_network', 'privatenetwork', 'private-network'
+              network = Builder::Network::PrivateNetwork.new(@machine_name, @network_data)
+            when 'forwarded', 'forwarded_port', 'forwarded-port', 'forwardedport', 'port'
+              network = Builder::Network::ForwardedPort.new(@machine_name, @network_data)
+            when 'public', 'puplic_network', 'publicnetwork', 'public-network'
+              network = Builder::Network::PublicNetwork.new(@machine_name, @network_data)
+            else
+              raise StandardError, "Network type unknown or not set. Please check the network configuration."
+            end
+
+            network.build
+
+            network.result
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-invade/generator/section/provider.rb

@@ -0,0 +1,36 @@
+module VagrantPlugins
+  module Invade
+    module Generator
+      module Section
+
+        class Provider
+
+          attr_accessor :machine_name, :type, :provider_data
+
+          def initialize(machine_name, type, provider_data)
+            @machine_name = machine_name
+            @type = type
+            @provider_data = provider_data
+          end
+
+          def generate
+            case @type
+            when 'virtualbox'
+              provider = Builder::Provider::VirtualBox.new(@machine_name, @provider_data)
+            when 'vmware'
+              provider = Builder::Provider::VMware.new(@machine_name, @provider_data)
+            else
+              raise StandardError, "Provider unknown or not set. Please check the provider configuration."
+            end
+
+            provider.build
+
+            provider.result
+          end
+
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-invade/generator/section/provision.rb

@@ -0,0 +1,38 @@
+module VagrantPlugins
+  module Invade
+    module Generator
+      module Section
+
+        class Provision
+
+          attr_accessor :machine_name, :type, :provision_data
+
+          def initialize(machine_name, type, provision_data)
+            @machine_name = machine_name
+            @type = type
+            @provision_data = provision_data
+          end
+
+          def generate
+            case @type
+            when 'shell'
+              provision = Builder::Provision::Shell.new(@machine_name, @provision_data)
+            when 'shell_inline', 'inline', 'shellinline'
+              provision = Builder::Provision::ShellInline.new(@machine_name, @provision_data)
+            when 'puppet'
+              provision = Builder::Provision::Puppet.new(@machine_name, @provision_data)
+            else
+              raise StandardError, "Provisioner unknown or not set. Please check the provision configuration."
+            end
+
+            provision.build
+
+            provision.result
+          end
+
+        end
+
+      end
+    end
+  end
+end