Removed capabilities from gunjs when a torrent is removed

Signed-off-by: Robert Gogete <[email protected]>

Author: gogeterobert

src/host.ts

@@ -572,7 +572,7 @@ export class FileHost implements IFileHost {
    * Remove a shared file
    * This removes the file from both HTTP and WebTorrent sharing
    */
-  public unshareFile(filename: string, deleteFile: boolean = false): boolean {
+  public async unshareFile(filename: string, deleteFile: boolean = false): Promise<boolean> {
     this.logger.debug(`📤 Unsharing file: ${filename}`);
 
     const wasShared = this.sharedFiles.has(filename);
@@ -592,6 +592,25 @@ export class FileHost implements IFileHost {
       }
     }
 
+    // Update Gun.js registry to remove the unshared file's magnet URI
+    if (this.gunRegistry && this.capabilities && wasShared) {
+      try {
+        this.logger.debug(`🔄 Updating Gun.js registry after unsharing ${filename}...`);
+
+        // Update the current capabilities with the updated magnet URI list
+        if (this.capabilities.webTorrent) {
+          this.capabilities.webTorrent.magnetUris = Array.from(
+            this.magnetUris.values()
+          );
+        }
+
+        await this.gunRegistry.register(this.capabilities);
+        this.logger.debug(`✅ Updated Gun.js registry after unsharing ${filename}`);
+      } catch (error) {
+        this.logger.warn(`⚠️ Failed to update Gun.js registry after unsharing:`, error);
+      }
+    }
+
     // Optionally delete the original file
     if (deleteFile && filePath && fs.existsSync(filePath)) {
       try {

src/interfaces.ts

@@ -53,9 +53,9 @@ export interface IFileHost {
    * Remove a shared file by its filename
    * @param filename filename of the file to unshare
    * @param deleteFile Whether to delete the original file from disk (default: false)
-   * @returns True if file was found and removed from tracking, false otherwise
+   * @returns Promise that resolves to true if file was found and removed from tracking, false otherwise
    */
-  unshareFile(filename: string, deleteFile?: boolean): boolean;
+  unshareFile(filename: string, deleteFile?: boolean): Promise<boolean>;
 
   /**
    * Get a list of currently shared filenames

src/webtorrent-manager.ts

@@ -99,10 +99,7 @@ class WebTorrentManager extends EventEmitter {
     this.logger.debug("🚀 Initializing shared WebTorrent client...");
 
     try {
-      this.webTorrentClient = new WebTorrent({
-        dht: { port: 30987 },
-        tracker: { port: 30987 }
-      });
+      this.webTorrentClient = new WebTorrent();
 
       // Set unlimited max listeners for the WebTorrent client (if method exists)
       if (typeof this.webTorrentClient.setMaxListeners === 'function') {

tests/host.test.ts

@@ -433,21 +433,21 @@ describe('FileHost', () => {
       it('should unshare a file', async () => {
         const hash = await fileHost.shareFile('/test/file.txt');
 
-        const result = fileHost.unshareFile(hash);
+        const result = await fileHost.unshareFile(hash);
 
         expect(result).toBe(true);
         expect(fileHost.getSharedFiles()).not.toContain(hash);
       });
 
-      it('should return false for non-existent file', () => {
-        const result = fileHost.unshareFile('non-existent-file.txt');
+      it('should return false for non-existent file', async () => {
+        const result = await fileHost.unshareFile('non-existent-file.txt');
         expect(result).toBe(false);
       });
 
       it('should delete file when deleteFile is true', async () => {
         const filename = await fileHost.shareFile('/test/file.txt');
 
-        fileHost.unshareFile(filename, true);
+        await fileHost.unshareFile(filename, true);
 
         expect(mockFs.unlinkSync).toHaveBeenCalledWith('/test/file.txt');
       });
@@ -458,7 +458,7 @@ describe('FileHost', () => {
 
         const filename = await host.shareFile('/test/file.txt');
 
-        host.unshareFile(filename);
+        await host.unshareFile(filename);
 
         expect(mockWebTorrentManager.removeTorrent).toHaveBeenCalledWith('magnet:?xt=urn:btih:test-hash&dn=test-file');
       }, 10000);