tnargav-aws 0.2.2

data/.gitignore

@@ -0,0 +1,14 @@
+# OS-specific
+.DS_Store
+
+# Bundler/Rubygems
+*.gem
+.bundle
+pkg/*
+tags
+Gemfile.lock
+
+# Vagrant
+.vagrant
+Vagrantfile
+!example_box/Vagrantfile

data/.gitsetup.yml

@@ -0,0 +1,5 @@
+---
+remotes:
+  vendor: https://github.com/mitchellh/vagrant-aws
+  primary: git@github.com:destructuring/vagrant-aws
+  secondary: git@bitbucket.org:destructuring/vagrant-aws

data/CHANGELOG.md

@@ -0,0 +1,39 @@
+# 0.2.2 (April 18, 2013)
+
+* Fix crashing bug with incorrect provisioner arguments.
+
+# 0.2.1 (April 16, 2013)
+
+* Got rid of extranneous references to old SSH settings.
+
+# 0.2.0 (April 16, 2013)
+
+* Add support for `vagrant ssh -c` [GH-42]
+* Ability to specify a timeout for waiting for instances to become ready. [GH-44]
+* Better error message if instance didn't become ready in time.
+* Connection can now be done using IAM profiles. [GH-41]
+
+# 0.1.3 (April 9, 2013)
+
+* The `AWS_ACCESS_KEY` and `AWS_SECRET_KEY` will be used if available
+  and no specific keys are set in the Vagrantfile. [GH-33]
+* Fix issues with SSH on VPCs, the correct IP is used. [GH-30]
+* Exclude the ".vagrant" directory from rsync.
+* Implement `:disabled` flag support for shared folders. [GH-29]
+* `aws.user_data` to specify user data on the instance. [GH-26]
+
+# 0.1.2 (March 22, 2013)
+
+* Choose the proper region when connecting to AWS. [GH-9]
+* Configurable SSH port. [GH-13]
+* Support other AWS-compatible API endpoints with `config.endpoint`
+  and `config.version`. [GH-6]
+* Disable strict host key checking on rsync so known hosts aren't an issue. [GH-7]
+
+# 0.1.1 (March 18, 2013)
+
+* Up fog dependency for Vagrant 1.1.1
+
+# 0.1.0 (March 14, 2013)
+
+* Initial release.

data/Gemfile

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+gemspec
+
+group :development do
+  # We depend on Vagrant for development, but we don't add it as a
+  # gem dependency because we expect to be installed within the
+  # Vagrant environment itself using `vagrant plugin`.
+  gem "vagrant", :git => "git://github.com/mitchellh/vagrant.git"
+end

data/LICENSE

@@ -0,0 +1,8 @@
+The MIT License (MIT)
+Copyright (c) 2013 Mitchell Hashimoto
+
+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,198 @@
+# Vagrant AWS Provider
+
+This is a [Vagrant](http://www.vagrantup.com) 1.2+ plugin that adds an [AWS](http://aws.amazon.com)
+provider to Vagrant, allowing Vagrant to control and provision machines in
+EC2 and VPC.
+
+**NOTE:** This plugin requires Vagrant 1.2+,
+
+## Features
+
+* Boot EC2 or VPC instances.
+* SSH into the instances.
+* Provision the instances with any built-in Vagrant provisioner.
+* Minimal synced folder support via `rsync`.
+* Define region-specifc configurations so Vagrant can manage machines
+  in multiple regions.
+
+## Usage
+
+Install using standard Vagrant 1.1+ plugin installation methods. After
+installing, `vagrant up` and specify the `aws` provider. An example is
+shown below.
+
+```
+$ vagrant plugin install vagrant-aws
+...
+$ vagrant up --provider=aws
+...
+```
+
+Of course prior to doing this, you'll need to obtain an AWS-compatible
+box file for Vagrant.
+
+## Quick Start
+
+After installing the plugin (instructions above), the quickest way to get
+started is to actually use a dummy AWS box and specify all the details
+manually within a `config.vm.provider` block. So first, add the dummy
+box using any name you want:
+
+```
+$ vagrant box add dummy https://github.com/mitchellh/vagrant-aws/raw/master/dummy.box
+...
+```
+
+And then make a Vagrantfile that looks like the following, filling in
+your information where necessary.
+
+```
+Vagrant.configure("2") do |config|
+  config.vm.box = "dummy"
+
+  config.vm.provider :aws do |aws, override|
+    aws.access_key_id = "YOUR KEY"
+    aws.secret_access_key = "YOUR SECRET KEY"
+    aws.keypair_name = "KEYPAIR NAME"
+
+    aws.ami = "ami-7747d01e"
+
+    override.ssh.username = "ubuntu"
+    override.ssh.private_key_path = "PATH TO YOUR PRIVATE KEY"
+  end
+end
+```
+
+And then run `vagrant up --provider=aws`.
+
+This will start an Ubuntu 12.04 instance in the us-east-1 region within
+your account. And assuming your SSH information was filled in properly
+within your Vagrantfile, SSH and provisioning will work as well.
+
+Note that normally a lot of this boilerplate is encoded within the box
+file, but the box file used for the quick start, the "dummy" box, has
+no preconfigured defaults.
+
+If you have issues with SSH connecting, make sure that the instances
+are being launched with a security group that allows SSH access.
+
+## Box Format
+
+Every provider in Vagrant must introduce a custom box format. This
+provider introduces `aws` boxes. You can view an example box in
+the [example_box/ directory](https://github.com/mitchellh/vagrant-aws/tree/master/example_box).
+That directory also contains instructions on how to build a box.
+
+The box format is basically just the required `metadata.json` file
+along with a `Vagrantfile` that does default settings for the
+provider-specific configuration for this provider.
+
+## Configuration
+
+This provider exposes quite a few provider-specific configuration options:
+
+* `access_key_id` - The access key for accessing AWS
+* `ami` - The AMI id to boot, such as "ami-12345678"
+* `availability_zone` - The availability zone within the region to launch
+  the instance. If nil, it will use the default set by Amazon.
+* `instance_ready_timeout` - The number of seconds to wait for the instance
+  to become "ready" in AWS. Defaults to 120 seconds.
+* `instance_type` - The type of instance, such as "m1.small". The default
+  value of this if not specified is "m1.small".
+* `keypair_name` - The name of the keypair to use to bootstrap AMIs
+   which support it.
+* `private_ip_address` - The private IP address to assign to an instance
+  within a [VPC](http://aws.amazon.com/vpc/)
+* `region` - The region to start the instance in, such as "us-east-1"
+* `secret_access_key` - The secret access key for accessing AWS
+* `security_groups` - An array of security groups for the instance. If this
+  instance will be launched in VPC, this must be a list of security group
+  IDs.
+* `subnet_id` - The subnet to boot the instance into, for VPC.
+* `tags` - A hash of tags to set on the machine.
+* `use_iam_profile` - If true, will use [IAM profiles](http://docs.aws.amazon.com/IAM/latest/UserGuide/instance-profiles.html)
+  for credentials.
+
+These can be set like typical provider-specific configuration:
+
+```ruby
+Vagrant.configure("2") do |config|
+  # ... other stuff
+
+  config.vm.provider :aws do |aws|
+    aws.access_key_id = "foo"
+    aws.secret_access_key = "bar"
+  end
+end
+```
+
+In addition to the above top-level configs, you can use the `region_config`
+method to specify region-specific overrides within your Vagrantfile. Note
+that the top-level `region` config must always be specified to choose which
+region you want to actually use, however. This looks like this:
+
+```ruby
+Vagrant.configure("2") do |config|
+  # ... other stuff
+
+  config.vm.provider :aws do |aws|
+    aws.access_key_id = "foo"
+    aws.secret_access_key = "bar"
+    aws.region = "us-east-1"
+
+    # Simple region config
+    aws.region_config "us-east-1", :ami => "ami-12345678"
+
+    # More comprehensive region config
+    aws.region_config "us-west-2" do |region|
+      region.ami = "ami-87654321"
+      region.keypair_name = "company-west"
+    end
+  end
+end
+```
+
+The region-specific configurations will override the top-level
+configurations when that region is used. They otherwise inherit
+the top-level configurations, as you would probably expect.
+
+## Networks
+
+Networking features in the form of `config.vm.network` are not
+supported with `vagrant-aws`, currently. If any of these are
+specified, Vagrant will emit a warning, but will otherwise boot
+the AWS machine.
+
+## Synced Folders
+
+There is minimal support for synced folders. Upon `vagrant up`,
+`vagrant reload`, and `vagrant provision`, the AWS provider will use
+`rsync` (if available) to uni-directionally sync the folder to
+the remote machine over SSH.
+
+This is good enough for all built-in Vagrant provisioners (shell,
+chef, and puppet) to work!
+
+## Development
+
+To work on the `vagrant-aws` plugin, clone this repository out, and use
+[Bundler](http://gembundler.com) to get the dependencies:
+
+```
+$ bundle
+```
+
+Once you have the dependencies, verify the unit tests pass with `rake`:
+
+```
+$ bundle exec rake
+```
+
+If those pass, you're ready to start developing the plugin. You can test
+the plugin without installing it into your Vagrant environment by just
+creating a `Vagrantfile` in the top level of this directory (it is gitignored)
+that uses it, and uses bundler to execute Vagrant:
+
+```
+$ bundle exec vagrant up --provider=aws
+```

data/Rakefile

@@ -0,0 +1,21 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'rspec/core/rake_task'
+
+# Immediately sync all stdout so that tools like buildbot can
+# immediately load in the output.
+$stdout.sync = true
+$stderr.sync = true
+
+# Change to the directory of this file.
+Dir.chdir(File.expand_path("../", __FILE__))
+
+# This installs the tasks that help with gem creation and
+# publishing.
+Bundler::GemHelper.install_tasks
+
+# Install the `spec` task so that we can run tests.
+RSpec::Core::RakeTask.new
+
+# Default task is to run the unit tests
+task :default => "spec"

data/example_box/README.md

@@ -0,0 +1,13 @@
+# Vagrant AWS Example Box
+
+Vagrant providers each require a custom provider-specific box format.
+This folder shows the example contents of a box for the `aws` provider.
+To turn this into a box:
+
+```
+$ tar cvzf aws.box ./metadata.json ./Vagrantfile
+```
+
+This box works by using Vagrant's built-in Vagrantfile merging to setup
+defaults for AWS. These defaults can easily be overwritten by higher-level
+Vagrantfiles (such as project root Vagrantfiles).

data/example_box/metadata.json

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

data/lib/vagrant-aws/action/connect_aws.rb

@@ -0,0 +1,46 @@
+require "fog"
+require "log4r"
+
+module VagrantPlugins
+  module AWS
+    module Action
+      # This action connects to AWS, verifies credentials work, and
+      # puts the AWS connection object into the `:aws_compute` key
+      # in the environment.
+      class ConnectAWS
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_aws::action::connect_aws")
+        end
+
+        def call(env)
+          # Get the region we're going to booting up in
+          region = env[:machine].provider_config.region
+
+          # Get the configs
+          region_config     = env[:machine].provider_config.get_region_config(region)
+
+          # Build the fog config
+          fog_config = {
+            :provider              => :aws,
+            :region                => region
+          }
+          if region_config.use_iam_profile
+            fog_config[:use_iam_profile] = True
+          else
+            fog_config[:aws_access_key_id] = region_config.access_key_id
+            fog_config[:aws_secret_access_key] = region_config.secret_access_key
+          end
+
+          fog_config[:endpoint] = region_config.endpoint if region_config.endpoint
+          fog_config[:version]  = region_config.version if region_config.version
+
+          @logger.info("Connecting to AWS...")
+          env[:aws_compute] = Fog::Compute.new(fog_config)
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-aws/action/is_created.rb

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module AWS
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is created and branch in the middleware.
+      class IsCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:result] = env[:machine].state.id != :not_created
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

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

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

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

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

@@ -0,0 +1,41 @@
+require "log4r"
+
+module VagrantPlugins
+  module AWS
+    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_aws::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env[:aws_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(aws, machine)
+          return nil if machine.id.nil?
+
+          # Find the machine
+          server = aws.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
+
+          # Read the DNS info
+          return {
+            :host => server.dns_name || server.private_ip_address,
+            :port => 22
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,38 @@
+require "log4r"
+
+module VagrantPlugins
+  module AWS
+    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_aws::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:aws_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_state(aws, machine)
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = aws.servers.get(machine.id)
+          if server.nil? || [:"shutting-down", :terminated].include?(server.state.to_sym)
+            # The machine can't be found
+            @logger.info("Machine not found or terminated, assuming it got destroyed.")
+            machine.id = nil
+            return :not_created
+          end
+
+          # Return the state
+          return server.state.to_sym
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-aws/action/run_instance.rb

@@ -0,0 +1,161 @@
+require "log4r"
+
+require 'vagrant/util/retryable'
+
+require 'vagrant-aws/util/timer'
+
+module VagrantPlugins
+  module AWS
+    module Action
+      # This runs the configured instance.
+      class RunInstance
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_aws::action::run_instance")
+        end
+
+        def call(env)
+          # Initialize metrics if they haven't been
+          env[:metrics] ||= {}
+
+          # Get the region we're going to booting up in
+          region = env[:machine].provider_config.region
+
+          # Get the configs
+          region_config      = env[:machine].provider_config.get_region_config(region)
+          ami                = region_config.ami
+          availability_zone  = region_config.availability_zone
+          instance_type      = region_config.instance_type
+          keypair            = region_config.keypair_name
+          private_ip_address = region_config.private_ip_address
+          security_groups    = region_config.security_groups
+          subnet_id          = region_config.subnet_id
+          tags               = region_config.tags
+          user_data          = region_config.user_data
+
+          # If there is no keypair then warn the user
+          if !keypair
+            env[:ui].warn(I18n.t("vagrant_aws.launch_no_keypair"))
+          end
+
+          # If there is a subnet ID then warn the user
+          if subnet_id
+            env[:ui].warn(I18n.t("vagrant_aws.launch_vpc_warning"))
+          end
+
+          # Launch!
+          env[:ui].info(I18n.t("vagrant_aws.launching_instance"))
+          env[:ui].info(" -- Type: #{instance_type}")
+          env[:ui].info(" -- AMI: #{ami}")
+          env[:ui].info(" -- Region: #{region}")
+          env[:ui].info(" -- Availability Zone: #{availability_zone}") if availability_zone
+          env[:ui].info(" -- Keypair: #{keypair}") if keypair
+          env[:ui].info(" -- Subnet ID: #{subnet_id}") if subnet_id
+          env[:ui].info(" -- Private IP: #{private_ip_address}") if private_ip_address
+          env[:ui].info(" -- Security Groups: #{security_groups.inspect}") if !security_groups.empty?
+
+          begin
+            options = {
+              :availability_zone  => availability_zone,
+              :flavor_id          => instance_type,
+              :image_id           => ami,
+              :key_name           => keypair,
+              :private_ip_address => private_ip_address,
+              :subnet_id          => subnet_id,
+              :tags               => tags,
+              :user_data          => user_data
+            }
+
+            if !security_groups.empty?
+              security_group_key = options[:subnet_id].nil? ? :groups : :security_group_ids
+              options[security_group_key] = security_groups
+            end
+
+            server = env[:aws_compute].servers.create(options)
+          rescue Fog::Compute::AWS::NotFound => e
+            # Invalid subnet doesn't have its own error so we catch and
+            # check the error message here.
+            if e.message =~ /subnet ID/
+              raise Errors::FogError,
+                :message => "Subnet ID not found: #{subnet_id}"
+            end
+
+            raise
+          rescue Fog::Compute::AWS::Error => e
+            raise Errors::FogError, :message => e.message
+          end
+
+          # Immediately save the ID since it is created at this point.
+          env[:machine].id = server.id
+
+          # Wait for the instance to be ready first
+          env[:metrics]["instance_ready_time"] = Util::Timer.time do
+            tries = region_config.instance_ready_timeout / 2
+
+            env[:ui].info(I18n.t("vagrant_aws.waiting_for_ready"))
+            begin
+              retryable(:on => Fog::Errors::TimeoutError, :tries => tries) do
+                # If we're interrupted don't worry about waiting
+                next if env[:interrupted]
+
+                # Wait for the server to be ready
+                server.wait_for(2) { ready? }
+              end
+            rescue Fog::Errors::TimeoutError
+              # Delete the instance
+              terminate(env)
+
+              # Notify the user
+              raise Errors::InstanceReadyTimeout,
+                timeout: region_config.instance_ready_timeout
+            end
+          end
+
+          @logger.info("Time to instance ready: #{env[:metrics]["instance_ready_time"]}")
+
+          if !env[:interrupted]
+            env[:metrics]["instance_ssh_time"] = Util::Timer.time do
+              # Wait for SSH to be ready.
+              env[:ui].info(I18n.t("vagrant_aws.waiting_for_ssh"))
+              while true
+                # If we're interrupted then just back out
+                break if env[:interrupted]
+                break if env[:machine].communicate.ready?
+                sleep 2
+              end
+            end
+
+            @logger.info("Time for SSH ready: #{env[:metrics]["instance_ssh_time"]}")
+
+            # Ready and booted!
+            env[:ui].info(I18n.t("vagrant_aws.ready"))
+          end
+
+          # Terminate the instance if we were interrupted
+          terminate(env) if env[:interrupted]
+
+          @app.call(env)
+        end
+
+        def recover(env)
+          return if env["vagrant.error"].is_a?(Vagrant::Errors::VagrantError)
+
+          if env[:machine].provider.state.id != :not_created
+            # Undo the import
+            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(Action.action_destroy, destroy_env)
+        end
+      end
+    end
+  end
+end