hr_deploy 0.0.2 → 0.0.6

data/lib/hr_deploy/helper.rb

@@ -0,0 +1,57 @@
+module HR_Deploy
+
+  class Helper
+
+    private
+
+    HELP_STRING = <<-eos
+Usage:            hr_deploy [command]
+
+The following commands are supported:
+
+generate          Generate a config file which specifies targets to deploy to.
+                  Won't overwrite existing config unless confirmed.
+
+deploy [target]   Deploy to [target]. If no [target] is specified, deploy
+                  to default target specified in config.
+                  This command supports the following options:
+
+                  --no-confirm    Skip confirmations while deploying.
+                                  This does not affect the initial
+                                  confirmation that triggers the deploy itself.
+                                  Default behavior is to confirm actions.
+
+                  --dry-run       Don't actually execute commands, only show
+                                  which commands would get executed on a real
+                                  deploy. This skips all confirmations,
+                                  including the initial confirmation that
+                                  triggers the deploy itself.
+                                  Default behavior is to actually run commands.
+
+                  --no-db-change  Skip database related commands: backing up,
+                                  migrating and restarting.
+
+version           Show gem version.
+
+help              Show help.
+
+Verbose documentation is at:
+https://github.com/enthrops/hr_deploy/blob/master/README.md
+    eos
+
+    public
+
+    def show_help(args)
+      after_failed_parse = args.fetch(:failed_parse, false)
+
+      if after_failed_parse
+        puts 'Unknown command'
+        puts
+      end
+
+      puts HELP_STRING
+
+      exit 1 if after_failed_parse
+    end
+  end
+end

data/lib/hr_deploy/parser.rb

@@ -0,0 +1,102 @@
+module HR_Deploy
+
+  class Parser
+
+    def parse_cmd_args
+      command = {}
+      command[:action] = nil
+      command[:options] = {}
+
+      cmd_options = ARGV
+
+      if cmd_options == [] || cmd_options == ['help']
+        command[:action] = :help
+        return command
+      end
+
+      if cmd_options == ['version']
+        command[:action] = :version
+        return command
+      end
+
+      if cmd_options == ['generate']
+        command[:action] = :generate
+        return command
+      end
+
+      if cmd_options.first == 'deploy'
+
+        deploy_args = cmd_options[1, cmd_options.length]
+
+        if deploy_args_valid?(deploy_args)
+
+          # Don't dry run by default
+          dry_run = deploy_args.delete('--dry-run')
+
+          # Confirm actions on deploys by default
+          # If '--dry-run' was provided, skip confirmations
+          confirm = !(deploy_args.delete('--no-confirm') || dry_run)
+
+          # Run database change actions by default
+          db_change = !deploy_args.delete('--no-db-change')
+
+          # If no target specified, it will be set to 'nil'
+          target_name = deploy_args.first
+
+          command[:action] = :deploy
+          command[:options][:target_name] = target_name
+          command[:options][:confirm] = confirm
+          command[:options][:dry_run] = dry_run
+          command[:options][:db_change] = db_change
+
+          return command
+        end
+      end
+
+      # Can't parse command
+      command[:action] = :help
+      command[:options][:failed_parse] = true
+      command
+    end
+
+    private
+
+    def deploy_args_valid?(args_to_deploy)
+
+      # We are going to mutate args later in the method
+      args_to_deploy = args_to_deploy.dup
+
+      # 0 to 3 additional options to 'deploy' command
+      # (target name, '--no-confirm', '--dry-run', '--no-db-change')
+      return false unless args_to_deploy.length <= 4
+
+      # Delete '--no-confirm' flag if it is present
+      if args_to_deploy.index('--no-confirm')
+        args_to_deploy.delete_at(args_to_deploy.index('--no-confirm'))
+      end
+
+      # Delete '--dry-run' flag if it is present
+      if args_to_deploy.index('--dry-run')
+        args_to_deploy.delete_at(args_to_deploy.index('--dry-run'))
+      end
+
+      # Delete '--no-db-change' flag if it is present
+      if args_to_deploy.index('--no-db-change')
+        args_to_deploy.delete_at(args_to_deploy.index('--no-db-change'))
+      end
+
+      # After we have removed '--no-confirm', '--dry-run' and '--no-db-change'
+      # we either have one param (target name) or nothing
+      return false unless args_to_deploy.length <= 1
+
+      # Explicit target should not be a command argument
+      if args_to_deploy.length == 1
+        if args_to_deploy.first.start_with?('--') || args_to_deploy.first.start_with?('-')
+          return false
+        end
+      end
+
+      true
+    end
+  end
+end

data/lib/hr_deploy/target.rb

@@ -3,53 +3,61 @@ module HR_Deploy
   class Target
 
     attr_reader :name
+    attr_reader :new_relic_disable_pinger_url
+    attr_reader :new_relic_enable_pinger_url
     attr_reader :description
-    attr_reader :disable_pinger_url
-    attr_reader :enable_pinger_url
 
-    def backup_db?
-      !!@backup_db
+    def initialize(args)
+
+      @name = args[:name]
+      @new_relic_disable_pinger_url = args[:new_relic_disable_pinger_url]
+      @new_relic_enable_pinger_url = args[:new_relic_enable_pinger_url]
+      @s3_asset_sync = args[:s3_asset_sync]
+      @default = args[:default]
+
+      # Use name for description, if no description given
+      @description = args[:description] || name
+
+      # Backup is off by default
+      @backup_db = args[:backup_db] == true
+
+      # Control maintenance mode is on by default
+      @maintenance = args[:maintenance] == false ? false : true
     end
 
-    def default?
-      !!@default
+    def pinging_interaction?
+      !!new_relic_disable_pinger_url
     end
 
     def requires_curl?
