capitate 0.1.7

data/lib/capitate/plugins/gem.rb

@@ -0,0 +1,25 @@
+module Capitate::Plugins::Gem
+  
+  # Install a gem.
+  #
+  # ==== Options
+  # +gems+:: List of gem names, or a single gem
+  #
+  # ==== Examples (in capistrano task)
+  #   gemc.install("raspell") 
+  #   gemc.install([ "raspell", "foo" ])
+  #
+  def install(gems)
+    # If a single object, wrap in array
+    gems = [ gems ] unless gems.is_a?(Array)
+    
+    # Install one at a time because we may need to pass install args (e.g. mysql)
+    # TODO: Fix this
+    gems.each do |gem|
+      sudo "gem install --no-rdoc --no-ri --no-verbose #{gem}"
+    end
+  end
+  
+end
+
+Capistrano.plugin :gemc, Capitate::Plugins::Gem

data/lib/capitate/plugins/package.rb

@@ -0,0 +1,30 @@
+# Package plugin.
+#
+# TODO: If type not set silently fails. NOT GOOD :(
+#
+module Capitate::Plugins::Package
+  
+  # Setup packager.
+  #
+  # ==== Options
+  # +packager_type+:: Packager type (:yum)
+  #
+  # ==== Examples (in capistrano task)
+  #   package.type = :yum
+  #   package.install [ "aspell", "foo" ]
+  #   package.remove [ "aspell", "foo" ]
+  #   package.update [ "aspell", "foo" ]
+  #   package.clean
+  #
+  def type=(packager_type)
+    case packager_type.to_sym
+    when :yum
+      include Capitate::Plugins::Yum
+    else
+      raise "Invalid packager type: #{packager_type}"
+    end    
+  end  
+    
+end
+
+Capistrano.plugin :package, Capitate::Plugins::Package

data/lib/capitate/plugins/profiles.rb

@@ -0,0 +1,33 @@
+# == Profiles
+# Useful for starting a profile, but your mileage may vary.
+#
+module Capitate::Plugins::Profiles
+  
+  ProfileDir = File.dirname(__FILE__) + "/../../profiles"
+  
+  # Load all profiles with prefix.
+  #
+  # ==== Options
+  # +prefix+:: Prefix to filter/search on
+  #  
+  def list(prefix = "")
+    Dir[ProfileDir + "/#{prefix}*.rb"].collect { |file| File.basename(file).gsub(File.extname(file), "") }
+  end
+  
+  # Choose a profile (via command line).
+  # 
+  # ==== Options
+  # +prefix+:: Prefix to filter/search on
+  #
+  def choose(prefix = "")
+    profile = HighLine.new.choose(*list(prefix)) do |menu|
+      menu.header = "Choose recipe profile"
+    end
+    
+    "#{ProfileDir}/#{profile}.rb"
+  end
+
+end
+    
+    
+Capistrano.plugin :profile, Capitate::Plugins::Profiles

data/lib/capitate/plugins/script.rb

