engineyard-local 0.2.1

data/lib/engineyard-local/command/vagrant_action.rb

@@ -0,0 +1,11 @@
+module Engineyard
+  module Local
+    module Command
+      class VagrantAction < Base
+        def exec(action)
+          run(Vagrant.actions.get(action))
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/errors.rb

@@ -0,0 +1,10 @@
+module Engineyard
+  module Local
+    module Errors
+      class PossibleIPCollision < Vagrant::Errors::VagrantError
+        status_code(101)
+        error_key(:possible_ip_collision, "eylocal.actions.network")
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware.rb

@@ -0,0 +1,28 @@
+require "engineyard-local/middleware/bundle"
+require "engineyard-local/middleware/dna"
+require "engineyard-local/middleware/chef"
+require "engineyard-local/middleware/rails"
+require "engineyard-local/middleware/exec"
+require "engineyard-local/middleware/tag"
+require "engineyard-local/middleware/network"
+require "engineyard-local/middleware/default_provisioner"
+
+# alter some of the core vagrant middleware stacks to make sure
+# the the default provisioner is run where necessary
+module Engineyard
+  module Local
+    module Middleware
+      registry = Vagrant.actions
+      up = registry.get(:up)
+      reload = registry.get(:reload)
+
+      up.insert(4, DefaultProvisioner)
+
+      # alter the reload middelware to check for default provisioning
+      reload.insert(1, DefaultProvisioner)
+
+      registry.register(:up, up)
+      registry.register(:reload, reload)
+    end
+  end
+end

data/lib/engineyard-local/middleware/bundle.rb

@@ -0,0 +1,40 @@
+module Engineyard
+  module Local
+    module Middleware
+      class Bundle
+        include Middleware::Helpers::Rvm
+        include Middleware::Helpers::Executable
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+        end
+
+        def call(env)
+          @env[:ui].info I18n.t("eylocal.setup.bundling")
+          ssh_exec!(env, commands)
+          @app.call(env)
+        end
+
+        def commands
+          rvm_env +
+            [
+              ensure_bundler_is_installed,
+              "cd #{project_dir}",
+              "sudo mkdir -p #{project_dir} /data/#{Engineyard::Local.config['app_name']}/",
+              "if [[ ! -e /data/#{Engineyard::Local.config['app_name']}/current ]]; then sudo ln -sf #{project_dir} /data/#{Engineyard::Local.config['app_name']}/current; fi",
+              if_gemfile_exists("sudo bundle install")
+            ]
+        end
+
+        def ensure_bundler_is_installed
+          "sudo bash -c 'command -v bundle &>/dev/null || { sudo gem install bundler; exit 0; }'"
+        end
+ 
+        def if_gemfile_exists(cmd)
+          "if [[ -e Gemfile || -e Gemfile.lock ]]; then #{cmd}; fi"
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/chef.rb

@@ -0,0 +1,44 @@
+module Engineyard
+  module Local
+    module Middleware
+      class Chef
+        include Middleware::Helpers::Executable
+        include Middleware::Helpers::Rvm
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+        end
+
+        def call(env)
+          @env[:ui].info I18n.t("eylocal.setup.chef")
+          ssh_exec!(env, commands)
+          @app.call(env)
+        end
+
+        def commands
+          rvm_env +
+          [
+            ensure_that_recipes_are_installed,
+            "cd /etc/chef",
+            "sudo env PATH=/usr/local/ey_resin/bin:/sbin:/bin:/usr/sbin:/usr/bin chef-solo -j /etc/chef/dna.json -c /etc/chef/solo.rb",
+            run_custom_chef_in_application_if_it_exists
+          ]
+        end
+
+        def ensure_that_recipes_are_installed
+          "if [[ ! -d /etc/chef/recipes ]]; then curl -s #{cookbook_uri} > /tmp/chef.tar.bz2; sudo mkdir -p /etc/chef/recipes && cd /etc/chef/recipes && sudo tar -xjf /tmp/chef.tar.bz2; rm /tmp/chef.tar.bz2; fi"
+        end
+
+        def run_custom_chef_in_application_if_it_exists
+          "if [[ -d /vagrant/deploy ]] ; then shopt -s nullglob; for rb in /vagrant/deploy/*.rb; do sudo env PATH=/usr/local/ey_resin/bin:/sbin:/bin:/usr/sbin:/usr/bin chef-solo -j /etc/chef/dna.json -c ${rb}; done; shopt -u nullglob; fi"
+        end
+
+        def cookbook_uri
+          Local.config[:box_defaults][:cookbook_uri]
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/default_provisioner.rb