-      !!disable_pinger_url
-    end
-
-    def requires_pinging_interaction?
-      # If URL to disable pinging is set, we want to disable and enable pinging when deploying
-      disable_pinger_url
-    end
-
-    def initialize(raw_target)
-      target_name = raw_target.keys.first
-      @name = target_name.to_s
-
-      options = raw_target[target_name]
-      # Target might have no options
-      if options
-        options.each do |option, value|
-          case option
-          when :description
-            @description = value
-          when :new_relic_disable_pinger_url
-            @disable_pinger_url = value
-          when :new_relic_enable_pinger_url
-            @enable_pinger_url = value
-          when :backup_db
-            @backup_db = value
-          when :default
-            @default = value
-          else
-            raise "Unknown option: #{option}"
-          end
-        end
-      end
-
-      @description = name unless description
+      pinging_interaction?
+    end
+
+    def requires_rake?
+      s3_asset_sync?
+    end
+
+    def s3_asset_sync?
+      !!s3_asset_sync
     end
+
+    def backup_db?
+      !!backup_db
+    end
+
+    def maintenance_interaction?
+      !!maintenance
+    end
+
+    def default?
+      !!default
+    end
+
+    private
+
+    attr_reader :s3_asset_sync
+    attr_reader :maintenance
+    attr_reader :backup_db
+    attr_reader :default
   end
 end

data/lib/hr_deploy/tasks/backup_db_task.rb

@@ -0,0 +1,25 @@
+require 'hr_deploy/tasks/task'
+
+module HR_Deploy
+
+  class BackupDBTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Backing up database...'
+
+      execute_system_command("heroku pgbackups:capture --expire --remote #{target.name}")
+
+      if dry_run? || !confirm?
+        self.success = true
+        return
+      end
+
+      if confirm_successful?('Database backed up?')
+        self.success = true
+      else
+        self.success = false
+        self.error = 'Could not back up database'
+      end
+    end
+  end
+end

data/lib/hr_deploy/tasks/check_heroku_task.rb

@@ -0,0 +1,64 @@
+require 'hr_deploy/tasks/task'
+require 'uri'
+require 'net/http'
+require 'net/https'
+require 'json'
+
+module HR_Deploy
+
+  class CheckHerokuTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Checking Heroku status...'
+
+      if dry_run?
+        self.success = true
+        return
+      end
+
+      uri = URI.parse('https://status.heroku.com/api/v3/current-status')
+
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+
+      request = Net::HTTP::Get.new(uri.request_uri)
+
+      begin
+        response = http.request(request)
+
+      rescue
+        self.success = false
+        self.error = 'Could not fetch Heroku status'
+
+      else
+        if response.code == '200'
+
+          status = JSON.parse(response.body)
+          status = status['status'] if status
+
+          if status && status['Production'] && status['Development']
+
+            production_ok = status['Production'] == 'green'
+            development_ok = status['Development'] == 'green'
+
+            if production_ok && development_ok
+              print_success 'Heroku is operating correctly'
+              self.success = true
+            else
+              self.success = false
+              self.error = 'Heroku is having problems'
+            end
+          else
+
+            self.success = false
+            self.error = 'Could not parse Heroku status'
+          end
+
+        else
+          self.success = false
+          self.error = 'Could not fetch Heroku status'
+        end
+      end
+    end
+  end
+end

data/lib/hr_deploy/tasks/check_network_task.rb

@@ -0,0 +1,29 @@
+require 'hr_deploy/tasks/task'
+require 'open-uri'
+
+module HR_Deploy
+
+  class CheckNetworkTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Checking Internet connection...'
+
+      if dry_run?
+        self.success = true
+        return
+      end
+
+      begin
+        open('http://www.google.com')
+
+      rescue
+        self.error = 'Not connected to the Internet'
+        self.success = false
+
+      else
+        print_success 'Internet connection is fine'
+        self.success = true
+      end
+    end
+  end
+end

data/lib/hr_deploy/tasks/clean_new_assets_task.rb

@@ -0,0 +1,25 @@
+require 'hr_deploy/tasks/task'
+
+module HR_Deploy
+
+  class CleanNewAssetsTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Cleaning up new assets...'
+
+      execute_system_command('rake assets:clean')
+
+      if dry_run? || !confirm?
+        self.success = true
+        return
+      end
+
+      if confirm_successful?('New assets cleaned up?')
+        self.success = true
+      else
+        self.success = false
+        self.error = 'Could not clean up new assets'
+      end
+    end
+  end
+end

data/lib/hr_deploy/tasks/clean_old_assets_task.rb

@@ -0,0 +1,25 @@
+require 'hr_deploy/tasks/task'
+
+module HR_Deploy
+
+  class CleanOldAssetsTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Cleaning up old assets...'
+
+      execute_system_command('rake assets:clean')
+
+      if dry_run? || !confirm?
+        self.success = true
+        return
+      end
+
+      if confirm_successful?('Old assets cleaned up?')
+        self.success = true
+      else
+        self.success = false
+        self.error = 'Could not clean up old assets'
+      end
+    end
+  end
+end

data/lib/hr_deploy/tasks/disable_maintenance_task.rb

@@ -0,0 +1,25 @@
+require 'hr_deploy/tasks/task'
+
+module HR_Deploy
+
+  class DisableMaintenanceTask < HR_Deploy::Task
+
+    def run
+      print_stage 'Disabling maintenance mode...'
+
+      execute_system_command("heroku maintenance:off --remote #{target.name}")
+
+      if dry_run? || !confirm?
+        self.success = true
+        return
+      end
+
+      if confirm_successful?('Maintenance mode disabled?')
+        self.success = true
+      else
+        self.success = false
+        self.error = 'Could not disable maintenance mode'
+      end
+    end
+  end
+end