Update main.cr

Author: michal92299

HackerOS-Updater/main.cr

@@ -2,6 +2,7 @@ require "option_parser"
 require "process"
 require "file_utils"
 require "colorize"
+
 # ANSI color codes
 RED = "\e[31m"
 GREEN = "\e[32m"
@@ -10,25 +11,30 @@ YELLOW = "\e[33m"
 CYAN = "\e[36m"
 MAGENTA = "\e[35m"
 RESET = "\e[0m"
+
 # Paths
 HACKEROS_UPDATE_SCRIPT = "/usr/share/HackerOS/Scripts/Bin/update-hackeros.sh"
 WALLPAPERS_UPDATE_SCRIPT = "/usr/share/HackerOS/Scripts/Bin/update-wallpapers.sh"
 BIN_PATH = Process.executable_path.not_nil!
 AUTO_SCRIPT_PATH = "#{ENV["HOME"]}/.hackeros/auto-update.sh" # Script to wait for internet
+
 def display_header(title : String)
   puts "<--------[ #{title} ]-------->".colorize(:yellow)
 end
+
 def run_command(cmd : String) : {Bool, String}
   status = Process.run(cmd, shell: true, input: Process::Redirect::Inherit, output: Process::Redirect::Inherit, error: Process::Redirect::Inherit)
   {status.success?, ""}
 end
+
 def get_status(success : Bool) : String
   if success
     "#{BLUE}COMPLETE#{RESET}"
   else
     "#{RED}FAILED#{RESET}"
   end
 end
+
 def perform_updates : {String, String, String, String, String, String, String, String, String}
   # APT Update
   display_header("System Update")
@@ -38,44 +44,54 @@ def perform_updates : {String, String, String, String, String, String, String, S
     apt_success &&= success
   end
   apt_status = get_status(apt_success)
+
   # Flatpak Update
   display_header("Flatpak Update")
   flatpak_success, _ = run_command("flatpak update -y")
   flatpak_status = get_status(flatpak_success)
+
   # Snap Update
   display_header("Snap Update")
   snap_success, _ = run_command("sudo snap refresh")
   snap_status = get_status(snap_success)
-  # Brew Update
+
+  # Brew Update (Expanded with cleanup)
   display_header("Brew Update")
   brew_success = true
-  ["brew update", "brew upgrade"].each do |cmd|
+  ["brew update", "brew upgrade", "brew cleanup"].each do |cmd|
     success, _ = run_command(cmd)
     brew_success &&= success
   end
   brew_status = get_status(brew_success)
+
   # Firmware Update
   display_header("Firmware Update")
   fw_success, _ = run_command("sudo fwupdmgr update")
   fw_status = get_status(fw_success)
+
   # Oh My Zsh Update
   display_header("Oh My Zsh Update")
   omz_success, _ = run_command("omz update")
   omz_status = get_status(omz_success)
+
   # Distrobox Update
   display_header("Distrobox Update")
   distrobox_success, _ = run_command("distrobox-upgrade --all")
   distrobox_status = get_status(distrobox_success)
+
   # HackerOS Update
   display_header("HackerOS Update")
   hacker_success, _ = run_command(HACKEROS_UPDATE_SCRIPT)
   hacker_status = get_status(hacker_success)
+
   # Wallpapers Update
   display_header("Wallpaper Updates")
   wall_success, _ = run_command(WALLPAPERS_UPDATE_SCRIPT)
   wall_status = get_status(wall_success)
+
   {apt_status, flatpak_status, snap_status, brew_status, fw_status, omz_status, distrobox_status, hacker_status, wall_status}
 end
+
 def show_summary(apt_status, flatpak_status, snap_status, brew_status, fw_status, omz_status, distrobox_status, hacker_status, wall_status)
   puts "\nSystem Updates - #{apt_status}"
   puts "Flatpak Updates - #{flatpak_status}"
@@ -87,6 +103,7 @@ def show_summary(apt_status, flatpak_status, snap_status, brew_status, fw_status
   puts "HackerOS Updates - #{hacker_status}"
   puts "Wallpaper Updates - #{wall_status}"
 end
+
 def enable_automatic_updates
   # Create a script that waits for internet and runs the updater
   auto_script = <<-SCRIPT
@@ -98,6 +115,7 @@ def enable_automatic_updates
   SCRIPT
   File.write(AUTO_SCRIPT_PATH, auto_script)
   File.chmod(AUTO_SCRIPT_PATH, 0o755)
+
   # Add to crontab
   current_crontab = `crontab -l`
   unless current_crontab.includes?("@reboot #{AUTO_SCRIPT_PATH}")
@@ -108,17 +126,20 @@ def enable_automatic_updates
   end
   puts "#{GREEN}Automatic updates enabled.#{RESET}"
 end
+
 def disable_automatic_updates
   # Remove from crontab
   current_crontab = `crontab -l`
   new_crontab = current_crontab.lines.reject { |line| line.includes?("@reboot #{AUTO_SCRIPT_PATH}") }.join("\n")
   File.write("/tmp/crontab.txt", new_crontab)
   run_command("crontab /tmp/crontab.txt")
   File.delete("/tmp/crontab.txt")
+
   # Remove script if exists
   File.delete(AUTO_SCRIPT_PATH) if File.exists?(AUTO_SCRIPT_PATH)
   puts "#{GREEN}Automatic updates disabled.#{RESET}"
 end
+
 def show_gui_menu
   loop do
     puts "\n#{YELLOW}=== HackerOS Updater Menu ===#{RESET}"
@@ -128,6 +149,7 @@ def show_gui_menu
     puts "#{GREEN}[L]og out#{RESET} #{CYAN}- Log out from current session#{RESET}"
     puts "#{GREEN}[T]erminal#{RESET} #{CYAN}- Open a new Alacritty terminal#{RESET}"
     puts "#{GREEN}[A]utomatic Updates#{RESET} #{CYAN}- Enable automatic updates on boot#{RESET}"
+    puts "#{GREEN}[D]isable Automatic Updates#{RESET} #{CYAN}- Disable automatic updates on boot#{RESET}"  # Added disable option
     print "#{MAGENTA}Enter your choice: #{RESET}"
     choice = ""
     STDIN.raw do |io|
@@ -151,11 +173,14 @@ def show_gui_menu
       Process.new("alacritty", input: Process::Redirect::Close, output: Process::Redirect::Close, error: Process::Redirect::Close)
     when "A"
       enable_automatic_updates
+    when "D"
+      disable_automatic_updates
     else
       puts "#{RED}Invalid choice. Try again.#{RESET}"
     end
   end
 end
+
 def main
   with_gui = false
   gui_mode = false
@@ -164,15 +189,19 @@ def main
     parser.on("--with-gui", "Run in GUI mode with Alacritty") { with_gui = true }
     parser.on("--gui-mode", "Internal GUI mode") { gui_mode = true }
   end
+
   if with_gui
     # Launch in Alacritty with gui-mode
     Process.new("alacritty", args: ["-e", BIN_PATH, "--gui-mode"], input: Process::Redirect::Close, output: Process::Redirect::Close, error: Process::Redirect::Close)
     return
   end
+
   apt_status, flatpak_status, snap_status, brew_status, fw_status, omz_status, distrobox_status, hacker_status, wall_status = perform_updates
   show_summary(apt_status, flatpak_status, snap_status, brew_status, fw_status, omz_status, distrobox_status, hacker_status, wall_status)
+
   if gui_mode
     show_gui_menu
   end
 end
+
 main