@@ -0,0 +1,118 @@
+module Capitate::Plugins::Script
+  
+  # Configure, make, make install.
+  #
+  # ==== Options
+  # +name+:: Name for app
+  # +options+:: Options 
+  # - +build_dest+:: Place to build from, ex. /usr/src, defaults to /tmp/name
+  # - +url+:: URL to download package from
+  # - +configure_options+:: Options for ./configure
+  # - +unpack_dir+:: Directory that is unpacked from tgz (if not matching the file name)
+  #
+  # ==== Examples (in capistrano task)
+  #   script.make_install("nginx", { :url => "http://sysoev.ru/nginx/nginx-0.5.35.tar.gz", ... })
+  #
+  def make_install(name, options)
+    install(name, options) do |dir|
+      configure_options = options[:configure_options] || ""
+      sudo "echo 'Configuring #{name}...' && cd #{dir} && ./configure #{configure_options} > configure.log"
+      sudo "echo 'Compiling #{name}...' && cd #{dir} && make > make.log"
+      sudo "echo 'Installing #{name}...' && cd #{dir} && make install > make_install.log"        
+    end
+  end
+  
+  # Download, unpack and yield (unpacked source director).
+  # 
+  # ==== Options
+  # +name+:: Name for app
+  # +options+:: Options
+  # - +build_dest+:: Place to build from, ex. /usr/src, defaults to /tmp/name
+  # - +url+:: URL to download package from
+  # 
+  # ==== Examples (in capistrano task)
+  #   script.make("rubygems", { :url => :url => "http://rubyforge.org/frs/download.php/29548/rubygems-1.0.1.tgz" }) do |dir|
+  #     sudo "echo 'Running setup...' && cd #{dir} && ruby #{dir}/setup.rb"
+  #   end
+  #
+  def install(name, options, &block)
+    build_dest = options[:build_dest]
+    build_dest ||= "/tmp/#{name}"
+    url = options[:url]    
+    dependencies = options[:dependencies]
+    
+    package.install(dependencies) if dependencies
+    
+    unpack(url, build_dest, options[:clean], options[:unpack_dir]) do |dir|
+      yield(dir) if block_given?
+    end      
+  end
+    
+  
+  # Run (sh) script. If script has .erb extension it will evaluate it.
+  #
+  # ==== Options
+  # +script+:: Name of sh file (relative to templates dir)
+  # +override_binding++:: Binding to override, otherwise defaults to current (task) context
+  #
+  # ==== Examples
+  #   script.sh("ruby/openssl_fix.sh")
+  #
+  def sh(script, override_binding = nil)
+        
+    if File.extname(script) == ".erb"
+      name = script[0...script.length-4]
+      dest = "/tmp/#{name}"
+      run "mkdir -p #{File.dirname(dest)}"
+      put template.load(script, override_binding || binding), dest
+    else
+      name = script
+      dest = "/tmp/#{name}"
+      run "mkdir -p #{File.dirname(dest)}"
+      put template.load(script), dest
+    end
+    
+    # If want verbose, -v
+    sudo "sh -v #{dest}"
+    
+    # Cleanup
+    sudo "rm -rf #{File.dirname(dest)}"    
+  end
+  
+  # Download and unpack URL.
+  # Yields path to unpacked source.
+  #
+  # ==== Options
+  # +url+:: URL to download
+  # +dest+:: Destination directory
+  # +clean+:: If true will remove the unpacked directory
+  # +unpack_dir+:: Directory that is unpacked from tgz (if not matching the file name)
+  #
+  # ==== Examples
+  #   script.unpack("http://rubyforge.org/frs/download.php/29548/rubygems-1.0.1.tgz", "/tmp/rubygems") do
+  #     sudo "ruby setup.rb"
+  #   end
+  #
+  def unpack(url, dest, clean = true, unpack_dir = nil, &block)        
+    file = url.split("/").last
+    
+    # TODO: Support other types
+    if file !~ /\.tar\.gz\Z|\.tgz\Z/
+      raise "Can't unpack this file: #{file}; only support tar.gz and tgz formats"
+    end
+    
+    unpack_dir ||= file.gsub(/\.tar\.gz|\.tgz/, "")
+    
+    sudo "echo 'Getting #{url}...' && mkdir -p #{dest} && cd #{dest} && wget -nv #{url}"
+    sudo "echo 'Unpacking...' && cd #{dest} && tar zxf #{file}"
+    
+    if block_given?
+      yield("#{dest}/#{unpack_dir}")
+      sudo "rm -f #{dest}/#{file}"
+      sudo "rm -rf #{dest}" if clean
+    end
+  end
+  
+end
+
+Capistrano.plugin :script, Capitate::Plugins::Script

data/lib/capitate/plugins/templates.rb

