ubistrano 1.2.7

data/lib/ubistrano/log.rb

@@ -0,0 +1,20 @@
+Capistrano::Configuration.instance(:must_exist).load do
+ 
+  namespace :log do
+    desc "Add logrotate entry for this application"
+    task :rotate, :roles => :app do
+      if yes(msg(:logrotate))
+        upload_from_erb '/etc/rotate.conf', binding, :folder => 'log'
+        sudo_each [
+          'cp -f /etc/logrotate.conf /etc/logrotate2.conf',
+          'chmod 777 /etc/logrotate2.conf',
+          'cat /etc/rotate.conf >> /etc/logrotate2.conf',
+          'cp -f /etc/logrotate2.conf /etc/logrotate.conf',
+          'rm -f /etc/logrotate2.conf',
+          'rm -f /etc/rotate.conf'
+        ]
+      end
+    end
+  end
+  
+end

data/lib/ubistrano/mysql.rb

@@ -0,0 +1,85 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :mysql do 
+    namespace :create do
+      desc "Create database and user"
+      task :default, :roles => :db do
+        mysql.create.user
+        mysql.create.db
+      end
+      
+      desc "Create database"
+      task :db, :roles => :db do
+        mysql_run "CREATE DATABASE #{db_table}"
+      end
+    
+      desc "Create database user"
+      task :user, :roles => :db do
+        mysql_run [
+          "CREATE USER '#{application}'@'localhost' IDENTIFIED BY '#{mysql_app_password}'",
+          "GRANT ALL PRIVILEGES ON #{db_table}.* TO '#{application}'@'localhost'"
+        ]
+      end
+    end
+  
+    namespace :destroy do
+      desc "Destroy database and user"
+      task :default, :roles => :db do
+        mysql.destroy.db
+        mysql.destroy.user
+      end
+      
+      desc "Destroy database"
+      task :db, :roles => :db do
+        mysql_run "DROP DATABASE #{db_table}"
+      end
+      
+      desc "Destroy database user"
+      task :user, :roles => :db do
+        mysql_run [
+          "REVOKE ALL PRIVILEGES, GRANT OPTION FROM '#{application}'@'localhost'",
+          "DROP USER '#{application}'@'localhost'"
+        ]
+      end
+    end
+    
+    namespace :backup do
+      desc "Upload local backup to remote"
+      task :local_to_server, :roles => :db do
+        from = File.expand_path("backups/#{backup_name}.bz2", FileUtils.pwd)
+        if File.exists?(from)
+          run_each "mkdir -p #{shared_path}/backups"
+          upload from, "#{shared_path}/backups/#{backup_name}.bz2"
+        else
+          puts "Does not exist: #{from}"
+        end
+      end
+      
+      desc "Restore remote database from backup"
+      task :restore, :roles => :db do
+        run_each "bunzip2 < #{shared_path}/backups/#{backup_name}.bz2 | mysql -u #{application} --password=#{mysql_app_password} #{db_table}"
+      end
+      
+      desc "Backup database to local"
+      task :to_local, :roles => :db do
+        to_server
+        system "mkdir -p ~/db_backups/#{stage}/#{application}"
+        get "#{shared_path}/db_backups/#{backup_name}.bz2", File.expand_path("~/db_backups/#{stage}/#{application}/#{backup_name}.bz2")
+      end
+      
+      desc "Backup database to remote"
+      task :to_server, :roles => :db do
+        run_each [
+          "mkdir -p #{shared_path}/db_backups",
+          "mysqldump --add-drop-table -u #{application} -p#{mysql_app_password} #{db_table} | bzip2 -c > #{shared_path}/db_backups/#{backup_name}.bz2"
+        ]
+      end
+      
+      def backup_name
+        now = Time.now
+        [ now.year, now.month, now.day ].join('-') + '.sql'
+      end
+    end
+  end
+
+end

data/lib/ubistrano/rails.rb

