1 files changed, 36 insertions, 4 deletions
diff --git a/packages/meshbay-client/src/main.js b/packages/meshbay-client/src/main.js
index 0a5723b..9ff0069 100644
--- a/packages/meshbay-client/src/main.js
+++ b/packages/meshbay-client/src/main.js
@@ -734,6 +734,19 @@ function registerBridge() {
   const completedPaths = new Map();
   let sinkId = 0;
 
+  // A clean quit still has to tidy up: the `.part` convention above means a
+  // crash leaves an obviously-unfinished file rather than a plausible one, but
+  // quitting deliberately should leave nothing at all. Synchronous on purpose —
+  // `before-quit` does not wait for promises, and an async cleanup here would
+  // race the process exiting and finish nothing.
+  app.on('before-quit', () => {
+    for (const [id, sink] of sinks) {
+      try { sink.stream.destroy(); } catch { /* already closed */ }
+      try { fs.unlinkSync(sink.partial); } catch { /* already gone */ }
+      sinks.delete(id);
+    }
+  });
+
   /** `name`, or the first "name (n).ext" that is not taken — never an overwrite. */
   function freeName(dir, filename) {
     if (!fs.existsSync(path.join(dir, filename))) return filename;
@@ -826,8 +839,17 @@ function registerBridge() {
       target = result.filePath;
     }
 
+    // Written to `<target>.part` and renamed on completion, never straight to
+    // the final name. `save:abort` already deleted a cancelled download, but
+    // nothing covered the app being quit, killed or crashing mid-transfer: the
+    // stream was simply abandoned and a truncated file kept the final name,
+    // which is the exact thing save:abort's own comment says is worse than no
+    // file at all — it looks complete to whoever opens it next. A leftover
+    // `.part` is self-evidently unfinished, and it is the same convention the
+    // node already uses for uploads (`_do_file_upload`).
     const id = String(++sinkId);
-    sinks.set(id, { stream: fs.createWriteStream(target), path: target });
+    const partial = target + '.part';
+    sinks.set(id, { stream: fs.createWriteStream(partial), path: target, partial });
     return { id, name: path.basename(target), path: target };
   });
 
@@ -847,8 +869,16 @@ function registerBridge() {
     const sink = sinks.get(String(id));
     if (!sink) return false;
     sinks.delete(String(id));
-    completedPaths.set(String(id), sink.path);
     await new Promise((resolve) => sink.stream.end(resolve));
+    // The rename is what publishes the download. Only after the stream has
+    // flushed, or the file bearing the final name would still be short.
+    try {
+      fs.renameSync(sink.partial, sink.path);
+    } catch (err) {
+      console.error('[MeshBay] could not finalise download:', err.message);
+      return false;
+    }
+    completedPaths.set(String(id), sink.path);
     return true;
   });
 
@@ -865,8 +895,10 @@ function registerBridge() {
     sinks.delete(String(id));
     await new Promise((resolve) => sink.stream.close(resolve));
     // A cancelled download leaves a truncated file, which is worse than none:
-    // it looks like a complete one to whoever opens it next.
-    try { fs.unlinkSync(sink.path); } catch { /* already gone */ }
+    // it looks like a complete one to whoever opens it next. Only the `.part`
+    // exists at this stage — the final name is only taken by the rename in
+    // save:end — so this removes that.
+    try { fs.unlinkSync(sink.partial); } catch { /* already gone */ }
     return true;
   });