Fix formatting for update system

Author: MichaelTaylor3D

src/cli/commands/update.rs

@@ -23,7 +23,7 @@ fn execute_interactive_mode(check_only: bool, force: bool) -> Result<()> {
         Err(e) => {
             println!("{} Failed to check for updates: {}", "✗".red(), e);
             return Ok(());
-        }
+        },
     };
 
     if !update_info.update_available {
@@ -73,20 +73,23 @@ fn execute_interactive_mode(check_only: bool, force: bool) -> Result<()> {
         match download_and_install_update(&download_url) {
             Ok(_) => {
                 println!();
-                println!("{}", "🎊 Update completed successfully!".bright_green().bold());
+                println!(
+                    "{}",
+                    "🎊 Update completed successfully!".bright_green().bold()
+                );
                 println!(
                     "  {} Version {} is now installed",
                     "→".cyan(),
                     update_info.latest_version.bright_white()
                 );
-            }
+            },
             Err(e) => {
                 println!("{} Update failed: {}", "✗".red(), e);
                 println!(
                     "  {} You can manually download from: https://github.com/DIG-Network/digstore/releases",
                     "→".cyan()
                 );
-            }
+            },
         }
     } else {
         println!("{} No installer available for your platform", "⚠".yellow());
@@ -128,7 +131,7 @@ fn execute_json_mode(check_only: bool, force: bool) -> Result<()> {
                     "status": "success"
                 });
                 println!("{}", serde_json::to_string_pretty(&output)?);
-            }
+            },
             Err(e) => {
                 let output = serde_json::json!({
                     "action": "update_failed",
@@ -138,7 +141,7 @@ fn execute_json_mode(check_only: bool, force: bool) -> Result<()> {
                     "manual_download": "https://github.com/DIG-Network/digstore/releases"
                 });
                 println!("{}", serde_json::to_string_pretty(&output)?);
-            }
+            },
         }
     } else {
         let output = serde_json::json!({

src/main.rs

@@ -202,7 +202,7 @@ fn main() -> Result<()> {
             force,
             json,
         } => cli::commands::update::execute(check_only, force, json),
-        
+
         Commands::Layer { command } => match command {
             LayerCommands::List {
                 json,
@@ -367,7 +367,7 @@ fn needs_wallet_initialization(command: &Commands) -> bool {
 fn check_and_prompt_for_updates() -> Result<()> {
     use colored::Colorize;
     use dialoguer::Confirm;
-    
+
     // Only check occasionally to avoid slowing down commands
     let update_info = match crate::update::check_for_updates() {
         Ok(info) => info,
@@ -398,12 +398,20 @@ fn check_and_prompt_for_updates() -> Result<()> {
         if let Some(download_url) = update_info.download_url {
             match crate::update::download_and_install_update(&download_url) {
                 Ok(_) => {
-                    println!("{}", "✓ Update completed! Please restart your terminal.".green().bold());
-                }
+                    println!(
+                        "{}",
+                        "✓ Update completed! Please restart your terminal."
+                            .green()
+                            .bold()
+                    );
+                },
                 Err(e) => {
                     println!("{} Update failed: {}", "✗".red(), e);
-                    println!("  {} Manual download: https://github.com/DIG-Network/digstore/releases", "→".cyan());
-                }
+                    println!(
+                        "  {} Manual download: https://github.com/DIG-Network/digstore/releases",
+                        "→".cyan()
+                    );
+                },
             }
         }
     } else {

src/update/installer.rs

@@ -19,15 +19,15 @@ pub enum InstallerType {
 /// Download and install an update
 pub fn download_and_install_update(download_url: &str) -> Result<()> {
     println!("{}", "Downloading update...".bright_blue());
-    
+
     // Determine installer type from URL
     let installer_type = determine_installer_type(download_url)?;
-    
+
     // Download the installer
     let temp_file = download_installer(download_url)?;
-    
+
     println!("{}", "Installing update...".bright_green());
-    
+
     // Install based on platform
     match installer_type {
         InstallerType::WindowsMsi => install_windows_msi(&temp_file),
@@ -36,13 +36,16 @@ pub fn download_and_install_update(download_url: &str) -> Result<()> {
         InstallerType::LinuxRpm => install_linux_rpm(&temp_file),
         InstallerType::LinuxAppImage => install_linux_appimage(&temp_file),
     }?;
-    
+
     // Clean up temp file
     let _ = std::fs::remove_file(&temp_file);
-    
+
     println!("{}", "✓ Update installed successfully!".green().bold());
-    println!("{}", "Please restart your terminal to use the new version.".yellow());
-    
+    println!(
+        "{}",
+        "Please restart your terminal to use the new version.".yellow()
+    );
+
     Ok(())
 }
 
@@ -74,34 +77,34 @@ fn download_installer(url: &str) -> Result<std::path::PathBuf> {
         .map_err(|e| DigstoreError::NetworkError {
             reason: format!("Failed to create HTTP client: {}", e),
         })?;
-    
+
     let response = client
         .get(url)
         .send()
         .map_err(|e| DigstoreError::NetworkError {
             reason: format!("Failed to download installer: {}", e),
         })?;
-    
+
     if !response.status().is_success() {
         return Err(DigstoreError::NetworkError {
             reason: format!("Download failed with status: {}", response.status()),
         });
     }
-    
+
     // Create temp file
     let temp_dir = std::env::temp_dir();
     let filename = url.split('/').last().unwrap_or("digstore-installer");
     let temp_file = temp_dir.join(filename);
-    
+
     // Write downloaded content
     let content = response.bytes().map_err(|e| DigstoreError::NetworkError {
         reason: format!("Failed to read download content: {}", e),
     })?;
-    
+
     std::fs::write(&temp_file, content)?;
-    
+
     println!("  {} Downloaded to: {}", "✓".green(), temp_file.display());
-    
+
     Ok(temp_file)
 }
 
@@ -113,14 +116,14 @@ fn install_windows_msi(msi_path: &Path) -> Result<()> {
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to run msiexec: {}", e),
         })?;
-    
+
     if !output.status.success() {
         let stderr = String::from_utf8_lossy(&output.stderr);
         return Err(DigstoreError::ConfigurationError {
             reason: format!("MSI installation failed: {}", stderr),
         });
     }
-    
+
     Ok(())
 }
 
@@ -133,13 +136,13 @@ fn install_macos_dmg(dmg_path: &Path) -> Result<()> {
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to mount DMG: {}", e),
         })?;
-    
+
     if !mount_output.status.success() {
         return Err(DigstoreError::ConfigurationError {
             reason: "Failed to mount DMG".to_string(),
         });
     }
-    
+
     // Extract mount point from output
     let mount_info = String::from_utf8_lossy(&mount_output.stdout);
     let mount_point = mount_info
@@ -149,30 +152,30 @@ fn install_macos_dmg(dmg_path: &Path) -> Result<()> {
         .ok_or_else(|| DigstoreError::ConfigurationError {
             reason: "Failed to determine mount point".to_string(),
         })?;
-    
+
     // Copy app to Applications
     let app_name = "Digstore Min.app";
     let source = format!("{}/{}", mount_point, app_name);
     let destination = format!("/Applications/{}", app_name);
-    
+
     let copy_output = Command::new("cp")
         .args(&["-r", &source, &destination])
         .output()
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to copy app: {}", e),
         })?;
-    
+
     // Unmount DMG
     let _ = Command::new("hdiutil")
         .args(&["detach", mount_point])
         .output();
-    
+
     if !copy_output.status.success() {
         return Err(DigstoreError::ConfigurationError {
             reason: "Failed to install app to Applications folder".to_string(),
         });
     }
-    
+
     Ok(())
 }
 
@@ -184,14 +187,14 @@ fn install_linux_deb(deb_path: &Path) -> Result<()> {
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to run dpkg: {}", e),
         })?;
-    
+
     if !output.status.success() {
         let stderr = String::from_utf8_lossy(&output.stderr);
         return Err(DigstoreError::ConfigurationError {
             reason: format!("DEB installation failed: {}", stderr),
         });
     }
-    
+
     Ok(())
 }
 
@@ -203,14 +206,14 @@ fn install_linux_rpm(rpm_path: &Path) -> Result<()> {
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to run rpm: {}", e),
         })?;
-    
+
     if !output.status.success() {
         let stderr = String::from_utf8_lossy(&output.stderr);
         return Err(DigstoreError::ConfigurationError {
             reason: format!("RPM installation failed: {}", stderr),
         });
     }
-    
+
     Ok(())
 }
 
