AppTower-ubistrano 1.0.2

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(:iptables))
+        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,94 @@
+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.db
+        mysql.create.user
+      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 '#{db_user}'@'localhost' IDENTIFIED BY '#{db_pass}'",
+          "GRANT ALL PRIVILEGES ON *.* TO '#{db_user}'@'localhost'"
+        ]
+      end
+    end
+    
+    namespace :update do
+      desc 'Update mysql root password'
+      task :root_password, :roles => :db do
+        old_pass = ask "Current root password? (default: none)"
+        new_pass = ask "New root password? (default: none)"
+        sudo "mysqladmin -u root #{old_pass.empty? ? '' : "--password=#{old_pass} "}password #{new_pass}"
+      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 '#{db_user}'@'localhost'",
+          "DROP USER '#{db_user}'@'localhost'"
+        ]
+      end
+    end
+    
+    namespace :backup do
+      desc "Upload local backup to remote"
+      task :local_to_server, :roles => :db do
+        from = File.expand_path("~/db_backups/#{stage}/#{application}/#{backup_name}.bz2")
+        if File.exists?(from)
+          run_each "mkdir -p #{shared_path}/db_backups"
+          upload from, "#{shared_path}/db_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}/db_backups/#{backup_name}.bz2 | mysql -u #{db_user} --password=#{db_pass} #{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 #{db_user} -p#{db_pass} #{db_table}_production | 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 "source ~/.bash_profile && 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,23 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :sinatra do
+    desc "Runs install.rb if exists"
+    task :install do
+      run_puts "if [ -e #{current_path}/install.rb ]; then sudo ruby #{current_path}/install.rb; fi"
+    end
+    
+    namespace :config do
+      desc "Creates config.ru in shared config"
+      task :default do
+        run "mkdir -p #{shared_path}/config"
+        upload_from_erb "#{shared_path}/config/config.ru", binding, :folder => 'sinatra'
+      end
+      
+      desc "Copies files in the shared config folder into our app"
+      task :to_app, :roles => :app do
+        run "cp -Rf #{shared_path}/config/* #{release_path}"
+      end
+    end
+  end
+  
+end

data/lib/ubistrano/ssh.rb

@@ -0,0 +1,51 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  
+  namespace :ssh do
+    desc 'Generate ssh keys and upload to server'
+    task :default do
+      ssh.keys.create
+      ssh.keys.upload
+    end
+    
+    namespace :keys do
+      desc "Creates an rsa ssh key pair (local or remote)"
+      task :create do
+        if ask(msg(:ssh_keys_create))[0..0].downcase == 'l'
+          system('ssh-keygen -t rsa') if yes(msg(:create_keys))
+        else
+          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 unless yes(msg(:have_keys))
+      end
+
+      desc "Uploads local ssh public keys into remote authorized_keys"
+      task :upload do
+        if yes(msg(:upload_keys))
+          keys = ask msg(:ssh_keys_upload), get_ssh_keys
+          if keys.empty?
+            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", keys.strip
+            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,30 @@
+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|
+      value = [ value ].flatten if key == :domain
+      set key, value
+    end
+
+    role :app, host, :primary => true
+    role :web, host
+    role :db,  host
+  end
+  
+end

data/lib/ubistrano/ubuntu.rb

@@ -0,0 +1,269 @@
+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))
+      if yes("Have you already created the user defined in deploy.rb?")
+        ssh.default
+        ubuntu.config.default
+        ubuntu.aptitude.default
+        ubuntu.install.default
+        ubuntu.restart
+        puts space(msg(:ubuntu_finished))
+      else
+        puts space(msg(:visudo))
+      end
+    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
+        else
+          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.sudoers
+        ubuntu.config.sshd_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'
+          sudo_puts '/etc/init.d/ssh reload'
+          set :port, ssh_port
+        end
+      end
+      
+      desc "Updates sudoers file"
+      task :sudoers do
+        if yes(msg(:sudoers))
+          add_line '/etc/sudoers', "#{user} ALL=NOPASSWD: ALL"
+        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
+      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 -q -y',
+            'a2enmod rewrite'
+          ]
+        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'
+          ROOT.mysql.create.user
+          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"
+          ]
+          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
+        install_source(:ruby) do |path|
+          sudo_puts [
+            "aptitude install libopenssl-ruby -q -y",
+            make_install(path)
+          ]
+        end if yes("May I install Ruby?")
+      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)?")
+          sudo_puts 'aptitude install apache2-prefork-dev -q -y'
+          gem_install :passenger
+          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',
+            './etc/init.d/god start'
+          ]
+        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