@@ -0,0 +1,76 @@
+Capistrano::Configuration.instance(:must_exist).load do
+
+  namespace :rails do
+    namespace :config do
+      desc "Creates database.yml in the shared config"
+      task :default, :roles => :app do
+        run "mkdir -p #{shared_path}/config"
+        Dir[File.expand_path('../../templates/rails/*', File.dirname(__FILE__))].each do |f|
+          upload_from_erb "#{shared_path}/config/#{File.basename(f, '.erb')}", binding, :folder => 'rails'
+        end
+      end
+      
+      desc "Set up app with app_helpers" 
+      task :app_helpers do
+        run "cd #{release_path} && script/plugin install git://github.com/winton/app_helpers.git"
+        run "cd #{release_path} && rake RAILS_ENV=production quiet=true app_helpers"
+      end
+      
+      desc "Configure attachment_fu"
+      task :attachment_fu, :roles => :app do
+        run_each [
+          "mkdir -p #{shared_path}/media",
+          "ln -sf #{shared_path}/media #{release_path}/public/media"
+        ]
+        sudo_each [
+          "mkdir -p #{release_path}/tmp/attachment_fu",
+          "chown -R #{user} #{release_path}/tmp/attachment_fu"
+        ]
+      end
+      
+      desc "Configure asset_packager" 
+      task :asset_packager do
+        run "cd #{release_path} && rake RAILS_ENV=production asset:packager:build_all"
+      end
+      
+      desc "Configure rails_widget"
+      task :rails_widget, :roles => :app do
+        run "cd #{release_path} && rake RAILS_ENV=production widget:production"
+      end
+      
+      desc "Copies yml files in the shared config folder into our app config"
+      task :to_app, :roles => :app do
+        run "cp -Rf #{shared_path}/config/* #{release_path}/config"
+      end
+      
+      namespace :thinking_sphinx do
+        desc "Configures thinking_sphinx"
+        task :default, :roles => :app do
+          sudo ";cd #{release_path} && rake RAILS_ENV=production ts:config"
+        end
+        
+        desc "Stop thinking_sphinx"
+        task :stop, :roles => :app do
+          sudo ";cd #{release_path} && rake RAILS_ENV=production ts:stop"
+        end
+        
+        desc "Start thinking_sphinx"
+        task :start, :roles => :app do
+          sudo ";cd #{release_path} && rake RAILS_ENV=production ts:start"
+        end
+        
+        desc "Restart thinking_sphinx"
+        task :restart, :roles => :app do
+          rails.config.thinking_sphinx.stop
+          rails.config.thinking_sphinx.start
+        end
+      end
+    end
+    
+    desc "Intialize Git submodules"
+    task :setup_git, :roles => :app do
+      run "cd #{release_path}; git submodule init; git submodule update"
+    end
+  end
+
+end

data/lib/ubistrano/sinatra.rb

@@ -0,0 +1,20 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :sinatra do    
+    namespace :config do
+      desc "Creates config.ru in shared config"
+      task :default do
+        run "mkdir -p #{shared_path}/config"
+        Dir[File.expand_path('../../templates/rails/*', File.dirname(__FILE__))].each do |f|
+          upload_from_erb "#{shared_path}/config/#{File.basename(f, '.erb')}", binding, :folder => 'rails'
+        end
+      end
+      
+      desc "Copies yml files in the shared config folder into our app config"
+      task :to_app, :roles => :app do
+        run "cp -Rf #{shared_path}/config/* #{release_path}/config"
+      end
+    end
+  end
+  
+end

data/lib/ubistrano/ssh.rb

