vagrant-arubacloud 0.0.2dev

data/lib/vagrant-arubacloud/action/read_ssh_info.rb

@@ -0,0 +1,43 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ArubaCloud
+    module Action
+
+      # This action reads the SSH info for the machine and puts it into
+      # ':machine_ssh_info' key in the environment
+      class ReadSSHInfo
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_arubacloud::action::read_ssh_info')
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:arubacloud_compute], env[:machine])
+          @app.call(env)
+        end
+
+        def read_ssh_info(arubacloud, machine)
+          return nil if machine.id.nil?
+
+          # Find the machine
+          server = arubacloud.servers.get(machine.id)
+          if server.nil?
+            # The machine can't be found
+            @logger.info("'Machine couldn't be found, assuming it got destroyed.")
+            machine.id = nil
+            nil
+          else
+            @logger.info("read_ssh_info: server.smart_ipv4: #{server.smart_ipv4}")
+            # Return the server object
+            {
+                :host => server.get_public_ip,
+                :port => 22,
+                :username => 'root'
+            }
+          end
+        end # read_ssh_info
+      end # ReadSSHInfo
+    end # Action
+  end # ArubaCloud
+end # VagrantPlugins

data/lib/vagrant-arubacloud/action/read_state.rb

@@ -0,0 +1,46 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ArubaCloud
+    module Action
+      # This action reads the state of the machine and puts it in the
+      # `:machine_state_id` key in the environment.
+      class ReadState
+        def initialize(app, env)
+          @app    = app
+          @env    = env
+          @logger = Log4r::Logger.new('vagrant_arubacloud::action::read_state')
+        end
+
+        def call(env)
+          # env[:ui].output(":machine_state_id: #{env[:machine_state_id]}")
+          # env[:ui].output(":arubacloud_compute: #{env[:arubacloud_compute]}")
+          # env[:ui].output(":machine: #{env[:machine]}")
+          env[:machine_state_id] = read_state(env[:arubacloud_compute], env[:machine])
+          @app.call(env)
+        end
+
+        def read_state(arubacloud, machine)
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = arubacloud.servers.get(machine.id)
+          unless server.instance_of? Fog::Compute::ArubaCloud::Server
+            msg = "VagrantPlugins::ArubaCloud::Action::ReadState.read_state, 'server' must be Fog::Compute::ArubaCloud::Server, got: #{server.class}"
+            @logger.critical("#{msg}")
+          end
+          if server.nil? || server.state == Fog::Compute::ArubaCloud::Server::DELETED
+            # The machine can't be found
+            @logger.info('Machine not found or deleted, assuming it got destroyed.')
+            machine.id = nil
+            return :not_created
+          end
+
+          @logger.debug("VagrantPlugins::ArubaCloud::Action::ReadState.read_state, server state : #{server.state}")
+          # Return the state
+          server.state
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-arubacloud/action.rb

@@ -0,0 +1,163 @@
+require 'vagrant/action/builder'
+require 'vagrant/action'
+
+module VagrantPlugins
+  module ArubaCloud
+    module Action
+
+      # Access top level stuff
+      include Vagrant::Action::Builtin
+
+      def self.action_destroy
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b1|
+            unless env[:result]
+              b1.use MessageNotCreated
+              next
+            end
+
+            b1.use Call, DestroyConfirm do |env1, b2|
+              if env1[:result]
+                b2.use ConnectArubaCloud
+                b2.use HaltServer
+                b2.use DeleteServer
+              else
+                b2.use Message, ' The server will not be deleted.'
+                next
+              end
+            end
+          end
+        end
+      end
+
+      def self.action_halt
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate #shame on you
+          b.use Call, IsCreated do |env, b1|
+            unless env[:result]
+              b1.use MessageNotCreated
+              next
+            end
+            b1.use ConnectArubaCloud
+            b1.use HaltServer
+          end
+        end
+      end
+
+      def self.action_provision
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            unless env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+            # Use our custom provisioning class
+            b2.use ArubaProvision
+            b2.use SyncedFolders
+          end
+        end
+      end
+
+      #def self.action_reload
+        #Vagrant::Action::Builder.new.tap do |b|
+
+        #end
+      #end
+
+      def self.action_read_ssh_info
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectArubaCloud
+          b.use ReadSSHInfo
+        end
+      end
+
+      # This action is called to read the state of the machine. The
+      # resulting state is expected to be put into the `:machine_state_id`
+      # key.
+      def self.action_read_state
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectArubaCloud
+          b.use ReadState
+        end
+      end
+
+      def self.action_ssh
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            unless env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+            b2.use SSHExec
+          end
+        end
+      end
+
+      def self.action_ssh_run
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            unless env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use SSHRun
+          end
+        end
+      end
+
+      def self.action_up
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if env[:result]
+              b2.use MessageAlreadyCreated
+              next
+            end
+
+            b2.use ConnectArubaCloud
+            # b2.use RunInitScript # temporarily removed
+            b2.use CreateServer
+            b2.use WaitForCommunicator
+          end
+        end
+      end
+
+      def self.action_list_servers
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConnectArubaCloud
+          b.use ListServers
+        end
+      end
+
+      def self.action_list_templates
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConnectArubaCloud
+          b.use ListTemplates
+        end
+      end
+
+      # The autoload farm
+      action_root = Pathname.new(File.expand_path('../action', __FILE__))
+      autoload :ConnectArubaCloud, action_root.join('connect_arubacloud')
+      autoload :CreateServer, action_root.join('create_server')
+      autoload :HaltServer, action_root.join('halt_server')
+      autoload :DeleteServer, action_root.join('delete_server')
+      autoload :IsCreated, action_root.join('is_created')
+      autoload :MessageNotCreated, action_root.join('message_not_created')
+      autoload :MessageAlreadyCreated, action_root.join('message_already_created')
+      autoload :ReadSSHInfo, action_root.join('read_ssh_info')
+      autoload :ReadState, action_root.join('read_state')
+      # autoload :RunInitScript, action_root.join("run_init_script")
+      autoload :ListServers, action_root.join('list_servers')
+      autoload :ListTemplates, action_root.join('list_templates')
+      autoload :ArubaProvision, action_root.join('aruba_provision')
+    end
+  end
+end

