capistrano-mb 0.22.0

data/lib/capistrano/tasks/fiftyfive.rake

@@ -0,0 +1,59 @@
+# Provides backward compatibility with the old "fiftyfive" task names.
+
+tasks = %w(
+  fiftyfive:aptitude:change_postgres_packages
+  fiftyfive:aptitude:install
+  fiftyfive:aptitude:install_postgres_repo
+  fiftyfive:aptitude:upgrade
+  fiftyfive:crontab
+  fiftyfive:delayed_job:init_d
+  fiftyfive:delayed_job:restart
+  fiftyfive:delayed_job:start
+  fiftyfive:delayed_job:stop
+  fiftyfive:dotenv:replace
+  fiftyfive:dotenv:update
+  fiftyfive:logrotate
+  fiftyfive:maintenance:disable
+  fiftyfive:maintenance:enable
+  fiftyfive:migrate:deploy
+  fiftyfive:nginx:configure
+  fiftyfive:nginx:restart
+  fiftyfive:nginx:start
+  fiftyfive:nginx:stop
+  fiftyfive:postgresql:create_database
+  fiftyfive:postgresql:create_user
+  fiftyfive:postgresql:database_yml
+  fiftyfive:postgresql:dump
+  fiftyfive:postgresql:logrotate_backup
+  fiftyfive:postgresql:pgpass
+  fiftyfive:postgresql:restore
+  fiftyfive:postgresql:tune
+  fiftyfive:rake
+  fiftyfive:rbenv:install
+  fiftyfive:rbenv:upgrade
+  fiftyfive:seed
+  fiftyfive:sidekiq:init_d
+  fiftyfive:sidekiq:restart
+  fiftyfive:sidekiq:start
+  fiftyfive:sidekiq:stop
+  fiftyfive:ssl:generate_csr
+  fiftyfive:ssl:generate_dh
+  fiftyfive:ssl:generate_self_signed_crt
+  fiftyfive:ufw:configure
+  fiftyfive:unicorn:config_rb
+  fiftyfive:unicorn:init_d
+  fiftyfive:unicorn:restart
+  fiftyfive:unicorn:start
+  fiftyfive:unicorn:stop
+  fiftyfive:user:add
+  fiftyfive:user:install_public_key
+  fiftyfive:version:write_initializer
+)
+
+tasks.each do |name|
+  task(name) do
+    mb_name = name.gsub(/^fiftyfive:/, "mb:")
+    compatibility_warning("The #{name} task has been renamed to #{mb_name}.")
+    invoke(mb_name)
+  end
+end

data/lib/capistrano/tasks/logrotate.rake

@@ -0,0 +1,16 @@
+mb_recipe :logrotate do
+  during :provision, "mb:logrotate"
+end
+
+namespace :mb do
+  desc "Configure logrotate for Rails logs"
+  task :logrotate do
+    privileged_on release_roles(:all) do
+      template "logrotate.erb",
+               "/etc/logrotate.d/#{application_basename}-logs",
+               :mode => 644,
+               :owner => "root:root",
+               :sudo => true
+    end
+  end
+end

data/lib/capistrano/tasks/maintenance.rake

@@ -0,0 +1,28 @@
+mb_recipe :maintenance do
+  # No hooks for this recipe
+end
+
+namespace :mb do
+  namespace :maintenance do
+    desc "Tell nginx to display a 503 page for all web requests, using the "\
+         "maintenance.html.erb template"
+    task :enable do
+      on roles(:web) do
+        reason = ENV["REASON"]
+        deadline = ENV["DEADLINE"]
+
+        template "maintenance.html.erb",
+                 "#{current_path}/public/system/maintenance.html",
+                 :binding => binding,
+                 :mode => "644"
+      end
+    end
+
+    desc "Remove the 503 page"
+    task :disable do
+      on roles(:web) do
+        execute :rm, "-f", "#{current_path}/public/system/maintenance.html"
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/migrate.rake