@@ -0,0 +1,98 @@
+module Capitate::Plugins::Templates
+  
+  # Get the absolute base templates path.
+  def gem_templates_root
+    File.expand_path(File.dirname(__FILE__) + "/../../templates")
+  end
+  
+  # Get full template path from relative path.
+  #
+  # Something like <tt>monit/monit.cnf => /usr/lib/..../capitate/lib/templates/monit/monit.cnf</tt>.
+  #
+  # ==== Options 
+  # +template_path+:: Relative path
+  #
+  def gem_template_path(template_path)
+    File.join(gem_templates_root, template_path)
+  end
+  
+  # Load template. If extension is erb will be evaluated with binding.
+  # 
+  # ==== Options
+  # +path+:: If starts with '/', absolute path, otherwise relative path to templates dir
+  # +override_binding+:: Binding to override, otherwise uses current (task) binding
+  #
+  # ==== Examples
+  #   template.load("memcached/memcached.monitrc.erb")
+  #   put template.load("memcached/memcached.monitrc.erb"), "/tmp/memcached.monitrc"
+  #
+  def load(path, override_binding = nil)
+    template_paths = [ path, gem_template_path(path) ]
+    
+    template_paths = template_paths.select { |file| File.exist?(file) }
+    
+    if template_paths.empty?
+      raise <<-EOS 
+      
+      
+      Template not found: #{path}
+      
+      EOS
+    end
+    
+    # Use first
+    template_path = template_paths.first
+    
+    template_data = IO.read(template_path)
+    
+    if File.extname(template_path) == ".erb"    
+      template = ERB.new(template_data)
+      template_data = template.result(override_binding || binding)
+    end
+    template_data
+  end
+  
+  # Load template from project.
+  # If extension is erb will be evaluated with binding.
+  # 
+  # ==== Options
+  # +path+:: If starts with '/', absolute path, otherwise relative path to templates dir
+  # +override_binding+:: Binding to override, otherwise uses current (task) binding
+  #
+  # ==== Examples
+  #   template.project("config/templates/sphinx.conf.erb")
+  #   put template.project("config/templates/sphinx.conf.erb"), "#{shared_path}/config/sphinx.conf"
+  #
+  def project(path, override_binding = nil)
+    load(project_root + "/" + path, override_binding || binding)
+  end
+    
+  # Write template at (relative path) with binding to LOCAL destination path.
+  #
+  # ==== Options
+  # +template_path+:: Path to template relative to templates path
+  # +dest_path+:: Local destination path to write to
+  # +overwrite_binding+:: Binding  
+  # +overwrite+:: Force overwrite
+  # +verbose+:: Verbose output
+  #
+  # ==== Examples
+  #   template.write("config/templates/sphinx.conf.erb", binding, "config/sphinx.conf")
+  #  
+  def write(template_path, dest_path, override_binding = nil, overwrite = false, verbose = true)    
+    # This is gnarly!
+    relative_dest_path = Pathname.new(File.expand_path(dest_path)).relative_path_from(Pathname.new(File.expand_path(".")))  
+    
+    if !overwrite && File.exist?(dest_path) 
+      puts "%10s %-40s (skipped)" % [ "create", relative_dest_path ] if verbose
+      return
+    end
+    
+    puts "%10s %-40s" % [ "create", relative_dest_path ] if verbose
+    template_data = load(template_path, override_binding)
+    File.open(dest_path, "w") { |file| file.puts(load(template_path, override_binding)) }
+  end
+    
+end
+
+Capistrano.plugin :template, Capitate::Plugins::Templates

data/lib/capitate/plugins/wget.rb

@@ -0,0 +1,23 @@
+require 'open-uri'
+
+module Capitate::Plugins::Wget
+  
+  # Download the uri, then upload it into the remote destination directory
+  # 
+  # ==== Options
+  # +uri+:: URI to get
+  # +remote_dest_dir+:: Remote destination directory, defaults to /tmp
+  #
+  def uri(uri, remote_dest_dir = "/tmp")
+    
+    uri = uri = URI.parse(uri)
+    name = uri.path.split("/").last
+    remote_dest_path = File.join(remote_dest_dir, name)
+    
+    logger.info "Downloading #{name} from #{uri}..."      
+    put open(uri).read, remote_dest_path    
+  end
+  
+end
+
+Capistrano.plugin :wget, Capitate::Plugins::Wget

data/lib/capitate/plugins/yum.rb

@@ -0,0 +1,87 @@
+# Packages must respond to update, remove, install and clean
+module Capitate::Plugins::Yum
+  
+  # Update all installed packages.
+  #
+  # ==== Options
+  # +packages+:: Packages to update, or empty to update all
+  #
+  # ==== Examples
+  #   yum.update
+  #   yum.update([ "aspell" ])
+  #
+  # Also can use package plugin with package.type = :yum, and then:
+  #   package.update
+  #   package.update([ "aspell" ])
+  #
+  def update(packages = [])
+    sudo "yum -y update #{packages.join(" ")}"
+  end
+    
+  # Remove via yum.
+  #
+  # ==== Options
+  # +packages+:: Packages to remove
+  #
+  # ==== Examples
+  #   yum.remove
+  #   yum.remove([ "aspell" ])
+  #
+  # Also can use package plugin with package.type = :yum, and then:
+  #   package.remove
+  #   package.remove([ "aspell" ])
+  #  
+  def remove(packages)
+    sudo "yum -y remove #{packages.join(" ")}"    
+  end
+  
+  # Install via yum.
+  # 
+  # ==== Options
+  # +packages+:: Packages to install, either String (for single package) or Array
+  # +update_existing+:: If package exists, where to yum update it, defaults to true
+  #  
+  # ==== Examples
+  #   yum.install
+  #   yum.install([ "aspell" ])
+  #
+  # Also can use package plugin with package.type = :yum, and then:
+  #   package.install
+  #   package.install([ "aspell" ])
+  #    
+  def install(packages, update_existing = true)    
+    
+    # If a single object, wrap in array
+    packages = [ packages ] unless packages.is_a?(Array)
+    
+    if update_existing
+      
+      installed_packages = []
+      
+      sudo "yum -d 0 list installed #{packages.join(" ")}" do |channel, stream, data|
+        installed_packages += data.split("\n")[1..-1].collect { |line| line.split(".").first }
+      end      
+    
+      packages -= installed_packages
+    
+      sudo "yum -y update #{installed_packages.join(" ")}" unless installed_packages.blank?
+    end
+    
+    sudo "yum -y install #{packages.join(" ")}" unless packages.blank?
+  end
+  
+  # Clean yum.
+  #
+  # ==== Examples
+  #   yum.clean
+  #
+  # Also can use package plugin with package.type = :yum, and then:
+  #   package.clean
+  #
+  def clean
+    sudo "yum -y clean all"
+  end
+  
+end
+
+Capistrano.plugin :yum, Capitate::Plugins::Yum

