vito 0.0.2 → 0.0.4

data/lib/vito/output.rb

@@ -0,0 +1,21 @@
+module Vito
+  class ConnectionOutput
+    def initialize(stdin, stdout, stderr, thread)
+      @stdout = stdout
+      @stderr = stderr
+      @thread = thread.value
+    end
+
+    def success?
+      @thread.exitstatus == 0
+    end
+
+    def result
+      if success?
+        @result ||= @stdout.read
+      else
+        @result ||= @stderr.read
+      end
+    end
+  end
+end

data/lib/vito/recipe.rb

@@ -1,4 +1,6 @@
 module Vito
+
+  # This is a parent class for recipes and shouldn't be used directly.
   class Recipe
     def initialize(options, connection)
       @options = options
@@ -10,7 +12,7 @@ module Vito
     end
 
     def depends_on_recipe(recipe, options = {})
-      Installation.new(recipe, options, @connection).install
+      Dsl::Installation.new(recipe, options, @connection).install
     end
 
     def run_command(command)
@@ -28,10 +30,11 @@ module Vito
       Vito::Utils::ProgramVersion.new(version_command, @connection)
     end
 
-    private
+    def os
+      @os ||= Vito::OperatingSystem.new(@connection).os
+    end
 
     def install_os_dependencies
-      os = Vito::OperatingSystem.new(@connection)
       os.update_packages
       os.install_dependencies(os_dependencies) if self.respond_to?(:os_dependencies)
     end

data/lib/vito/recipes/postgres.rb

@@ -0,0 +1,104 @@
+module Vito
+  module Recipes
+    class Postgres < Vito::Recipe
+      def run
+        if installed?
+          Vito::Log.write "Postgres already installed."
+        else
+          Vito::Log.write "Installing Postgres"
+          install_os_dependencies
+          install
+        end
+      end
+
+      private
+
+      def postgres_installed?
+        @memo ||= query("psql --version").success?
+      end
+
+      def user_exists?
+        query("sudo su - postgres -c \"psql -tAc \\\\\"SELECT 1 FROM pg_roles WHERE rolname='#{username}'\\\\\" | grep -q 1 || echo 'hey'\"").result == ""
+      end
+
+      def utf_template?
+        query("sudo su - postgres -c \"psql -d template1 -c \\\\\"SHOW SERVER_ENCODING;\\\\\" | grep -q UTF || echo 'hey'\"").result == ""
+      end
+
+      def installed?
+        # Only verifies Postgres configuration if it's installed
+        if postgres_installed?
+          Vito::Log.write "User doesn't exist. I will create one." unless user_exists?
+          Vito::Log.write "template1 database's encoding should be UTF-8. I'll adjust this." unless utf_template?
+        end
+
+        postgres_installed? &&
+          user_exists? &&
+          utf_template?
+      end
+
+      def username
+        "vitouser"
+      end
+
+      def password
+        "123456"
+      end
+
+      def install
+        string = []
+        run_command "sudo apt-get install -y postgresql-9.1 libpq-dev"
+
+        # Create user only if it doesn't exist yet
+        unless user_exists?
+          string << "sudo su - postgres -c \"createuser -d -r -s #{username}\""
+        end
+
+        run_command(string.join(" && ")) unless string.empty?
+        convert_template1_to_utf8 unless utf_template?
+        string = []
+
+        if os.is?(:ubuntu, "10")
+          string << "sudo /etc/init.d/postgresql restart"
+        else
+          string << "sudo service postgresql restart"
+        end
+        run_command "sudo su - postgres -c \"psql -d template1 -c \\\\\"alter user #{username} with password '#{password}'\\\\\";\""
+
+        run_command(string.join(" && "))
+      end
+
+      def install_os_dependencies
+        return if postgres_installed?
+        Vito::Log.write "Installing some OS dependencies first"
+        if os.is?(:ubuntu, "10")
+          run_command("sudo apt-get remove -y postgresql")
+          run_command("sudo apt-get autoremove -y")
+          run_command("sudo apt-get install -y python-software-properties")
+          run_command("sudo add-apt-repository ppa:pitti/postgresql && sudo apt-get update")
+        end
+
+        super
+      end
+
+      def os_dependencies
+        %w(libpq-dev postgresql-contrib)
+      end
+
+      def convert_template1_to_utf8
+        # In some cases, the OS locale won't be compatible with UTF-8. When
+        # creating a database, Rails will try to use Postgres' template1 with
+        # UTF-8.
+        #
+        # Here, we delete template1, then login in template0 to recreate
+        # template1 with UTF-8 encoding.
+        run_command "sudo su - postgres -c \"psql -d template1 -c \\\\\"UPDATE pg_database SET datallowconn = TRUE where datname = 'template0';\\\\\"\""
+        run_command "sudo su - postgres -c \"psql -d template0 -c \\\\\"UPDATE pg_database SET datistemplate = FALSE where datname = 'template1'; \\\\\"\""
+        run_command "sudo su - postgres -c \"psql -d template0 -c \\\\\"drop database template1; \\\\\"\""
+        run_command "sudo su - postgres -c \"psql -d template0 -c \\\\\"create database template1 with template = template0 encoding = 'UNICODE'  LC_CTYPE = 'en_US.UTF-8' LC_COLLATE = 'C'; \\\\\"\""
+        run_command "sudo su - postgres -c \"psql -d template0 -c \\\\\" UPDATE pg_database SET datistemplate = TRUE where datname = 'template1'; \\\\\"\""
+        run_command "sudo su - postgres -c \"psql -d template1 -c \\\\\" UPDATE pg_database SET datallowconn = FALSE where datname = 'template0'; \\\\\"\""
+      end
+    end
+  end
+end

