do_snapshot 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f4c4999ce47e4ce7b2466c14205933af0cbdc961
-  data.tar.gz: 0403072d2dd494b8f03090e7c4b7624414c9d7b8
+  metadata.gz: 33604b56b4b901f20645dd6a6ac8f6530bb3c3b7
+  data.tar.gz: 57b7c547a40af585b4d28314a274d759ffad9559
 SHA512:
-  metadata.gz: 6c9866805ad171483316b5dd4fbb1e674dff9d779b097e23757167af90f99cb4acd39fd9827b1fbe3d52a5f4a8aae06578c49fa81ad9defea6f535be4183bb26
-  data.tar.gz: 5ac4d9c275029df7a2300c258ba7b75268657353e906c01cfc375189a12c4d5be64e27253bb49e52df39a5656e67259f126ca28c4659a58a78cc5922837c6eee
+  metadata.gz: 72ea4af26a82e76649593595ca079ef18ea651b153344915ac246fc915a2ddbec8330ff7ed42d89a7e23c56828b6644e213eddc3b0aa137045a298687200b41e
+  data.tar.gz: b2afa75459cfe9fff4dd52aedb5c1b05ba509532433132f149ac539d311addfea1fe881d5851ab98ae280f5121dd762facffcd571a87ea7d70ae7e26b78b0765

data/README.md

@@ -1,16 +1,17 @@
 # DoSnapshot
 
-You can use this gem to backup's DigitalOcean droplet's via snapshot.
+You can use this gem to backup's DigitalOcean droplet's via snapshot method.
 
 Here some features:
 
