vagrant-simple_cloud 0.0.5

data/lib/vagrant-simple_cloud/helpers/client_service.rb

@@ -0,0 +1,81 @@
+require 'vagrant-simple_cloud/helpers/result'
+require 'faraday'
+require 'json'
+
+module VagrantPlugins
+  module SimpleCloud
+    module Helpers
+      module ClientService
+        def clientservice
+          @clientservice ||= ApiClientService.new(@machine)
+        end
+      end
+
+      class ApiClientService
+
+        def initialize(machine)
+          @logger = Log4r::Logger.new('vagrant::simple_cloud::apiclientservice')
+          @config = machine.provider_config
+          @clientservice = Faraday.new({
+            :url => @config.serviceaddr,
+            :ssl => {
+              :ca_file => @config.ca_path
+            }
+          })
+        end
+
+        def post(path, params = {}, method = :post)
+          @clientservice.headers['Content-Type'] = 'application/json'
+          begin
+            @logger.info "Request: #{path}"
+            result = @clientservice.send(method) do |req|
+              req.url path
+              req.headers['Authorization'] = "#{@config.token}"
+              req.body = params.to_json
+            end
+          rescue Faraday::Error::ConnectionFailed => e
+            # TODO this is suspect but because farady wraps the exception
+            #      in something generic there doesn't appear to be another
+            #      way to distinguish different connection errors :(
+            if e.message =~ /certificate verify failed/
+              raise Errors::CertificateError
+            end
+
+            raise e
+          end
+
+          begin
+            body = JSON.parse(result.body)
+            @logger.info "Response: #{body}"
+            next_page = body["links"]["pages"]["next"] rescue nil
+            unless next_page.nil?
+              uri = URI.parse(next_page)
+              new_path = path.split("?")[0]
+              next_result = self.request("#{new_path}?#{uri.query}")
+              req_target = new_path.split("/")[-1]
+              body["#{req_target}"].concat(next_result["#{req_target}"])
+            end
+          rescue JSON::ParserError => e
+            raise(Errors::JSONError, {
+              :message => e.message,
+              :path => path,
+              :params => params,
+              :response => result.body
+            })
+          end
+
+          unless /^2\d\d$/ =~ result.status.to_s
+            raise(Errors::APIStatusError, {
+              :path => path,
+              :params => params,
+              :status => result.status,
+              :response => body.inspect
+            })
+          end
+
+          Result.new(body)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/helpers/result.rb

@@ -0,0 +1,40 @@
+module VagrantPlugins
+  module SimpleCloud
+    module Helpers
+      class Result
+        def initialize(body)
+          @result = body
+        end
+
+        def [](key)
+          @result[key.to_s]
+        end
+
+        def find_id(sub_obj, search) #:ssh_keys, {:name => 'ijin (vagrant)'}
+          find(sub_obj, search)["id"]
+        end
+
+        def find(sub_obj, search)
+          key = search.keys.first #:slug
+          value = search[key].to_s #sfo1
+          key = key.to_s #slug
+
+          result = @result[sub_obj.to_s].inject(nil) do |result, obj|
+            obj[key] == value ? obj : result
+          end
+
+          result || error(sub_obj, key, value)
+        end
+
+        def error(sub_obj, key, value)
+          raise(Errors::ResultMatchError, {
+           :key => key,
+           :value => value,
+           :collection_name => sub_obj.to_s,
+           :sub_obj => @result[sub_obj.to_s]
+          })
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/plugin.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+  module SimpleCloud
+    class Plugin < Vagrant.plugin('2')
+      name 'SimpleCloud'
+      description <<-DESC
+        This plugin installs a provider that allows Vagrant to manage
+        machines using SimpleCloud's API.
+      DESC
+
+      config(:simple_cloud, :provider) do
+        require_relative 'config'
+        Config
+      end
+
+      provider(:simple_cloud, parallel: true, defaultable: false) do
+        require_relative 'provider'
+        Provider
+      end
+
+      command(:rebuild) do
+        require_relative 'commands/rebuild'
+        Commands::Rebuild
+      end
+
+      command("simplecloud-list", primary: false) do
+        require_relative 'commands/list'
+        Commands::List
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/provider.rb