@@ -0,0 +1,29 @@
+mb_recipe :migrate do
+  during   "deploy:migrate_and_restart", "deploy"
+  prior_to "deploy:migrate",             "enable_maintenance_before"
+  during   "deploy:published",           "disable_maintenance_after"
+end
+
+namespace :mb do
+  namespace :migrate do
+    desc "Deploy the app, stopping it and showing a 503 maintenance page "\
+         "while database migrations are being performed; then start the app"
+    task :deploy do
+      set(:mb_restart_during_migrate, true)
+      invoke :deploy
+    end
+
+    task :enable_maintenance_before do
+      if fetch(:mb_restart_during_migrate)
+        invoke_if_defined "mb:maintenance:enable"
+        invoke_if_defined "deploy:stop"
+      end
+    end
+
+    task :disable_maintenance_after do
+      if fetch(:mb_restart_during_migrate)
+        invoke_if_defined "mb:maintenance:disable"
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/nginx.rake

@@ -0,0 +1,31 @@
+mb_recipe :nginx do
+  during :provision, "configure"
+end
+
+namespace :mb do
+  namespace :nginx do
+    desc "Install nginx.conf files and restart nginx"
+    task :configure do
+      privileged_on roles(:web) do
+        template("nginx.erb", "/etc/nginx/nginx.conf", :sudo => true)
+
+        template "nginx_unicorn.erb",
+                 "/etc/nginx/sites-enabled/#{application_basename}",
+                 :sudo => true
+
+        execute "sudo rm -f /etc/nginx/sites-enabled/default"
+        execute "sudo mkdir -p /etc/nginx/#{application_basename}-locations"
+        execute "sudo service nginx restart"
+      end
+    end
+
+    %w(start stop restart).each do |command|
+      desc "#{command} nginx"
+      task command.intern do
+        privileged_on roles(:web) do
+          execute "sudo service nginx #{command}"
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/postgresql.rake

