ventriloquist 0.0.1 → 0.1.0

data/lib/ventriloquist/cap/debian/ventriloquist_containers_upstart.rb

@@ -0,0 +1,33 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Debian
+        module VentriloquistContainersUpstart
+          def self.ventriloquist_containers_upstart(machine)
+            machine.communicate.tap do |comm|
+              if ! comm.test('test -f /etc/init/ventriloquist.conf')
+                machine.env.ui.info('Configuring Ventriloquist services upstart')
+                machine.communicate.sudo '
+cat<<EOF > /etc/init/ventriloquist.conf
+description "Restart configured Ventriloquist services after reboot"
+
+start on (started docker)
+
+script
+  if [ -d /var/lib/ventriloquist/cids ]; then
+    sleep 1 # Give Docker some time
+    for cidfile in \$(ls /var/lib/ventriloquist/cids/*); do
+      docker start \$(cat \$cidfile)
+    done
+  fi
+end script
+respawn
+EOF'
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/download.rb

@@ -0,0 +1,26 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module Download
+          # FIXME: Use vagrant downloader and upload to guest machine
+          def self.download(machine, src, destination)
+            machine.communicate.tap do |comm|
+              if comm.test('which wget')
+                machine.env.ui.info("Downloading #{src} to #{destination}")
+                comm.execute("wget #{src} -O #{destination}")
+
+              elsif comm.test('which curl')
+                machine.env.ui.info("Downloading #{src} to #{destination}")
+                comm.execute("curl #{src} -o #{destination}")
+
+              else
+                raise 'Unable to download file for guest VM!'
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/make.rb

@@ -0,0 +1,14 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module Make
+          def self.make(machine, workdir, target = 'all')
+            machine.env.ui.info("Running `make #{target}` on #{workdir}")
+            machine.communicate.execute("cd #{workdir} && make #{target}")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/mysql_configure_client.rb

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module MySqlConfigureClient
+          def self.mysql_configure_client(machine)
+            machine.communicate.tap do |comm|
+              if ! comm.test('grep -q client $HOME/.my.cnf 2>/dev/null')
+                machine.env.ui.info('Setting default MySQL configs')
+                comm.execute('echo -e "[client]\nprotocol=tcp\npassword=vagrant" >> $HOME/.my.cnf')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/pg_export_pghost.rb

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module PgExportPghost
+          def self.pg_export_pghost(machine)
+            machine.communicate.tap do |comm|
+              if ! comm.test('grep -q PGHOST /etc/profile.d/ventriloquist.sh 2>/dev/null')
+                machine.env.ui.info('Setting default PGHOST')
+                comm.sudo('echo "export PGHOST=localhost" >> /etc/profile.d/ventriloquist.sh')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/rvm_install.rb

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module RvmInstall
+          def self.rvm_install(machine)
+            if ! machine.communicate.test('test -d $HOME/.rvm')
+              machine.env.ui.info('Installing RVM')
+              machine.communicate.execute('\curl -L https://get.rvm.io | bash -s stable --autolibs=enabled')
+              #machine.communicate.execute('rvm requirements')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/rvm_install_ruby.rb

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module RvmInstallRuby
+          def self.rvm_install_ruby(machine, version)
+            if ! machine.communicate.test("rvm list | grep #{version}")
+              machine.env.ui.info("Installing Ruby #{version}")
+              machine.communicate.execute("rvm install #{version}")
+              # FIXME: THIS IS DEBIAN SPECIFIC
+              machine.communicate.sudo("apt-get install -y libxslt1-dev")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/cap/linux/untar.rb

@@ -0,0 +1,14 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Cap
+      module Linux
+        module Untar
+          def self.untar(machine, src, workdir)
+            machine.env.ui.info("Extracting #{src} to #{workdir}")
+            machine.communicate.execute("cd #{workdir} && tar xvfz #{src}")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/config.rb

@@ -0,0 +1,12 @@
+module VagrantPlugins
+  module Ventriloquist
+    class Config < Vagrant.plugin("2", :config)
+      attr_reader :services, :platforms
+
+      def initialize
+        @services  = []
+        @platforms = []
+      end
+    end
+  end
+end

data/lib/ventriloquist/errors.rb

@@ -0,0 +1,8 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Errors
+      class BaseError < ::Vagrant::Errors::VagrantError
+      end
+    end
+  end
+end

data/lib/ventriloquist/platform.rb

@@ -0,0 +1,11 @@
+module VagrantPlugins
+  module Ventriloquist
+    class Platform
+      attr_reader :name, :config
+
+      def initialize(name, config)
+        @name, @config = name, config
+      end
+    end
+  end
+end

data/lib/ventriloquist/platforms/go.rb

@@ -0,0 +1,13 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Platforms
+      class Go < Platform
+        def provision(machine)
+          @config[:version] = '1.1.2' if @config[:version] == 'latest'
+          machine.guest.capability(:mercurial_install)
+          machine.guest.capability(:go_install, @config[:version])
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/platforms/nodejs.rb

@@ -0,0 +1,11 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Platforms
+      class NodeJS < Platform
+        def provision(machine)
+          machine.guest.capability(:nodejs_install)
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/platforms/phantomjs.rb

@@ -0,0 +1,12 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Platforms
+      class PhantomJS < Platform
+        def provision(machine)
+          @config[:version] = '1.9.1' if config[:version] == 'latest'
+          machine.guest.capability(:phantomjs_install, @config[:version])
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/platforms/ruby.rb

@@ -0,0 +1,15 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Platforms
+      class Ruby < Platform
+        def provision(machine)
+          @config[:version] = '2.0.0' if @config[:version] == 'latest'
+          machine.guest.tap do |guest|
+            guest.capability(:rvm_install)
+            guest.capability(:rvm_install_ruby, @config[:version])
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/platforms_builder.rb

@@ -0,0 +1,56 @@
+require_relative 'platform'
+require_relative 'platforms/ruby'
+require_relative 'platforms/nodejs'
+require_relative 'platforms/phantomjs'
+require_relative 'platforms/go'
+
+module VagrantPlugins
+  module Ventriloquist
+    class PlatformsBuilder
+      MAPPING = {
+        'ruby'      => Platforms::Ruby,
+        'nodejs'    => Platforms::NodeJS,
+        'phantomjs' => Platforms::PhantomJS,
+        'go'        => Platforms::Go,
+      }
+
+      def initialize(platforms, mapping = MAPPING)
+        @platforms = platforms.flatten
+        @mapping   = mapping
+      end
+
+      def self.build(platforms)
+        new(platforms).build
+      end
+
+      def build
+        @platforms.each_with_object([]) do |cfg, built_platforms|
+          case cfg
+            when Hash
+              built_platforms.concat build_platforms(cfg)
+            when String, Symbol
+              built_platforms << create_platform_provisioner(cfg, {})
+            else
+              raise "Unknown cfg type: #{cfg.class}"
+          end
+        end
+      end
+
+      private
+
+      def build_platforms(cfg_hash)
+        cfg_hash.map do |name, config|
+          create_platform_provisioner(name, config)
+        end
+      end
+
+      def create_platform_provisioner(name, config)
+        name, version = name.to_s.split(':')
+        config[:version] ||= (version || 'latest')
+
+        klass = @mapping.fetch(name.to_s)
+        klass.new(name, config)
+      end
+    end
+  end
+end

data/lib/ventriloquist/plugin.rb

@@ -0,0 +1,119 @@
+require "vagrant"
+
+Vagrant.require_plugin "vocker"
+
+require_relative "version"
+
+I18n.load_path << File.expand_path(File.dirname(__FILE__) + '/../../locales/en.yml')
+I18n.reload!
+
+module VagrantPlugins
+  module Ventriloquist
+    class Plugin < Vagrant.plugin("2")
+      name "Ventriloquist"
+      description <<-DESC
+      Vagrant development environments made easy
+      DESC
+
+      provisioner(:ventriloquist) do
+        require_relative "provisioner"
+        Provisioner
+      end
+
+      config(:ventriloquist, :provisioner) do
+        require_relative "config"
+        Config
+      end
+
+      guest_capability("debian", "pg_install_client") do
+        require_relative "cap/debian/pg_install_client"
+        Cap::Debian::PgInstallClient
+      end
+
+      guest_capability("debian", "pg_install_headers") do
+        require_relative "cap/debian/pg_install_headers"
+        Cap::Debian::PgInstallHeaders
+      end
+
+      guest_capability("linux", "pg_export_pghost") do
+        require_relative "cap/linux/pg_export_pghost"
+        Cap::Linux::PgExportPghost
+      end
+
+      guest_capability("debian", "mysql_install_client") do
+        require_relative "cap/debian/mysql_install_client"
+        Cap::Debian::MySqlInstallClient
+      end
+
+      guest_capability("debian", "mysql_install_headers") do
+        require_relative "cap/debian/mysql_install_headers"
+        Cap::Debian::MySqlInstallHeaders
+      end
+
+      guest_capability("linux", "mysql_configure_client") do
+        require_relative "cap/linux/mysql_configure_client"
+        Cap::Linux::MySqlConfigureClient
+      end
+
+      guest_capability("debian", "git_install") do
+        require_relative "cap/debian/git_install"
+        Cap::Debian::GitInstall
+      end
+
+      guest_capability("debian", "mercurial_install") do
+        require_relative "cap/debian/mercurial_install"
+        Cap::Debian::MercurialInstall
+      end
+
+      guest_capability("linux", "make") do
+        require_relative "cap/linux/make"
+        Cap::Linux::Make
+      end
+
+      guest_capability("debian", "install_build_tools") do
+        require_relative "cap/debian/install_build_tools"
+        Cap::Debian::InstallBuildTools
+      end
+
+      guest_capability("debian", "ventriloquist_containers_upstart") do
+        require_relative "cap/debian/ventriloquist_containers_upstart"
+        Cap::Debian::VentriloquistContainersUpstart
+      end
+
+      guest_capability("linux", "download") do
+        require_relative "cap/linux/download"
+        Cap::Linux::Download
+      end
+
+      guest_capability("linux", "untar") do
+        require_relative "cap/linux/untar"
+        Cap::Linux::Untar
+      end
+
+      guest_capability("debian", "nodejs_install") do
+        require_relative "cap/debian/nodejs_install"
+        Cap::Debian::NodejsInstall
+      end
+
+      guest_capability("linux", "rvm_install") do
+        require_relative "cap/linux/rvm_install"
+        Cap::Linux::RvmInstall
+      end
+
+      guest_capability("debian", "phantomjs_install") do
+        require_relative "cap/debian/phantomjs_install"
+        Cap::Debian::PhantomjsInstall
+      end
+
+      guest_capability("debian", "go_install") do
+        require_relative "cap/debian/go_install"
+        Cap::Debian::GoInstall
+      end
+
+      guest_capability("linux", "rvm_install_ruby") do
+        require_relative "cap/linux/rvm_install_ruby"
+        Cap::Linux::RvmInstallRuby
+      end
+    end
+  end
+end

data/lib/ventriloquist/provisioner.rb

@@ -0,0 +1,59 @@
+require "vocker/docker_client"
+require "vocker/docker_installer"
+
+require_relative "errors"
+require_relative "services_builder"
+require_relative "platforms_builder"
+
+module VagrantPlugins
+  module Ventriloquist
+    # TODO: Improve handling of vagrant-lxc specifics (like checking for apparmor
+    #       profile stuff + autocorrection)
+    class Provisioner < Vagrant.plugin("2", :provisioner)
+      def initialize(machine, config, installer = nil, client = nil)
+        super(machine, config)
+        @installer = installer || Vocker::DockerInstaller.new(@machine)
+        @client    = client    || Vocker::DockerClient.new(@machine)
+      end
+
+      def provision
+        @logger = Log4r::Logger.new("vagrant::provisioners::ventriloquist")
+
+        provision_services
+        provision_platforms
+      end
+
+      protected
+
+      def provision_services
+        return if config.services.empty?
+
+        @logger.info("Checking for Docker installation...")
+        @installer.ensure_installed
+
+        if @machine.guest.capability?(:ventriloquist_containers_upstart)
+          @machine.guest.capability(:ventriloquist_containers_upstart)
+        end
+
+        unless @client.daemon_running?
+          raise Vocker::Errors::DockerNotRunning
+        end
+
+        ServicesBuilder.build(config.services, @client).each do |service|
+          service.provision(@machine)
+        end
+      end
+
+      def provision_platforms
+        return if config.platforms.empty?
+
+        # Install git so we can install "stuff" from sources
+        @machine.guest.capability(:git_install)
+
+        PlatformsBuilder.build(config.platforms).each do |platform|
+          platform.provision(@machine)
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/service.rb

@@ -0,0 +1,23 @@
+module VagrantPlugins
+  module Ventriloquist
+    class Service
+      CONTAINER_IDS_PATH = '/var/lib/ventriloquist/cids'
+
+      attr_reader :name, :config, :docker_client
+
+      def initialize(name, config, docker_client)
+        @name, @config, @docker_client = name, config, docker_client
+      end
+
+      def provision(machine)
+        machine.env.ui.info("Starting #{@name} service")
+        machine.communicate.sudo("mkdir -p #{CONTAINER_IDS_PATH}")
+        # Reduce network latency, see https://groups.google.com/d/msg/docker-user/Z3zQuRawIsE/2AEkl30WpTQJ
+        # for more info
+        @config[:dns]     = '127.0.0.1'
+        @config[:cidfile] = "#{CONTAINER_IDS_PATH}/#{@name}"
+        @docker_client.run_container(@config)
+      end
+    end
+  end
+end

data/lib/ventriloquist/services/mysql.rb

@@ -0,0 +1,39 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Services
+      class MySql < Service
+        def provision(machine)
+          super
+
+          @machine = machine
+
+          install_client
+          install_headers
+          configure_client
+        end
+
+        protected
+
+        def install_client
+          if @machine.guest.capability?(:mysql_install_client)
+            @machine.guest.capability(:mysql_install_client)
+          else
+            @machine.env.ui.warn 'Unable to install the MySQL client'
+          end
+        end
+
+        def install_headers
+          if @machine.guest.capability?(:mysql_install_headers)
+            @machine.guest.capability(:mysql_install_headers)
+          else
+            @machine.env.ui.warn 'Unable to install MySQL header files'
+          end
+        end
+
+        def configure_client
+          @machine.guest.capability(:mysql_configure_client)
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/services/postgresql.rb

@@ -0,0 +1,39 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Services
+      class PostgreSQL < Service
+        def provision(machine)
+          super
+
+          @machine = machine
+
+          install_client
+          install_headers
+          export_host
+        end
+
+        protected
+
+        def install_client
+          if @machine.guest.capability?(:pg_install_client)
+            @machine.guest.capability(:pg_install_client)
+          else
+            @machine.env.ui.warn 'Unable to install the PostgreSQL client'
+          end
+        end
+
+        def install_headers
+          if @machine.guest.capability?(:pg_install_headers)
+            @machine.guest.capability(:pg_install_headers)
+          else
+            @machine.env.ui.warn 'Unable to install PostgreSQL header files'
+          end
+        end
+
+        def export_host
+          @machine.guest.capability(:pg_export_pghost)
+        end
+      end
+    end
+  end
+end

data/lib/ventriloquist/services/redis.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+  module Ventriloquist
+    module Services
+      class Redis < Service
+        def provision(machine)
+          super
+          install_client(machine)
+        end
+
+        protected
+
+        # TODO: Use the same version as the configured service
+        def install_client(machine)
+          return if machine.communicate.test('which redis-cli > /dev/null')
+
+          machine.guest.tap do |guest|
+            guest.capability(:install_build_tools)
+            guest.capability(:download, 'http://download.redis.io/redis-stable.tar.gz', '/tmp/redis-stable.tar.gz')
+            guest.capability(:untar, '/tmp/redis-stable.tar.gz', '/tmp')
+            guest.capability(:make, '/tmp/redis-stable', 'redis-cli')
+          end
+
+          machine.communicate.tap do |comm|
+            comm.sudo('cp /tmp/redis-stable/src/redis-cli /usr/local/bin')
+            comm.execute('rm -rf /tmp/redis-stable*')
+          end
+        end
+      end
+    end
+  end
+end