capchef 0.0.1

data/lib/capchef.rb

@@ -0,0 +1,57 @@
+module Capchef
+  extend self
+
+  # Default to /tmp but allow change in case it is noexec
+  def tmpdir
+    @tmpdir ||= '/tmp'
+  end
+
+  def tmpdir=(tmpdir)
+    @tmpdir = tmpdir
+  end
+
+  def path
+    @path ||= '/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin'
+  end
+
+  def path=(path)
+    @path = path
+  end
+
+  def prepend_path(dir)
+    @path = "#{dir}:#{path}"
+  end
+
+  # Runs +command+ as root invoking the command with su -c
+  # and handling the root password prompt.
+  #
+  #   surun "/etc/init.d/apache reload"
+  #   # Executes
+  #   # su - -c '/etc/init.d/apache reload'
+  #
+  def surun(cap, command, options={})
+    @root_password ||= cap.fetch(:root_password, Capistrano::CLI.password_prompt("root password: "))
+    cap.run("su -c 'cd; PATH=#{path}; #{command}'", options) do |channel, stream, output|
+      puts "[#{channel[:host]}] #{output}" if output
+      channel.send_data("#{@root_password}\n") if output && output =~ /^Password:/
+      yield channel, stream, output if block_given?
+    end
+  end
+
+  def surun_script(cap, script, options={})
+    # TBD - We can change this to use /tmp as its no longer noexec
+    raise "No such file: #{script}" unless File.exist?(script)
+    basename = File.basename(script)
+    tmpdir = "/tmp/#{basename}.#$$"
+    remote_script = "#{tmpdir}/#{basename}"
+    cap.run "mkdir #{tmpdir}", options
+    cap.upload script, remote_script, options
+    cap.run "chmod 0755 #{remote_script}", options
+    if block_given?
+      yield remote_script
+    else
+      surun cap, remote_script, options
+    end
+    cap.run "rm -rf #{tmpdir}", options
+  end
+end

data/lib/capchef_recipes.rb

@@ -0,0 +1,83 @@
+require 'erb'
+require 'yaml'
+require 'json'
+require 'tempfile'
+require 'zlib'
+require 'archive/tar/minitar'
+require 'capchef'
+include Archive::Tar
+  
+Capistrano::Configuration.instance.load do
+  # User settings
+  #set :user, 'deploy' unless exists?(:user)
+  #set :group,'www-data' unless exists?(:group)
+  
+  # Git settings for capistrano
+  default_run_options[:pty] = true
+  ssh_options[:forward_agent] = true
+
+  namespace :chef do
+    desc 'Install chef recipes on remote machines.'
+    task :default do
+      Capchef.prepend_path(chef_solo_path) if exists?(:chef_solo_path)
+
+      raise 'No nodes.yml' unless File.exist?('nodes.yml')
+      config = YAML.load(ERB.new(File.read('nodes.yml')).result(binding))
+
+      remote_tmpdir = "/tmp/chef_solo.#$$"
+      run "mkdir #{remote_tmpdir}"
+      remote_node_file = "#{remote_tmpdir}/node.json"
+      remote_solo_file = "#{remote_tmpdir}/solo.rb"
+      remote_cookbooks_tgz = "#{remote_tmpdir}/cookbooks.tgz"
+      remote_roles_tgz = "#{remote_tmpdir}/roles.tgz"
+
+      valid_hosts = []
+      find_servers_for_task(current_task).each do |s|
+        host_config = config[s.host]
+        if host_config
+          valid_hosts << s.host
+          put host_config.to_json, remote_node_file, :hosts => s.host
+        else
+          $stderr.puts "WARNING: #{s.host} not configured in nodes.yml, skipping"
+        end
+      end
+      if valid_hosts.empty?
+        $stderr.puts 'No configured hosts'
+        exit 1
+      end
+
+      solo_rb = "file_cache_path '/etc/chef'\ncookbook_path '/etc/chef/cookbooks'\nrole_path '/etc/chef/roles'\n"
+      put solo_rb, remote_solo_file, :hosts => valid_hosts
+      begin
+        tmp_cookbooks_tgz = Tempfile.new('cookbooks')
+        Dir.chdir('chef-repo') do
+          Minitar.pack('cookbooks', Zlib::GzipWriter.new(tmp_cookbooks_tgz))
+          tmp_cookbooks_tgz.close
+          upload tmp_cookbooks_tgz.path, remote_cookbooks_tgz, :hosts => valid_hosts
+
+          sio_roles_tgz = StringIO.new
+          gzip = Zlib::GzipWriter.new(sio_roles_tgz)
+          Minitar::Writer.open(gzip) do |tar|
+          #Minitar::Writer.open(sio_roles_tgz) do |tar|
+            Find.find('roles') do |role_file|
+              # TODO: This does not work (how do I specify a role in JSON?
+              if role_file.match(/\.yml$/)
+                data = YAML.load(ERB.new(File.read(role_file)).result(binding)).to_json
+                new_role_file = role_file.sub(/\.yml$/, '.json')
+                tar.add_file_simple(new_role_file, :size=>data.size, :mode=>0644, :mtime=>File.mtime(role_file)) { |f| f.write(data) }
+              else
+                Minitar.pack_file(role_file, tar)
+              end
+            end
+          end
+          gzip.close
+          put sio_roles_tgz.string, remote_roles_tgz, :hosts => valid_hosts
+        end
+        Capchef.surun(self, "mkdir -p /etc/chef; cd /etc/chef; rm -rf cookbooks roles; tar zxf #{remote_cookbooks_tgz}; tar zxf #{remote_roles_tgz}; chef-solo -c #{remote_solo_file} -j #{remote_node_file}", :hosts => valid_hosts)
+      ensure
+        tmp_cookbooks_tgz.unlink
+        run "rm -rf #{remote_tmpdir}"
+      end
+    end
+  end
+end

