techinform 1.3.2 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f187fd82d7128b0595866009e8bad5e738b19de1cf444135e3277500ca1234d1
-  data.tar.gz: 8f32da705e93d97e1dee77f333d1407d9bb2adae77be2169101d4ddb81951624
+  metadata.gz: 617bad8f7da2970e52597896b96ed96a14c138024fa5e4df81e80603d7c8cd3e
+  data.tar.gz: b624229f38a639b5b1dbf47b0a292487c1f6d38a2703da9e22060a0ee9b16a5b
 SHA512:
-  metadata.gz: ecf874371250f17dc854a57c0d9cf5f3c48f802d0e1f38c764d2b95080517b89c96d050b2e49ba67ec39b5c5a25e118d3668e3487b9522ef4cfaa1ddb675fb73
-  data.tar.gz: cea4da064fe4eed7c7993d62f900530441d906a1929630307ad443ec0ae2fd286073ea1fee9d0b03ed78e9b779b79509c00f92d5c549defd0462f6f2db86a2b0
+  metadata.gz: 0ab59696b7cab4f857ebae00a442f09c67860306a7f9406c8a1a51b4d6c1f08e34f518dbc161eea78b369696f363b13acb2190deea1be9efcba5f7ce12b35711
+  data.tar.gz: 2846c545b9e08e6c7765b3969aeb8626658d57b98d00fbf6050aeb2c1c47693428046d6cf88d3f971da8df32f4c0a830ca5080f92af38914ce7d398463d4a4f1

data/lib/backup/backup.rb

@@ -3,11 +3,13 @@ require_relative '../techinform/defaults'
 require 'highline'
 
 class Backup
+  attr_reader :dry_run
   DATE_FORMAT = '%Y-%m-%d-%H-%M'
 
-  def initialize compress: true, encrypt: nil
+  def initialize(compress: true, encrypt: nil, dry_run: true)
     @compress = compress
     @encrypt = encrypt.nil? ? !ENV['GPGKEY'].nil? : encrypt
+    @dry_run = dry_run
     raise "GPGKEY environment variable should be specified for encryption" if encrypt? && ENV['GPGKEY'].nil?
     ensure_path
   end
@@ -78,7 +80,6 @@ class Backup
   end
 
   def get_datetime_from_filename filename
-    puts 'Filename: ' + filename.split('.').first.split('-')[2..6].join('-')
     DateTime.strptime(filename.split('.').first.split('-')[2..6].join('-'), DATE_FORMAT)
   end
 
@@ -105,51 +106,94 @@ class Backup
     File.size(filepath) > minimum_backup_size
   end
 
-  def clean_files(prefix = 'local', dry_run = true)
+  def clean_files(prefix = 'local')
     puts(dry_run ? 'Performing dry run' : '!!!! Cleaning backups !!!!')
+    if prefix =~ /^\//
+      process_clean_files(prefix)
+    else
+      # Process each backup type
+      Dir["#{Techinform::BACKUPS_PREFIX}/#{prefix}/*"].each do |type|
+        puts "Type: #{type}"
+        # Get all backup name
+        Dir["#{type}/*"].each do |path|
+          process_clean_files(path)
+        end
+      end
+    end
+  end
+
+  def process_clean_files(path)
+    puts "Process files in path #{path}"
+    # Get all files
+    mark_files = mark_file_to_delete(path)
+    mark_files.each do |file, delete|
+      puts "#{file} #{"*" if delete}"
+    end
+    show_statistics(mark_files)
+    unless dry_run
+      return unless HighLine.new.agree "Delete #{mark_files.select{|file, delete| delete}.keys.size} files - Are you sure? (yes/no)"
+      # Actually delete files
+      mark_files.select{|file, delete| delete}.keys.each do |file|
+        puts `rm #{file}`
+      end
+    end
+  end
+
+  def show_statistics(mark_files)
+    files_to_delete, files_to_store, size_to_delete, size_to_store  = 0, 0, 0, 0
+    mark_files.each do |file, is_delete|
+      if is_delete
+        files_to_delete += 1
+        size_to_delete += File.size(file)
+      else
+        files_to_store += 1
+        size_to_store += File.size(file)
+      end
+    end
+    puts "\u{1f480} Files to delete - #{files_to_delete}, files size - #{human_filesize(size_to_delete)} \u{1f480}"
+    puts "Files to store - #{files_to_store}, files size - #{human_filesize(size_to_store)}"
+  end
+
+  # Mark files to delete
+  # { 'filenpath' => true, 'filepath2' => false }
+  # true - delete file, false - do not delete
+  def mark_file_to_delete(path)
+    result = {}
     months_taken = []
     days_taken = []
