@nightowne/tas-cli 2.0.0 → 2.3.0

package/src/utils/compression.js

@@ -8,6 +8,7 @@ import path from 'path';
 
 const gzip = promisify(zlib.gzip);
 const gunzip = promisify(zlib.gunzip);
+import { PassThrough } from 'stream';
 
 // File extensions that are already compressed (skip compression for these)
 const SKIP_COMPRESSION = new Set([
@@ -71,6 +72,24 @@ export class Compressor {
         }
     }
 
+    /**
+     * Get a compression transform stream
+     * Returns: { stream: Transform, compressed: boolean }
+     */
+    getCompressStream(filename = '') {
+        if (this.shouldSkip(filename)) {
+            return {
+                stream: new PassThrough(),
+                compressed: false
+            };
+        }
+
+        return {
+            stream: zlib.createGzip({ level: 6 }),
+            compressed: true
+        };
+    }
+
     /**
      * Decompress gzip data
      */
@@ -81,4 +100,15 @@ export class Compressor {
 
         return await gunzip(data);
     }
+
+    /**
+     * Get a decompression transform stream
+     */
+    getDecompressStream(wasCompressed) {
+        if (!wasCompressed) {
+            return new PassThrough();
+        }
+
+        return zlib.createGunzip();
+    }
 }

package/src/utils/progress.js

@@ -76,10 +76,10 @@ export class ProgressBar {
      * Format bytes to human readable
      */
     formatBytes(bytes) {
-        if (bytes === 0) return '0 B';
+        if (!Number.isFinite(bytes) || bytes <= 0) return '0 B';
         const k = 1024;
-        const sizes = ['B', 'KB', 'MB', 'GB'];
-        const i = Math.floor(Math.log(bytes) / Math.log(k));
+        const sizes = ['B', 'KB', 'MB', 'GB', 'TB'];
+        const i = Math.min(Math.floor(Math.log(bytes) / Math.log(k)), sizes.length - 1);
         return (bytes / Math.pow(k, i)).toFixed(1) + ' ' + sizes[i];
     }
 

package/src/utils/throttle.js

@@ -0,0 +1,26 @@
+import { Transform } from 'stream';
+
+export class Throttle extends Transform {
+    constructor(bytesPerSecond) {
+        super();
+        this.bytesPerSecond = bytesPerSecond;
+        this.passed = 0;
+        this.startTime = Date.now();
+    }
+
+    _transform(chunk, encoding, callback) {
+        this.passed += chunk.length;
+        const elapsed = Date.now() - this.startTime;
+        const expectedTime = (this.passed / this.bytesPerSecond) * 1000;
+
+        if (expectedTime > elapsed) {
+            setTimeout(() => {
+                this.push(chunk);
+                callback();
+            }, expectedTime - elapsed);
+        } else {
+            this.push(chunk);
+            callback();
+        }
+    }
+}