Implement leo updater

leo/lib.rs

@@ -25,3 +25,4 @@ pub mod config;
 pub mod errors;
 pub mod logger;
 pub mod synthesizer;
+pub mod updater;

leo/updater.rs

@@ -0,0 +1,86 @@
+use crate::config::Config;
+
+use colored::Colorize;
+use self_update::{backends::github, version::bump_is_greater, Status};
+
+pub struct Updater;
+
+// TODO Add logic for users to easily select release versions.
+impl Updater {
+    const LEO_BIN_NAME: &'static str = "leo";
+    const LEO_REPO_NAME: &'static str = "leo";
+    const LEO_REPO_OWNER: &'static str = "AleoHQ";
+
+    /// Show all available releases for `leo`.
+    pub fn show_available_releases() -> Result<(), self_update::errors::Error> {
+        let releases = github::ReleaseList::configure()
+            .repo_owner(Self::LEO_REPO_OWNER)
+            .repo_name(Self::LEO_REPO_NAME)
+            .build()?
+            .fetch()?;
+
+        tracing::info!("List of available Leo's versions");
+        for release in releases {
+            tracing::info!("* {}", release.version);
+        }
+        Ok(())
+    }
+
+    /// Update `leo` to the latest release.
+    pub fn update_to_latest_release(show_output: bool) -> Result<Status, self_update::errors::Error> {
+        let status = github::Update::configure()
+            .repo_owner(Self::LEO_REPO_OWNER)
+            .repo_name(Self::LEO_REPO_NAME)
+            .bin_name(Self::LEO_BIN_NAME)
+            .current_version(&include_str!("./leo-version").replace('v', ""))
+            .show_download_progress(true)
+            .no_confirm(true)
+            .show_output(show_output)
+            .build()?
+            .update()?;
+
+        Ok(status)
+    }
+
+    /// Check if there is an available update for `leo` and return the newest release.
+    pub fn update_available() -> Result<Option<String>, self_update::errors::Error> {
+        let updater = github::Update::configure()
+            .repo_owner(Self::LEO_REPO_OWNER)
+            .repo_name(Self::LEO_REPO_NAME)
+            .bin_name(Self::LEO_BIN_NAME)
+            .current_version(&include_str!("./leo-version").replace('v', ""))
+            .build()?;
+
+        let current_version = updater.current_version();
+        let latest_release = updater.get_latest_release()?;
+
+        if bump_is_greater(&current_version, &latest_release.version)? {
+            Ok(Some(latest_release.version))
+        } else {
+            Ok(None)
+        }
+    }
+
+    /// Display the CLI message, if the Leo configuration allows.
+    pub fn print_cli() {
+        let config = Config::read_config().unwrap();
+
+        if config.auto_update {
+            // If the auto update configuration is on, attempt to update the version.
+            if let Ok(status) = Self::update_to_latest_release(false) {
+                if status.updated() {
+                    tracing::info!("Successfully updated to {}", status.version());
+                }
+            }
+        } else {
+            // If the auto update configuration is off, notify the user to update leo.
+            if let Some(latest_version) = Self::update_available().unwrap() {
+                let mut message = format!("{}", "🟢 A new version is available! Run".bold().green());
+                message += &format!("{}", " `leo update` ".bold().white());
+                message += &format!("{}", &(format!("to update to v{}.", latest_version)).bold().green());
+
+                tracing::info!("\n{}\n", message);
+            }
+        }
+    }
+}