-    # Process each backup type
-    Dir["#{Techinform::BACKUPS_PREFIX}/#{prefix}/*"].each do |type|
-      puts "Type: #{type}"
-      # Get all backup name
-      Dir["#{type}/*"].each do |name|
-        puts "Name: #{name}"
-        # Get all files
-        result = {} # true - delete file, false - do not delete
-        Dir["#{name}/*"].sort.each do |file|
-          datetime = get_datetime_from_filename(file)
-          day_id = datetime.strftime('%Y-%m-%d')
-          month_id = datetime.strftime('%Y-%m')
-          if datetime < (DateTime.now << 2)     # 2 month ago
-            if datetime.day == 1 && !(months_taken.include? month_id)
-              result[file] = false
-              months_taken << month_id
-            else
-              result[file] = true
-            end
-          elsif datetime < DateTime.now - 14 # 2 weeks ago
-            if [1, 7, 14, 21].include?(datetime.day) && !(days_taken.include?(day_id))
-              result[file] = false
-              days_taken << day_id
-            else
-              result[file] = true
-            end
-          end
-        end
-        result.each do |file, delete|
-          puts "#{file} #{"*" if delete}"
+    Dir["#{path}/*"].sort.each do |file|
+      datetime = get_datetime_from_filename(file)
+      day_id = datetime.strftime('%Y-%m-%d')
+      month_id = datetime.strftime('%Y-%m')
+      if datetime < (DateTime.now << 2)     # 2 month ago
+        if datetime.day == 1 && !(months_taken.include? month_id)
+          result[file] = false
+          months_taken << month_id
+        else
+          result[file] = true
         end
-        unless dry_run
-          return unless HighLine.new.agree "Delete #{result.select{|file, delete| delete}.keys.size} files - Are you sure? (yes/no)"
-          # Actually delete files
-          result.select{|file, delete| delete}.keys.each do |file|
-            puts `rm #{file}`
-          end
+      elsif datetime < DateTime.now - 14 # 2 weeks ago
+        if [1, 7, 14, 21].include?(datetime.day) && !(days_taken.include?(day_id))
+          result[file] = false
+          days_taken << day_id
+        else
+          result[file] = true
         end
+      else
+        result[file] = false
       end
     end
+    return result
+  end
 
+  def human_filesize(size)
+    {
+        ' B'  => 1024,
+        ' KB' => 1024 * 1024,
+        ' MB' => 1024 * 1024 * 1024,
+        ' GB' => 1024 * 1024 * 1024 * 1024,
+        ' TB' => 1024 * 1024 * 1024 * 1024 * 1024
+    }.each_pair { |e, s| return "#{(size.to_f / (s / 1024)).round(2)}#{e}" if size < s }
   end
 end
 

data/lib/techinform/version.rb

@@ -1,3 +1,3 @@
 module Techinform
-  VERSION = "1.3.2"
+  VERSION = "1.4.0"
 end

data/lib/techinform.rb

@@ -43,7 +43,7 @@ module Techinform
     desc "clean [prefix]", 'Clean old backup files (default prefix: local)'
     option :delete, type: :boolean, desc: 'Actually delete files'
     def clean(prefix = 'local')
-      puts Backup.new.clean_files(prefix, !options[:delete])
+      puts Backup.new(dry_run: !options[:delete]).clean_files(prefix)
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: techinform
 version: !ruby/object:Gem::Version
-  version: 1.3.2
+  version: 1.4.0
 platform: ruby
 authors:
 - Alexey Chernikov
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-06-09 00:00:00.000000000 Z
+date: 2019-06-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor