vagrant-chassis-digitalocean 1.0.0

data/.gitignore

@@ -0,0 +1,19 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+test/.vagrant
+tmp
+.idea/

data/CHANGELOG.md

@@ -0,0 +1,8 @@
+Changelog
+=========
+
+0.5.1
+-----
+- added confirmation for `vagrant destroy` [tmatilai]
+- added `ProvisionerCleanup` middleware to destroy action [tmatilai]
+- fixed break from destroy action [tmatilai]

data/Gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+gem 'vagrant', :git => 'git://github.com/mitchellh/vagrant.git', :tag => 'v1.4.0'
+gem 'rake'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,23 @@
+Copyright (c) 2013 John Bender
+Copyright (c) 2013 Shawn Dahlen
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,95 @@
+Digital Ocean Vagrant Provider For Chassis
+==========================================
+`vagrant-chassis-digitalocean` is a provider plugin for Vagrant that supports the
+management of [Digital Ocean](https://www.digitalocean.com/) droplets
+(instances) that setup WordPress using [Chassis](https://github.com/Chassis/Chassis/).
+
+**NOTE:** This plugin is based on the amazing work of the [vagrant-digitalocean
+](https://github.com/smdahlen/vagrant-digitalocean) plugin
+
+Current features include:
+- create and destroy droplets
+- power on and off droplets
+- rebuild a droplet
+- setup a SSH public key for authentication
+- create a new user account during droplet creation
+
+The provider has been tested with Vagrant 1.1.5+ using Ubuntu 12.04 and
+CentOS 6.3 guest operating systems.
+
+Install
+-------
+Installation of the provider requires two steps:
+
+1. Install the provider plugin using the Vagrant command-line interface:
+
+        $ vagrant plugin install vagrant-chassis-digitalocean
+
+
+**NOTE:** If you are using a Mac, you may need to install a CA bundle to enable SSL
+communication with the Digital Ocean API. It is recommended to first install
+[Homebrew](http://mxcl.github.io/homebrew/). With Homebrew installed, run
+the following command to install the bundle:
+
+    $ brew install curl-ca-bundle
+
+Once the bundle is installed, add the following environment variable to your
+`.bash_profile` script and `source` it:
+
+```bash
+export SSL_CERT_FILE=/usr/local/opt/curl-ca-bundle/share/ca-bundle.crt
+```
+
+Configure
+---------
+Once the provider has been installed, you will need to configure your [Chassis](https://github.com/Chassis/Chassis/) project
+to use it. You can use your we recommend adding your Digital Ocean details in your projects `config.yaml` or `config.local.yaml` file:
+```bash
+digitalocean:
+  client_id: YOUR_CLIENT_ID
+  api_key: YOUR_API_KEY
+  region: 'Singapore 1'
+  size: '512MB'
+  image: 'Ubuntu 12.10 x32'
+```
+
+Run
+---
+After creating your project's `Vagrantfile` with the required configuration
+attributes described above, you may create a new droplet with the following
+command:
+
+    $ vagrant up --provider=digital_ocean
+
+This command will create a new droplet, setup your SSH key for authentication,
+create a new user account, and run the provisioners you have configured.
+
+**Supported Commands**
+
+The provider supports the following Vagrant sub-commands:
+- `vagrant destroy` - Destroys the droplet instance.
+- `vagrant ssh` - Logs into the droplet instance using the configured user
+  account.
+- `vagrant halt` - Powers off the droplet instance.
+- `vagrant provision` - Runs the configured provisioners and rsyncs any
+  specified `config.vm.synced_folder`.
+- `vagrant reload` - Reboots the droplet instance.
+- `vagrant rebuild` - Destroys the droplet instance and recreates it with the
+  same IP address is was assigned to previously.
+- `vagrant status` - Outputs the status (active, off, not created) for the
+  droplet instance.
+
+Contribute
+----------
+To contribute, clone the repository, and use [Bundler](http://gembundler.com)
+to install dependencies:
+
+    $ bundle
+
+To run the provider's tests:
+
+    $ bundle exec rake test
+
+You can now make modifications. Running `vagrant` within the Bundler
+environment will ensure that plugins installed in your Vagrant
+environment are not loaded.

data/Rakefile

@@ -0,0 +1,22 @@
+require 'bundler/gem_helper'
+
+namespace :gem do
+  Bundler::GemHelper.install_tasks
+end
+
+task :test do
+  result = sh 'bash test/test.sh'
+
+  if result
+    puts 'Success!'
+  else
+    puts 'Failure!'
+    exit 1
+  end
+end
+
+def env
+  ['DO_CLIENT_ID', 'DO_API_KEY', 'VAGRANT_LOG'].inject('') do |acc, key|
+    acc += "#{key}=#{ENV[key] || 'error'} "
+  end
+end

data/box/metadata.json

@@ -0,0 +1,3 @@
+{
+  "provider": "digital_ocean"
+}

data/lib/vagrant-chassis-digitalocean.rb

@@ -0,0 +1,20 @@
+require 'vagrant-digitalocean/version'
+require 'vagrant-digitalocean/plugin'
+require 'vagrant-digitalocean/errors'
+
+module VagrantPlugins
+  module DigitalOcean
+    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/lib/vagrant-chassis-digitalocean/actions.rb

@@ -0,0 +1,160 @@
+require 'vagrant-digitalocean/actions/check_state'
+require 'vagrant-digitalocean/actions/create'
+require 'vagrant-digitalocean/actions/destroy'
+require 'vagrant-digitalocean/actions/power_off'
+require 'vagrant-digitalocean/actions/power_on'
+require 'vagrant-digitalocean/actions/rebuild'
+require 'vagrant-digitalocean/actions/reload'
+require 'vagrant-digitalocean/actions/setup_user'
+require 'vagrant-digitalocean/actions/setup_sudo'
+require 'vagrant-digitalocean/actions/setup_key'
+require 'vagrant-digitalocean/actions/sync_folders'
+require 'vagrant-digitalocean/actions/modify_provision_path'
+
+module VagrantPlugins
+  module DigitalOcean
+    module Actions
+      include Vagrant::Action::Builtin
+
+      def self.destroy
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            else
+              b.use Call, DestroyConfirm do |env2, b2|
+                if env2[:result]
+                  b2.use Destroy
+                  b2.use ProvisionerCleanup if defined?(ProvisionerCleanup)
+                end
+              end
+            end
+          end
+        end
+      end
+
+      def self.ssh
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use SSHExec
+            when :off
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.ssh_run
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use SSHRun
+            when :off
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.provision
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use Provision
+              b.use ModifyProvisionPath
+              b.use SyncFolders
+            when :off
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.up
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.already_active')
+            when :off
+              b.use PowerOn
+              b.use provision
+            when :not_created
+              b.use SetupKey
+              b.use Create
+              b.use SetupSudo
+              b.use SetupUser
+              b.use provision
+            end
+          end
+        end
+      end
+
+      def self.halt
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use PowerOff
+            when :off
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.already_off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.reload
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active
+              b.use Reload
+              b.use provision
+            when :off
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.off')
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+
+      def self.rebuild
+        return Vagrant::Action::Builder.new.tap do |builder|
+          builder.use ConfigValidate
+          builder.use Call, CheckState do |env, b|
+            case env[:machine_state]
+            when :active, :off
+              b.use Rebuild
+              b.use SetupSudo
+              b.use SetupUser
+              b.use provision
+            when :not_created
+              env[:ui].info I18n.t('vagrant_digital_ocean.info.not_created')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-chassis-digitalocean/actions/check_state.rb

@@ -0,0 +1,19 @@
+module VagrantPlugins
+  module DigitalOcean
+    module Actions
+      class CheckState
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @logger = Log4r::Logger.new('vagrant::digitalocean::check_state')
+        end
+
+        def call(env)
+          env[:machine_state] = @machine.state.id
+          @logger.info "Machine state is '#{@machine.state.id}'"
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-chassis-digitalocean/actions/create.rb

@@ -0,0 +1,96 @@
+require 'vagrant-digitalocean/helpers/client'
+
+module VagrantPlugins
+  module DigitalOcean
+    module Actions
+      class Create
+        include Helpers::Client
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @client = client
+          @logger = Log4r::Logger.new('vagrant::digitalocean::create')
+        end
+
+        def call(env)
+          ssh_key_id = env[:ssh_key_id]
+
+          size_id = @client
+            .request('/sizes')
+            .find_id(:sizes, :name => @machine.provider_config.size)
+
+          image_id = @client
+            .request('/images')
+            .find_id(:images, :name => @machine.provider_config.image)
+
+          region_id = @client
+            .request('/regions')
+            .find_id(:regions, :name => @machine.provider_config.region)
+
+          # submit new droplet request
+          result = @client.request('/droplets/new', {
+            :size_id => size_id,
+            :region_id => region_id,
+            :image_id => image_id,
+            :name => @machine.config.vm.hostname || @machine.name,
+            :ssh_key_ids => ssh_key_id,
+            :private_networking => @machine.provider_config.private_networking,
+            :backups_enabled => @machine.provider_config.backups_enabled
+          })
+
+          # wait for request to complete
+          env[:ui].info I18n.t('vagrant_digital_ocean.info.creating') 
+          @client.wait_for_event(env, result['droplet']['event_id'])
+
+          # assign the machine id for reference in other commands
+          @machine.id = result['droplet']['id'].to_s
+
+          # refresh droplet state with provider and output ip address
+          droplet = Provider.droplet(@machine, :refresh => true)
+          env[:ui].info I18n.t('vagrant_digital_ocean.info.droplet_ip', {
+            :ip => droplet['ip_address']
+          })
+          if droplet['private_ip_address']
+            env[:ui].info I18n.t('vagrant_digital_ocean.info.droplet_private_ip', {
+              :ip => droplet['private_ip_address']
+            })
+          end
+
+          # wait for ssh to be ready
+          switch_user = @machine.provider_config.setup?
+          user = @machine.config.ssh.username
+          @machine.config.ssh.username = 'root' if switch_user
+
+          retryable(:tries => 120, :sleep => 10) do
+            next if env[:interrupted]
+            raise 'not ready' if !@machine.communicate.ready?
+          end
+
+          @machine.config.ssh.username = user
+
+          @app.call(env)
+        end
+
+        # Both the recover and terminate are stolen almost verbatim from
+        # the Vagrant AWS provider up action
+        def recover(env)
+          return if env['vagrant.error'].is_a?(Vagrant::Errors::VagrantError)
+
+          if @machine.state.id != :not_created
+            terminate(env)
+          end
+        end
+
+        def terminate(env)
+          destroy_env = env.dup
+          destroy_env.delete(:interrupted)
+          destroy_env[:config_validate] = false
+          destroy_env[:force_confirm_destroy] = true
+          env[:action_runner].run(Actions.destroy, destroy_env)
+        end
+      end
+    end
+  end
+end