From e320d68aea907af74ab366ad72bd75602978117f Mon Sep 17 00:00:00 2001
From: raychu86 <raychu86@gmail.com>
Date: Wed, 2 Sep 2020 17:19:23 -0700
Subject: [PATCH] Implement leo updater

---
 leo/lib.rs     |  1 +
 leo/updater.rs | 86 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 87 insertions(+)
 create mode 100644 leo/updater.rs

diff --git a/leo/lib.rs b/leo/lib.rs
index 5de76c348c..1fa729229c 100644
--- a/leo/lib.rs
+++ b/leo/lib.rs
@@ -25,3 +25,4 @@ pub mod config;
 pub mod errors;
 pub mod logger;
 pub mod synthesizer;
+pub mod updater;
diff --git a/leo/updater.rs b/leo/updater.rs
new file mode 100644
index 0000000000..5fdb7b01b0
--- /dev/null
+++ b/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);
+            }
+        }
+    }
+}