@@ -0,0 +1,102 @@
+require 'vagrant-simple_cloud/actions'
+
+module VagrantPlugins
+  module SimpleCloud
+    class Provider < Vagrant.plugin('2', :provider)
+
+      # This class method caches status for all droplets within
+      # the SimpleCloud account. A specific droplet's status
+      # may be refreshed by passing :refresh => true as an option.
+      def self.droplet(machine, opts = {})
+        client = Helpers::ApiClient.new(machine)
+
+        # load status of droplets if it has not been done before
+        if !@droplets
+          result = client.request('/v2/vps')
+          @droplets = result['vps']
+        end
+
+        if opts[:refresh] && machine.id
+          # refresh the droplet status for the given machine
+          @droplets.delete_if { |d| d['id'].to_s == machine.id }
+          result = client.request("/v2/vps/#{machine.id}")
+          @droplets << droplet = result['vps']
+        else
+          # lookup droplet status for the given machine
+          droplet = @droplets.find { |d| d['id'].to_s == machine.id }
+        end
+
+        # if lookup by id failed, check for a droplet with a matching name
+        # and set the id to ensure vagrant stores locally
+        # TODO allow the user to configure this behavior
+        if !droplet
+          name = machine.config.vm.hostname || machine.name
+          droplet = @droplets.find { |d| d['name'] == name.to_s }
+          machine.id = droplet['id'].to_s if droplet
+        end
+
+        droplet ||= {'status' => 'not_created'}
+      end
+
+      def initialize(machine)
+        @machine = machine
+      end
+
+      def action(name)
+        return Actions.send(name) if Actions.respond_to?(name)
+        nil
+      end
+
+      # This method is called if the underying machine ID changes. Providers
+      # can use this method to load in new data for the actual backing
+      # machine or to realize that the machine is now gone (the ID can
+      # become `nil`). No parameters are given, since the underlying machine
+      # is simply the machine instance given to this object. And no
+      # return value is necessary.
+      def machine_id_changed
+      end
+
+      # This should return a hash of information that explains how to
+      # SSH into the machine. If the machine is not at a point where
+      # SSH is even possible, then `nil` should be returned.
+      #
+      # The general structure of this returned hash should be the
+      # following:
+      #
+      #     {
+      #       :host => "1.2.3.4",
+      #       :port => "22",
+      #       :username => "mitchellh",
+      #       :private_key_path => "/path/to/my/key"
+      #     }
+      #
+      # **Note:** Vagrant only supports private key based authenticatonion,
+      # mainly for the reason that there is no easy way to exec into an
+      # `ssh` prompt with a password, whereas we can pass a private key
+      # via commandline.
+      def ssh_info
+        droplet = Provider.droplet(@machine)
+
+        return nil if droplet['status'].to_sym != :active
+
+        public_network = droplet['networks']['v4'].find { |network| network['type'] == 'public' }
+
+        return {
+          :host => public_network['ip_address'],
+          :port => '22',
+          :username => 'root',
+          :private_key_path => nil
+        }
+      end
+
+      # This should return the state of the machine within this provider.
+      # The state must be an instance of {MachineState}. Please read the
+      # documentation of that class for more information.
+      def state
+        state = Provider.droplet(@machine)['status'].to_sym
+        long = short = state.to_s
+        Vagrant::MachineState.new(state, short, long)
+      end
+    end
+  end
+end

data/lib/vagrant-simple_cloud/version.rb

@@ -0,0 +1,5 @@
+module VagrantPlugins
+  module SimpleCloud
+    VERSION = '0.0.5'
+  end
+end

data/lib/vagrant-simple_cloud.rb

@@ -0,0 +1,20 @@
+require 'vagrant-simple_cloud/version'
+require 'vagrant-simple_cloud/plugin'
+require 'vagrant-simple_cloud/errors'
+
+module VagrantPlugins
+  module SimpleCloud
+    def self.source_root
+      @source_root ||= Pathname.new(File.expand_path('../../', __FILE__))
+    end
+
+    def self.public_key(private_key_path)
+      File.read("#{private_key_path}.pub")
+    rescue
+      raise Errors::PublicKeyError, :path => "#{private_key_path}.pub"
+    end
+
+    I18n.load_path << File.expand_path('locales/en.yml', source_root)
+    I18n.reload!
+  end
+end

data/locales/en.yml

@@ -0,0 +1,92 @@
+en:
+  vagrant_simple_cloud:
+    info:
+      off: "Droplet is off"
+      not_created: "Droplet has not been created"
+      already_active: "Droplet is already active"
+      already_off: "Droplet is already off"
+      creating: "Creating a new droplet..."
+      droplet_ip: "Assigned IP address: %{ip}"
+      droplet_private_ip: "Private IP address: %{ip}"
+      destroying: "Destroying the droplet..."
+      shutting_down: "Shutting down the droplet..."
+      powering_off: "Powering off the droplet..."
+      powering_on: "Powering on the droplet..."
+      rebuilding: "Rebuilding the droplet..."
+      reloading: "Rebooting the droplet..."
+      creating_user: "Creating user account: %{user}..."
+      late_sudo_install_deb8:  "SimpleCloud's debian-8 image lacks sudo. Installing now."
+      modifying_sudo: "Modifying sudoers file to remove tty requirement..."
+      using_key: "Using existing SSH key: %{name}"
+      creating_key: "Creating new SSH key: %{name}..."
+      trying_rsync_install: "Rsync not found, attempting to install with yum..."
+      rsyncing: "Rsyncing folder: %{hostpath} => %{guestpath}..."
+      rsync_missing: "The rsync executable was not found in the current path."
+      images: "Description                                        Slug                           ID\n\n%{images}"
+      images_with_regions: "Description                                        Slug                 ID                 Regions\n\n%{images}"
+      regions: "Description                    Slug\n\n%{regions}"
+      sizes: "Memory          CPUs            Slug\n\n%{sizes}"
+      list_error: 'Could not contact the SimpleCloud API: %{message}'
+    config:
+      token: "Token is required"
+      serviceaddr: "Service address is required"
+      private_key: "SSH private key path is required"
+      public_key: "SSH public key not found: %{key}"
+    errors:
+      public_key: |-
+        There was an issue reading the public key at:
+
+        Path: %{path}
+
+        Please check the file's permissions.
+      api_status: |-
+        There was an issue with the request made to the SimpleCloud
+        API at:
+
+        Path: %{path}
+        URI Params: %{params}
+
+        The response status from the API was:
+
+        Status: %{status}
+        Response: %{response}
+      rsync: |-
+        There was an error when attemping to rsync a share folder.
+        Please inspect the error message below for more info.
+
+        Host path: %{hostpath}
+        Guest path: %{guestpath}
+        Error: %{stderr}
+      json: |-
+        There was an issue with the JSON response from the SimpleCloud
+        API at:
+
+        Path: %{path}
+        URI Params: %{params}
+
+        The response JSON from the API was:
+
+        Response: %{response}
+      result_match: |-
+        The result collection for %{collection_name}:
+
+        %{sub_obj}
+
+        Contained no object with the value "%{value}" for the the
+        key "%{key}".
+
+        Please ensure that the configured value exists in the collection.
+      certificate: |-
+        The secure connection to the SimpleCloud API has failed. Please
+        ensure that your local certificates directory is defined in the
+        provider config.
+
+            config.vm.provider :simple_cloud do |vm|
+              vm.ca_path = "/path/to/ssl/ca/cert.crt"
+            end
+
+        This is generally caused by the OpenSSL configuration associated
+        with the Ruby install being unaware of the system specific ca
+        certs.
+      local_ip: |-
+        The SimpleCloud provider was unable to determine the host's IP.

data/test/Vagrantfile

@@ -0,0 +1,24 @@
+Vagrant.require_plugin('vagrant-simple_cloud')
+
+Vagrant.configure('2') do |config|
+
+  config.ssh.username = 'tester'
+  config.ssh.private_key_path = 'test_id_rsa'
+
+  config.vm.synced_folder '.', '/vagrant', :disabled => true
+
+  config.vm.provider :simple_cloud do |provider, override|
+    override.vm.box = 'simple_cloud'
+    provider.token = ENV['DO_TOKEN']
+    provider.serviceaddr = 'http://localhost/'
+    provider.ssh_key_name = 'Test Key'
+  end
+
+  config.vm.provision :shell, :path => 'scripts/provision.sh'
+
+  config.vm.define :ubuntu do |ubuntu|
+    ubuntu.vm.provider :simple_cloud do |provider|
+      provider.image = '26'
+    end
+  end
+end

data/test/scripts/provision.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+echo 'Testing 1 2 3!'