@@ -0,0 +1,177 @@
+mb_recipe :postgresql do
+  during :provision, %w(
+    create_user
+    create_database
+    database_yml
+    pgpass
+    logrotate_backup
+  )
+end
+
+namespace :mb do
+  namespace :postgresql do
+    desc "Update postgresql.conf using pgtune"
+    task :tune do
+      privileged_on primary(:db), :in => :sequence do
+        pgtune_dir = "/tmp/pgtune"
+        pgtune_output = "/tmp/postgresql.conf.pgtune"
+        pg_conf = "/etc/postgresql/9.1/main/postgresql.conf"
+
+        execute :sudo, "rm", "-rf", pgtune_dir
+        execute :sudo, "git",
+                "clone",
+                "-q",
+                "https://github.com/gregs1104/pgtune.git",
+                pgtune_dir
+
+        execute "sudo #{pgtune_dir}/pgtune",
+                "--input-config", pg_conf,
+                "--output-config", pgtune_output,
+                "--type", "Web",
+                "--connections", fetch(:mb_postgresql_max_connections)
+
+        # Log diff for informational purposes
+        execute :sudo, "diff", pg_conf, pgtune_output, "|| true"
+
+        execute :sudo, "cp", pgtune_output, pg_conf
+        execute :sudo, "service", "postgresql", "restart"
+      end
+    end
+
+    desc "Create user if it doesn't already exist"
+    task :create_user do
+      privileged_on primary(:db) do
+        user = fetch(:mb_postgresql_user)
+
+        unless test("sudo -u postgres psql -c '\\du' | grep -q #{user}")
+          passwd = fetch(:mb_postgresql_password)
+          md5 = Digest::MD5.hexdigest(passwd + user)
+          execute "sudo -u postgres psql -c " +
+                  %Q["CREATE USER #{user} PASSWORD 'md5#{md5}';"]
+        end
+      end
+    end
+
+    desc "Create database if it doesn't already exist"
+    task :create_database do
+      privileged_on primary(:db) do
+        user = fetch(:mb_postgresql_user)
+        db = fetch(:mb_postgresql_database)
+
+        unless test("sudo -u postgres psql -l | grep -w -q #{db}")
+          execute "sudo -u postgres createdb -O #{user} #{db}"
+        end
+      end
+    end
+
+    desc "Generate database.yml"
+    task :database_yml do
+      yaml = {
+        fetch(:rails_env).to_s => {
+          "adapter" => "postgresql",
+          "encoding" => "unicode",
+          "database" => fetch(:mb_postgresql_database).to_s,
+          "pool" => fetch(:mb_postgresql_pool_size).to_i,
+          "username" => fetch(:mb_postgresql_user).to_s,
+          "password" => fetch(:mb_postgresql_password).to_s,
+          "host" => fetch(:mb_postgresql_host).to_s
+        }
+      }
+      fetch(:mb_postgresql_password)
+      on release_roles(:all) do
+        put YAML.dump(yaml),
+            "#{shared_path}/config/database.yml",
+            :mode => "600"
+      end
+    end
+
+    desc "Generate pgpass file (needed by backup scripts)"
+    task :pgpass do
+      fetch(:mb_postgresql_password)
+      on release_roles(:all) do
+        template "pgpass.erb",
+                 fetch(:mb_postgresql_pgpass_path),
+                 :mode => "600"
+      end
+    end
+
+    desc "Configure logrotate to back up the database daily"
+    task :logrotate_backup do
+      on roles(:backup) do
+        backup_path = fetch(:mb_postgresql_backup_path)
+        execute :mkdir, "-p", File.dirname(backup_path)
+        execute :touch, backup_path
+      end
+
+      privileged_on roles(:backup) do |host, user|
+        template\
+          "postgresql-backup-logrotate.erb",
+          "/etc/logrotate.d/postgresql-backup-#{application_basename}",
+          :owner => "root:root",
+          :mode => "644",
+          :binding => binding,
+          :sudo => true
+      end
+    end
+
+    desc "Dump the database to FILE"
+    task :dump do
+      on primary(:db) do
+        with_pgpassfile do
+          execute :pg_dump,
+            "-Fc -Z9 -O",
+            "-x", fetch(:mb_postgresql_dump_options),
+            "-f", remote_dump_file,
+            connection_flags,
+            fetch(:mb_postgresql_database)
+        end
+
+        download!(remote_dump_file, local_dump_file)
+
+        info(
+          "Exported #{fetch(:mb_postgresql_database)} "\
+          "to #{local_dump_file}."
+          )
+      end
+    end
+
+    desc "Restore database from FILE"
+    task :restore do
+      on primary(:db) do
+        exit 1 unless agree(
+          "\nErase existing #{fetch(:rails_env)} database "\
+          "and restore from local file: #{local_dump_file}? "
+          )
+
+        upload!(local_dump_file, remote_dump_file)
+
+        with_pgpassfile do
+          execute :pg_restore,
+            "-O -c",
+            connection_flags,
+            "-d", fetch(:mb_postgresql_database),
+            remote_dump_file
+        end
+      end
+    end
+
+    def local_dump_file
+      ENV.fetch("FILE", "#{fetch(:mb_postgresql_database)}.dmp")
+    end
+
+    def remote_dump_file
+      "/tmp/#{fetch(:mb_postgresql_database)}.dmp"
+    end
+
+    def connection_flags
+      [
+        "-U", fetch(:mb_postgresql_user),
+        "-h", fetch(:mb_postgresql_host)
+      ].join(" ")
+    end
+
+    def with_pgpassfile(&block)
+      with(:pgpassfile => fetch(:mb_postgresql_pgpass_path), &block)
+    end
+  end
+end

data/lib/capistrano/tasks/provision.rake

@@ -0,0 +1,18 @@
+# Define empty provision tasks.
+# These will be filled in by other recipes that contribute additional
+# `before` and `during` tasks.
+
+desc "Install and set up all app prerequisites (assumes Ubuntu 12.04)"
+task :provision
+
+namespace :provision do
+  desc "Install and set up all app prerequisites for Ubuntu 12.04"
+  task :"12_04" do
+    invoke "provision"
+  end
+
+  desc "Install and set up all app prerequisites for Ubuntu 14.04"
+  task :"14_04" do
+    invoke "provision"
+  end
+end

