From eb875363e10f74e2f685223249c1b268e86201ea Mon Sep 17 00:00:00 2001
From: Tristan Ancelet <tristanancelet@yahoo.com>
Date: Sat, 19 Oct 2024 10:55:30 -0500
Subject: [PATCH] added feature to allow user to name their backup

---
 install-discord-linux.rb | 82 +++++++++++++++++++++++++---------------
 1 file changed, 51 insertions(+), 31 deletions(-)

diff --git a/install-discord-linux.rb b/install-discord-linux.rb
index 772945d..d015def 100755
--- a/install-discord-linux.rb
+++ b/install-discord-linux.rb
@@ -11,6 +11,7 @@ require 'rubygems/package'
 require 'fileutils'
 require 'optparse'
 require 'zlib'
+require 'find'
 
 # END: Includes & Requires
 
@@ -59,6 +60,7 @@ $CONFIG[:desktop_file_path] = $CONFIG[:discord_path] / 'discord.desktop'
 $CONFIG[:discord_version_file] = $CONFIG[:discord_path] / 'resources' / 'build_info.json'
 $CONFIG[:action] = 'install'
 $CONFIG[:keep_installer] = false
+$CONFIG[:debug] = false
 DISCORD_DESKTOP_CONTENTS=<<EOF
 [Desktop Entry]
 Name=Discord
@@ -95,9 +97,13 @@ OptionParser.new do |parser|
     $CONFIG[:action] = 'uninstall'
   end
 
-  parser.on('-b', '--backup', "Backup Discord Installation (not-implemented)") do 
+  parser.on('-b', '--backup [FILENAME]', "Backup Discord Installation") do |backup_filename|
     $CONFIG[:action] = 'backup'
-    $CONFIG[:backup_filename] = 'discord-backup-' + Time.now.strftime('%Y-%m-%d') + '.tar.gz'
+    if backup_filename.nil?
+      $CONFIG[:backup_filename] = 'discord-backup-' + Time.now.strftime('%Y-%m-%d') + '.tar.gz'
+    else
+      $CONFIG[:backup_filename] = backup_filename + ".tar.gz"
+    end
   end
 
   parser.separator ""
@@ -107,6 +113,11 @@ OptionParser.new do |parser|
     puts parser
     exit
   end
+
+  parser.on('-d', '--debug', "Turn on debug logging") do 
+    $CONFIG[:debug] = true
+  end
+
   parser.on('--keep', 'Keep discord tar.gz file (default: False)') do 
     $CONFIG[:keep_installer] = true
   end
@@ -201,45 +212,54 @@ def make_backup
     symlinks: Array.new
   }
 
-  Dir.glob("#{$CONFIG[:discord_config_path].path}/**/**/**/**").each{ |item| 
-    if File.file?(item) 
-      files[:files] << item
-    elsif File.symlink?(item)
-      files[:symlinks] << item
-    elsif File.directory?(item)
-      files[:dirs] << item
-    end
-  }
-
-  Dir.glob("#{$CONFIG[:discord_path].path}/**/**/**/**").each {|item| 
-    if File.file?(item)
-      files[:files] << item
-    elsif File.symlink?(item)
-      files[:symlinks] << item
-    elsif File.directory?(item)
-      files[:dirs] << item
-    end
-  }
+  [ $CONFIG[:discord_config_path].path, $CONFIG[:discord_path].path ].each do |backup_dir|
+    puts "make_backup: Getting Files from #{backup_dir})"
+    Find.find("#{backup_dir}").each{ |item| 
+      puts("#{item} was retrieved from #{backup_dir}") if $CONFIG[:debug]
+      if File.file?(item) 
+        puts("#{item} was found to be a file") if $CONFIG[:debug]
+        files[:files] << item
+      elsif File.symlink?(item)
+        puts("#{item} was found to be a symlink") if $CONFIG[:debug]
+        files[:symlinks] << item
+      elsif File.directory?(item)
+        puts("#{item} was found to be a directory") if $CONFIG[:debug]
+        files[:dirs] << item
+      end
+    }
+  end
 
   Dir.chdir(Dir.home) do 
+    puts "make_backup: Beginning backup process (OUTPUT_FILE = #{Dir.getwd}/#{$CONFIG[:backup_filename]})"
     File.open($CONFIG[:backup_filename], "wb") do |backup_file|
       Zlib::GzipWriter.wrap(backup_file) do |gzip|
         Gem::Package::TarWriter.new(gzip) do |tar|
-          files[:files].each do |file|
+
+          puts "\nmake_backup : files : Beginning backup process for files (Count: #{files[:files].count})"
+          file_count = files[:files].count
+          files[:files].each_with_index { |file, i|
+            puts "make_backup : files : Backing up #{file} (#{i+1}/#{file_count})" if $CONFIG[:debug]
             File.open(file, 'rb') { |file_to_archive|
               contents = file_to_archive.read()
-              tar.add_file_simple(file_to_archive.path.sub("#{Dir.home}/",''), file_to_archive.stat.mode, contents.length) do |io|
+              tar.add_file_simple(file_to_archive.path.sub("#{Dir.home}/",''), file_to_archive.stat.mode, contents.length) { |io|
                 io.write(contents)
-              end
+              }
             }
-          end
-          files[:dirs].each do |dir|
-            tar.mkdir(dir.sub("#{Dir.home}/",""), File.stat(dir).mode)
-          end
+          }
 
-         files[:symlinks].each {|symlink|
-           tar.add_symlink(symlink.sub("#{Dir.home}/",""), File.readlink(symlink), File.lstat(symlink).mode)
-         }
+          puts "\nmake_backup : directories : Beginning backup process for directories (Count: #{files[:dirs].count})"
+          dir_count = files[:dirs].count
+          files[:dirs].each_with_index {|dir,i|
+            puts "make_backup : directories : Backing up #{dir} (#{i+1}/#{dir_count})" if $CONFIG[:debug]
+            tar.mkdir(dir.sub("#{Dir.home}/",""), File.stat(dir).mode)
+          }
+
+          puts "\nmake_backup : symlinks : Beginning backup process for symlinks (Count: #{files[:symlinks].count})"
+          symlink_count = files[:symlinks].count
+          files[:symlinks].each_with_index {|symlink,i|
+            puts "make_backup : symlinks : Backing up #{symlink} (#{i+1}/#{symlink_count})" if $CONFIG[:debug]
+            tar.add_symlink(symlink.sub("#{Dir.home}/",""), File.readlink(symlink), File.lstat(symlink).mode)
+          }
 
         end
       end