From afa0cd0366712d6990390de87d36e471f8bc6a96 Mon Sep 17 00:00:00 2001
From: Gravita <gravita@gravit.pro>
Date: Sun, 11 Jul 2021 17:07:54 +0700
Subject: [PATCH] [FEATURE] Downloader ProgressBar

---
 LaunchServer/build.gradle                     |  1 +
 .../gravit/launchserver/command/Command.java  | 57 +++++++++++++++++++
 2 files changed, 58 insertions(+)

diff --git a/LaunchServer/build.gradle b/LaunchServer/build.gradle
index 680d2052..af6dffac 100644
--- a/LaunchServer/build.gradle
+++ b/LaunchServer/build.gradle
@@ -72,6 +72,7 @@ task cleanjar(type: Jar, dependsOn: jar) {
 dependencies {
     pack project(':LauncherAPI')
     pack group: 'org.apache.maven', name: 'maven-artifact', version: '3.8.1'
+    bundle group: 'me.tongfei', name: 'progressbar', version: '0.9.2'
     bundle group: 'org.fusesource.jansi', name: 'jansi', version: rootProject['verJansi']
     bundle group: 'org.jline', name: 'jline', version: rootProject['verJline']
     bundle group: 'org.jline', name: 'jline-reader', version: rootProject['verJline']
diff --git a/LaunchServer/src/main/java/pro/gravit/launchserver/command/Command.java b/LaunchServer/src/main/java/pro/gravit/launchserver/command/Command.java
index 684d79e8..e7daa403 100644
--- a/LaunchServer/src/main/java/pro/gravit/launchserver/command/Command.java
+++ b/LaunchServer/src/main/java/pro/gravit/launchserver/command/Command.java
@@ -1,8 +1,19 @@
 package pro.gravit.launchserver.command;
 
+import me.tongfei.progressbar.ProgressBar;
+import me.tongfei.progressbar.ProgressBarBuilder;
+import me.tongfei.progressbar.ProgressBarStyle;
+import pro.gravit.launcher.AsyncDownloader;
 import pro.gravit.launchserver.LaunchServer;
+import pro.gravit.utils.Downloader;
 
+import java.io.IOException;
+import java.nio.file.Path;
+import java.util.List;
+import java.util.Locale;
 import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.atomic.AtomicLong;
 
 public abstract class Command extends pro.gravit.utils.command.Command {
     protected final LaunchServer server;
@@ -17,4 +28,50 @@ public Command(Map<String, pro.gravit.utils.command.Command> childCommands, Laun
         super(childCommands);
         this.server = server;
     }
+
+    protected boolean showApplyDialog(String text) throws IOException {
+        System.out.printf("%s [Y/N]:", text);
+        String response = server.commandHandler.readLine().toLowerCase(Locale.ROOT);
+        return response.equals("y");
+    }
+
+    protected Downloader downloadWithProgressBar(String taskName, List<AsyncDownloader.SizedFile> list, String baseUrl, Path targetDir) throws Exception {
+        long total = 0;
+        for (AsyncDownloader.SizedFile file : list) {
+            total += file.size;
+        }
+        long totalFiles = list.size();
+        AtomicLong current = new AtomicLong(0);
+        AtomicLong currentFiles = new AtomicLong(0);
+        ProgressBar bar = (new ProgressBarBuilder()).setTaskName(taskName)
+                .setInitialMax(total)
+                .showSpeed()
+                .setStyle(ProgressBarStyle.COLORFUL_UNICODE_BLOCK)
+                .setUnit("MB", 1024 * 1024)
+                .build();
+        bar.setExtraMessage(String.format(" [0/%d]", totalFiles));
+        Downloader downloader = Downloader.downloadList(list, baseUrl, targetDir, new Downloader.DownloadCallback() {
+            @Override
+            public void apply(long fullDiff) {
+                current.addAndGet(fullDiff);
+                bar.stepBy(fullDiff);
+            }
+
+            @Override
+            public void onComplete(Path path) {
+                bar.setExtraMessage(String.format(" [%d/%d]", currentFiles.incrementAndGet(), totalFiles));
+            }
+        }, null, 4);
+        downloader.getFuture().handle((v, e) -> {
+            CompletableFuture<Void> future = new CompletableFuture<>();
+            bar.close();
+            if (e != null) {
+                future.completeExceptionally(e);
+            } else {
+                future.complete(null);
+            }
+            return future;
+        });
+        return downloader;
+    }
 }