@@ -0,0 +1,56 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :ssh do
+    desc 'Generate ssh keys and upload to server'
+    task :default do
+      ssh.keys.create.local
+      ssh.keys.upload
+      ssh.keys.create.remote
+    end
+    
+    namespace :keys do
+      namespace :create do
+        desc "Creates an rsa ssh key pair (local)"
+        task :local do
+          system('ssh-keygen -t rsa') if yes(msg(:create_keys))
+        end
+        
+        desc "Creates an rsa ssh key pair (remote)"
+        task :remote do
+          if yes(msg(:create_server_keys))
+            pass = ask "Enter a password for this key:"
+            sudo_each [
+              "ssh-keygen -t rsa -N '#{pass}' -q -f /home/#{user}/.ssh/id_rsa",
+              "chmod 0700 /home/#{user}/.ssh",
+              "chown -R #{user} /home/#{user}/.ssh"
+            ]
+            sudo_puts "tail -1 /home/#{user}/.ssh/id_rsa.pub"
+          end
+        end
+      end
+
+      desc "Uploads local ssh public keys into remote authorized_keys"
+      task :upload do
+        if yes(msg(:upload_keys))
+          key = ask msg(:upload_keys_2)
+          key = get_ssh_key key
+          if key.nil?
+            ssh.setup if yes("No keys found. Generate ssh keys now?")
+          else
+            sudo_each [
+              "mkdir -p /home/#{user}/.ssh",
+              "touch /home/#{user}/.ssh/authorized_keys"
+            ]
+            add_line "/home/#{user}/.ssh/authorized_keys", key
+            sudo_each [
+              "chmod 0700 /home/#{user}/.ssh",
+              "chmod 0600 /home/#{user}/.ssh/authorized_keys",
+              "chown -R #{user} /home/#{user}/.ssh"
+            ]
+          end
+        end
+      end
+    end
+  end
+
+end

data/lib/ubistrano/stage.rb

@@ -0,0 +1,29 @@
+Capistrano::Configuration.instance(:must_exist).load do
+
+  desc 'Set the target stage to staging'
+  task :staging do
+    set :stage, :staging
+  end
+
+  desc 'Set the target stage to test'
+  task :testing do
+    set :stage, :test
+  end
+  
+  # None of this works in a namespace
+  desc 'Set up stage-dependent properties'
+  task :setup_stage do
+    set :base_dir,  "#{ubistrano[:base_dir]}/#{stage}"
+    set :deploy_to, "#{base_dir}/#{application}"
+    set :db_table,  "#{application}#{stage == :staging ? "_#{stage}" : ''}"    
+    
+    ubistrano[stage].each do |key, value|
+      set key, value
+    end
+
+    role :app, host, :primary => true
+    role :web, host, :primary => true
+    role :db,  host, :primary => true
+  end
+  
+end

data/lib/ubistrano/ubuntu.rb