@@ -223,28 +226,32 @@ fn install_linux_appimage(appimage_path: &Path) -> Result<()> {
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to make AppImage executable: {}", e),
         })?;
-    
+
     if !chmod_output.status.success() {
         return Err(DigstoreError::ConfigurationError {
             reason: "Failed to make AppImage executable".to_string(),
         });
     }
-    
+
     // Copy to /usr/local/bin (requires sudo)
     let copy_output = Command::new("sudo")
-        .args(&["cp", appimage_path.to_str().unwrap(), "/usr/local/bin/digstore"])
+        .args(&[
+            "cp",
+            appimage_path.to_str().unwrap(),
+            "/usr/local/bin/digstore",
+        ])
         .output()
         .map_err(|e| DigstoreError::ConfigurationError {
             reason: format!("Failed to install AppImage: {}", e),
         })?;
-    
+
     if !copy_output.status.success() {
         let stderr = String::from_utf8_lossy(&copy_output.stderr);
         return Err(DigstoreError::ConfigurationError {
             reason: format!("AppImage installation failed: {}", stderr),
         });
     }
-    
+
     Ok(())
 }
 
@@ -260,14 +267,14 @@ fn find_platform_download_url(assets: &[GitHubAsset]) -> Option<String> {
     } else {
         vec![]
     };
-    
+
     for pattern in platform_patterns {
         for asset in assets {
             if asset.name.contains(pattern) {
                 return Some(asset.browser_download_url.clone());
             }
         }
     }
-    
+
     None
 }

src/update/mod.rs

@@ -2,8 +2,8 @@
 //!
 //! This module handles version checking and automatic updates
 
-pub mod version_check;
 pub mod installer;
+pub mod version_check;
 
-pub use version_check::{check_for_updates, UpdateInfo};
 pub use installer::{download_and_install_update, InstallerType};
+pub use version_check::{check_for_updates, UpdateInfo};