From 4a36b9ee920cf71ce1656cfbb471b11931c7a82e Mon Sep 17 00:00:00 2001
From: Klemek <klemek@outlook.fr>
Date: Mon, 17 Mar 2025 16:27:28 +0100
Subject: [PATCH] download speed

---
 README.md  |  2 +-
 index.html |  3 ++-
 main.js    | 59 +++++++++++++++++++++++++++++++++++++++++++++---------
 3 files changed, 52 insertions(+), 12 deletions(-)

diff --git a/README.md b/README.md
index cabd16d..fb0ba23 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@
 * [x] Link sharing and better workflow
 * [x] Multiple peers
 * [x] Get peer data
-* [ ] See download speed
+* [x] See download speed
 * [ ] DaisyUI redesign
 * [ ] QRCode
 * [ ] Multiple files
diff --git a/index.html b/index.html
index 1386b3f..79d1d93 100644
--- a/index.html
+++ b/index.html
@@ -58,7 +58,8 @@
           <br>
         </template>
         <template v-if="downloading">
-          <progress :value="downloadProgress" :max="downloadTotal"></progress>
+          <progress :value="downloadProgress" :max="downloadTotal"></progress><br>
+          <label>{{ prettyDownloadSpeed }}</label><label v-if="!client.downloadEnd"> - {{ prettyRemainingTime }}</label>
           <br>
           <br>
         </template>
diff --git a/main.js b/main.js
index f497049..47ffbb4 100644
--- a/main.js
+++ b/main.js
@@ -23,7 +23,35 @@ const utils = {
       }
     }
     return 'Unknown'
-  }
+  },
+  prettyBytes(byteCount) {
+    let size = byteCount / 1000;
+    if (size < 1000) {
+      return `${size.toFixed(2)} KB`;
+    }
+    size /= 1000;
+    if (size < 1000) {
+      return `${size.toFixed(2)} MB`;
+    }
+    size /= 1000;
+    return `${size.toFixed(2)} GB`;
+  },
+
+  prettyTime(deltaMillisecond) {
+    let time = deltaMillisecond / 1000;
+    if (time <= 1) {
+      return `<1s`;
+    }
+    if (time < 60) {
+      return `${Math.floor(time).toFixed(0).padStart(2, '0')}s`;
+    }
+    time /= 60;
+    if (time < 60) {
+      return `${Math.floor(time).toFixed(0).padStart(2, '0')}m${((time * 60) % 60).toFixed(0).padStart(2, '0')}s`;
+    }
+    time /= 60;
+    return `${Math.floor(time).toFixed(0).padStart(2, '0')}h${((time * 60) % 60).toFixed(0).padStart(2, '0')}m`;
+  },
 };
 
 const MESSAGE_TYPE = {
@@ -57,6 +85,7 @@ const app = createApp({
         connection: null,
         connected: false,
         downloadStart: null,
+        downloadEnd: null,
         received: [],
         buffer: null, // TODO multiple files
       },
@@ -113,22 +142,31 @@ const app = createApp({
       if (! this.downloading) {
         return 'Waiting for file info...';
       }
-      if (this.downloadProgress > this.downloadTotal) {
+      if (this.client.downloadEnd) {
         return 'File downloaded';
       }
       return 'Downloading...';
     },
     prettyFileSize() {
-      let size = this.fileSize / 1000;
-      if (size < 1000) {
-        return `${size.toFixed(2)} KB`;
+      return utils.prettyBytes(this.fileSize);
+    },
+    prettyDownloadSpeed() {
+      if (! this.client.downloadStart) {
+        return '';
       }
-      size /= 1000;
-      if (size < 1000) {
-        return `${size.toFixed(2)} MB`;
+      const time = (this.client.downloadEnd ?? (new Date())) - this.client.downloadStart;
+      const speed = 1000 * this.downloadProgress / time;
+      return `${utils.prettyBytes(speed)}/s`;
+    },
+    prettyRemainingTime() {
+      if (! this.client.downloadStart || this.client.downloadEnd) {
+        return '';
       }
-      size /= 1000;
-      return `${size.toFixed(2)} GB`;
+      const time = (this.client.downloadEnd ?? (new Date())) - this.client.downloadStart;
+      const speed = this.downloadProgress / time;
+      const remainingBytes = this.downloadTotal - this.downloadProgress;
+      const remainingTime = remainingBytes / speed;
+      return `${utils.prettyTime(remainingTime)}`;
     },
   },
   watch: {},
@@ -364,6 +402,7 @@ const app = createApp({
       if (indexes.length) {
         this.sendClientSeek(indexes);
       } else {
+        this.client.downloadEnd = new Date();
         this.sendClientDone();
         this.clientDownloadFile();
       }