From e91c00c9c07bac63a2d183b297052826eb67858e Mon Sep 17 00:00:00 2001
From: Boris Smidt <155442777+boris-smidt-klarrio@users.noreply.github.com>
Date: Tue, 20 Aug 2024 03:18:27 +0200
Subject: [PATCH] Add aqua tool installer cli (#889)

* Add aqua cli

* Move aqua cli to generic.rs

* Add a dry-run support to aqua

* style: format code

---------

Co-authored-by: Steve Lau <stevelauc@outlook.com>
---
 src/config.rs        |  1 +
 src/main.rs          |  1 +
 src/steps/generic.rs | 14 ++++++++++++++
 3 files changed, 16 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index 017365e6..c3e29153 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -53,6 +53,7 @@ pub enum Step {
     AppMan,
     Asdf,
     Atom,
+    Aqua,
     Audit,
     AutoCpufreq,
     Bin,
diff --git a/src/main.rs b/src/main.rs
index 11438d0f..4b743bd0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -420,6 +420,7 @@ fn run() -> Result<()> {
     })?;
     runner.execute(Step::Poetry, "Poetry", || generic::run_poetry(&ctx))?;
     runner.execute(Step::Zvm, "ZVM", || generic::run_zvm(&ctx))?;
+    runner.execute(Step::Aqua, "aqua", || generic::run_aqua(&ctx))?;
 
     if should_run_powershell {
         runner.execute(Step::Powershell, "Powershell Modules Update", || {
diff --git a/src/steps/generic.rs b/src/steps/generic.rs
index dce89ec7..f2c5e1e8 100644
--- a/src/steps/generic.rs
+++ b/src/steps/generic.rs
@@ -224,6 +224,20 @@ pub fn run_apm(ctx: &ExecutionContext) -> Result<()> {
         .status_checked()
 }
 
+pub fn run_aqua(ctx: &ExecutionContext) -> Result<()> {
+    let aqua = require("aqua")?;
+
+    print_separator("Aqua");
+    if ctx.run_type().dry() {
+        println!("Updating aqua ...");
+        println!("Updating aqua installed cli tools ...");
+        Ok(())
+    } else {
+        ctx.run_type().execute(&aqua).arg("update-aqua").status_checked()?;
+        ctx.run_type().execute(&aqua).arg("update").status_checked()
+    }
+}
+
 pub fn run_rustup(ctx: &ExecutionContext) -> Result<()> {
     let rustup = require("rustup")?;