railsmachine 0.1.0

data/README

@@ -0,0 +1,6 @@
+== Railsmachine
+
+Doc coming soon.
+
+
+

data/Rakefile

@@ -0,0 +1,52 @@
+require 'rake'
+require 'rake/testtask'
+require 'rake/clean'
+require 'rake/gempackagetask'
+require 'rake/rdoctask'
+require 'tools/rakehelp'
+require 'fileutils'
+include FileUtils
+
+setup_tests
+setup_clean ["pkg", "lib/*.bundle", "*.gem", ".config"]
+
+setup_rdoc ['README', 'LICENSE', 'COPYING', 'lib/**/*.rb', 'doc/**/*.rdoc']
+
+desc "Does a full compile, test run"
+task :default => [:test, :package]
+
+version="0.1.0"
+name="railsmachine"
+
+setup_gem(name, version) do |spec|
+  spec.summary = "The Rails Machine task library"
+  spec.description = spec.summary
+  spec.author="Bradley Taylor"
+  spec.add_dependency('capistrano', '>= 1.1.0')
+  spec.add_dependency('mongrel_cluster', '>= 0.2.0')
+  spec.has_rdoc = false
+  spec.files += Dir.glob("bin/*")
+  spec.files += Dir.glob("resources/**/*")
+  spec.default_executable = "railsmachine"
+  spec.executables = ["railsmachine"]
+end
+
+
+task :install => [:test, :package] do
+  sh %{sudo gem install pkg/#{name}-#{version}.gem}
+end
+
+task :uninstall => [:clean] do
+  sh %{sudo gem uninstall #{name}}
+end
+
+task :gem_source do
+  mkdir_p "pkg/gems"
+
+  FileList["**/*.gem"].each { |gem| mv gem, "pkg/gems" }
+  FileList["pkg/*.tgz"].each {|tgz| rm tgz }
+  rm_rf "pkg/#{name}-#{version}"
+
+  sh %{ generate_yaml_index.rb -d pkg }
+  sh %{ scp -r pkg/* #{ENV['SSH_USER']}@rubyforge.org:/var/www/gforge-projects/railsmachine/releases/ }
+end

data/bin/railsmachine

@@ -0,0 +1,35 @@
+#!/usr/bin/env ruby
+
+begin
+  require 'rubygems'
+rescue LoadError
+  # no rubygems to load, so we fail silently
+end
+
+require 'optparse'
+
+@options = {}
+OptionParser.new do |opts|
+  opts.banner = "Usage: #{$0} [options] [args]"
+
+  opts.on("-A", "--apply-to DIRECTORY",
+    "Create a minimal set of scripts and recipes",
+    "for use with capistrano to configure servers."
+  ) { |value| @options[:apply_to] = value }
+  opts.on("-n", "--name APPLICATION_NAME",
+    "Name of application."
+  ) { |value| @options[:application] = value }
+  opts.on("-d", "--domain DOMAIN_NAME",
+    "Domain name for application."
+  ) { |value| @options[:domain] = value }
+ 
+  if ARGV.empty?
+    puts opts
+      exit
+  else
+    opts.parse!(ARGV)
+  end
+end
+
+require 'railsmachine/generators/loader'
+RailsMachine::Generators::RailsLoader.load! @options

data/lib/railsmachine/generators/loader.rb

@@ -0,0 +1,20 @@
+module RailsMachine
+  module Generators
+    class RailsLoader
+      def self.load!(options)
+        require "#{options[:apply_to]}/config/environment"
+        require "rails_generator"
+        require "rails_generator/scripts/generate"
+
+        Rails::Generator::Base.sources << Rails::Generator::PathSource.new(
+          :railsmachine, File.dirname(__FILE__))
+
+        args = ["railsmachine"]
+        args << (options[:application] || "Application")
+        args << (options[:domain] || "my.railsmachina.com")
+
+        Rails::Generator::Scripts::Generate.new.run(args)
+      end
+    end
+  end
+end

data/lib/railsmachine/generators/railsmachine/USAGE

@@ -0,0 +1,12 @@
+NAME
+     railsmachine - creates configuration and SwitchTower tasks for server configuration.
+
+SYNOPSIS
+     railsmachine [Application name]
+
+DESCRIPTION
+     This generator creates rakefiles and deployment recipes.
+
+            
+EXAMPLE
+      ./script/generate railsmachine MyRockinApp

data/lib/railsmachine/generators/railsmachine/railsmachine_generator.rb

@@ -0,0 +1,24 @@
+class RailsmachineGenerator < Rails::Generator::NamedBase
+  attr_reader :application_name
+  attr_reader :domain_name
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    @application_name = self.file_name
+    @domain_name = @args[0]
+  end
+
+  def manifest
+    record do |m|
+      m.directory "config"
+      m.template "deploy.rb", File.join("config", "deploy.rb")
+    end
+  end
+
+  protected
+
+    # Override with your own usage banner.
+    def banner
+      "Usage: #{$0} railsmachine ApplicationName DomainName"
+    end
+end

data/lib/railsmachine/generators/railsmachine/templates/deploy.rb

@@ -0,0 +1,76 @@
+require 'railsmachine/recipes'
+
+# This defines a deployment "recipe" that you can feed to capistrano
+# (http://manuals.rubyonrails.com/read/book/17). It allows you to automate
+# (among other things) the deployment of your application.
+
+# =============================================================================
+# REQUIRED VARIABLES
+# =============================================================================
+# You must always specify the application and repository for every recipe. The
+# repository must be the URL of the repository you want this recipe to
+# correspond to. The deploy_to path must be the path on each machine that will
+# form the root of the application path.
+
+set :application, "<%= singular_name %>"
+set :deploy_to, "/var/www/apps/#{application}"
+set :domain, "<%= domain_name %>"
+
+set :user, "deploy"
+set :repository, "svn+ssh://#{user}@#{domain}#{deploy_to}/repos/trunk"
+set :rails_env, "production"
+
+# Automatically symlink these directories from curent/public to shared/public.
+# set :app_symlinks, %w{photo, document, asset}
+
+# =============================================================================
+# ROLES
+# =============================================================================
+# You can define any number of roles, each of which contains any number of
+# machines. Roles might include such things as :web, or :app, or :db, defining
+# what the purpose of each machine is. You can also specify options that can
+# be used to single out a specific subset of boxes in a particular role, like
+# :primary => true.
+
+role :web, domain
+role :app, domain
+role :db,  domain, :primary => true
+role :scm, domain
+
+# =============================================================================
+# APACHE OPTIONS
+# =============================================================================
+# set :apache_server_name, domain
+# set :apache_server_aliases, %w{alias1 alias2}
+# set :apache_default_vhost, true # force use of apache_default_vhost_config
+# set :apache_default_vhost_conf, "/etc/httpd/conf/default.conf"
+# set :apache_conf, "/etc/httpd/conf/apps/#{application}.conf"
+# set :apache_ctl, "/etc/init.d/httpd"
+# set :apache_proxy_port, 8000
+# set :apache_proxy_servers, 2
+# set :apache_proxy_address, "127.0.0.1"
+# set :apache_ssl_enabled, false
+# set :apache_ssl_ip, "127.0.0.1"
+# set :apache_ssl_forward_all, false
+
+# =============================================================================
+# MONGREL OPTIONS
+# =============================================================================
+# set :mongrel_servers, apache_proxy_servers
+# set :mongrel_port, apache_proxy_port
+# set :mongrel_address, apache_proxy_address
+# set :mongrel_environment, "production"
+# set :mongrel_config, "/etc/mongrel_cluster/#{application}.conf"
+# set :mongrel_user, user
+# set :mongrel_group, group
+
+# =============================================================================
+# MYSQL OPTIONS
+# =============================================================================
+
+
+# =============================================================================
+# SSH OPTIONS
+# =============================================================================
+# ssh_options[:keys] = %w(/path/to/my/key /path/to/another/key)
+# ssh_options[:port] = 25

data/lib/railsmachine/recipes.rb

@@ -0,0 +1,104 @@
+require 'railsmachine/recipes/svn'
+require 'railsmachine/recipes/mysql'
+require 'railsmachine/recipes/apache'
+require 'railsmachine/recipes/mongrel'
+
+Capistrano.configuration(:must_exist).load do
+  set :app_symlinks, nil
+  
+  desc "Setup servers."
+  task :setup_servers  do
+    setup_db
+    setup_app
+    setup_web
+  end
+  
+  desc "Setup application server."
+  task :setup_app, :roles => :app  do
+    set :mongrel_environment, rails_env
+    set :mongrel_port, apache_proxy_port
+    set :mongrel_servers, apache_proxy_servers
+    configure_mongrel_cluster
+  end
+  
+  desc "Restart application server."
+  task :restart_app, :roles => :app  do
+    restart_mongrel_cluster
+  end
+  
+  desc "Start application server."
+  task :start_app, :roles => :app  do
+    start_mongrel_cluster
+  end
+  
+  desc "Stop application server."
+  task :stop_app, :roles => :app  do
+    stop_mongrel_cluster
+  end
+  
+  desc "Setup web server."
+  task :setup_web, :roles => :web  do
+    set :apache_server_name, domain
+    configure_apache
+  end
+  
+  desc "Restart web server."
+  task :restart_web, :roles => :web  do
+    restart_apache
+  end
+  
+  desc "Reload web server configuration."
+  task :reload_web, :roles => :web  do
+    reload_apache
+  end
+  
+  desc "Start web server."
+  task :start_web, :roles => :web  do
+    start_apache
+  end
+  
+  desc "Stop web server."
+  task :stop_web, :roles => :web  do
+    stop_apache
+  end
+  
+  desc "Setup database server."
+  task :setup_db, :roles => :db, :only => { :primary => true } do
+    setup_mysql
+  end
+  
+  desc "Setup source control server."
+  task :setup_scm, :roles => :scm  do
+    setup_svn
+    import_svn
+  end
+  
+  desc <<-DESC
+  Restart the processes on the application server by calling restart_app.
+  DESC
+  task :restart, :roles => :app do
+    restart_app
+  end
+  
+  desc <<-DESC
+  Start the processes on the application server by calling start_app.
+  DESC
+  task :spinner, :roles => :app do
+    start_app
+  end
+  
+  desc "Link up any directories."
+  task :after_symlink, :roles => :app do
+    if app_symlinks
+      app_symlinks.each { |link| run "ln -nfs #{shared_path}/public/#{link} #{current_path}/public/#{link}" }
+    end
+ end
+  
+  desc "Setup symlink dirs"
+  task :after_setup, :roles => :app do
+    if app_symlinks
+      app_symlinks.each { |link| run "mkdir #{shared_path}/#{link}" }
+    end
+  end
+  
+end

data/lib/railsmachine/recipes/apache.rb

@@ -0,0 +1,70 @@
+Capistrano.configuration(:must_exist).load do
+  
+  set :apache_server_name, nil
+  set :apache_conf, nil
+  set :apache_default_vhost, false
+  set :apache_default_vhost_conf, nil
+  set :apache_ctl, "/etc/init.d/httpd"
+  set :apache_server_aliases, []
+  set :apache_proxy_port, 8000
+  set :apache_proxy_servers, 2
+  set :apache_proxy_address, "127.0.0.1"
+  set :apache_ssl_enabled, false
+  set :apache_ssl_ip, nil
+  set :apache_ssl_forward_all, false
+  
+  desc "Configure Apache. This uses the :use_sudo
+  variable to determine whether to use sudo or not. By default, :use_sudo is
+  set to true."
+  task :configure_apache, :roles => :web do
+    set_apache_conf
+    set :apache_server_name, domain unless apache_server_name
+    
+    server_aliases = []
+    server_aliases << "www.#{domain}"
+    server_aliases.concat apache_server_aliases
+    set :apache_server_aliases_array, server_aliases
+    
+    file = File.join(File.dirname(__FILE__), "templates", "httpd.conf")
+    buffer = render :template => File.read(file)
+    
+    if apache_ssl_enabled
+      file = File.join(File.dirname(__FILE__), "templates", "httpd-ssl.conf")
+      ssl_buffer = render :template => File.read(file)
+      buffer += ssl_buffer
+    end
+    
+    put buffer, "#{shared_path}/httpd.conf"
+    send(run_method, "cp #{shared_path}/httpd.conf #{apache_conf}")
+    delete "#{shared_path}/httpd.conf"
+  end
+  
+  desc "Start Apache "
+  task :start_apache, :roles => :web do
+    send(run_method, "#{apache_ctl} start")
+  end
+  
+  desc "Restart Apache "
+  task :restart_apache, :roles => :web do
+    send(run_method, "#{apache_ctl} restart")
+  end
+  
+  desc "Stop Apache "
+  task :stop_apache, :roles => :web do
+    send(run_method, "#{apache_ctl} stop")
+  end
+  
+  desc "Reload Apache "
+  task :reload_apache, :roles => :web do
+    send(run_method, "#{apache_ctl} reload")
+  end
+  
+  def set_apache_conf
+    if apache_default_vhost
+      set :apache_conf, "/etc/httpd/conf/default.conf" unless apache_default_vhost_conf
+    else 
+      set :apache_conf, "/etc/httpd/conf/apps/#{application}.conf" unless apache_conf
+    end
+  end
+  
+end

data/lib/railsmachine/recipes/mongrel.rb

@@ -0,0 +1 @@
+require 'mongrel_cluster/recipes'

data/lib/railsmachine/recipes/mysql.rb

@@ -0,0 +1,58 @@
+require 'yaml'
+require 'capistrano'
+require 'capistrano/cli'
+
+module MySQLMethods
+  
+  def execute(sql, user)
+    run "mysql --user=#{user} -p --execute=\"#{sql}\"" do |channel, stream, data|
+      handle_mysql_password(user, channel, stream, data)
+    end
+  end
+  
+  private
+  def handle_mysql_password(user, channel, stream, data)
+    logger.info data, "[database on #{channel[:host]} asked for password]"
+    if data =~ /^Enter password:/
+      pass = Capistrano::CLI.password_prompt "Enter database password for '#{user}':"
+      channel.send_data "#{pass}\n" 
+    end
+  end
+end
+
+Capistrano.plugin :mysql, MySQLMethods
+
+Capistrano.configuration(:must_exist).load do
+  
+  set :mysql_admin, nil
+  
+  desc "Execute MySQL statements using --execute option. Set the 'sql' variable."
+  task :execute_mysql, :roles => :db, :only => { :primary => true } do
+    set_mysql_admin
+    mysql.execute sql, mysql_admin
+  end
+  
+  desc "Create MySQL database and user based on config/database.yml"
+  task :setup_mysql, :roles => :db, :only => { :primary => true } do
+    # on_rollback {}
+    
+    set_mysql_admin
+    read_config
+    
+    sql = "CREATE DATABASE #{db_name};"
+    sql += "GRANT ALL PRIVILEGES ON #{db_name}.* TO #{db_user}@localhost IDENTIFIED BY '#{db_password}';"  
+    mysql.execute sql, mysql_admin
+  end
+  
+  def read_config
+    db_config = YAML.load_file('config/database.yml')
+    set :db_user, db_config[rails_env]["username"]
+    set :db_password, db_config[rails_env]["password"] 
+    set :db_name, db_config[rails_env]["database"]
+  end
+  
+  def set_mysql_admin
+    set :mysql_admin, user unless mysql_admin
+  end
+  
+end