data/lib/vito/recipes/rbenv.rb

@@ -22,18 +22,23 @@ module Vito
       end
 
       def install_rbenv
-        puts "Installing rbenv..."
+        puts "Installing Rbenv..."
         string = []
-        string << "git clone git://github.com/sstephenson/rbenv.git ~/.rbenv"
-        string << "echo 'export PATH=\"\\$HOME/.rbenv/bin:\\$PATH\"' >> ~/.profile"
-        string << "echo 'eval \"$(rbenv init -)\"' >> ~/.profile"
-        string << "exec \\$SHELL -l"
+        # TODO verify if this folder already exists
+        run_command "git clone git://github.com/sstephenson/rbenv.git ~/.rbenv"
+
+        # TODO verify if .bashrc already have these lines
+        string = []
+        string << "echo 'eval \"$(rbenv init -)\"' | cat - ~/.bashrc > vitotemp && mv vitotemp ~/.bashrc"
+        string << "echo 'export PATH=\"\\$HOME/.rbenv/bin:\\$PATH\"' | cat - ~/.bashrc > vitotemp && mv vitotemp ~/.bashrc"
+
         run_command string.join(" && ")
       end
 
       def install_ruby_build
         puts "Installing ruby-build embedded into rbenv..."
         string = []
+        # TODO verify if this dir already exists
         string << "git clone https://github.com/sstephenson/ruby-build.git ~/.rbenv/plugins/ruby-build"
         run_command string.join(" && ")
       end

data/lib/vito/shell_initializer.rb

@@ -2,25 +2,20 @@ require "vito/recipes/ruby"
 
 module Vito
   class ShellInitializer
-    attr_reader :ssh
-
-    def initialize(args)
-      @installation_specs = File.open("vito.rb").read
+    def initialize(argv)
+      @argv = argv
     end
 
     def run
-      run_vito_file
-    end
-
-    # called by the vito.rb file
-    def server(*args, &block)
-      Vito::Server.new.instance_eval &block
+      Vito::DslFile.new(argv).run(vito_file)
     end
 
     private
 
-    def run_vito_file
-      instance_eval @installation_specs
+    attr_reader :argv
+
+    def vito_file
+      File.open("vito.rb").read
     end
   end
 end

data/lib/vito/version.rb

@@ -1,3 +1,3 @@
 module Vito
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

data/spec/acceptance/recipes/git_acceptance_spec.rb

@@ -0,0 +1,20 @@
+require "spec_helper"
+
+describe "Git recipe" do
+  before do
+    setup_vagrant
+  end
+
+  it "installs Git" do
+    assert_installation("git --version").should be_false
+
+    Vito::DslFile.new.run do
+      server do
+        connection :ssh, command: 'ssh -i ~/.vagrant.d/insecure_private_key vagrant@localhost -p2222', silent: true
+        install :git
+      end
+    end
+
+    assert_installation("git --version").should be_true
+  end
+end

data/spec/acceptance/recipes/postgres_acceptance_spec.rb

@@ -0,0 +1,20 @@
+require "spec_helper"
+
+describe "Postgres recipe" do
+  before do
+    setup_vagrant
+  end
+
+  it "installs Postgres" do
+    assert_installation("psql --version").should be_false
+
+    Vito::DslFile.new.run do
+      server do
+        connection :ssh, command: 'ssh -i ~/.vagrant.d/insecure_private_key vagrant@localhost -p2222', silent: true
+        install :postgres
+      end
+    end
+
+    assert_installation("psql --version").should be_true
+  end
+end

data/spec/acceptance/recipes/rbenv_acceptance_spec.rb