data/lib/capistrano/tasks/rake.rake

@@ -0,0 +1,20 @@
+mb_recipe :rake do
+  # No hooks
+end
+
+namespace :mb do
+  desc "Remotely execute a rake task"
+  task :rake do
+    if ENV['COMMAND'].nil?
+      raise "USAGE: cap #{fetch(:stage)} mb:rake COMMAND=my:task"
+    end
+
+    on primary(:app) do
+      within current_path do
+        with :rails_env => fetch(:rails_env) do
+          execute :rake, ENV['COMMAND']
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/rbenv.rake

@@ -0,0 +1,93 @@
+mb_recipe :rbenv do
+  during :provision, %w(install write_vars)
+end
+
+namespace :mb do
+  namespace :rbenv do
+    desc "Install rbenv and compile ruby"
+    task :install do
+      invoke "mb:rbenv:run_installer"
+      invoke "mb:rbenv:modify_bashrc"
+      invoke "mb:rbenv:bootstrap_ubuntu_for_ruby_compile"
+      invoke "mb:rbenv:compile_ruby"
+    end
+
+    desc "Install the latest version of Ruby"
+    task :upgrade do
+      invoke "mb:rbenv:update_rbenv"
+      invoke "mb:rbenv:bootstrap_ubuntu_for_ruby_compile"
+      invoke "mb:rbenv:compile_ruby"
+    end
+
+    task :write_vars do
+      on release_roles(:all) do
+        execute :mkdir, "-p ~/.rbenv"
+        execute :touch, "~/.rbenv/vars"
+        execute :chmod, "0600 ~/.rbenv/vars"
+
+        vars = ""
+
+        fetch(:mb_rbenv_vars).each do |name, value|
+          execute :sed, "--in-place '/^#{name}=/d' ~/.rbenv/vars"
+          vars << "#{name}=#{value}\n"
+        end
+
+        tmp_file = "/tmp/rbenv_vars"
+        put vars, tmp_file
+        execute :cat, tmp_file, ">> ~/.rbenv/vars"
+        execute :rm, tmp_file
+      end
+    end
+
+    task :run_installer do
+      on release_roles(:all) do
+        execute :curl,
+                "-L https://raw.github.com/fesplugas/rbenv-installer/master/bin/rbenv-installer",
+                "|", :bash
+      end
+    end
+
+    task :modify_bashrc do
+      on release_roles(:all) do
+        unless test("grep -qs 'rbenv init' ~/.bashrc")
+          template("rbenv_bashrc", "/tmp/rbenvrc")
+          execute :cat, "/tmp/rbenvrc ~/.bashrc > /tmp/bashrc"
+          execute :mv, "/tmp/bashrc ~/.bashrc"
+          execute %q{export PATH="$HOME/.rbenv/bin:$PATH"}
+          execute %q{eval "$(rbenv init -)"}
+        end
+      end
+    end
+
+    task :bootstrap_ubuntu_for_ruby_compile do
+      privileged_on release_roles(:all) do |host, user|
+        with :debian_frontend => "noninteractive" do
+          execute "sudo ~#{user}/.rbenv/plugins/rbenv-bootstrap/bin/rbenv-bootstrap-ubuntu-12-04"
+          execute "sudo aptitude -y -q install libffi-dev"
+        end
+      end
+    end
+
+    task :compile_ruby do
+      ruby_version = fetch(:mb_rbenv_ruby_version)
+      on release_roles(:all) do
+        force = ENV["RBENV_FORCE_INSTALL"] || begin
+          ! test("rbenv versions | grep -q '#{ruby_version}'")
+        end
+
+        if force
+          execute "CFLAGS=-O3 rbenv install --force #{ruby_version}"
+          execute "rbenv global #{ruby_version}"
+          execute "gem install bundler psych --no-document"
+          execute "rbenv rehash"
+        end
+      end
+    end
+
+    task :update_rbenv do
+      on release_roles(:all) do
+        execute "rbenv update"
+      end
+    end
+  end
+end