data/test/test.sh

@@ -0,0 +1,10 @@
+cd test
+
+bundle exec vagrant up --provider=simple_cloud
+bundle exec vagrant up
+bundle exec vagrant provision
+bundle exec vagrant rebuild
+bundle exec vagrant halt
+bundle exec vagrant destroy
+
+cd ..

data/test/test_id_rsa

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEowIBAAKCAQEAmxZRyfvgXFxlPW9ivoxffdK9erqpCp1oitghUtDxbzPSCbNw
+qBoiJcnPVA/TuCxMnruUcNEXYgKfTL8lD3A1Hom8N1pTAhSed5m4qAGqTMubT15s
+cSR+SnDdriShErB/9YSb9LVn1aR0MsFS3H/+x1j4w5d6hBas8BhDfuVd16shvoaA
+OKy0ywy+NBuvGy/6Au3q3t7M9wdelODRnYLSWWqaLeYExRKxWWc7ape+oduQoe4r
+BNVwGmIOjWOM9aFPEPVHdLGO+LQyPExdeuS0rW96a39U4p8GjGzsrkNcKzVOGjM3
+pIsGs3qOi7RzJ3z48HiBj9NT8I2fFpGHERerbQIDAQABAoIBABXsIcObhyuHJAh7
+JkopLZZro70lhZ+qgIyf4JYEUxyVBqu4YcRhbVJKJLSNSDBQksQdX+5SoCuKk1oV
+6vcztU6Lyb9JVVKF96CQajnVgm04msutXUbhEbkUG0Hyi5JIwM3D4QfGXNcmWAaU
+rVHeBfXH7eI4F2l0ix2lUGUvpwRFRDq9HgpOjXzyc57B4jeF7na/UTnt+Uoi4hzZ
+FjjQ7nSLqEJLXtQBqt4EnAZu6/9JlAApunyMOX2oTqRNn8XGmD0Rc+AouipHM+Mc
+9/fN9oqVxxXw2MdJA6S/sEFLEDrbifmyyHOereuZtOjdWLqsCdZwewYl8nuBnYEU
+GjVzYgECgYEAx+efis7xma28HWqtW9GLvjBcFAD/f+MDDeqX9TKFwf+91tUq0QZi
+SqXvmIvCnpsO8I70WEskT+pPwJWReAbZBrCbCVDbH34KEkAHywH9sK6chWnB8OpU
+0mp0gH89A4bq/tedKVHCQ2sAbKgbIc1zf3zpmMQiV+smMDQXU1fTg/kCgYEAxpst
+BD2cYftFjxFZE1v8fx6t6oHtzYRtNNFTYzfxzzRBaTTRRzdhSfh0tLFueyg/fcKR
+oCXUxbfCYRLk+zHP2p/AyyN9R5p2AMAc6lOZPpBj7u9kjjDVnk76DYnLDqP3Da2s
+i7b0DNYxm2gt1VSZfOuJHv7z85SLcJQsg+3ymBUCgYBrOpFX0d3Cw3COjvRitiox
+YJtjl411uf2fb2EHg4xAHcBlBn8rFDOROyUkPIOutBn1a5kh61yVCWiyMwiOy42K
+ixz+iEKhx+f7FiGYAX9lUKRg4/PGGMxa+gN4EchWpf5TqLCCw3pi03is0BeNsDjt
+/8EF0t9hLZ+UZ7zDVe79cQKBgGTPi5AlfeW2V96BHcfX31jfR8RLY1v4pj4zKrKo
+SRO2IKW4a6pMkBOuC/9UORJGocPCKY0y5sfduMrxfk2LQUhl4sS6JPNdkhxbZ9IB
+0T2SqUc1OMN8QlJzIDYTBYFO9S56Q6U/nq2NY+zQesNYh/iCzj1viIDRm93vOJFX
+DNbpAoGBALlQvzzMsT3/fPYn8moQiUCJ9XRZ4X2qwYy5Q8J8QvutI+j9o9+pJBhc
+3zSlB8HHa7asf27GUbYtv7oFDpqqcC6EFtvfp1OCiX/OjBIJA1YXTFG3YWC5ngC4
+JPxyTn4MdoX0enm8PRDg7CSZwa4AK1MIYetbiuJgWJ2wKXDFxuGH
+-----END RSA PRIVATE KEY-----