vagrant-profitbricks 1.0.0

data/lib/vagrant-profitbricks/action/message_already_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Action
+      class MessageAlreadyCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_profitbricks.already_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/action/message_not_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Action
+      class MessageNotCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_profitbricks.not_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,49 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProfitBricks
+    module Action
+      # This action reads the SSH info for the machine and puts it into the
+      # `:machine_ssh_info` key in the environment.
+      class ReadSSHInfo
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_profitbricks::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:profitbricks_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(profitbricks, machine)
+          return nil if machine.id.nil?
+
+          # Find the machine
+          server = profitbricks.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
+            return nil
+          end
+
+          if server.interfaces.empty? || server.internet_access == false
+            # The machine can't be found
+            @logger.info("Server has no internet access.")
+            machine.id = nil
+            return nil
+          end
+
+          # Read the DNS info
+          return {
+            :host => server.nics[0].ips,
+            :port => 22,
+            :username => "root"
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProfitBricks
+    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
+          @logger = Log4r::Logger.new("vagrant_profitbricks::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:profitbricks_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_state(profitbricks, machine)
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = profitbricks.servers.get(machine.provider_config.datacenter_id, machine.id)
+          rescue Excon::Error::NotFound
+          if server.nil? || server.state == "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
+
+          # Return the state
+          return server.state.downcase.to_sym
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/action/run_init_script.rb

@@ -0,0 +1,28 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProfitBricks
+    module Action
+      class RunInitScript
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_profitbricks::action::run_init_script")
+        end
+
+        def call(env)
+          config           = env[:machine].provider_config
+          machine_config   = env[:machine].config
+          begin
+            communicator = machine_config.vm.communicator ||= :ssh
+          rescue NoMethodError
+            communicator = :ssh
+          end
+
+          # Can we handle Windows config here?
+          @app.call(env)
+          env[:machine].communicate.sudo config.init_script if config.init_script && communicator == :ssh
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/create_image.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class CreateImage < Vagrant.plugin("2", :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant profitbricks images create [options]"
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          with_target_vms(argv, :provider => :profitbricks) do |machine|
+            machine.action('create_image')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/flavors.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class Flavors < Vagrant.plugin("2", :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant profitbricks flavors [options]"
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          with_target_vms(argv, :provider => :profitbricks) do |machine|
+            machine.action('list_flavors')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/images.rb

@@ -0,0 +1,33 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class Images < 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_images", __FILE__)
+            ListImages
+          end
+
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?("-h") || @main_args.include?("--help")
+            # Print the help for all the profitbricks 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
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/keypairs.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class KeyPairs < Vagrant.plugin("2", :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant profitbricks keypairs [options]"
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          with_target_vms(argv, :provider => :profitbricks) do |machine|
+            machine.action('list_keypairs')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/list_images.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class ListImages < Vagrant.plugin("2", :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant profitbricks images list [options]"
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          with_target_vms(argv, :provider => :profitbricks) do |machine|
+            machine.action('list_images')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-profitbricks/command/networks.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class Networks < Vagrant.plugin("2", :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant profitbricks networks [options]"
+          end
+
+          argv = parse_options(opts)
+          return if !argv
+
+          with_target_vms(argv, :provider => :profitbricks) do |machine|
+            machine.action('list_networks')
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,79 @@
+require 'vagrant-profitbricks/action'
+
+module VagrantPlugins
+  module ProfitBricks
+    module Command
+      class Root < Vagrant.plugin("2", :command)
+        def self.synopsis
+          I18n.t('vagrant_profitbricks.command.synopsis')
+        end
+
+        def initialize(argv, env)
+          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
+
+          @subcommands = Vagrant::Registry.new
+
+          @subcommands.register(:images) do
+            require File.expand_path("../images", __FILE__)
+            Images
+          end
+          @subcommands.register(:flavors) do
+            require File.expand_path("../flavors", __FILE__)
+            Flavors
+          end
+          @subcommands.register(:keypairs) do
+            require File.expand_path("../keypairs", __FILE__)
+            KeyPairs
+          end
+          @subcommands.register(:networks) do
+            require File.expand_path("../networks", __FILE__)
+            Networks
+          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")
+            # Print the help for all the profitbricks 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 |opts|
+            opts.banner = "Usage: vagrant profitbricks <subcommand> [<args>]"
+            opts.separator ""
+            opts.separator I18n.t('vagrant_profitbricks.command.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 I18n.t('vagrant_profitbricks.command.help_subcommands') +
+                           " `vagrant profitbricks <subcommand> -h`"
+          end
+
+          @env.ui.info(opts.help, :prefix => false)
+        end
+      end
+    end
+  end
+end

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

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

data/lib/vagrant-profitbricks/config.rb

@@ -0,0 +1,152 @@
+require "vagrant"
+require "fog/profitbricks"
+
+module VagrantPlugins
+  module ProfitBricks
+    class Config < Vagrant.plugin("2", :config)
+      # The API key to access ProfitBricks - profitbricks_password
+      # @return [String]
+      attr_accessor :password
+
+      # The username to access ProfitBricks.
+      #
+      # @return [String]
+      attr_accessor :username
+
+      # The profitbricks_url to access ProfitBricks. 
+      #
+      # expected to be a string url -
+      # 'https://api.profitbricks.com/cloudapi/v3/'
+      attr_accessor :profitbricks_url
+
+      # The flavor of server to launch, either the ID or name. This
+      # can also be a regular expression to partially match a name.
+      attr_accessor :flavor
+
+      # The name or ID of the image to use. This can also be a regular
+      # expression to partially match a name.
+      attr_accessor :image
+
+
+      # The name of the server. This defaults to the name of the machine
+      # defined by Vagrant (via `config.vm.define`), but can be overriden
+      # here.
+      attr_accessor :server_name
+
+      # Id or Name (name can be regular expression ) of datacenter. If provided with Name that doesn't match
+      # new datacenter will be created in location specified by location option
+      attr_accessor :datacenter_id
+
+      #  Specify location for datacenter if one needs to be created, default value is 'us/las'
+      attr_accessor :location
+      
+      #  Disk volume size in GB
+      attr_accessor :profitbricks_volume_size
+
+      #  Disk volume type (SSD or HDD)
+      #  Will default to HDD if not specified
+      attr_accessor :volume_type
+
+      #  The licence type of the volume. Options: LINUX, WINDOWS, UNKNOWN, OTHER
+      #  You will need to provide either the image or the licence_type parameters. 
+      #  licence_type is required, but if image is supplied, it is already set and cannot be changed
+      attr_accessor :volume_licence_type
+
+      #  The storage availability zone assigned to the volume. 
+      #  Valid values: AUTO, ZONE_1, ZONE_2, or ZONE_3. This only applies to HDD volumes. 
+      #  Leave blank or set to AUTO when provisioning SSD volumes.
+      #  Will default to AUTO if not specified
+      attr_accessor :volume_availability_zone
+
+      #  One or more SSH keys to allow access to the volume via SSH. 
+      #  Works with ProfitBricks supplied Linux images.
+      attr_accessor :volume_ssh_keys
+
+      # POne-time password is set on the Image for the appropriate account.
+      # This field may only be set in creation requests. When reading, it always returns null.
+      # Password has to contain 8-50 characters. Only these characters are allowed: [abcdefghjkmnpqrstuvxABCDEFGHJKLMNPQRSTUVX23456789]
+      attr_accessor :image_password
+
+      # number of cores in profitbricks server
+      attr_accessor :profitbricks_cores
+
+      # Required, amount of memory in GB allocated to virtual server
+      attr_accessor :profitbricks_ram
+
+      # Sets the CPU type. "AMD_OPTERON" or "INTEL_XEON". Defaults to "AMD_OPTERON".
+      attr_accessor :cpu_family
+
+      # The LAN ID the network interface (NIC) will sit on. If the LAN ID does not exist it will be created.
+      attr_accessor :lan_id
+
+      # Indicates the private IP address has outbound access to the public internet. Defaults to 'false'.
+      attr_accessor :nat
+
+      def initialize
+        @password  = UNSET_VALUE
+        @profitbricks_url = UNSET_VALUE
+        @flavor   = UNSET_VALUE
+        @profitbricks_cores = UNSET_VALUE
+        @profitbricks_ram = UNSET_VALUE
+        @image    = UNSET_VALUE
+        @server_name = UNSET_VALUE
+        @datacenter_id = UNSET_VALUE
+        @username = UNSET_VALUE
+        @profitbricks_volume_size = UNSET_VALUE
+        @location = UNSET_VALUE
+        @volume_type = UNSET_VALUE
+        @volume_licence_type = UNSET_VALUE
+        @volume_availability_zone = UNSET_VALUE
+        @volume_ssh_keys = UNSET_VALUE
+        @cpu_family = UNSET_VALUE
+        @lan_id = UNSET_VALUE
+        @nat = UNSET_VALUE
+        @image_password = UNSET_VALUE
+      end
+
+      def finalize!
+        @password  = nil if @password == UNSET_VALUE
+        @profitbricks_url = nil if @profitbricks_url == UNSET_VALUE
+        @flavor    = nil if @flavor == UNSET_VALUE
+        @profitbricks_cores = 1 if @profitbricks_cores == UNSET_VALUE
+        @profitbricks_ram   = 256 if profitbricks_ram == UNSET_VALUE
+        @image    = /ubuntu/ if @image == UNSET_VALUE
+        @server_name = nil if @server_name == UNSET_VALUE
+        @datacenter_id = nil if @datacenter_id == UNSET_VALUE
+        @username = nil if @username == UNSET_VALUE
+        @profitbricks_volume_size = 5 if @profitbricks_volume_size == UNSET_VALUE
+        @location = 'us/las' if @location == UNSET_VALUE
+        @volume_type = 'HDD' if @volume_type == UNSET_VALUE
+        @volume_licence_type = 'UNKNOWN' if @volume_licence_type == UNSET_VALUE
+        @volume_availability_zone = 'AUTO' if @volume_availability_zone == UNSET_VALUE
+        @volume_ssh_keys = nil if @volume_ssh_keys == UNSET_VALUE
+        @cpu_family = 'AMD_OPTERON' if @cpu_family == UNSET_VALUE
+        @lan_id = nil if @lan_id == UNSET_VALUE
+        @nat = false if @nat == UNSET_VALUE
+        @image_password = nil if @image_password == UNSET_VALUE
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        errors << I18n.t("vagrant_profitbricks.config.password_required") if !@password
+        errors << I18n.t("vagrant_profitbricks.config.username_required") if !@username
+        errors << I18n.t("vagrant_profitbricks.config.datacenter_required") if !@datacenter_id
+        {
+            :profitbricks_url => @profitbricks_url
+        }.each_pair do |key, value|
+          errors << I18n.t("vagrant_profitbricks.config.invalid_uri", :key => key, :uri => value) unless value.nil? || valid_uri?(value)
+        end
+
+        { "ProfitBricks Provider" => errors }
+      end
+
+      private
+
+      def valid_uri? value
+        uri = URI.parse value
+        uri.kind_of?(URI::HTTP)
+      end
+    end
+  end
+end