data/lib/capitate/recipes.rb

@@ -0,0 +1,9 @@
+# Load recipes
+# This is only loadable from within Capistrano
+recipes = Dir[File.dirname(__FILE__) + "/../recipes/**/*.rb"].collect { |recipe| File.expand_path(recipe) }
+recipes.each do |recipe|
+  Capistrano::Configuration.instance.load recipe
+end
+
+# Reset the password var
+Capistrano::Configuration.instance.load(:string => "reset_password")

data/lib/capitate/version.rb

@@ -0,0 +1,9 @@
+module Capitate #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 7
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/lib/capitate.rb

@@ -0,0 +1,32 @@
+require 'rubygems'
+require 'active_support'
+require 'highline'
+require 'capistrano'
+
+HighLine.track_eof = false
+
+# Add this path to ruby load path
+$:.unshift File.dirname(__FILE__)
+
+module Capitate # :nodoc:
+ module Plugins # :nodoc:
+ end  
+end
+
+require 'capitate/plugins/base'
+require 'capitate/plugins/gem'
+require 'capitate/plugins/package'
+require 'capitate/plugins/profiles'
+require 'capitate/plugins/script'
+require 'capitate/plugins/templates'
+require 'capitate/plugins/wget'
+require 'capitate/plugins/yum'
+
+require "capitate/cap_ext/connections"
+require "capitate/cap_ext/extension_proxy"
+require "capitate/cap_ext/variables"
+        
+class Capistrano::Configuration   
+  include Capitate::CapExt::Variables
+end
+

data/lib/doc/centos/memcached.yml

@@ -0,0 +1,18 @@
+install:
+  memcached_memory: |
+    Memcached memory (in MB).
+  
+    set :memcached_memory, 64
+  
+  memcached_pid_path: |
+    Path to memcached pid file.
+    Defaults to /var/run/memcached.pid
+    
+    set :memcached_pid_path, "/var/run/memcached.pid"
+  
+  memcached_port: |
+    Memcached port.
+    Defaults to 11211.
+    
+    set :memcached_port, 11211
+  

data/lib/doc/centos/mongrel_cluster.yml

@@ -0,0 +1,11 @@
+setup:
+  mongrel_size: |
+    Number of mongrels.
+    
+    set :mongrel_size, 3
+    
+  mongrel_port: |
+    Starting port for mongrels.
+    If there are 3 mongrels with port 9000, then instances will be at 9000, 9001, and 9002
+    
+    set :mongrel_port, 9000

data/lib/doc/centos/monit.yml

@@ -0,0 +1,12 @@
+install:
+  monit_port: |
+    Monit port.
+    Defaults to 2812.
+    
+    set :monit_port, 2812
+    
+  monit_password: |
+    Monit password.
+    Defaults to password prompt.
+    
+    set :monit_password, Proc.new { Capistrano::CLI.ui.ask('Monit admin password (to set): ') })

data/lib/doc/centos/mysql.yml

@@ -0,0 +1,6 @@
+install:
+  mysql_admin_password_set: |
+    Mysql admin password (to set).
+    Defaults to password prompt.
+    
+    set :mysql_admin_password_set, Proc.new { Capistrano::CLI.ui.ask('Mysql admin password (to set): ') }

data/lib/doc/centos/nginx.yml

