show downloaded bytes + speed on hyperdrive mirror

scripts/bootstrap.js

@@ -14,6 +14,8 @@ const byteSize = require('tiny-byte-size')
 const { decode } = require('hypercore-id-encoding')
 const safetyCatch = require('safety-catch')
 const Rache = require('rache')
+const speedometer = require('speedometer')
+const isTTY = isBare ? false : process.stdout.isTTY // TODO: support Bare
 
 const argv = global.Pear?.config.args || global.Bare?.argv || global.process.argv
 
@@ -78,12 +80,8 @@ function advise () {
 }
 
 async function download (key, all = false) {
-  for await (const output of downloader(key, all)) console.log(output)
-}
-
-async function * downloader (key, all) {
-  if (all) yield '🍐 Fetching all runtimes from: \n   ' + key
-  else yield '🍐 [ localdev ] - no local runtime: fetching runtime'
+  if (all) console.log('🍐 Fetching all runtimes from: \n   ' + key)
+  else console.log('🍐 [ localdev ] - no local runtime: fetching runtime')
 
   const store = CORESTORE || path.join(PEAR, 'corestores', 'platform')
 
@@ -99,6 +97,9 @@ async function * downloader (key, all) {
   swarm.on('connection', (socket) => { runtimes.corestore.replicate(socket) })
 
   await runtimes.ready()
+  if (isTTY) {
+    monitorDrive(runtimes)
+  }
 
   swarm.join(runtimes.discoveryKey, { server: false, client: true })
   const done = runtimes.corestore.findingPeers()
@@ -109,30 +110,52 @@ async function * downloader (key, all) {
   runtimes = runtimes.checkout(runtimes.version)
   goodbye(() => runtimes.close())
 
-  yield `\n  Extracting platform runtime${all ? 's' : ''} to disk\n`
+  console.log(`\n  Extracting platform runtime${all ? 's' : ''} to disk\n`)
 
   const runtime = runtimes.mirror(new Localdrive(SWAP), {
     prefix: '/by-arch' + (all ? '' : '/' + ADDON_HOST)
   })
 
   for await (const { op, key, bytesAdded } of runtime) {
     if (op === 'add') {
-      yield '\x1B[32m+\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']'
+      console.log('\x1B[32m+\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']')
     } else if (op === 'change') {
-      yield '\x1B[33m~\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']'
+      console.log('\x1B[33m~\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']')
     } else if (op === 'remove') {
-      yield '\x1B[31m-\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']'
+      console.log('\x1B[31m-\x1B[39m ' + key + ' [' + byteSize(bytesAdded) + ']')
     }
   }
 
-  yield '\x1B[2K\x1B[200D  Runtime extraction complete\x1b[K\n'
+  console.log('\x1B[2K\x1B[200D  Runtime extraction complete\x1b[K\n')
 
   await runtimes.close()
   await swarm.destroy()
   await corestore.close()
 
   const tick = isWindows ? '^' : '✔'
 
-  if (all) yield '\x1B[32m' + tick + '\x1B[39m Download complete\n'
-  else yield '\x1B[32m' + tick + '\x1B[39m Download complete, initalizing...\n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n\n'
+  if (all) console.log('\x1B[32m' + tick + '\x1B[39m Download complete\n')
+  else console.log('\x1B[32m' + tick + '\x1B[39m Download complete, initalizing...\n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n\n')
+}
+
+/**
+ * @param {Hyperdrive} drive
+ */
+async function monitorDrive (drive) {
+  const downloadSpeedometer = speedometer()
+  let downloadedBytes = 0
+  let speed = 0
+  const blobs = await drive.getBlobs()
+  blobs.core.on('download', (_index, bytes) => {
+    downloadedBytes += bytes
+    speed = downloadSpeedometer(bytes)
+  })
+  const interval = setInterval(() => {
+    process.stdout.clearLine();
+    process.stdout.cursorTo(0);
+    process.stdout.write(`Downloaded: ${byteSize(downloadedBytes)} - Speed: ${byteSize(speed)}/s`);
+  }, 500)
+  return () => {
+    clearInterval(interval)
+  }
 }