data/lib/vagrant-arubacloud/cap/disable_requiretty.rb

@@ -0,0 +1,13 @@
+module Cap
+  class DisableRequireTty
+    def self.disable_requiretty(machine)
+      output = ''
+      command = 'sed -i "s/^.*requiretty/#Defaults requiretty/" /etc/sudoers'
+
+      machine.communicate.execute(command) do |type, data|
+        output += data if type == :stdout
+      end
+      output.chomp!
+    end
+  end
+end

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

@@ -0,0 +1,67 @@
+require 'vagrant-arubacloud/action'
+
+module VagrantPlugins
+  module ArubaCloud
+    module Command
+      class Root < Vagrant.plugin('2', :command)
+        def self.synopsis
+          'query ArubaCloud for servers and templates'
+        end
+
+        def initialize(argv, env)
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+          @env = env
+
+          @subcommands = Vagrant::Registry.new
+          @subcommands.register(:templates) do
+            require File.expand_path('../templates', __FILE__)
+            Templates
+          end
+          @subcommands.register(:servers) do
+            require File.expand_path('../servers', __FILE__)
+            Servers
+          end
+
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?('-h') || @main_args.include?('--help')
+            return help
+          end
+
+          # Set command_class to default nil
+          command_class = nil
+          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
+          return help if command_class.nil? || !@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 |opts|
+            opts.banner = 'Usage: vagrant arubacloud <subcommand> [<args>]'
+            opts.separator ''
+            opts.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|
+              opts.separator "     #{key}"
+            end
+
+            opts.separator ''
+            opts.separator 'For help on any individual subcommand run `vagrant arubacloud <subcommand> -h`'
+          end
+
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-arubacloud/command/servers.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ArubaCloud
+    module Command
+      class Servers < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant arubacloud templates [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          with_target_vms(argv, :provider => :arubacloud) do |machine|
+            machine.action('list_servers')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-arubacloud/command/templates.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ArubaCloud
+    module Command
+      class Templates < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant arubacloud templates [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          with_target_vms(argv, :provider => :arubacloud) do |machine|
+            machine.action('list_templates')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-arubacloud/config.rb

@@ -0,0 +1,124 @@
+require 'vagrant'
+require 'fog'
+
+module VagrantPlugins
+  module ArubaCloud
+
+    class Config < Vagrant.plugin('2', :config)
+      # ArubaCloud Username
+      # @return [String]
+      attr_accessor :arubacloud_username
+
+      # ArubaCloud Password
+      # @return [String]
+      attr_accessor :arubacloud_password
+
+      # Ws EndPoint Url
+      # Expected to be the url of the web service to use
+      # @return [String]
+      attr_accessor :url
+
+      # The name of the server. This defaults to the name of the machine
+      # defined by Vagrant (via 'config.vm.define'), but can be override here.
+      # @return [String]
+      attr_accessor :server_name
+
+      # The ID of the template to use, use vagrant arubacloud templates to obtain
+      # the complete list.
+      # @return [Integer]
+      attr_accessor :template_id
+
+      # The smart vm type expressed in ID:
+      # 1 = small
+      # 2 = medium
+      # 3 = large
+      # 4 = extra-large
+      # @return [Integer]
+      attr_accessor :package_id
+
+      # The admin password of the vm (root user) .
+      # @return [String]
+      attr_accessor :admin_password
+
+      # Service Type expressed in ID [Integer]:
+      # 1 = Pro Hyper-V
+      # 2 = Pro VMWare
+      # 3 = Pro Hyper-V Low Cost
+      # 4 = Smart
+      # @return [Integer]
+      attr_accessor :service_type
+
+      # Number of Virtual CPU to be assigned to the VM
+      # Pro VMWare:  1 < n < 8
+      # Pro Hyper-V: 1 < n < 4
+      # @return [Integer]
+      attr_accessor :cpu_number
+
+      # Amount of GB of RAM to be assigned to the VM
+      # n <= 16
+      # @return [Integer]
+      attr_accessor :ram_qty
+
+      # Array containing hard disk Configuration
+      # Example configuration (size is expressed in GB):
+      # Hds = [{:type => 0, :size => 100}, {:type => 1, :size => 200}]
+      # Hd type 0 is required because specify the first hard disk, max size per hd: 500 GB
+      # Hd type > 0 < 4 are 3 additional hard disks (optional)
+      # @return [Array]
+      attr_accessor :hds
+
+      def initialize
+        @arubacloud_username = UNSET_VALUE
+        @arubacloud_password = UNSET_VALUE
+        @admin_password = UNSET_VALUE
+        @url = UNSET_VALUE
+        @server_name = UNSET_VALUE
+        @template_id = UNSET_VALUE
+        @package_id = UNSET_VALUE
+        @service_type = UNSET_VALUE
+        @cpu_number = UNSET_VALUE
+        @ram_qty = UNSET_VALUE
+        @hds = UNSET_VALUE
+      end
+
+      def finalize!
+        @arubacloud_username = nil if @arubacloud_username == UNSET_VALUE
+        @arubacloud_password = nil if @arubacloud_password == UNSET_VALUE
+        @admin_password = nil if @admin_password == UNSET_VALUE
+        @url = nil if @url == UNSET_VALUE
+        @server_name = nil if @server_name == UNSET_VALUE
+        @template_id = nil if @template_id == UNSET_VALUE
+        @package_id = nil if @package_id == UNSET_VALUE
+        @service_type = nil if @service_type == UNSET_VALUE
+        @cpu_number = nil if @cpu_number == UNSET_VALUE
+        @ram_qty = nil if @ram_qty == UNSET_VALUE
+        @hds = nil if @hds == UNSET_VALUE
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        # Global configurations needed by all service types
+        errors << I18n.t('vagrant_arubacloud.config.arubacloud_username_required') unless @arubacloud_username
+        errors << I18n.t('vagrant_arubacloud.config.arubacloud_password_required') unless @arubacloud_password
+        errors << I18n.t('vagrant_arubacloud.config.admin_password_required') unless @admin_password
+        errors << I18n.t('vagrant_arubacloud.config.template_id_required') unless @template_id
+        errors << I18n.t('vagrant_arubacloud.config.service_type_required') unless @service_type
+
+        if @service_type.eql? 4
+          errors << I18n.t('vagrant_arubacloud.config.package_id_required') unless @package_id
+        else
+          errors << I18n.t('vagrant_arubacloud.config.cpu_number_required') unless @cpu_number
+          errors << I18n.t('vagrant_arubacloud.config.ram_qty_required') unless @ram_qty
+          if @hds
+            errors << I18n.t('vagrant_arubacloud.config.hds_conf_must_be_array') unless @hds.kind_of?(Array)
+          elsif
+            errors << I18n.t('vagrant_arubacloud.config.hds_conf_required')
+          end
+        end
+        {'ArubaCloud Provider' => errors}
+      end
+
+    end # Config
+  end # ArubaCloud
+end # VagrantPlugins

data/lib/vagrant-arubacloud/errors.rb

@@ -0,0 +1,19 @@
+require 'vagrant'
+
+module VagrantPlugins
+  module ArubaCloud
+    module Errors
+
+      class ArubaCloudError < Vagrant::Errors::VagrantError
+        error_namespace('vagrant_aruba_cloud.errors')
+      end
+
+      class MachineAlreadyPresent < ArubaCloudError
+      end
+
+      class BadServerResponse < ArubaCloudError
+      end
+
+    end
+  end
+end

data/lib/vagrant-arubacloud/plugin.rb

@@ -0,0 +1,48 @@
+begin
+  require 'vagrant'
+rescue LoadError
+  raise 'The ArubaCloud IaaS provider must be run within Vagrant.'
+end
+
+if Vagrant::VERSION < '1.2.0'
+  raise 'ArubaCloud IaaS provider is only compatible with Vagrant 1.2+'
+end
+
+module VagrantPlugins
+  module ArubaCloud
+    class Plugin < Vagrant.plugin('2')
+      # noinspection RubyArgCount
+      name 'ArubaCloud'
+      description <<-DESC
+      This plugin enable Vagrant to manage machines in ArubaCloud IaaS service.
+      DESC
+
+      config(:arubacloud, :provider) do
+        require_relative 'config'
+        Config
+      end
+
+      provider(:arubacloud, { :box_optional => true, :parallel => true }) do
+        ArubaCloud.init_i18n
+        ArubaCloud.init_logging
+
+        require_relative 'provider'
+        Provider
+      end
+
+      command('arubacloud') do
+        ArubaCloud.init_i18n
+        ArubaCloud.init_logging
+
+        require_relative 'command/root'
+        Command::Root
+      end
+
+      # Disable require tty for centOS
+      guest_capability 'redhat', 'disable_requiretty' do
+        require_relative 'cap/disable_requiretty'
+        Cap::DisableRequireTty
+      end
+    end # Plugin
+  end # ArubaCloud
+end # VagrantPlugins

data/lib/vagrant-arubacloud/provider.rb

@@ -0,0 +1,51 @@
+require 'vagrant'
+
+require 'vagrant-arubacloud/action'
+
+
+module VagrantPlugins
+  module ArubaCloud
+    class Provider < Vagrant.plugin('2', :provider)
+      def initialize(machine)
+        @machine = machine
+      end
+
+      def action(name)
+        # Attempt to get the action method from the Action class if it
+        # exists, otherwise return nil to show that we don't support the
+        # given action.
+        action_method = "action_#{name}"
+        return Action.send(action_method) if Action.respond_to?(action_method)
+        nil
+      end
+
+      def ssh_info
+        # Run a custom action called "read_ssh_info" which does what it
+        # says and puts the resulting SSH info into the `:machine_ssh_info`
+        # key in the environment.
+        env = @machine.action("read_ssh_info")
+        env[:machine_ssh_info]
+      end
+
+      def state
+        # Run a custom action we define called "read_state" which does
+        # what it says. It puts the state in the `:machine_state_id`
+        # key in the environment.
+        env = @machine.action('read_state')
+
+        state_id = env[:machine_state_id]
+
+        # Get the short and long description
+        short = I18n.t("vagrant_arubacloud.states.short_#{state_id}")
+        long = I18n.t("vagrant_arubacloud.states.long_#{state_id}")
+
+        # Return the MachineState object
+        Vagrant::MachineState.new(state_id, short, long)
+      end
+
+      def to_s
+        'ArubaCloud IaaS'
+      end
+    end # Provider
+  end # ArubaCloud
+end # VagrantPlugins

data/lib/vagrant-arubacloud/version.rb

@@ -0,0 +1,5 @@
+module Vagrant
+  module ArubaCloud
+    VERSION = '0.0.2dev'
+  end
+end

data/lib/vagrant-arubacloud.rb

@@ -0,0 +1,53 @@
+require 'vagrant-arubacloud/version'
+require 'vagrant-arubacloud/plugin'
+
+require 'pathname'
+
+module VagrantPlugins
+  module ArubaCloud
+
+    lib_path = Pathname.new(File.expand_path('../vagrant-arubacloud', __FILE__))
+    autoload :Errors, lib_path.join('errors')
+
+    def self.init_i18n
+      I18n.load_path << File.expand_path('locales/en.yml', source_root)
+      I18n.reload!
+    end
+
+    # This initializes the logging so that our logs are outputted at
+    # the same level as Vagrant core logs.
+    def self.init_logging
+      # Initialize logging
+      level = nil
+      begin
+        level = Log4r.const_get(ENV['VAGRANT_LOG'].upcase)
+      rescue NameError
+        # This means that the logging constant wasn't found,
+        # which is fine. We just keep `level` as `nil`. But
+        # we tell the user.
+        level = nil
+      end
+
+      # Some constants, such as "true" resolve to booleans, so the
+      # above error checking doesn't catch it. This will check to make
+      # sure that the log level is an integer, as Log4r requires.
+      level = nil unless level.is_a?(Integer)
+
+      # Set the logging level on all "vagrant" namespaced
+      # logs as long as we have a valid level.
+      if level
+        logger = Log4r::Logger.new('vagrant_arubacloud')
+        logger.outputters = Log4r::Outputter.stderr
+        logger.level = level
+        logger = nil
+      end
+    end
+
+    # This returns the path to the source of this plugin.
+    #
+    # @return [Pathname]
+    def self.source_root
+      @source_root ||= Pathname.new(File.expand_path('../../', __FILE__))
+    end
+  end
+end