@@ -0,0 +1,25 @@
+install:
+  nginx_bin_path: |
+    Nginx sbin path.
+    Defaults to /sbin/nginx
+    
+    set :nginx_bin_path, "/sbin/nginx"
+    
+  nginx_conf_path: |
+    Path to nginx conf.
+    Defaults to /etc/nginx/nginx.conf
+    
+    set :nginx_conf_path, "/etc/nginx/nginx.conf"
+    
+  nginx_pid_path: |
+    Path to nginx pid file.
+    Defaults to /var/run/nginx.pid
+    
+    set :nginx_pid_path, "/var/run/nginx.pid"
+    
+  nginx_prefix_path: |
+    Nginx install prefix.
+    Defaults to /var/nginx
+    
+    set :nginx_prefix_path, "/var/nginx"
+  

data/lib/doc/centos/sphinx.yml

@@ -0,0 +1,14 @@
+install:  
+  sphinx_prefix: |
+    Sphinx install prefix.
+    Defaults to "/usr/local/sphinx"
+    
+    set :sphinx_prefix, "/usr/local/sphinx"
+    
+setup:
+  sphinx_prefix: |
+    Sphinx install prefix.
+    Defaults to "/usr/local/sphinx"
+    
+    set :sphinx_prefix, "/usr/local/sphinx"
+    

data/lib/doc/gems.yml

@@ -0,0 +1,7 @@
+install:
+  gem_list: |
+    set :gem_list, [ 
+      "rake", 
+      "mysql -- --with-mysql-include=/usr/include/mysql --with-mysql-lib=/usr/lib/mysql --with-mysql-config",
+      "raspell"
+    ]

data/lib/doc/install.yml

@@ -0,0 +1,11 @@
+default:
+  recipes: |
+    Recipes to run as part of install.
+    
+    set :recipes, [ "centos:setup_for_web", "packages:install", "centos:ruby:install" ]
+    
+  install_user: |
+    The user to run as for install recipes.
+    Defaults to root.
+    
+    set :install_user, "root"

data/lib/doc/memcached.yml

@@ -0,0 +1,13 @@
+install_monit:  
+  memcached_pid_path: |
+    Path to memcached pid file.
+    Defaults to /var/run/memcached.pid
+    
+    set :memcached_pid_path, "/var/run/memcached.pid"
+  
+  memcached_port: |
+    Memcached port.
+    Defaults to 11211.
+    
+    set :memcached_port, 11211
+  

data/lib/doc/mongrel_cluster.yml

@@ -0,0 +1,12 @@
+setup_monit:
+  mongrel_size: |
+    Number of mongrels.
+    
+    set :mongrel_size, 3
+    
+  mongrel_port: |
+    Starting port for mongrels.
+    If there are 3 mongrels with port 9000, then instances will be at 9000, 9001, and 9002
+    
+    set :mongrel_port, 9000
+  

data/lib/doc/mysql.yml

@@ -0,0 +1,46 @@
+setup:
+  db_name: |
+    Database name (application).
+    
+    set :db_name, "app_db_name"
+  
+  db_user: |
+    Database user (application).
+    
+    set :db_user, "app_db_user"
+    
+  db_pass: |
+    Database password (application).
+    
+    set :db_pass, "the_password"
+    
+  web_host: |
+    Web host to provide access privileges to (if recipe used in context of web app).
+    Defaults to nil
+    TODO: Support multiple web hosts
+    
+    set :web_host, 10.0.6.100
+    
+  db_host: |
+    Database host (to provide itself with access).
+    
+    set :db_host, 10.0.6.101
+    
+  mysql_admin_password: |
+    Mysql admin password (to use to connect).
+    Defaults to password prompt.
+
+    set :mysql_admin_password, Proc.new { Capistrano::CLI.ui.ask('Mysql admin password: ') })
+  
+setup_monit:
+  db_port: |
+    Mysql port.
+    Defaults to 3306.
+    
+    set :db_port, 3306
+    
+  mysql_pid_path: |
+    Path to mysql pid file.
+    Defaults to /var/run/mysqld/mysqld.pid
+    
+    set :mysql_pid_path, "/var/run/mysqld/mysqld.pid"

data/lib/doc/nginx.yml

@@ -0,0 +1,23 @@
+install_monit:
+  nginx_pid_path: |
+    Path to nginx pid file.
+    Defaults to /var/run/nginx.pid
+    
+    set :nginx_pid_path, "/var/run/nginx.pid"
+    
+setup_mongrel:
+  mongrel_size: |
+    Number of mongrels.
+    
+    set :mongrel_size, 3
+    
+  mongrel_port: |
+    Starting port for mongrels.
+    If there are 3 mongrels with port 9000, then instances will be at 9000, 9001, and 9002
+    
+    set :mongrel_port, 9000
+
+  domain_name: |
+    Domain name for nginx virtual host, (without www prefix).
+    
+    set :domain_name, "foo.com"