-- Multiple threads out of the box.
-- Binary for cron and command-line.
-- Mail notification when maximum of snapshots for droplet reached.
+- Multiple threads out of the box. No matter how much droplet's you have.
+- Auto-cleanup for old snapshots.
+- Binary special for cron and command-line.
+- Mail notifications when fail or maximum of snapshots is reached for one or multiple droplets.
 - Custom mail settings (You can set [Pony](https://github.com/benprew/pony) mail settings).
 - Stop mode (when you don't want to create new snapshots when maximum is reached).
 - Logging into selected directory.
-- Trace mode for research.
+- Verbose mode for research.
 - Quiet mode for silence.
 
 There not so much of dependencies:
@@ -35,64 +36,96 @@ Or install it yourself as:
 
 ## Usage
 
-First you may need to set DigitalOcean API keys:
- 
-
-    $ export DIGITAL_OCEAN_CLIENT_ID = "SOMEID"
-    $ export DIGITAL_OCEAN_API_KEY = "SOMEKEY"
+First you may need to set DigitalOcean API keys: 
 
+    $ export DIGITAL_OCEAN_CLIENT_ID="SOMEID"
+    $ export DIGITAL_OCEAN_API_KEY="SOMEKEY"
  
-If you want to set keys without environment, than set it via options:
+If you want to set keys without environment, than set it via options when you run do_snapshot:
 
     $ do_snapshot --digital-ocean-client-id YOURLONGAPICLIENTID --digital-ocean-api-key YOURLONGAPIKEY
 
+#### Basic usage
+
+Keep latest 5 and cleanup older if maximum is reached.
+Here we `keeping` only 5 `latest` snapshots and `cleanup` older after new one is created.
+**If creation of snapshots failed no one will be deleted.**
+
+**By default we `keep` 10 droplets.**
+
+    $ do_snapshot --keep 5 -c
+  
+Keep latest 3 from selected droplets:
+  
+    $ do_snapshot --only 123456 1234567 --keep 3
+  
+Working with all except selected droplets:
+  
+    $ do_snapshot --exclude 123456 123457
+  
+Keep latest 5 snapshots, send mail notification instead of creating new one:
+  
+    $ do_snapshot --keep 10 --stop --mail to:yourmail@example.com
+    
 E-mail notifications disabled out of the box. 
 For working mailer you need to set e-mail settings via run options.
 
-```shell
-Usage:
-  do_snapshot create
-
-Options:
-  -o, [--only=123456 123456 123456]                              # Use only selected droplets.
-  -e, [--exclude=123456 123456 123456]                           # Except some droplets.
-  -k, [--keep=5]                                                 # How much snapshots you want to keep?
-                                                                 # Default: 10
-  -s, [--stop], [--no-stop]                                      # Stop creating snapshots if maximum is reached.
-  -m, [--mail=to:yourmail@example.com]                           # Receive mail if maximum is reached.
-  -t, [--smtp=user_name:yourmail@example.com password:password]  # SMTP options.
-  -l, [--log=/Users/someone/.do_snapshot/main.log]               # Log file path. By default logging is disabled.
-  -d, [--trace], [--no-trace]                                    # Debug mode.
-  -q, [--quiet], [--no-quiet]                                    # Quiet mode. If don't need any messages and log's
-      [--digital-ocean-client-id=YOURLONGAPICLIENTID]            # DIGITAL_OCEAN_CLIENT_ID. if you can't use environment.
-      [--digital-ocean-api-key=YOURLONGAPIKEY]                   # DIGITAL_OCEAN_API_KEY. if you can't use environment.
-
-Description:
-  `do_snapshot create` will create and cleanup snapshots on your droplets.
-
-  You can optionally specify parameters to select or exclude some droplets.
-
-  Advanced options example for MAIL feature:
-
-  --mail to:mail@somehost.com from:from@host.com 
-  --smtp address:smtp.gmail.com user_name:someuser password:somepassword
-
-  For more details look here: [https://github.com/benprew/pony](https://github.com/benprew/pony)
-
-  Example:
-
-  > $ do_snapshot --keep 5 --quiet
-
-  > $ do_snapshot --only 123456 1234567 --store 3
-
-  > $ do_snapshot --exclude 123456 123457
-
-  > $ do_snapshot --keep 10 --stop true --mail to:yourmail@example.com
-```
+    --mail to:mail@somehost.com from:from@host.com --smtp address:smtp.gmail.com port:25 user_name:someuser password:somepassword
+
+#### Cron example
+
+    0 4 * * 7 /.../bin/do_snapshot -k 5 -m to:TO from:FROM -t address:HOST user_name:LOGIN password:PASSWORD port:2525 -q -c
+
+#### Real world example
+
+    $ bin/do_snapshot --only 123456 -k 3 -c -m to:TO from:FROM -t address:HOST user_name:LOGIN password:PASSWORD port:2525 -v                                  
+    
+    Checking DigitalOcean Id's.
+    Start performing operations
+    Setting DigitalOcean Id's.
+    Loading list of DigitalOcean droplets
+    Working with list of DigitalOcean droplets
+    Preparing droplet id: 123456 name: mrcr.ru to take snapshot.
+    Shutting down droplet.
+    Start creating snapshot for droplet id: 123456 name: mrcr.ru.
+    Wait until snapshot will be created.
+    Snapshot name: mrcr.ru_2014_07_18 created successfully.
+    Droplet id: 123456 name: mrcr.ru snapshots: 4.
+    For droplet with id: 123456 and name: mrcr.ru the maximum number 3 of snapshots is reached.
+    Cleaning up snapshots for droplet id: 123456 name: mrcr.ru.
+    Snapshot name: mrcr.ru_2014_07_17 delete requested.
+    All operations has been finished.
+    Sending e-mail notification.
+
+### All options:    
+
+    > $ do_snapshot c  
+    
+    aliases: s, snap, create
+    
+    Options:
+      -o, [--only=123456 123456 123456]                              # Select some droplets.
+      -e, [--exclude=123456 123456 123456]                           # Except some droplets.
+      -k, [--keep=5]                                                 # How much snapshots you want to keep?
+                                                                     # Default: 10
+      -m, [--mail=to:yourmail@example.com]                           # Receive mail if fail or maximum is reached.
+      -t, [--smtp=user_name:yourmail@example.com password:password]  # SMTP options.
+      -l, [--log=/Users/someone/.do_snapshot/main.log]               # Log file path. By default logging is disabled.
+      -c, [--clean], [--no-clean]                                    # Cleanup snapshots after create. If you have more images than you want to `keep`, older will be deleted.
+      -s, [--stop], [--no-stop]                                      # Stop creating snapshots if maximum is reached.
+      -v, [--trace], [--no-trace]                                    # Verbose mode.
+      -q, [--quiet], [--no-quiet]                                    # Quiet mode. If don't need any messages in console.
+          [--digital-ocean-client-id=YOURLONGAPICLIENTID]            # DIGITAL_OCEAN_CLIENT_ID. if you can't use environment.
+          [--digital-ocean-api-key=YOURLONGAPIKEY]                   # DIGITAL_OCEAN_API_KEY. if you can't use environment.    
+    
+    Description:
+      `do_snapshot` able to create and cleanup snapshots on your droplets.
+    
+      You can optionally specify parameters to select or exclude some droplets.   
 
 ## Contributing
 
-1. Fork it ( https://github.com/[my-github-username]/do_snapshot/fork )
+1. Fork it ( https://github.com/merqlove/do_snapshot/fork )
 2. Create your feature branch (`git checkout -b my-new-feature`)
 3. Commit your changes (`git commit -am 'Add some feature'`)
 4. Push to the branch (`git push origin my-new-feature`)

data/do_snapshot.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |spec|
   spec.version       = DoSnapshot::VERSION
   spec.authors       = ['Alexander Merkulov']
   spec.email         = ['sasha@merqlove.ru']
-  spec.summary       = %q{Snapshot creator for Digital Ocean droplets. Multi-threading inside. Use it with Cron or other tools.}
-  spec.description   = %q{Snapshot creator for Digital Ocean droplets. Multi-threading inside. Use it with Cron or other tools.}
+  spec.summary       = %q{Snapshot creator for Digital Ocean droplets. Multi-threading. Auto-cleanup. Cron optimized.}
+  spec.description   = %q{Snapshot creator for Digital Ocean droplets. Multi-threading inside. Auto-cleanup feature. No matter how much droplets you have. Cron optimized.}
   spec.homepage      = 'http://github.com/merqlove/do_snapshot'
   spec.license       = 'MIT'
 

data/lib/do_snapshot/cli.rb

@@ -1,23 +1,27 @@
 require 'do_snapshot'
 require 'do_snapshot/command'
-require 'thor'
 
 module DoSnapshot
   # CLI is here
   #
   class CLI < Thor
-    default_task :create
+    default_task :snap
+
+    map %w( c s create )  => :snap
+    map %w( -V ) => :version
 
     def initialize(*args)
       super
 
-      # Use Thor log especially
-      Log.thor_log = Thor::Shell::Color.new unless options.include?('quiet')
+      Log.quiet = options['quiet']
+      # Use Thor shell
+      Log.shell = shell unless Log.quiet
+      Log.verbose = options['trace']
 
-      if options.include?('log')
-        Log.logger       = Logger.new(options['log'])
-        Log.logger.level = options.include?('trace') ? Logger::DEBUG : Logger::INFO
-      end
+      logger if options.include?('log')
+
+      Log.mail = options['mail']
+      Log.smtp = options['smtp']
 
       # Check for keys via options
       ENV['DIGITAL_OCEAN_CLIENT_ID'] = options['digital_ocean_client_id'] if options.include? 'digital_ocean_client_id'
@@ -26,49 +30,84 @@ module DoSnapshot
       try_keys_first
     end
 
-    desc 'create', 'create and cleanup snapshot\'s'
+    desc 'c / s / snap / create', 'DEFAULT. Create and cleanup snapshot\'s'
     long_desc <<-LONGDESC
-    `do_snapshot create` will create and cleanup snapshots on your droplets.
+    `do_snapshot` able to create and cleanup snapshots on your droplets.
 
     You can optionally specify parameters to select or exclude some droplets.
 
-    Advanced options example for MAIL feature:
+    ### Advanced options example for MAIL feature:
 
-    --mail to:mail@somehost.com from:from@host.com --smtp address:smtp.gmail.com user_name:someuser password:somepassword
+    --mail to:mail@somehost.com from:from@host.com --smtp address:smtp.gmail.com port:25 user_name:someuser password:somepassword
 
     For more details look here: https://github.com/benprew/pony
 
-    Example:
+    ### Examples
+
+    Keep latest 5 and cleanup older if maximum is reached:
+
+    $ do_snapshot --keep 5 -c
+
+    Keep latest 3 from selected droplets:
+
+    $ do_snapshot --only 123456 1234567 --keep 3
+
+    Working with all except selected droplets:
+
+    $ do_snapshot --exclude 123456 123457
 
-    > $ do_snapshot --keep 5
+    Keep latest 5 snapshots, not create new if we have more, send mail notification instead:
 
-    > $ do_snapshot --only 123456 1234567 --store 3
+    $ do_snapshot --keep 10 --stop --mail to:yourmail@example.com
 
-    > $ do_snapshot --exclude 123456 123457
+    ### Cron example
 
-    > $ do_snapshot --keep 10 --stop true --mail to:yourmail@example.com
+    0 4 * * 7 /.../bin/do_snapshot -k 5 -m to:TO from:FROM -t address:HOST user_name:LOGIN password:PASSWORD port:2525 -q -c
+
+    VERSION: #{DoSnapshot::VERSION}
     LONGDESC
-    option :only, type: :array, default: [], aliases: ['-o'], banner: '123456 123456 123456', desc: 'Use only selected droplets.'
-    option :exclude, type: :array, default: [], aliases: ['-e'], banner: '123456 123456 123456', desc: 'Except some droplets.'
-    option :keep, type: :numeric, default: 10, aliases: ['-k'], banner: '5', desc: 'How much snapshots you want to keep?'
-    option :stop, type: :boolean, aliases: ['-s'], desc: 'Stop creating snapshots if maximum is reached.'
-    option :mail, type: :hash, default: {}, aliases: ['-m'], banner: 'to:yourmail@example.com', desc: 'Receive mail if maximum is reached.'
-    option :smtp, type: :hash, default: {}, aliases: ['-t'], banner: 'user_name:yourmail@example.com password:password', desc: 'SMTP options.'
-    option :log, type: :string, aliases: ['-l'], banner: '/Users/someone/.do_snapshot/main.log', desc: 'Log file path. By default logging is disabled.'
-    option :trace, type: :boolean, aliases: ['-d'], desc: 'Debug mode.'
-    option :quiet, type: :boolean, aliases: ['-q'], desc: 'Quiet mode. If don\'t need any messages and log\'s'
-    option :digital_ocean_client_id, type: :string, banner: 'YOURLONGAPICLIENTID', desc: 'DIGITAL_OCEAN_CLIENT_ID. if you can\'t use environment.'
-    option :digital_ocean_api_key, type: :string, banner: 'YOURLONGAPIKEY', desc: 'DIGITAL_OCEAN_API_KEY. if you can\'t use environment.'
-
-    def create
-      DoSnapshot::Command.execute options, %w( log trace quiet digital_ocean_client_id digital_ocean_api_key )
+
+    method_option :only,    type: :array,   default: [], aliases: %w( -o ), banner: '123456 123456 123456', desc: 'Select some droplets.'
+    method_option :exclude, type: :array,   default: [], aliases: %w( -e ), banner: '123456 123456 123456', desc: 'Except some droplets.'
+    method_option :keep,    type: :numeric, default: 10, aliases: %w( -k ), banner: '5', desc: 'How much snapshots you want to keep?'
+    method_option :delay,   type: :numeric, default: 10, aliases: %w( -d ), banner: '5', desc: 'Delay between snapshot operation status requests.'
+    method_option :mail,    type: :hash,    aliases: %w( -m ), banner: 'to:yourmail@example.com', desc: 'Receive mail if fail or maximum is reached.'
+    method_option :smtp,    type: :hash,    aliases: %w( -t ), banner: 'user_name:yourmail@example.com password:password', desc: 'SMTP options.'
+    method_option :log,     type: :string,  aliases: %w( -l ), banner: '/Users/someone/.do_snapshot/main.log', desc: 'Log file path. By default logging is disabled.'
+    method_option :clean,   type: :boolean, aliases: %w( -c ), desc: 'Cleanup snapshots after create. If you have more images than you want to `keep`, older will be deleted.'
+    method_option :stop,    type: :boolean, aliases: %w( -s),  desc: 'Stop creating snapshots if maximum is reached.'
+    method_option :trace,   type: :boolean, aliases: %w( -v ), desc: 'Verbose mode.'
+    method_option :quiet,   type: :boolean, aliases: %w( -q ), desc: 'Quiet mode. If don\'t need any messages and in console.'
+
+    method_option :digital_ocean_client_id, type: :string, banner: 'YOURLONGAPICLIENTID', desc: 'DIGITAL_OCEAN_CLIENT_ID. if you can\'t use environment.'
+    method_option :digital_ocean_api_key,   type: :string, banner: 'YOURLONGAPIKEY',      desc: 'DIGITAL_OCEAN_API_KEY. if you can\'t use environment.'
+
+    def snap
+      Command.execute options, %w( log trace digital_ocean_client_id digital_ocean_api_key )
     rescue => e
+
+      Command.fail_power_on(e.id) if e && e.class == SnapshotCreateError && e.respond_to?('id')
       Log.error e.message
       backtrace(e) if options.include? 'trace'
+      if Log.mail
+        Log.mail[:subject] = 'Digital Ocean: Error.'
+        Log.mail[:body] = 'Please check your droplets.'
+        Log.notify
+      end
+    end
+
+    desc 'version, -V', 'Shows the version of the currently installed DoSnapshot gem'
+    def version
+      puts DoSnapshot::VERSION
     end
 
     protected
 
+    def logger
+      Log.logger = Logger.new(options['log'])
+      Log.logger.level = Log.verbose ? Logger::DEBUG : Logger::INFO
+    end
+
     def backtrace(e)
       e.backtrace.each do |t|
         Log.error t

data/lib/do_snapshot/command.rb

@@ -1,7 +1,4 @@
 require 'digitalocean'
-require 'date'
-require 'pony'
-require 'core_ext/hash'
 require 'thread'
 
 module DoSnapshot
@@ -16,26 +13,41 @@ module DoSnapshot
           send("#{key}=", option) unless skip.include? key
         end
 
-        self.notify  = false
-        self.threads = []
-
+        Log.info 'Start performing operations'
         work_droplets
-        email_message if notify && mail
+        Log.info 'All operations has been finished.'
+
+        Log.notify if notify && !quiet
+      end
+
+      def fail_power_on(id)
+        return unless id
 
-        Log.info 'All snapshots requested'
+        set_id
+        instance = Digitalocean::Droplet.find(id)
+        fail instance.message unless instance.status.include? 'OK'
+        if instance.droplet.status.include? 'active'
+          Log.info "Droplet id: #{id} failed to snapshot. But it still running."
+        else
+          Digitalocean::Droplet.power_on(id)
+          Log.info "Droplet id: #{id} failed to snapshot. POWER ON has been requested."
+        end
       end
 
       protected
 
-      attr_accessor :droplets
-      attr_accessor :exclude
-      attr_accessor :only
-      attr_accessor :keep
-      attr_accessor :mail
-      attr_accessor :smtp
-      attr_accessor :stop
-      attr_accessor :notify
-      attr_accessor :threads
+      attr_accessor :droplets, :mail, :smtp, :exclude, :only
+      attr_accessor :delay, :keep, :quiet, :stop, :clean
+
+      attr_writer :notify, :threads
+
+      def notify
+        @notify ||= false
+      end
+
+      def threads
+        @threads ||= []
+      end
 
       # Getting droplets list from API.
       # And store into object.
@@ -44,7 +56,7 @@ module DoSnapshot
         set_id
         Log.debug 'Loading list of DigitalOcean droplets'
         droplets = Digitalocean::Droplet.all
-        fail droplets.message unless droplets.status == 'OK'
+        fail droplets.message unless droplets.status.include? 'OK'
         self.droplets = droplets.droplets
       end
 
@@ -59,7 +71,7 @@ module DoSnapshot
           next if !only.empty? && !only.include?(id)
 
           instance = Digitalocean::Droplet.find(id)
-          fail instance.message unless instance.status == 'OK'
+          fail instance.message unless instance.status.include? 'OK'
 
           prepare_instance instance.droplet
         end
@@ -69,7 +81,7 @@ module DoSnapshot
       # Threads review
       #
       def thread_chain
-        threads.each {|t| t.join}
+        threads.each { |t| t.join }
       end
 
       # Preparing instance to take snapshot.
@@ -77,9 +89,9 @@ module DoSnapshot
       #
       def prepare_instance(instance)
         return unless instance
-        Log.debug "Preparing droplet id: #{instance.id} name: #{instance.name} to snapshot."
+        Log.info "Preparing droplet id: #{instance.id} name: #{instance.name} to take snapshot."
 
-        warning_size = "For droplet with id: #{instance.id} and name: #{instance.name} the maximum #{keep} is reached."
+        warning_size = "For droplet with id: #{instance.id} and name: #{instance.name} the maximum number #{keep} of snapshots is reached."
 
         if instance.snapshots.size >= keep && stop
           Log.warning warning_size
@@ -90,11 +102,15 @@ module DoSnapshot
         # Stopping instance.
         Log.debug 'Shutting down droplet.'
         threads << Thread.new do
-          unless instance.status.include? 'off'
-            event = Digitalocean::Droplet.power_off(instance.id)
-            if event.status.include? 'OK'
-              sleep 1.3 until get_event_status(event.event_id)
+          begin
+            unless instance.status.include? 'off'
+              event = Digitalocean::Droplet.power_off(instance.id)
+              if event.status.include? 'OK'
+                sleep delay until get_event_status(event.event_id)
+              end
             end
+          rescue => e
+            raise DropletShutdownError.new(instance.id), e.message, e.backtrace
           end
 
           # Create snapshot.
@@ -105,7 +121,7 @@ module DoSnapshot
       # Trying to create a snapshot.
       #
       def create_snapshot(instance, warning_size)
-        Log.debug "Start creating snapshot for droplet id: #{instance.id} name: #{instance.name}."
+        Log.info "Start creating snapshot for droplet id: #{instance.id} name: #{instance.name}."
 
         today         = DateTime.now
         name          = "#{instance.name}_#{today.strftime('%Y_%m_%d')}"
@@ -120,7 +136,7 @@ module DoSnapshot
 
         Log.debug 'Wait until snapshot will be created.'
 
-        sleep 10 until get_event_status(event.event_id)
+        sleep delay until get_event_status(event.event_id)
 
         snapshot_size += 1
 
@@ -130,9 +146,40 @@ module DoSnapshot
         if snapshot_size > keep
           Log.warning warning_size if snapshot_size > keep
           self.notify = true
+
+          # Cleanup snapshots.
+          cleanup_snapshots instance, (snapshot_size - keep - 1) if clean
+        end
+      rescue => e
+        case e.class
+        when SnapshotCleanupError
+          raise
+        else
+          raise SnapshotCreateError.new(instance.id), e.message, e.backtrace
         end
       end
 
+      # Cleanup our snapshots.
+      #
+      def cleanup_snapshots(instance, size)
+        Log.debug "Cleaning up snapshots for droplet id: #{instance.id} name: #{instance.name}."
+
+        (0..size).each do |i|
+          snapshot = instance.snapshots[i]
+          event = Digitalocean::Image.destroy(snapshot.id)
+
+          if !event
+            fail 'Something wrong with DigitalOcean or with your connection :)'
+          elsif event && !event.status.include?('OK')
+            fail event.message
+          end
+
+          Log.info "Snapshot name: #{snapshot.name} delete requested."
+        end
+      rescue => e
+        raise SnapshotCleanupError.new(instance.id), e.message, e.backtrace
+      end
+
       # Looking for event status.
       #
       # Before snapshot we to know that machine has powered off.
@@ -143,29 +190,6 @@ module DoSnapshot
         event.event.percentage && event.event.percentage.include?('100') ? true : false
       end
 
-      # Sending message via Hash params.
-      #
-      # Options:: -m to:mail@somehost.com from:from@host.com -t address:smtp.gmail.com user_name:someuser password:somepassword
-      #
-      def email_message
-        mail.symbolize_keys!
-        smtp.symbolize_keys!
-
-        mail[:subject] = 'DigitalOcean Snapshot maximum is reached.' unless mail[:subject]
-        mail[:body] = "Please cleanup your Digital Ocean account. \nSnapshot maximum is reached." unless mail[:body]
-        mail[:from] = 'noreply@someonelse.com' unless mail[:from]
-        mail[:via] = :smtp unless mail[:via]
-
-        smtp[:domain] = 'localhost.localdomain' unless smtp[:domain]
-        smtp[:port] = '25' unless smtp[:port]
-
-        mail[:via_options] = smtp
-
-        Log.debug 'Sending e-mail notification.'
-        # Look into your inbox :)
-        Pony.mail(mail)
-      end
-
       # Set id's of Digital Ocean API.
       #
       def set_id

data/lib/do_snapshot/version.rb

@@ -1,5 +1,5 @@
 # Current version
 #
 module DoSnapshot
-  VERSION = '0.0.4'
+  VERSION = '0.0.5'
 end

data/lib/do_snapshot.rb

@@ -1,44 +1,120 @@
 require 'do_snapshot/version'
+require 'thor'
 require 'logger'
+require 'date'
+require 'pony'
+require 'core_ext/hash'
 
 # Used primary for creating snapshot's as backups for DigitalOcean
 #
 module DoSnapshot
+  # Set multiple Errors with `id` @param
+  #
+  class DigitalOceanError < StandardError
+    attr_reader :id
+    # @param [Object] id
+    def initialize(id)
+      @id = id
+    end
+  end
+  class DropletShutdownError < DigitalOceanError; end
+  class SnapshotCreateError < DigitalOceanError; end
+  class SnapshotCleanupError < DigitalOceanError; end
+
   # Shared logger
   #
   class Log
     class << self
       attr_accessor :logger
-      attr_accessor :thor_log
+      attr_accessor :shell
+      attr_accessor :mail
+      attr_accessor :quiet
+      attr_accessor :verbose
+      attr_writer :smtp
+
+      def smtp
+        @smtp ||= {}
+      end
 
       def log(type, message)
+        buffer << message
         logger.send(type, message) if logger
+
+        say message, color(type) unless type == :debug && !debug?
+      end
+
+      def color(type)
+        case type
+        when :debug
+          :white
+        when :error
+          :red
+        when :warn
+          :yellow
+        else
+          :green
+        end
       end
 
       def info(message)
         log :info, message
-        say message, :green
       end
 
       def warning(message)
         log :warn, message
-        say message, :yellow
       end
 
       def error(message)
         log :error, message
-        say message, :red
       end
 
       def debug(message)
         log :debug, message
-        say message, :white if logger && logger.level == Logger::DEBUG
+      end
+
+      def say(message, color)
+        shell.say message, color if shell
+      end
+
+      def debug?
+        logger && logger.level == Logger::DEBUG
+      end
+
+      # Sending message via Hash params.
+      #
+      # Options:: --mail to:mail@somehost.com from:from@host.com --smtp address:smtp.gmail.com user_name:someuser password:somepassword
+      #
+      def notify
+        return unless mail
+
+        mail.symbolize_keys!
+        smtp.symbolize_keys!
+
+        notify_init
+
+        Log.debug 'Sending e-mail notification.'
+        # Look into your inbox :)
+        Pony.mail(mail)
       end
 
       protected
 
-      def say(message, color)
-        thor_log.say message, color if thor_log
+      def notify_init
+        mail[:subject] = 'Digital Ocean: maximum snapshots is reached.' unless mail[:subject]
+        mail[:body]    = "Please cleanup your Digital Ocean account.\nSnapshot maximum is reached." unless mail[:body]
+        mail[:from]    = 'noreply@someonelse.com' unless mail[:from]
+        mail[:to]      = 'to@someonelse.com' unless mail[:to]
+        mail[:via]     = :smtp unless mail[:via]
+        mail[:body]    = "#{mail[:body]}\n\nTrace: #{DateTime.now}\n#{buffer.join("\n")}"
+        smtp[:domain]  = 'localhost.localdomain' unless smtp[:domain]
+        smtp[:port]    = '25' unless smtp[:port]
+
+        mail[:via_options] = smtp
+      end
+
+      attr_writer :buffer
+      def buffer
+        @buffer ||= %w()
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: do_snapshot
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Alexander Merkulov
@@ -81,7 +81,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 description: Snapshot creator for Digital Ocean droplets. Multi-threading inside.
-  Use it with Cron or other tools.
+  Auto-cleanup feature. No matter how much droplets you have. Cron optimized.
 email:
 - sasha@merqlove.ru
 executables:
@@ -127,7 +127,7 @@ rubyforge_project:
 rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
-summary: Snapshot creator for Digital Ocean droplets. Multi-threading inside. Use
-  it with Cron or other tools.
+summary: Snapshot creator for Digital Ocean droplets. Multi-threading. Auto-cleanup.
+  Cron optimized.
 test_files:
 - test/.keep