data/lib/capchef_utility_recipes.rb

@@ -0,0 +1,15 @@
+require 'capchef'
+
+Capistrano::Configuration.instance.load do
+  default_run_options[:pty] = true
+  ssh_options[:forward_agent] = true
+
+  namespace :utility do
+    desc "execute basic command line functions. ie. cap exec cmd='ps -eaf | grep redis' optional arg: HOSTS=ip1,ip2 (replace cmd with sudo for sudo commands)"
+    task :default do
+      run(ENV['cmd']) if ENV['cmd']
+      Capchef.surun(self, ENV['sudo']) if ENV['sudo']
+    end
+  end
+end
+

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: capchef
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+  - Brad Pardee
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-09-03 00:00:00 -04:00
+default_executable: 
+dependencies: 
+  - !ruby/object:Gem::Dependency 
+    name: capistrano
+    prerelease: false
+    requirement: &id001 !ruby/object:Gem::Requirement 
+      none: false
+      requirements: 
+        - - ">="
+          - !ruby/object:Gem::Version 
+            version: "0"
+    type: :runtime
+    version_requirements: *id001
+  - !ruby/object:Gem::Dependency 
+    name: json
+    prerelease: false
+    requirement: &id002 !ruby/object:Gem::Requirement 
+      none: false
+      requirements: 
+        - - ">="
+          - !ruby/object:Gem::Version 
+            version: "0"
+    type: :runtime
+    version_requirements: *id002
+  - !ruby/object:Gem::Dependency 
+    name: minitar
+    prerelease: false
+    requirement: &id003 !ruby/object:Gem::Requirement 
+      none: false
+      requirements: 
+        - - ">="
+          - !ruby/object:Gem::Version 
+            version: "0"
+    type: :runtime
+    version_requirements: *id003
+description: Chef capistrano recipes so you can configure your machines without a server
+email: 
+  - bradpardee@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+  - lib/capchef.rb
+  - lib/capchef_recipes.rb
+  - lib/capchef_utility_recipes.rb
+has_rdoc: true
+homepage: http://github.com/ClarityServices/capchef
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+  - lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.1
+signing_key: 
+specification_version: 3
+summary: Chef capistrano recipes
+test_files: []
+