@@ -0,0 +1,34 @@
+module Engineyard
+  module Local
+    module Middleware
+      class DefaultProvisioner
+        def initialize(app, env)
+          @app, @env = app, env
+        end
+
+        # if the default recipe cookbooks/main/recipes/default.rb exists
+        # and the user doesn't already have a provisioner configured use
+        # the default recipe
+        def call(env)
+          if File.exists?(default_recipe_path) && !provisioner_configured?
+            @env[:ui].info I18n.t("eylocal.provisioner", :recipe_path => default_recipe_path)
+            @env[:vm].config.vm.provision :chef_solo  do |chef|
+              chef.cookbooks_path = "cookbooks"
+              chef.add_recipe "main"
+            end
+          end
+
+          @app.call(env)
+        end
+
+        def default_recipe_path
+          File.join(@env[:root_path], "cookbooks", "main", "recipes", "default.rb")
+        end
+
+        def provisioner_configured?
+          !@env[:vm].config.vm.provisioners.empty?
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/dna.rb

@@ -0,0 +1,80 @@
+require 'fileutils'
+
+module Engineyard
+  module Local
+    module Middleware
+      class DNA
+        include Middleware::Helpers::Uploadable
+        include Middleware::Helpers::Executable
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+        end
+
+        def call(env)
+          @env[:ui].info I18n.t("eylocal.setup.dna")
+
+          create_dna
+          ssh_upload!(env, dna_tmp_to, dna_tmp_to)
+          ssh_upload!(env, solo_from, solo_tmp_to)
+          ssh_exec!(env, commands)
+          FileUtils.rm dna_tmp_to
+          @app.call(env)
+        end
+
+        def create_dna
+          dna_data = File.read(dna_from)
+
+          app_url = %Q{#{`git config remote.origin.url`}}.strip
+          app_url = app_url.empty? ? "file://#{Dir.pwd}" : app_url
+          Engineyard::Local.config['app_git_url'] = app_url
+
+          Engineyard::Local.config['app_name'] = Engineyard::Local.config['app_git_url'].sub(/^.*\/([^\/\.]+)(?:\.git)?/,'\1').strip.gsub(/\s/,'_')
+
+          Engineyard::Local.config['app_revision'] = %Q{#{`git show --format=oneline`.gsub(/^(\w+).*$/m,'\1')}}.strip
+          app_type = 'rack'
+
+          dna_data.gsub!(/APPNAME/,Engineyard::Local.config['app_name'])
+          dna_data.gsub!(/APPURL/,Engineyard::Local.config['app_git_url'])
+          dna_data.gsub!(/APPREVISION/,Engineyard::Local.config['app_revision'])
+          dna_data.gsub!(/APPTYPE/,app_type)
+
+          File.open(dna_tmp_to, "w+") {|fh| fh.write dna_data}
+        end
+
+        def commands
+            [
+              "sudo mv #{solo_tmp_to} #{solo_to}",
+              "sudo mv #{dna_tmp_to} #{dna_to}",
+            ]
+        end
+
+        def dna_from
+          File.expand_path("config/dna.json", Engineyard::Local.project_root)
+        end
+
+        def dna_tmp_to
+          "/tmp/dna.json.#{$$}"
+        end
+
+        def dna_to
+          "/etc/chef/dna.json"
+        end
+
+        def solo_from
+          File.expand_path("config/solo.rb", Engineyard::Local.project_root)
+        end
+
+        def solo_tmp_to
+          "/tmp/solo.rb.#{$$}"
+        end
+
+        def solo_to
+          "/etc/chef/solo.rb"
+        end
+
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/exec.rb

@@ -0,0 +1,27 @@
+module Engineyard
+  module Local
+    module Middleware
+      class Exec
+        include Middleware::Helpers::Rvm
+        include Middleware::Helpers::Executable
+
+        def initialize(app, env)
+          @app, @env = app, env
+        end
+
+        def call(env)
+          ssh_exec!(env, commands, :prefix => false)
+          @app.call(env)
+        end
+
+        def commands
+          rvm_env + [ "cd #{project_dir}", command_args ]
+        end
+
+        def command_args
+          @env["eylocal.exec.command_args"].join(" ")
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/helpers.rb

@@ -0,0 +1,4 @@
+require "engineyard-local/middleware/helpers/executable"
+require "engineyard-local/middleware/helpers/network"
+require "engineyard-local/middleware/helpers/rvm"
+require "engineyard-local/middleware/helpers/uploadable"

data/lib/engineyard-local/middleware/helpers/executable.rb

@@ -0,0 +1,27 @@
+module Engineyard
+  module Local
+    module Middleware
+      module Helpers
+        module Executable
+          # Depends soley on the commands method
+          def ssh_exec!(env, bash_commands, opts = {})
+            commands = bash_commands.join(";\n")
+
+            env[:vm].channel.execute(commands) do |stream, data|
+              if stream == :stdout
+                env[:vm].ui.info(data.strip)
+              else
+                env[:vm].ui.error(data.strip)
+              end
+            end
+          end
+
+          # TODO might belong in a general helpers mixin
+          def project_dir
+            @env[:vm].config.vm.shared_folders["v-root"][:guestpath]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/helpers/network.rb

@@ -0,0 +1,20 @@
+module Engineyard
+  module Local
+    module Middleware
+      module Helpers
+        module Network
+          def proposed_ip
+            # networks is an array of pairs, snd is the ip, fst is type
+            # TODO this generally sucks and is tightly coupled to vagrant internals
+            (networks.first || [[]]).last.first || Local.config[:network] || '10.0.0.1'
+          end
+
+          def networks
+            @env[:vm].config.vm.networks
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/engineyard-local/middleware/helpers/rvm.rb

@@ -0,0 +1,37 @@
+module Engineyard
+  module Local
+    module Middleware
+      module Helpers
+        # TODO this whole helper should be removed by setting up rvm properly
+        # for the vagrant ssh session user inside the vm
+        module Rvm
+          def ruby_env_const
+            Local.config[:rvm][:ruby_env_const]
+          end
+
+          # get the first ruby listed by rvm list
+          # TODO fragile
+          def export_ruby
+            %Q(export #{ ruby_env_const }=`rvm list | grep "^   " | awk '{ print $1 }' | tail -1`)
+          end
+
+          # TODO determine gemset and app name and export
+          def use_ruby_with_gemset
+            "rvm use $#{ ruby_env_const } > /dev/null"
+          end
+
+          def bash_rvm_setup
+            "source /etc/profile.d/*"
+          end
+
+          # everything needed to setup the proper gemset env
+          def rvm_env
+            [ bash_rvm_setup,
+              export_ruby,
+              use_ruby_with_gemset ]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/helpers/uploadable.rb

@@ -0,0 +1,14 @@
+module Engineyard
+  module Local
+    module Middleware
+      module Helpers
+        module Uploadable
+          # Depends soley on the commands method
+          def ssh_upload!(env, to, from)
+            env[:vm].channel.upload(to, from)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/engineyard-local/middleware/network.rb

@@ -0,0 +1,64 @@
+module Engineyard
+  module Local
+    module Middleware
+      class Network
+        include Helpers::Network
+
+        def initialize(app, env, options)
+          @app, @env, @options = app, env, options
+        end
+
+        def call(env)
+          Virtualbox.uuid_map.each do |name, uuid|
+            # skip the current vm
+            next if uuid == env[:vm].uuid
+
+            # get the configured ip from the tag data stored by ey-local
+            existing_ip = Virtualbox.extra_data(uuid, Local.config[:network_ip_key])
+
+            # if the proposed ip matches a different vms tagged ip
+            # let the user decide if this is intended or needs to be fixed
+            if(!@options[:silent] && existing_ip == proposed_ip)
+              while !["Y", "N"].include?(result ||= nil)
+                result = check_with_user(name)
+              end
+
+              if result == "N"
+                raise(Errors::PossibleIPCollision,
+                      :proposed_ip => proposed_ip,
+                      :vm_name => name)
+              end
+
+              if result == "Y"
+                break
+              end
+            end
+          end
+
+          @app.call(env)
+        end
+
+        def check_with_user(name)
+           @env[:ui].ask(prompt(name), :color => Thor::Shell::Color::YELLOW)
+        end
+
+        def prompt(name)
+          @prompt ||= I18n.t("eylocal.actions.network.prompt",
+                             :vm_name => name,
+                             :proposed_ip => proposed_ip)
+        end
+
+        #taken directly from Vagrant source
+        def recover(env)
+          if env[:vm].created? && env["vagrant.error"].is_a?(Errors::PossibleIPCollision)
+            # Interrupted, destroy the VM. We note that we don't want to
+            # validate the configuration here.
+            destroy_env = env.clone
+            destroy_env[:validate] = false
+            env[:action_runner].run(:destroy, destroy_env)
+          end
+        end
+      end
+    end
+  end
+end