vagrant-linode 0.1.3 → 0.2.0

data/lib/vagrant-linode/actions/read_ssh_info.rb

@@ -0,0 +1,48 @@
+require 'log4r'
+
+module VagrantPlugins
+  module Linode
+    module Actions
+      # 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)
+	  @env = _env
+          @app    = app
+	  @machine = _env[:machine]
+          @logger = Log4r::Logger.new('vagrant_linode::action::read_ssh_info')
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:linode_api], @machine)
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(_linode, machine)
+          return nil if machine.id.nil?
+          server = Provider.linode(machine, refresh: true)
+	  @env[:ui].info "Machine State ID: %s" % [machine.state.id]
+
+          #return nil if machine.state.id != :active # @todo this seems redundant to the next line. may be more correct.
+          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
+
+          public_network = server.network.find { |network| network['ispublic'] == 1 }
+	  @env[:ui].info "IP Address: %s" % public_network['ipaddress']
+
+          {
+            host: public_network['ipaddress'],
+            port: '22',
+            username: 'root',
+            private_key_path: machine.config.ssh.private_key_path
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/actions/read_state.rb

@@ -0,0 +1,38 @@
+module VagrantPlugins
+  module Linode
+    module Actions
+      class ReadState
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger = Log4r::Logger.new('vagrant_linode::action::read_state')
+        end
+
+        def call(env)
+          env[:machine_state] = read_state(env[:linode_api], @machine)
+          @logger.info "Machine state is '#{env[:machine_state]}'"
+          @app.call(env)
+        end
+
+        def read_state(_linode, machine)
+          return :not_created if machine.id.nil?
+          server = Provider.linode(machine)
+
+          return :not_created if server.nil?
+	  status = server.status
+          return :not_created if status.nil?
+          states = {
+            ''  => :not_created,
+            '-2' => :boot_failed,
+            '-1' => :being_created,
+            '0' => :brand_new,
+            '1' => :active, # running
+            '2' => :off, # powered off
+            '3' => :shutting_down
+          }
+          states[status.to_s]
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/actions/rebuild.rb

@@ -1,20 +1,21 @@
 require 'vagrant-linode/helpers/client'
+require 'vagrant-linode/helpers/waiter'
 
 module VagrantPlugins
   module Linode
     module Actions
       class Rebuild
-        include Helpers::Client
         include Vagrant::Util::Retryable
+        include Helpers::Waiter
 
         def initialize(app, env)
           @app = app
           @machine = env[:machine]
-          @client = client
           @logger = Log4r::Logger.new('vagrant::linode::rebuild')
         end
 
         def call(env)
+          @client = env[:linode_api]
           # @todo find a convenient way to send provider_config back to the create action, reusing the diskid or configid
           fail 'not implemented'
           # look up image id

data/lib/vagrant-linode/actions/reload.rb

@@ -1,19 +1,20 @@
 require 'vagrant-linode/helpers/client'
+require 'vagrant-linode/helpers/waiter'
 
 module VagrantPlugins
   module Linode
     module Actions
       class Reload
-        include Helpers::Client
+        include Helpers::Waiter
 
         def initialize(app, env)
           @app = app
           @machine = env[:machine]
-          @client = client
           @logger = Log4r::Logger.new('vagrant::linode::reload')
         end
 
         def call(env)
+          @client = env[:linode_api]
           # submit reboot linode request
           result = @client.linode.reboot(linodeid: @machine.id)
 

data/lib/vagrant-linode/commands/create_image.rb

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

data/lib/vagrant-linode/commands/datacenters.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module Linode
+    module Commands
+      class Datacenters < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant linode datacenters [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          with_target_vms(argv, provider: :linode) do |machine|
+            machine.action('list_datacenters')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/commands/distributions.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module Linode
+    module Commands
+      class Distributions < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant linode distributions [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          with_target_vms(argv, provider: :linode) do |machine|
+            machine.action('list_distributions')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/commands/images.rb

@@ -0,0 +1,59 @@
+module VagrantPlugins
+  module Linode
+    module Commands
+      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
+          @subcommands.register(:create) do
+            require File.expand_path('../create_image', __FILE__)
+            CreateImage
+          end
+
+          super(argv, env)
+        end
+
+        def execute
+          if @main_args.include?('-h') || @main_args.include?('--help')
+            # Print the help for all the rackspace 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 linode images <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 linode images <subcommand> -h`'
+          end
+
+          @env.ui.info(opts.help, prefix: false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/commands/list_images.rb

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

data/lib/vagrant-linode/commands/networks.rb

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

data/lib/vagrant-linode/commands/plans.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module Linode
+    module Commands
+      class Plans < Vagrant.plugin('2', :command)
+        def execute
+          options = {}
+          opts = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant linode plans [options]'
+          end
+
+          argv = parse_options(opts)
+          return unless argv
+
+          with_target_vms(argv, provider: :linode) do |machine|
+            machine.action('list_plans')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/commands/root.rb

@@ -0,0 +1,81 @@
+require 'vagrant-linode/actions'
+
+module VagrantPlugins
+  module Linode
+    module Commands
+      class Root < Vagrant.plugin('2', :command)
+        def self.synopsis
+          'query Linode for available images or plans'
+        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(:plans) do
+            require File.expand_path('../plans', __FILE__)
+            Plans
+          end
+          @subcommands.register(:distributions) do
+            require File.expand_path('../distributions', __FILE__)
+            Distributions
+          end
+          @subcommands.register(:datacenters) do
+            require File.expand_path('../datacenters', __FILE__)
+            Datacenters
+          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 linode 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 linode <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 linode <subcommand> -h`'
+          end
+
+          @env.ui.info(opts.help, prefix: false)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-linode/commands/servers.rb

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

data/lib/vagrant-linode/config.rb

@@ -1,7 +1,8 @@
 module VagrantPlugins
   module Linode
     class Config < Vagrant.plugin('2', :config)
-      attr_accessor :token
+      attr_accessor :token # deprecated
+      attr_accessor :api_key
       attr_accessor :api_url
       attr_accessor :distribution
       attr_accessor :datacenter
@@ -21,7 +22,10 @@ module VagrantPlugins
       alias_method :setup?, :setup
 
       def initialize
+        # @logger  = Log4r::Logger.new('vagrant::linode::config')
+
         @token              = UNSET_VALUE
+        @api_key            = UNSET_VALUE
         @api_url            = UNSET_VALUE
         @distribution       = UNSET_VALUE
         @datacenter         = UNSET_VALUE
@@ -40,7 +44,9 @@ module VagrantPlugins
       end
 
       def finalize!
+        @api_key            = ENV['LINODE_API_KEY'] if @api_key == UNSET_VALUE
         @token              = ENV['LINODE_TOKEN'] if @token == UNSET_VALUE
+        @api_key            = @token if ((@api_key == nil) and (@token != nil))
         @api_url            = ENV['LINODE_URL'] if @api_url == UNSET_VALUE
         @distribution       = 'Ubuntu 14.04 LTS' if @distribution == UNSET_VALUE
         @datacenter         = 'dallas' if @datacenter == UNSET_VALUE
@@ -59,8 +65,10 @@ module VagrantPlugins
 
       def validate(machine)
         errors = []
-        errors << I18n.t('vagrant_linode.config.token') unless @token
-
+        errors << I18n.t('vagrant_linode.config.api_key') unless @api_key
+        # Log4r::Logger.new('vagrant_linode.config.token') if @token
+        # env[:ui].info I18n.t('vagrant_linode.config.token') if @token
+        # errors << I18n.t('vagrant_linode.config.token') if @token
         key = machine.config.ssh.private_key_path
         key = key[0] if key.is_a?(Array)
         if !key