app-tools 1.16.2 → 1.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b64fc53d70475212a0ffcd12eb559f6a4bd50575
-  data.tar.gz: 00cb6fe266f70fb152bc88e20597644d91e35738
+  metadata.gz: 1b1349b6e56f652dd2e10565c9fe956cde148239
+  data.tar.gz: 0f08c79203bcca5b91a5b66866478a8af38eea88
 SHA512:
-  metadata.gz: 7df2ca8af806b0da102dc58500cccc932b05b856a72b89b9da842214859b5dac93ede14739389b49b242d9c3876b97c10ac78a5ccdc77220a9af5afcd04e25bb
-  data.tar.gz: bbfa4542f413484392366daa4e1d5c43f1f622dd8d785ce2ab608baeaf3712890be5c363ba67ae8935dcffdb77366c18eecc666d121e9cb57b2a4caf0288bde1
+  metadata.gz: 6c3669c778ab3acc2ac98c2e88a927f44c38d3976b23b3a8b46b5ea1dd1e42be8fd493983ceb26d30b03c54f4a404c24e9a0b3f21a2bc4c4ebf3440c7a665b58
+  data.tar.gz: 7fe90f20aa1972ba86f5882d6d9b2fc8ce01ec4a9ed5a846e5fb61fb1cd0416a64a9871de7d6fbec67cf75d84cdf13a2db3e1d518c471458279964b9f2e1d9d2

data/bin/mongo-backup

@@ -6,6 +6,7 @@ require 'app_tools/version'
 require 'fileutils'
 require 'mongo'
 require 'tmpdir'
+require 'slack-ruby-client'
 
 module AppTools
   module MongoBackup
@@ -29,6 +30,8 @@ module AppTools
       bucket_name = options[:bucket]
       max_backups = options[:max].to_i
       mongo_path = options[:mongo]
+      slack_api_token = options[:api]
+      slack_channel = options[:channel]
 
       if profile.nil?
         exit_now! "Must specify AWS profile"
@@ -47,6 +50,10 @@ module AppTools
         exit_now! "Must specify a database to backup!"
       end
 
+      if slack_api_token.nil? or slack_channel.nil?
+        puts "WARNING: No Slack API and/or channel token given, no notification will be sent"
+      end
+
       mongo_uri = Mongo::URI.new(mongo_path)
       database = mongo_uri.database
       user = mongo_uri.credentials[:user]
@@ -58,67 +65,111 @@ module AppTools
       scutil_tool = find_tool('scutil', ['/usr/sbin'])
       tar_tool = find_tool('tar', ['/usr/bin'])
 