@@ -0,0 +1,273 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :ubuntu do
+    desc "Set up a fresh Ubuntu server"
+    task :default do
+      puts space(msg(:about_templates))
+      exit unless yes(msg(:visudo))
+      if yes("Create the remote deploy user?")
+        exit unless yes(msg(:add_user))
+      end
+      ssh.default
+      ubuntu.config.default
+      ubuntu.aptitude.default
+      puts space(msg(:run_ubuntu_install))
+    end
+    
+    desc "Restart Ubuntu server"
+    task :restart do
+      if yes(msg(:ubuntu_restart))
+        sudo_each 'shutdown -r now'
+      end
+    end
+    
+    namespace :aptitude do
+      desc 'Run all aptitude tasks'
+      task :default do
+        if yes(msg(:aptitude_default))
+          aptitude.update
+          aptitude.upgrade
+          aptitude.essential
+          ubuntu.restart
+        else
+          exit unless yes(msg(:aptitude_instructions))
+        end
+      end
+      
+      desc 'Aptitude update'
+      task :update do
+        sudo_puts 'aptitude update -q -y'
+      end
+      
+      desc 'Aptitude upgrade'
+      task :upgrade do
+        sudo_puts 'aptitude upgrade -q -y'
+      end
+      
+      desc 'Aptitude install build-essential'
+      task :essential do
+        sudo_puts 'aptitude install build-essential -q -y'
+      end
+    end
+    
+    namespace :config do
+      desc 'Run all tasks'
+      task :default do
+        ubuntu.config.sshd_config
+        ubuntu.config.ssh_config
+        ubuntu.config.iptables
+      end
+
+      desc "Updates server iptables"
+      task :iptables do
+        if yes(msg(:iptables))
+          upload_from_erb '/etc/iptables.rules', binding, :folder => 'ubuntu'
+          sudo_each [
+            'iptables-restore < /etc/iptables.rules',
+            'rm /etc/iptables.rules'
+          ]
+        end
+      end
+      
+      desc "Updates sshd_config"
+      task :sshd_config do
+        if yes(msg(:sshd_config))
+          set :ssh_port, port
+          set :port, 22
+          change_line '/etc/ssh/sshd_config', 'Port 22',              "Port #{port}"
+          change_line '/etc/ssh/sshd_config', 'PermitRootLogin yes',  'PermitRootLogin no'
+          change_line '/etc/ssh/sshd_config', 'X11Forwarding yes',    'X11Forwarding no'
+          change_line '/etc/ssh/sshd_config', 'UsePAM yes',           'UsePAM no'
+          remove_line '/etc/ssh/sshd_config', 'UseDNS .*'
+          add_line    '/etc/ssh/sshd_config', 'UseDNS no'
+          remove_line '/etc/ssh/sshd_config', 'StrictHostKeyChecking .*'
+          sudo_puts '/etc/init.d/ssh reload'
+          set :port, ssh_port
+        end
+      end
+      
+      desc "Updates ssh_config"
+      task :ssh_config do
+        if yes(msg(:ssh_config))
+          remove_line '/etc/ssh/ssh_config', 'StrictHostKeyChecking .*'
+          add_line    '/etc/ssh/ssh_config', 'StrictHostKeyChecking no'
+        end
+      end
+    end
+    
+    namespace :install do
+      desc 'Run all install tasks'
+      task :default do
+        ubuntu.install.apache
+        ubuntu.install.git
+        ubuntu.install.mysql
+        ubuntu.install.mysqltuner
+        ubuntu.install.perl
+        ubuntu.install.php
+        ubuntu.install.postfix
+        ubuntu.install.ruby
+        ubuntu.install.rubygems
+        ubuntu.install.passenger
+        ubuntu.install.god
+        ubuntu.install.rails
+        ubuntu.install.sinatra
+        ubuntu.install.sphinx
+        ubuntu.restart
+        puts space(msg(:ubuntu_finished))
+      end
+      
+      desc 'Install Apache'
+      task :apache, :roles => :web do
+        if yes("May I install Apache?")
+          sudo_puts [
+            'aptitude install apache2 apache2-mpm-prefork apache2-utils apache2.2-common libapr1 libaprutil1 ssl-cert apache2-prefork-dev -q -y',
+            'a2enmod rewrite',
+            'a2enmod ssl',
+            'a2dissite default'
+          ]
+        end
+      end
+      
+      desc 'Install Git'
+      task :git, :roles => :app do
+        install_source(:git) do |path|
+          sudo_puts [
+            'apt-get build-dep git-core -q -y',
+            make_install(path)
+          ]
+        end if yes("May I install Git?")
+      end
+      
+      desc 'Install MySQL'
+      task :mysql, :roles => :db do
+        if yes("May I install MySQL?")
+          sudo_puts 'aptitude install mysql-client-5.0 mysql-common mysql-server mysql-server-5.0 libmysqlclient15-dev libmysqlclient15off -q -y'
+          sudo "mysqladmin -u root password #{mysql_root_password || ''}"
+          exit unless yes(msg(:secure_mysql))
+        end
+      end
+      
+      desc "Install MySQLTuner"
+      task :mysqltuner, :roles => :db do
+        if yes(msg(:mysqltuner))
+          bin = "/usr/local/bin"
+          run_each [
+            "cd #{bin} && sudo wget --quiet #{sources[:mysqltuner]}",
+            "cd #{bin} && sudo chmod 0700 mysqltuner.pl",
+            "cd #{bin} && sudo mv mysqltuner.pl mysqltuner"
+          ]
+          exit unless yes(msg(:mysqltuner_instructions))
+        end
+      end
+      
+      desc 'Install Perl'
+      task :perl, :roles => :web do
+        if yes("May I install Perl?")
+          sudo_puts 'aptitude install libdbi-perl libnet-daemon-perl libplrpc-perl libdbd-mysql-perl -q -y'
+        end
+      end
+      
+      desc 'Install PHP'
+      task :php, :roles => :web do
+        if yes("May I install PHP?")
+          sudo_puts 'aptitude install php5-common php5-mysql libapache2-mod-php5 php-pear php-mail php-net-smtp -q -y'
+        end
+      end
+      
+      desc 'Install Postfix'
+      task :postfix, :roles => :web do
+        if yes("May I install Postfix and set it up as a relay?")
+          smtp  = ask 'What is your SMTP server address?'
+          login = ask 'What is your SMTP server username?'
+          pass  = ask 'What is your SMTP server password?'
+          sudo_puts 'aptitude install postfix -q -y'
+          add_line  '/etc/postfix/main.cf',
+            'smtp_sasl_auth_enable = yes',
+            'smtp_sasl_security_options = noanonymous',
+            'smtp_sasl_password_maps = hash:/etc/postfix/saslpasswd',
+            'smtp_always_send_ehlo = yes',
+            "relayhost = #{smtp}"
+          sudo_each 'touch /etc/postfix/saslpasswd'
+          add_line  '/etc/postfix/saslpasswd', "#{smtp} #{login}:#{pass}"
+          sudo_each [
+            'postmap /etc/postfix/saslpasswd',
+            '/etc/init.d/postfix restart'
+          ]
+        end
+      end
+      
+      desc 'Install Ruby'
+      task :ruby, :roles => :app do
+        if yes("May I install Ruby?")
+          sudo_puts "aptitude install libopenssl-ruby -q -y"
+          install_source(:ruby) do |path|
+            sudo_puts make_install(path)
+          end
+        end
+      end
+      
+      desc 'Install RubyGems'
+      task :rubygems, :roles => :app do
+        if yes("May I install RubyGems?")
+          install_source(:rubygems) do |path|
+            run_puts "cd #{path} && sudo ruby setup.rb"
+          end
+          gems.update
+        end
+      end
+      
+      desc 'Install Passenger'
+      task :passenger, :roles => :app do
+        if yes("May I install Passenger (mod_rails)?")
+          gem_install :passenger
+          ROOT.apache.reload if yes(msg(:passenger))
+        end
+      end
+      
+      desc 'Install God'
+      task :god, :roles => :app do
+        if yes(msg(:god))
+          gem_install 'god'
+          upload_from_erb '/etc/init.d/god', binding, :folder => 'ubuntu'
+          sudo_each [
+            ';cd /etc/init.d && sudo chmod +x god',
+            'mkdir -p /usr/local/etc/god'
+          ]
+          upload_from_erb('/usr/local/etc/god.god',        binding, :folder => 'ubuntu')
+          upload_from_erb('/usr/local/etc/god/apache.god', binding, :folder => 'ubuntu') if yes(msg(:god_apache))
+          upload_from_erb('/usr/local/etc/god/mysql.god',  binding, :folder => 'ubuntu') if yes(msg(:god_mysql))
+          upload_from_erb('/usr/local/etc/god/sshd.god',   binding, :folder => 'ubuntu') if yes(msg(:god_sshd))
+          sudo_puts [
+            'update-rc.d god defaults'
+          ]
+          exit unless yes(msg(:god_finished))
+        end
+      end
+      
+      desc 'Install Rails'
+      task :rails, :roles => :app do
+        if yes("May I install Rails?")
+          gem_install :mysql
+          gem_install :rails
+        end
+      end
+      
+      desc 'Install Sinatra'
+      task :sinatra, :roles => :app do
+        if yes("May I install Sinatra?")
+          gem_install :do_mysql  # Datamapper
+          gem_install 'dm-core'
+          gem_install :sinatra   # Sinatra
+        end
+      end
+      
+      desc 'Install Sphinx'
+      task :sphinx, :roles => :app do
+        install_source(:sphinx) do |path|
+          sudo_puts make_install(path)
+        end if yes("May I install Sphinx?")
+      end
+    end
+  end
+
+end