From 4efedcbdaff4ffa0886dbe7b7c7274e64322b2d8 Mon Sep 17 00:00:00 2001
From: ThaUnknown <6506529+ThaUnknown@users.noreply.github.com>
Date: Sat, 12 Mar 2022 22:52:23 +0100
Subject: [PATCH] torrent stats

---
 src/renderer/src/lib/pages/Player.svelte | 28 ++++++++++++++++++++++--
 src/renderer/src/modules/util.js         |  9 +++++++-
 2 files changed, 34 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/lib/pages/Player.svelte b/src/renderer/src/lib/pages/Player.svelte
index 79cefe8..69ba7a7 100644
--- a/src/renderer/src/lib/pages/Player.svelte
+++ b/src/renderer/src/lib/pages/Player.svelte
@@ -30,10 +30,11 @@
 
 <script>
   import { alEntry } from '@/modules/anilist.js'
+  import { client } from '@/modules/torrent.js'
   import { resolveFileMedia } from '@/modules/anime.js'
   import Peer from '@/modules/Peer.js'
   import Subtitles from '@/modules/subtitles.js'
-  import { toTS, videoRx } from '@/modules/util.js'
+  import { toTS, videoRx, fastPrettyBytes } from '@/modules/util.js'
   import Keyboard from './Keyboard.svelte'
 
   async function mediaChange(current, image) {
@@ -629,6 +630,13 @@
       }
     }
   }
+  const torrent = {}
+  function updateStats() {
+    torrent.peers = (client?.torrents.length && client?.torrents[0].numPeers) || 0
+    torrent.up = (client?.torrents.length && client?.torrents[0].uploadSpeed) || 0
+    torrent.down = (client?.torrents.length && client?.torrents[0].downloadSpeed) || 0
+  }
+  setInterval(updateStats, 200)
 </script>
 
 <svelte:window on:keydown={handleKeydown} bind:innerWidth bind:innerHeight />
@@ -691,7 +699,16 @@
         Buffer health: {stats.buffer || 0}
       </div>
     {/if}
-    <div class="top z-40 d-flex flex-row-reverse">
+    <div class="top z-40 d-flex justify-content-between">
+      <div />
+      <div class="d-flex">
+        <span class="material-icons" data-name="peers"> people </span>
+        <span class="stats">{torrent.peers}</span>
+        <span class="material-icons"> arrow_downward </span>
+        <span class="stats">{fastPrettyBytes(torrent.up)}/s</span>
+        <span class="material-icons"> arrow_upward </span>
+        <span class="stats">{fastPrettyBytes(torrent.down)}/s</span>
+      </div>
       <span class="material-icons ctrl font-size-12 p-10" title="Keybinds [`]" data-name="togglePopout" on:click={() => (showKeybinds = true)}> help_outline </span>
     </div>
     <div class="middle d-flex align-items-center justify-content-center flex-grow-1 z-40 position-relative">
@@ -799,6 +816,13 @@
 {/if}
 
 <style>
+  .stats {
+    font-size: 1.8rem !important;
+    white-space: nowrap;
+    align-self: center;
+    font-weight: 600;
+    font-family: Roboto, Arial, Helvetica, sans-serif;
+  }
   .miniplayer {
     transition: width 0.2s ease;
     width: 25vw !important;
diff --git a/src/renderer/src/modules/util.js b/src/renderer/src/modules/util.js
index ab56371..5dfddc0 100644
--- a/src/renderer/src/modules/util.js
+++ b/src/renderer/src/modules/util.js
@@ -10,7 +10,14 @@ export function countdown (s) {
   if (d || h) tmp.push(h + 'h')
   if (d || h || m) tmp.push(m + 'm')
   return tmp.join(' ')
-  
+}
+
+const units = [' B', ' kB', ' MB', ' GB', ' TB']
+export function fastPrettyBytes (num) {
+  if (isNaN(num)) return '0 B'
+  if (num < 1) return num + ' B'
+  const exponent = Math.min(Math.floor(Math.log(num) / Math.log(1000)), units.length - 1)
+  return Number((num / Math.pow(1000, exponent)).toFixed(2)) + units[exponent]
 }
 
 export const DOMPARSER = DOMParser.prototype.parseFromString.bind(new DOMParser())