-      Dir.mktmpdir {|dump_dir_name|
-        local_host_name = `#{scutil_tool} --get LocalHostName`.strip
-
-        puts "Backing up mongodb://#{mongo_server}/#{database} from #{local_host_name}"
-
-        if !(user.nil? and password.nil?)
-          `#{mongodump_tool} -h #{mongo_server} -u #{user} -p #{password} -d #{database} -o #{dump_dir_name}`
-        else
-          `#{mongodump_tool} -h #{mongo_server} -d #{database} -o #{dump_dir_name}`
-        end
-
-        if $? != 0
-          raise "Unable to create dump of mongodb://#{mongo_server}/#{database}"
-        end
-
-        date_time = DateTime.now.strftime("%Y%m%d-%H%M%SZ")
-
-        backup_filename = "#{local_host_name}-#{database}-#{date_time}.tar.gz"
-
-        `cd #{dump_dir_name}; #{tar_tool} -czvf #{backup_filename} #{database}/*`
-
-        if $? != 0
-          raise "Unable to create tar zip file '#{backup_filename}'"
-        end
-
-        `cd #{dump_dir_name}; #{aws_tool} s3 cp #{backup_filename} s3://#{bucket_name}/ --profile #{profile}`
-
-        if $? != 0
-          raise "Unable to upload '#{backup_filename}' to s3://#{bucket_name}/"
-        end
-
-        # Reduce backups to desired maximum
-        backup_filenames = `#{aws_tool} s3 ls s3://#{bucket_name} --profile #{profile}`.split('\n')
-
-        if $? != 0
-          raise "Unable to get list of existing backups from s3://#{bucket_name}"
-        end
-
-        # backup_filenames = %w(
-        #   'host-database-20160624-002055Z.tar.gz',
-        #   'host-database-20160620-002055Z.tar.gz',
-        #   'host-database-20160621-002055Z.tar.gz',
-        #   'host-database-20160708-002055Z.tar.gz',
-        #   'host-database-20160601-002055Z.tar.gz.save',
-        # )
-        backup_filenames.sort!
-        backup_filenames.delete_if { |f| f.match(/save/) }
-        num_backups_to_delete = backup_filenames.count - max_backups
-
-        if num_backups_to_delete > 0
-          (0...num_backups_to_delete).each { |i|
-            `#{aws_tool} s3 rm s3://#{bucket_name}/#{backup_filenames[i]} --profile #{profile}`
-
-            if $? != 0
-              puts "WARNING: Unable to delete s3://#{bucket_name}/#{backup_filenames[i]}"
+      backup_status = :failed
+
+      Dir.mktmpdir do |dump_dir_name|
+        begin
+          local_host_name = `#{scutil_tool} --get LocalHostName`.strip
+
+          puts "Backing up mongodb://#{mongo_server}/#{database} from #{local_host_name}"
+
+          if !(user.nil? and password.nil?)
+            `#{mongodump_tool} -h #{mongo_server} -u #{user} -p #{password} -d #{database} -o #{dump_dir_name}`
+          else
+            `#{mongodump_tool} -h #{mongo_server} -d #{database} -o #{dump_dir_name}`
+          end
+
+          if $? != 0
+            raise "Unable to create dump of mongodb://#{mongo_server}/#{database}"
+          end
+
+          date_time = DateTime.now.strftime("%Y%m%d-%H%M%SZ")
+
+          backup_filename = "#{local_host_name}-#{database}-#{date_time}.tar.gz"
+
+          `cd #{dump_dir_name}; #{tar_tool} -czvf #{backup_filename} #{database}/*`
+
+          if $? != 0
+            raise "Unable to create tar zip file '#{backup_filename}'"
+          end
+
+          `cd #{dump_dir_name}; #{aws_tool} s3 cp #{backup_filename} s3://#{bucket_name}/ --profile #{profile}`
+
+          if $? != 0
+            raise "Unable to upload '#{backup_filename}' to s3://#{bucket_name}/"
+          end
+
+          # Reduce backups to desired maximum
+          backup_filenames = `#{aws_tool} s3 ls s3://#{bucket_name} --profile #{profile}`.split('\n')
+
+          if $? != 0
+            raise "Unable to get list of existing backups from s3://#{bucket_name}"
+          end
+
+          # backup_filenames = %w(
+          #   'host-database-20160624-002055Z.tar.gz',
+          #   'host-database-20160620-002055Z.tar.gz',
+          #   'host-database-20160621-002055Z.tar.gz',
+          #   'host-database-20160708-002055Z.tar.gz',
+          #   'host-database-20160601-002055Z.tar.gz.save',
+          # )
+          backup_filenames.sort!
+          backup_filenames.delete_if { |f| f.match(/save/) }
+          num_backups_to_delete = backup_filenames.count - max_backups
+
+          if num_backups_to_delete > 0
+            (0...num_backups_to_delete).each { |i|
+              `#{aws_tool} s3 rm s3://#{bucket_name}/#{backup_filenames[i]} --profile #{profile}`
+
+              if $? != 0
+                puts "WARNING: Unable to delete s3://#{bucket_name}/#{backup_filenames[i]}"
+              end
+            }
+          end
+
+          backup_status = :succeeded
+        ensure
+          Slack.configure do |config|
+            config.token = slack_api_token
+          end
+
+          unless slack_api_token.nil? or slack_channel.nil?
+            if backup_status == :succeeded
+              message = "MongoDB backup complete"
+              attachments = [
+                {
+                    "color": "good",
+                    "title": "MongoDB Backup Succeeded",
+                    "text": "A backup of database mongodb://#{mongo_server}/#{database} was made from #{local_host_name} to s3://#{bucket_name}/#{backup_filename}",
+                    "ts": Time.now.utc.to_i
+                }
+              ]
+            else
+              message = "MongoDB backup failed"
+              attachments = [
+                  {
+                      "color": "danger",
+                      "title": "MongoDB Backup Failed",
+                      "text": "A backup of database mongodb://#{mongo_server}/#{database} from #{local_host_name} failed.",
+                      "ts": Time.now.utc.to_i
+                  }
+              ]
+            end
+            begin
+              slack = Slack::Web::Client.new
+              slack.chat_postMessage(channel: slack_channel, text: message, attachments: attachments, as_user: true)
+            rescue
+              puts "Unable to send Slack message to #{slack_channel}"
             end
-          }
+          end
         end
-      }
+      end
 
-      puts "Backup completed successfully"
+      if backup_status == :succeeded
+        puts "Backup was successful"
+      else
+        puts "Backup failed"
+      end
     end
 
     description 'mongo-backup - Perform backup for a MongoDB database'
@@ -128,6 +179,8 @@ module AppTools
     on("-b", "--bucket BUCKET_NAME", "AWS bucket")
     on("-n", "--max MAX_BACKUPS", "Maximum number of backups to keep")
     on("-m", "--mongo MONGO_URI", "Full URI of MongoDB to backup")
+    on("-t", "--api SLACK_API_TOKEN", "Slack API token")
+    on("-c", "--channel SLACK_CHANNEL", "Slack channel")
 
     use_log_level_option :toggle_debug_on_signal => 'USR1'
 

data/lib/app_tools/version.rb

@@ -1,3 +1,3 @@
 module AppTools
-  VERSION = "1.16.2"
+  VERSION = "1.17.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: app-tools
 version: !ruby/object:Gem::Version
-  version: 1.16.2
+  version: 1.17.0
 platform: ruby
 authors:
 - John Lyon-Smith
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-05 00:00:00.000000000 Z
+date: 2016-09-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -136,6 +136,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.3'
+- !ruby/object:Gem::Dependency
+  name: slack-ruby-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
 description: |
   Generate IPA files with correct Swift and symbol files for uploading to iTunesConnect.
   Resign IPA files with correct certificate and provisioning profiles for distribution.  Upload IPA files to iTunesConnect.