teyyare 0.2.6 → 0.2.7

package/bin/teyyare.js

@@ -56,7 +56,7 @@ async function main() {
   }
 
   if (command === '--version' || command === '-v' || command === 'version') {
-    console.log('0.2.6');
+    console.log('0.2.7');
     process.exit(0);
   }
 
@@ -170,7 +170,7 @@ Peak RSS:         ${(snap.peakRSS / (1024 * 1024)).toFixed(2)} MB
 `);
     process.exit(0);
   } else if (command === 'status') {
-    console.log('\x1b[1m\x1b[36mTeyyare\x1b[0m v0.2.6 status: ready');
+    console.log('\x1b[1m\x1b[36mTeyyare\x1b[0m v0.2.7 status: ready');
   } else {
     console.error(`Unknown command: ${command}`);
     printUsage();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "teyyare",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "High-performance resumable file transfer engine and CLI built on muttafa and raptiye",
   "main": "src/index.js",
   "type": "module",

package/src/index.js

@@ -141,8 +141,14 @@ TCPTransport.prototype._attachSocketReader = function(socket) {
         if (idx !== -1) {
           pool.splice(idx, 1);
         }
+        if (pool.length === 0 && this.onPeerDisconnect) {
+          this.onPeerDisconnect(peer);
+        }
       }
     }
+    if (this._inboundSockets.size === 0 && this.onPeerDisconnect) {
+      this.onPeerDisconnect(1);
+    }
   });
 
   socket.on('error', () => {
@@ -458,6 +464,15 @@ export async function serve({
 
   let activeSender = null;
 
+  transport.onPeerDisconnect = async (peer) => {
+    if (activeSender) {
+      const s = activeSender;
+      activeSender = null;
+      await s.abort();
+    }
+    replication.reset();
+  };
+
   replication.onBatch(async (msg) => {
     if (!msg.buffers || msg.buffers.length < 2) return;
     const [metaBuf, payloadBuf] = msg.buffers;
@@ -481,8 +496,9 @@ export async function serve({
 
       // Abort any existing running sender before starting new transfer
       if (activeSender) {
-        activeSender.abort();
+        const s = activeSender;
         activeSender = null;
+        await s.abort();
       }
 
       const manifest = await TransferManifest.build(resolved);
@@ -534,6 +550,12 @@ export async function serve({
     receiver,
     metrics,
     async close() {
+      if (activeSender) {
+        const s = activeSender;
+        activeSender = null;
+        await s.abort();
+      }
+      await receiver.close();
       metrics.stop();
       await transport.close();
     }
@@ -600,6 +622,7 @@ export async function send({
   try {
     await sender.transfer();
   } finally {
+    await sender.abort();
     metrics.stop();
     await transport.close();
   }
@@ -687,6 +710,7 @@ export async function pull({
     await receiver.waitForCompletion();
   } finally {
     if (progress) progress.stop();
+    await receiver.close();
     metrics.stop();
     await transport.close();
   }

package/src/receiver/pipeline.js

@@ -370,6 +370,7 @@ export class ReceiverPipeline {
     // Sync disk buffers
     await file.fileHandle.sync();
     await file.fileHandle.close();
+    file.fileHandle = null;
 
     // Verify whole-file checksum if available
     const expectedSha = file.sha256;
@@ -389,13 +390,29 @@ export class ReceiverPipeline {
     this.completedFiles.add(file.destPath);
   }
 
+  async close() {
+    for (const [fileId, file] of this.activeFiles.entries()) {
+      if (file.fileHandle) {
+        try {
+          await file.fileHandle.close();
+        } catch {}
+        file.fileHandle = null;
+      }
+    }
+    this.activeFiles.clear();
+    this._pendingChunks.clear();
+  }
+
   async _computeFileSha256(filePath) {
     return new Promise((resolve, reject) => {
       const hash = crypto.createHash('sha256');
       const stream = createReadStream(filePath);
       stream.on('data', (d) => hash.update(d));
       stream.on('end', () => resolve(hash.digest('hex')));
-      stream.on('error', reject);
+      stream.on('error', (err) => {
+        stream.destroy(err);
+        reject(err);
+      });
     });
   }
 }

package/src/sender/pipeline.js

@@ -141,14 +141,27 @@ export class SenderPipeline {
         }
       }
     });
+    this._activeHandle = null;
+    this._abortWaiters = new Set();
     this.aborted = false;
   }
 
-  abort() {
+  async abort() {
     this.aborted = true;
+    for (const waiter of this._abortWaiters) {
+      waiter();
+    }
+    this._abortWaiters.clear();
+    if (this._activeHandle) {
+      const h = this._activeHandle;
+      this._activeHandle = null;
+      try {
+        await h.close();
+      } catch {}
+    }
     if (this.engine) {
       try {
-        this.engine.halt();
+        this.engine.reset();
       } catch {}
     }
   }
@@ -157,7 +170,19 @@ export class SenderPipeline {
     if (this.aborted) return;
     while (this.engine.isBackpressured()) {
       if (this.aborted) return;
-      await this.engine.ready();
+      await new Promise((resolve) => {
+        let done = false;
+        const waiter = () => {
+          if (!done) {
+            done = true;
+            this._abortWaiters.delete(waiter);
+            resolve();
+          }
+        };
+        this._abortWaiters.add(waiter);
+        this.engine.ready().then(waiter, waiter);
+      });
+      if (this.aborted) return;
       await new Promise((r) => setImmediate(r));
     }
     if (this.aborted) return;
@@ -216,6 +241,7 @@ export class SenderPipeline {
 
       if (file.size > 0) {
         const handle = await fs.open(filePath, 'r');
+        this._activeHandle = handle;
         try {
           for (let i = 0; i < totalFileChunks; i++) {
             if (this.aborted) break;
@@ -282,7 +308,10 @@ export class SenderPipeline {
             }
           }
         } finally {
-          await handle.close();
+          this._activeHandle = null;
+          try {
+            await handle.close();
+          } catch {}
         }
       }