@@ -0,0 +1,22 @@
+require "spec_helper"
+
+describe "Rbenv recipe" do
+  before do
+    setup_vagrant
+  end
+
+  it "installs Git and then Rbenv" do
+    assert_installation("git --version").should be_false
+    assert_installation("rbenv --version").should be_false
+
+    Vito::DslFile.new.run do
+      server do
+        connection :ssh, command: 'ssh -i ~/.vagrant.d/insecure_private_key vagrant@localhost -p2222', silent: true
+        install :rbenv
+      end
+    end
+
+    assert_installation("git --version").should be_true
+    assert_installation("rbenv --version").should be_true
+  end
+end

data/spec/acceptance/recipes/ruby_acceptance_spec.rb

@@ -0,0 +1,24 @@
+require "spec_helper"
+
+describe "Ruby recipe" do
+  before do
+    setup_vagrant
+  end
+
+  it "installs Git, Rbenv and then Ruby" do
+    assert_installation("git --version").should be_false
+    assert_installation("rbenv --version").should be_false
+    assert_installation("ruby -v").should be_false
+
+    Vito::DslFile.new.run do
+      server do
+        connection :ssh, command: 'ssh -i ~/.vagrant.d/insecure_private_key vagrant@localhost -p2222', silent: true
+        install :ruby, version: "1.9.3-p125"
+      end
+    end
+
+    assert_installation("git --version").should be_true
+    assert_installation("rbenv --version").should be_true
+    assert_installation("ruby -v").should be_true
+  end
+end

data/spec/spec_helper.rb

@@ -1,2 +1,12 @@
+ENV["env"] ||= "test"
+$:.unshift File.expand_path('../../lib', __FILE__)
 require "rubygems"
 require "bundler/setup"
+require "open3"
+require "vito"
+
+Dir["./spec/support/*.rb"].each { |f| require f }
+
+RSpec.configure do |config|
+  config.include RSpecSupport::Vagrant
+end

data/spec/support/vagrant.rb

@@ -0,0 +1,34 @@
+module RSpecSupport
+  module Vagrant
+    def connection
+      'ssh -i ~/.vagrant.d/insecure_private_key vagrant@localhost -p2222'
+    end
+
+    def setup_vagrant
+      system_command("vagrant up")
+      raise "Box not working" unless vagrant_command("ls")
+      reboot_vagrant_box
+    end
+
+    def assert_installation(command)
+      vagrant_command(command)
+    end
+
+    def reboot_vagrant_box
+      system_command("vagrant snapshot go initial_box")
+    end
+
+    private
+
+    def system_command(command)
+      stdin, stdout, stderr, thread = Open3.popen3(command)
+      thread.value.exitstatus == 0
+    end
+
+    def vagrant_command(command)
+      command = "#{connection} #{command}"
+      stdin, stdout, stderr, thread = Open3.popen3(command)
+      thread.value.exitstatus == 0
+    end
+  end
+end

data/spec/vito/connection_spec.rb

@@ -1,6 +1,4 @@
 require "spec_helper"
-require "vito/connection"
-require "vito/log"
 
 describe Vito::Connection do
   before do
@@ -20,8 +18,13 @@ describe Vito::Connection do
     context "invalid command" do
       let(:command) { "harrr" }
 
-      its(:success?) { should be_false }
-      its(:result)   { should == "sh: harrr: command not found\n" }
+      it "raises error and logs its messages" do
+        Vito::Log.should_receive(:raise).with("An error occurred. Here's the stacktrace:")
+        Vito::Log.should_receive(:raise).with("sh: harrr: command not found\n")
+        Vito::Log.stub(:raise)
+
+        expect{ subject }.to raise_error "Error."
+      end
     end
   end
 end

data/spec/vito/dsl/installation_spec.rb

@@ -0,0 +1,17 @@
+require "spec_helper"
+
+describe Vito::Dsl::Installation do
+  let(:connection) { double }
+  let(:ruby)       { double }
+  let(:options)    { {version: "2.0.0"} }
+
+  subject { described_class.new(:ruby, options, connection) }
+
+  describe "#install" do
+    it "installs each recipes" do
+      ruby.should_receive(:run)
+      Vito::Recipes::Ruby.stub(:new).with(options, connection) { ruby }
+      subject.install
+    end
+  end
+end

data/spec/vito/dsl_file_spec.rb

@@ -0,0 +1,37 @@
+require "vito/dsl_file"
+
+describe Vito::DslFile do
+  let(:server) { double }
+
+  subject { described_class.new }
+
+  before do
+    stub_const("Vito::Dsl::Server", Class.new)
+  end
+
+  describe "#run" do
+    before do
+      Vito::Dsl::Server.stub(:new).with([:hey]) { server }
+    end
+
+    it "runs the server block evaluating it as string" do
+      server.should_receive(:instance_eval)
+
+      subject.run <<-str
+        server :hey do
+
+        end
+      str
+    end
+
+    it "runs the server as a Ruby block" do
+      server.should_receive(:instance_eval)
+
+      subject.run do
+        server :hey do
+
+        end
+      end
+    end
+  end
+end