capistrano-atlas 0.0.1

data/lib/capistrano/tasks/migrate.rake

@@ -0,0 +1,29 @@
+atlas_recipe :migrate do
+  during   "deploy:migrate_and_restart", "deploy"
+  prior_to "deploy:migrate",             "enable_maintenance_before"
+  during   "deploy:published",           "disable_maintenance_after"
+end
+
+namespace :atlas 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(:atlas_restart_during_migrate, true)
+      invoke :deploy
+    end
+
+    task :enable_maintenance_before do
+      if fetch(:atlas_restart_during_migrate)
+        invoke_if_defined "atlas:maintenance:enable"
+        invoke_if_defined "deploy:stop"
+      end
+    end
+
+    task :disable_maintenance_after do
+      if fetch(:atlas_restart_during_migrate)
+        invoke_if_defined "atlas:maintenance:disable"
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/nginx.rake

@@ -0,0 +1,25 @@
+atlas_recipe :nginx do
+  during :provision, "configure"
+end
+
+namespace :atlas 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)
+
+        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,149 @@
+atlas_recipe :postgresql do
+  during :provision, %w(
+    create_user
+    create_database
+    database_yml
+    pgpass
+    logrotate_backup
+  )
+end
+
+namespace :atlas do
+  namespace :postgresql do
+    desc "Create user if it doesn't already exist"
+    task :create_user do
+      privileged_on primary(:db) do
+        user = fetch(:atlas_postgresql_user)
+
+        unless test("sudo -u postgres psql -c '\\du' | grep -q #{user}")
+          passwd = fetch(:atlas_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(:atlas_postgresql_user)
+        db = fetch(:atlas_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(:atlas_postgresql_database).to_s,
+          "pool" => fetch(:atlas_postgresql_pool_size).to_i,
+          "username" => fetch(:atlas_postgresql_user).to_s,
+          "password" => fetch(:atlas_postgresql_password).to_s,
+          "host" => fetch(:atlas_postgresql_host).to_s
+        }
+      }
+      fetch(:atlas_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(:atlas_postgresql_password)
+      on release_roles(:all) do
+        template "pgpass.erb",
+                 fetch(:atlas_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(:atlas_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(:atlas_postgresql_dump_options),
+            "-f", remote_dump_file,
+            connection_flags,
+            fetch(:atlas_postgresql_database)
+        end
+
+        download!(remote_dump_file, local_dump_file)
+
+        info(
+          "Exported #{fetch(:atlas_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(:atlas_postgresql_database),
+            remote_dump_file
+        end
+      end
+    end
+
+    def local_dump_file
+      ENV.fetch("FILE", "#{fetch(:atlas_postgresql_database)}.dmp")
+    end
+
+    def remote_dump_file
+      "/tmp/#{fetch(:atlas_postgresql_database)}.dmp"
+    end
+
+    def connection_flags
+      [
+        "-U", fetch(:atlas_postgresql_user),
+        "-h", fetch(:atlas_postgresql_host)
+      ].join(" ")
+    end
+
+    def with_pgpassfile(&block)
+      with(:pgpassfile => fetch(:atlas_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/puma.rake

@@ -0,0 +1,67 @@
+atlas_recipe :puma do
+  during "deploy:starting", "starting"
+  during :provision, %w(init_d nginx_site config_rb)
+  during "deploy:start", "start"
+  during "deploy:stop", "stop"
+  during "deploy:restart", "restart"
+  during "deploy:publishing", "restart"
+end
+
+namespace :atlas do
+  namespace :puma do
+    task :starting do
+      fetch(:linked_files, []) << "config/puma.rb"
+    end
+
+    desc "Install service script for puma"
+    task :init_d do
+      privileged_on roles(:app) do |host, user|
+        puma_user = fetch(:atlas_puma_user) || user
+
+        template "puma_init.erb",
+                 "/etc/init.d/puma_#{application_basename}",
+                 :mode => "a+rx",
+                 :binding => binding
+
+        execute "update-rc.d -f puma_#{application_basename} defaults"
+      end
+    end
+
+    desc "Install puma proxy into nginx sites and restart nginx"
+    task :nginx_site do
+      set(:atlas_server_name, "puma")
+
+      privileged_on roles(:web) do
+        template "nginx_site.erb",
+                 "/etc/nginx/sites-enabled/#{application_basename}"
+        execute "rm -f /etc/nginx/sites-enabled/default"
+        execute "mkdir -p /etc/nginx/#{application_basename}-locations"
+        execute "service nginx restart"
+      end
+    end
+
+    desc "Create config/puma.rb"
+    task :config_rb do
+      on release_roles(:all) do
+        template "puma.rb.erb", "#{shared_path}/config/puma.rb"
+      end
+    end
+
+    %w[start stop].each do |command|
+      desc "#{command} puma"
+      task command do
+        on roles(:app) do
+          execute "service puma_#{application_basename} #{command}"
+        end
+      end
+    end
+
+    desc "restart puma"
+    task :restart do
+      on roles(:app) do
+        execute "service puma_#{application_basename} restart || "\
+                "service puma_#{application_basename} start"
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/rake.rake

@@ -0,0 +1,20 @@
+atlas_recipe :rake do
+  # No hooks
+end
+
+namespace :atlas do
+  desc "Remotely execute a rake task"
+  task :rake do
+    if ENV['COMMAND'].nil?
+      raise "USAGE: cap #{fetch(:stage)} atlas: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,104 @@
+atlas_recipe :rbenv do
+  during :provision, %w(install write_vars)
+end
+
+namespace :atlas do
+  namespace :rbenv do
+    desc "Install rbenv and compile ruby"
+    task :install do
+      invoke "atlas:rbenv:run_installer"
+      invoke "atlas:rbenv:add_plugins"
+      invoke "atlas:rbenv:modify_bashrc"
+      invoke "atlas:rbenv:compile_ruby"
+    end
+
+    desc "Install the latest version of Ruby"
+    task :upgrade do
+      invoke "atlas:rbenv:add_plugins"
+      invoke "atlas:rbenv:update_rbenv"
+      invoke "atlas: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(:atlas_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
+      installer_url = \
+      "https://github.com/rbenv/rbenv-installer/raw/master/bin/rbenv-installer"
+
+      on release_roles(:all) do
+        with :path => "$HOME/.rbenv/bin:$HOME/.rbenv/shims:$PATH" do
+          execute :curl, "-fsSL", installer_url, "| bash"
+        end
+      end
+    end
+
+    task :add_plugins do
+      plugins = %w(
+        sstephenson/rbenv-vars
+        sstephenson/ruby-build
+        rkh/rbenv-update
+      )
+      plugins.each do |plugin|
+        git_repo = "https://github.com/#{plugin}.git"
+        plugin_dir = "$HOME/.rbenv/plugins/#{plugin.split('/').last}"
+
+        on release_roles(:all) do
+          unless test("[ -d #{plugin_dir} ]")
+            execute :git, "clone", git_repo, plugin_dir
+          end
+        end
+      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 :compile_ruby do
+      ruby_version = fetch(:atlas_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 --no-document"
+        end
+      end
+    end
+
+    task :update_rbenv do
+      on release_roles(:all) do
+        execute "rbenv update"
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/seed.rake

@@ -0,0 +1,16 @@
+atlas_recipe :seed do
+  prior_to "deploy:publishing", "atlas:seed"
+end
+
+namespace :atlas do
+  desc "Run rake db:seed"
+  task :seed do
+    on primary(:app) do
+      within release_path do
+        with :rails_env => fetch(:rails_env) do
+          execute :rake, "db:seed"
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/tasks/sidekiq.rake

@@ -0,0 +1,42 @@
+atlas_recipe :sidekiq do
+  during :provision, "init_d"
+  during "deploy:start", "start"
+  during "deploy:stop", "stop"
+  during "deploy:restart", "restart"
+  during "deploy:publishing", "restart"
+end
+
+namespace :atlas do
+  namespace :sidekiq do
+    desc "Install sidekiq service script"
+    task :init_d do
+      privileged_on roles(fetch(:atlas_sidekiq_role)) do |host, user|
+        template "sidekiq_init.erb",
+                 "/etc/init.d/sidekiq_#{application_basename}",
+                 :mode => "a+rx",
+                 :binding => binding,
+                 :sudo => true
+
+        execute "sudo update-rc.d -f sidekiq_#{application_basename} defaults"
+      end
+    end
+
+    %w[start stop].each do |command|
+      desc "#{command} sidekiq"
+      task command do
+        on roles(fetch(:atlas_sidekiq_role)) do
+          execute "service sidekiq_#{application_basename} #{command}"
+        end
+      end
+    end
+
+    desc "restart sidekiq"
+    task :restart do
+      # Re-enable the "stop" task, just in case we called it once already
+      # (as would happen during deploy:migrate_and_restart).
+      Rake::Task["atlas:sidekiq:stop"].reenable
+      invoke "atlas:sidekiq:stop"
+      invoke "atlas:sidekiq:start"
+    end
+  end
+end