sous_chef 0.0.1

data/.gitignore

@@ -0,0 +1,25 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+gbin
+coverage
+doc
+rdoc
+pkg
+vendor/gems
+.yardoc
+
+## PROJECT::SPECIFIC

data/Gemfile

@@ -0,0 +1,18 @@
+bin_path 'gbin'
+disable_system_gems
+
+only :runtime do
+end
+ 
+only :development do
+  gem 'bundler'
+  gem 'jeweler'
+  gem 'rspec', :require_as => 'spec'
+  gem 'ZenTest'
+  gem 'rake'
+  gem 'rcov'
+  gem 'ruby-debug'
+  gem 'yard'
+  gem 'reek'
+  gem 'roodi'
+end

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Martin Emde
+
+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.rdoc

@@ -0,0 +1,6 @@
+= Sous Chef
+
+Chef's prep-cook
+
+Create bash scripts with chef-like syntax
+

data/Rakefile

@@ -0,0 +1,76 @@
+require 'bundler'
+Bundler.require_env
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "sous_chef"
+    gem.summary = %Q{Chef's prep-cook}
+    gem.description = %Q{Create bash scripts using a simple chef-like syntax}
+    gem.email = ["memde@engineyard.com", "bdonovan@engineyard.com"]
+    gem.homepage = "http://github.com/engineyard/sous_chef"
+    gem.authors = ["Martin Emde", "Brian Donovan"]
+
+    bundle = Bundler::Bundle.load(File.dirname(__FILE__) + '/Gemfile')
+    bundle.environment.dependencies.each do |d|
+      if d.only && d.only.include?('runtime')
+        gem.add_dependency d.name, d.version.to_s
+      end
+    end
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+  spec.rcov_opts << '--exclude' << 'spec,vendor,Library'
+end
+
+task :spec => :check_dependencies
+
+begin
+  require 'reek/adapters/rake_task'
+  Reek::RakeTask.new do |t|
+    t.fail_on_error = true
+    t.verbose = false
+    t.source_files = 'lib/**/*.rb'
+  end
+rescue LoadError
+  task :reek do
+    abort "Reek is not available. In order to run reek, you must: sudo gem install reek"
+  end
+end
+
+begin
+  require 'roodi'
+  require 'roodi_task'
+  RoodiTask.new do |t|
+    t.verbose = false
+  end
+rescue LoadError
+  task :roodi do
+    abort "Roodi is not available. In order to run roodi, you must: sudo gem install roodi"
+  end
+end
+
+task :default => :spec
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION.yml

@@ -0,0 +1,5 @@
+--- 
+:major: 0
+:minor: 0
+:patch: 1
+

data/lib/sous_chef/recipe.rb

@@ -0,0 +1,114 @@
+module SousChef
+  class Recipe
+    attr_accessor :node, :verbose, :shebang
+    alias_method :verbose?, :verbose
+    alias_method :shebang?, :shebang
+
+    def initialize(*flags, &block)
+      flags.each {|flag| __send__("#{flag}=", true)}
+      @resources = []
+      @block = block
+    end
+
+    def self.load(file)
+      new { instance_eval(File.read(file), file, 1) }
+    end
+
+    def setup
+      if @block
+        instance_eval &@block
+        @block = nil
+      end
+    end
+
+    def to_script
+      @script ||= begin
+        setup
+        lines = []
+        lines << "#!/bin/bash" if shebang?
+        lines += @resources.map do |resource|
+          # @context = resource
+          script = ""
+          script << %{# #{resource.name}\n} if verbose? && resource.name
+          script << resource.to_script
+        end
+        lines.join("\n\n")
+      end
+    end
+
+    def echo(string)
+      command "echo '#{escape_string(string)}'"
+    end
+
+    def execute(*args, &block)
+      add_resource Resource::Execute.new(self, *args, &block)
+    end
+
+    def file(*args, &block)
+      add_resource Resource::File.new(self, *args, &block)
+    end
+
+    def directory(*args, &block)
+      add_resource Resource::Directory.new(self, *args, &block)
+    end
+
+    def log(*args, &block)
+      add_resource Resource::Log.new(self, *args, &block)
+    end
+
+    def gemfile(*args, &block)
+      add_resource Resource::Gemfile.new(self, *args, &block)
+    end
+
+    def command(cmd)
+      if context
+        context.command(cmd)
+      else
+        execute { command cmd }
+      end
+    end
+
+    def halt_on_failed_command
+      execute "halt on failed command" do
+        command "set -e"
+      end
+    end
+
+    protected
+      def add_resource(resource)
+        with_context(resource) do
+          @resources << resource
+          resource.setup
+        end
+        resource
+      end
+
+      def context
+        @context
+      end
+
+      def with_context(resource)
+        @context = resource
+        yield
+        @context = nil
+      end
+
+      def escape_path(path)
+        path
+      end
+
+      def escape_string(string)
+        # many slashes because single-quote has some sort of
+        # special meaning in regexp replacement strings
+        string && string.gsub(/'/, %q{\\\\'})
+      end
+
+      def method_missing(meth, *args, &block)
+        if context && context.resource_respond_to?(meth)
+          context.__send__(meth, *args, &block)
+        else
+          super
+        end
+      end
+  end
+end

data/lib/sous_chef/resource/base.rb

@@ -0,0 +1,79 @@
+module SousChef
+  module Resource
+    class Base
+      attr_reader :name
+
+      def initialize(context, name, &block)
+        @context = context
+        @name = name
+        @block = block
+        @only_if = nil
+        @commands = []
+      end
+
+      def setup
+        if @block
+          instance_eval &@block
+          @block = nil
+        end
+      end
+
+      def not_if(cmd=nil)
+        only_if "! #{cmd}"
+      end
+
+      def only_if(cmd=nil)
+        if cmd
+          @only_if = cmd
+        else
+          @only_if
+        end
+      end
+
+      alias_method :resource_respond_to?, :respond_to?
+      def respond_to?(meth)
+        super || context.respond_to?(meth)
+      end
+
+      def prepend(cmd)
+        @commands.unshift(cmd)
+      end
+
+      def append(cmd)
+        @commands.push(cmd)
+      end
+
+      alias_method :command, :append
+
+      def to_script
+        setup
+        if only_if
+          @commands.compact!
+          @commands.map! { |line| "  #{line}" }
+          prepend "if #{only_if}; then"
+          append  "fi"
+        end
+        @commands.join("\n").strip
+      end
+
+      protected
+        attr_reader :context
+
+        def set_or_return(attr, val)
+          if val.nil?
+            instance_variable_get("@#{attr}")
+          else
+            instance_variable_set("@#{attr}", val)
+          end
+        end
+
+        def method_missing(meth, *args, &block)
+          if context.respond_to?(meth)
+            context.__send__(meth, *args, &block)
+          else
+            super
+          end
+        end
+    end
+  end
+end

data/lib/sous_chef/resource/directory.rb

@@ -0,0 +1,52 @@
+module SousChef
+  module Resource
+    class Directory < Base
+      ACTIONS = %w[create delete]
+
+      def initialize(*args)
+        action :create
+        super
+      end
+
+      def path(path=nil)
+        set_or_return(:path, path) || name
+      end
+
+      def mode(mode=nil)
+        set_or_return(:mode, mode)
+      end
+
+      def action(action=nil)
+        set_or_return(:action, action && validate_action(action))
+      end
+
+      def to_script
+        @script ||= begin
+          setup
+          __send__(action)
+          [super, mode_command].compact.join("\n")
+        end
+      end
+
+      protected
+        def create
+          command %{mkdir -p #{escape_path(path)}}
+        end
+
+        def delete
+          command %{rmdir #{escape_path(path)}}
+        end
+
+        def validate_action(action)
+          return action if ACTIONS.include?(action.to_s)
+          raise ArgumentError, "Invalid action #{action}, only #{ACTIONS.join(', ')} allowed"
+        end
+
+        def mode_command
+          if mode
+            sprintf(%{chmod %04o %s}, mode, escape_path(path))
+          end
+        end
+    end
+  end
+end

data/lib/sous_chef/resource/execute.rb

@@ -0,0 +1,33 @@
+module SousChef
+  module Resource
+    class Execute < Base
+      def initialize(context, name=nil, &block)
+        super
+
+        @cwd = nil
+      end
+
+      def to_script
+        @script ||= begin
+          prepend %{cd #{escape_path(@cwd)}} if @cwd
+          @commands = @commands.inject([]) do |result, line|
+            result + line.split("\n")
+          end
+          super
+        end
+      end
+
+      def cwd(dir=nil)
+        set_or_return(:cwd, dir)
+      end
+
+      def command(cmd)
+        @commands << cmd
+      end
+
+      def creates(path)
+        not_if %{test -e #{escape_path(path)}}
+      end
+    end
+  end
+end

data/lib/sous_chef/resource/file.rb

@@ -0,0 +1,36 @@
+module SousChef
+  module Resource
+    class File < Directory
+      def content(content=nil)
+        set_or_return(:content, content)
+      end
+
+      protected
+        def escaped_content
+          escape_string(content)
+        end
+
+        def create
+          not_if file_exist_command
+          command create_file_command
+        end
+
+        def delete
+          only_if file_exist_command
+          command "rm #{escape_path(path)}"
+        end
+
+        def file_exist_command
+          %{test -e #{escape_path(path)}}
+        end
+
+        def create_file_command
+          if content
+            %{echo '#{escaped_content}' > #{escape_path(path)}}
+          else
+            %{touch #{escape_path(path)}}
+          end
+        end
+    end
+  end
+end

data/lib/sous_chef/resource/gemfile.rb

@@ -0,0 +1,56 @@
+require 'stringio'
+
+module SousChef
+  module Resource
+    class Gemfile < File
+      def initialize(context, name, &block)
+        super
+        @gems = []
+        @sources = []
+      end
+
+      def gem(name, version=nil)
+        @gems << [name, version]
+      end
+
+      def source(url)
+        @sources << url
+      end
+
+      # override
+      def path(path=nil)
+        if path
+          super
+        else
+          gemfile = super
+          gemfile = "#{gemfile}/Gemfile" unless gemfile.split('/').last == "Gemfile"
+          gemfile
+        end
+      end
+
+      protected
+        # override
+        def content
+          result = StringIO.new
+          if @sources.any?
+            @sources.each { |url| result.puts %{source "#{url}"} }
+            result.puts
+          end
+
+          max_name_size = 0
+          @gems.each {|name,| max_name_size = [max_name_size, name.size].max}
+
+          @gems.sort_by {|name,| name.downcase}.each do |name, version|
+            if version.nil?
+              result.puts %{gem "#{name}"}
+            else
+              width = max_name_size + 3 # 2 quotes + comma
+              result.printf %{gem %-#{width}s "%s"\n}, %{"#{name}",}, version
+            end
+          end
+
+          result.string.strip
+        end
+    end
+  end
+end

data/lib/sous_chef/resource/log.rb

@@ -0,0 +1,39 @@
+module SousChef
+  module Resource
+    class Log < Base
+      def initialize(context, name=nil, &block)
+        super
+
+        @stdout = nil
+        @stderr = nil
+      end
+
+      def stdout(stdout=nil)
+        set_or_return(:stdout, stdout)
+      end
+
+      def stderr(stderr=nil)
+        set_or_return(:stderr, stderr)
+      end
+
+      def to_script
+        @script ||= begin
+          setup
+          unless @stdout || @stderr
+            @stdout, @stderr = name, "&1"
+          end
+          append exec_command(escape_path(@stdout), escape_path(@stderr))
+          super
+        end
+      end
+
+      protected
+        def exec_command(stdout, stderr)
+          args = []
+          args << "1>#{escape_path(stdout)}" if stdout
+          args << "2>#{escape_path(stderr)}" if stderr
+          "exec #{args.join(' ')}"
+        end
+    end
+  end
+end

data/lib/sous_chef/resource.rb

@@ -0,0 +1,10 @@
+module SousChef
+  module Resource
+    autoload :Base,      'sous_chef/resource/base'
+    autoload :Directory, 'sous_chef/resource/directory'
+    autoload :Execute,   'sous_chef/resource/execute'
+    autoload :File,      'sous_chef/resource/file'
+    autoload :Gemfile,   'sous_chef/resource/gemfile'
+    autoload :Log,       'sous_chef/resource/log'
+  end
+end

data/lib/sous_chef.rb

@@ -0,0 +1,9 @@
+module SousChef
+  autoload :Recipe,   'sous_chef/recipe'
+  autoload :Resource, 'sous_chef/resource'
+
+  def self.prep(*flags, &block)
+    recipe = Recipe.new(*flags, &block)
+    recipe.to_script
+  end
+end

data/spec/fixtures/deploy_command.rb

@@ -0,0 +1,102 @@
+ruby_version = 'ruby-1.8.6'
+rubygems_version = '1.3.5'
+home_dir = '/home/sous_chef'
+
+halt_on_failed_command
+
+log do
+  stdout "/root/stdout.log"
+  stderr "/root/stderr.log"
+end
+
+file "/root/report.log" do
+  action :delete
+end
+
+def report(message)
+  command "echo '#{escape_string(message)}' >> /root/report.log"
+end
+
+file "/etc/config.yml" do
+  content node[:config].to_yaml
+  mode 0600
+end
+
+execute 'rvm' do
+  report "Installing ruby version manager"
+  creates "/usr/local/rvm/scripts/rvm"
+  command "gem install rvm && rvm-install"
+end
+
+execute "source rvm" do
+  command <<-EOS
+RUBYOPT=""
+source /usr/local/rvm/scripts/rvm
+  EOS
+end
+
+file "/etc/profile.d/rvm.sh" do
+  debugger
+  echo "Setting up rvm source"
+  content <<-EOS
+# rvm configuration
+RUBYOPT=""
+if [[ -s /usr/local/rvm/scripts/rvm ]] ; then source /usr/local/rvm/scripts/rvm ; fi
+  EOS
+end
+
+echo 'installing ruby'
+
+execute 'install ruby' do
+  command %{rvm install #{ruby_version}}
+  not_if "rvm list | grep #{ruby_version}"
+end
+
+execute 'use ruby' do
+  command "rvm use #{ruby_version}"
+end
+
+execute 'update rubygems' do
+  report "upgrading rubygems"
+  command <<-EOS
+gem uninstall rubygems-update
+gem install rubygems-update -v #{rubygems_version} --no-ri --no-rdoc
+update_rubygems
+gem uninstall rubygems-update
+  EOS
+  not_if "gem -v | grep #{rubygems_version}"
+end
+
+directory home_dir do
+  mode 0755
+end
+
+gemfile home_dir do
+  source "http://gemcutter.org/"
+
+  gem 'chef'
+  gem 'dbi',          '0.4.3'
+  gem 'dbd-mysql',    '0.4.3'
+  gem 'open4',        '0.9.6'
+end
+
+execute 'install bundler' do
+  command "gem install bundler --no-ri --no-rdoc"
+end
+
+execute 'bundle gems' do
+  cwd home_dir
+  command "gem bundle"
+end
+
+file "/etc/profile.d/bin-path.sh" do
+  echo "Setting up bin-path"
+  content <<-EOS
+# bundled gem bin path configuration
+export PATH=#{home_dir}/bin:\$PATH
+  EOS
+end
+
+execute 'run chef' do
+  command "nohup #{home_dir}/bin/chef #{node[:chef_args]} &"
+end