@trap_stevo/filetide 0.0.38 → 0.0.40

package/dist/cjs/FileMessager.js

@@ -24,6 +24,7 @@ class FileMessager {
     this.transferQueue = new HUDTargetQueue();
     this.currentTransferBarrierTransfers = new Map();
     this.clientTransferBarriers = new Map();
+    this.currentTidingTransfers = new Map();
     this.fileNet = new IoTide(serverOptions.port, serverOptions, true, this.onConnect.bind(this), this.onDisconnect.bind(this));
     this.onlineClients = new Map();
     return;
@@ -40,7 +41,9 @@ class FileMessager {
   }
   start() {
     this.fileNet.on("get-client-transfer-barrier-response", this.handleClientTransferBarrierResponse.bind(this));
+    this.fileNet.on("get-client-transfer-state", this.handleGetClientTransferState.bind(this));
     this.fileNet.on("notify-transfer-barrier", this.handleNotifyTransferBarrier.bind(this));
+    this.fileNet.on("send-current-transfer-state", this.handleSendCurrentTransferState.bind(this));
     this.fileNet.on("send-file-list-to-client", this.handleSendFileListToClient.bind(this));
     this.fileNet.on("list-client-files", this.handleListClientFiles.bind(this));
     this.fileNet.on("transfer-start", this.handleFileTransferStart.bind(this));
@@ -51,6 +54,7 @@ class FileMessager {
     this.fileNet.on("client-offline", this.handleClientOffline.bind(this));
     this.fileNet.on("clients-online", this.handleClientsOnline.bind(this));
     this.fileNet.on("client-online", this.handleClientOnline.bind(this));
+    this.fileNet.on("client-tiding", this.handleClientTiding.bind(this));
     return;
   }
   getClientFromID(clientID) {
@@ -84,7 +88,7 @@ class FileMessager {
    * @param {String} fileName - The name of the file
    * @param {Number} chunkIndex - The index of the current chunk
    */
-  sendFileChunkToClient(senderData, clientData, fileChunk, fileName, chunkIndex, totalChunks, filePath = process.cwd(), fileSize = 100) {
+  sendFileChunkToClient(senderData, clientData, fileChunk, fileName, chunkIndex, chunkSize, totalChunks, filePath = process.cwd(), fileSize = 100) {
     const senderClient = this.getClientFromID(senderData.id);
     const client = this.getClientFromID(clientData.id);
     if (!client) {
@@ -102,10 +106,12 @@ class FileMessager {
       });
     }
     this.fileNet.emitToTide(client.tideID, "transfer-progress", {
+      senderName: senderData.userID,
       senderID: senderData.id,
       fileName,
       fileChunk,
       chunkIndex,
+      chunkSize,
       path: filePath,
       fileSize
     });
@@ -114,7 +120,13 @@ class FileMessager {
       this.fileNet.emitToTide(senderClient.tideID, "transfer-status", {
         status: `Chunk ${chunkIndex} of ${fileName} sent to client ${clientData.userID}!`,
         recipientID: clientData.userID,
-        success: true
+        success: true,
+        fileName,
+        chunkIndex,
+        totalChunks,
+        chunkSize,
+        path: filePath,
+        fileSize
       });
     }
   }
@@ -130,20 +142,26 @@ class FileMessager {
   handleSendFileListToClient(listData) {
     const {
       listedFiles,
+      totalItems,
       recipientID,
       senderID
     } = listData;
-    const recipientConnectionID = FileNetClientManager.getOnlineClient(recipientID).id;
-    const senderConnectionID = FileNetClientManager.getOnlineClient(senderID).id;
-    const senderClient = this.getClientFromID(senderConnectionID);
-    const client = this.getClientFromID(recipientConnectionID);
-    if (!client || !senderClient) {
+    try {
+      const recipientConnectionID = FileNetClientManager.getOnlineClient(recipientID).id;
+      const senderConnectionID = FileNetClientManager.getOnlineClient(senderID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      const client = this.getClientFromID(recipientConnectionID);
+      if (!client || !senderClient) {
+        return;
+      }
+      this.fileNet.emitToTide(client.tideID, "client-file-list", {
+        senderID,
+        totalItems,
+        listedFiles
+      });
+    } catch (error) {
       return;
     }
-    this.fileNet.emitToTide(client.tideID, "client-file-list", {
-      senderID,
-      listedFiles
-    });
   }
   handleListClientFiles(clientData) {
     const {
@@ -151,80 +169,172 @@ class FileMessager {
       recipientID,
       depth = 1
     } = clientData;
-    const recipientConnectionID = FileNetClientManager.getOnlineClient(recipientID).id;
-    const senderConnectionID = FileNetClientManager.getOnlineClient(requesterID).id;
-    const senderClient = this.getClientFromID(senderConnectionID);
-    const client = this.getClientFromID(recipientConnectionID);
-    if (!client || !senderClient) {
+    try {
+      const recipientConnectionID = FileNetClientManager.getOnlineClient(recipientID).id;
+      const senderConnectionID = FileNetClientManager.getOnlineClient(requesterID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      const client = this.getClientFromID(recipientConnectionID);
+      if (!client || !senderClient) {
+        return;
+      }
+      this.fileNet.emitToTide(client.tideID, "list-files", {
+        senderID: requesterID,
+        recipientID,
+        depth
+      });
+    } catch (error) {
       return;
     }
-    this.fileNet.emitToTide(client.tideID, "list-files", {
-      senderID: requesterID,
-      recipientID,
-      depth
-    });
   }
   handleClientToClientTransfer(transferData) {
-    const {
-      senderID,
-      recipientId,
-      fileName,
-      fileChunk,
-      fileSize,
-      filePath,
-      chunkIndex,
-      totalChunks
-    } = transferData;
-    console.log(`Transferring file ~ ${fileName} | chunk ${chunkIndex + 1} of ${totalChunks} to client (${recipientId})`);
-    this.sendFileChunkToClient({
-      userID: senderID,
-      id: FileNetClientManager.getOnlineClient(senderID).id
-    }, {
-      userID: recipientId,
-      id: FileNetClientManager.getOnlineClient(recipientId).id
-    }, fileChunk, fileName, chunkIndex, totalChunks, filePath, fileSize);
-    if (chunkIndex + 1 === totalChunks) {
-      this.fileNet.emit("transfer-complete", {
+    try {
+      const {
+        chunkBatch
+      } = transferData;
+      if (chunkBatch && chunkBatch.length > 0) {
+        for (const chunkData of chunkBatch) {
+          const {
+            senderID,
+            recipientId,
+            fileName,
+            fileChunk,
+            fileSize,
+            filePath,
+            chunkIndex,
+            chunkSize,
+            completedChunks,
+            totalChunks
+          } = chunkData;
+          console.log(`[FileTide ~ File Messager] ~ Transferring file ~ ${fileName} | chunk ${chunkIndex + 1} >> ${completedChunks} of ${totalChunks} to client (${recipientId})`);
+          this.sendFileChunkToClient({
+            userID: senderID,
+            id: FileNetClientManager.getOnlineClient(senderID).id
+          }, {
+            userID: recipientId,
+            id: FileNetClientManager.getOnlineClient(recipientId).id
+          }, fileChunk, fileName, chunkIndex, chunkSize, totalChunks, filePath, fileSize);
+          if (completedChunks === totalChunks) {
+            this.fileNet.emit("transfer-complete", {
+              senderID,
+              recipientId,
+              fileName,
+              filePath
+            });
+            console.log(`[FileTide ~ File Messager] ~ File transfer completed: ${fileName} from ${senderID} to ${recipientId}`);
+            const tidingTransfers = this.currentTidingTransfers.get(senderID);
+            if (tidingTransfers) {
+              tidingTransfers.delete(recipientId);
+            }
+          }
+        }
+        return;
+      }
+      const {
         senderID,
         recipientId,
         fileName,
-        filePath
-      });
-      console.log(`[FileTide ~ File Messager] ~ File transfer completed: ${fileName} from ${senderID} to ${recipientId}`);
+        fileChunk,
+        fileSize,
+        filePath,
+        chunkIndex,
+        chunkSize,
+        completedChunks,
+        totalChunks
+      } = transferData;
+      console.log(`[FileTide ~ File Messager] ~ Transferring file ~ ${fileName} | chunk ${chunkIndex + 1} >> ${completedChunks} of ${totalChunks} to client (${recipientId})`);
+      this.sendFileChunkToClient({
+        userID: senderID,
+        id: FileNetClientManager.getOnlineClient(senderID).id
+      }, {
+        userID: recipientId,
+        id: FileNetClientManager.getOnlineClient(recipientId).id
+      }, fileChunk, fileName, chunkIndex, chunkSize, totalChunks, filePath, fileSize);
+      if (chunkIndex + 1 === totalChunks) {
+        this.fileNet.emit("transfer-complete", {
+          senderID,
+          recipientId,
+          fileName,
+          filePath
+        });
+        console.log(`[FileTide ~ File Messager] ~ File transfer completed: ${fileName} from ${senderID} to ${recipientId}`);
+        const tidingTransfers = this.currentTidingTransfers.get(senderID);
+        if (tidingTransfers) {
+          tidingTransfers.delete(recipientId);
+        }
+      }
+    } catch (error) {
+      console.log(error);
+      return;
     }
   }
   handleClientTransferBarrierResponse(data) {
-    console.log(`Checking ${data.recipientID}'s transfer barrier for transfer ~ ${data.transferID}...`);
-    const recipientConnectionID = FileNetClientManager.getOnlineClient(data.recipientID).id;
-    const senderConnectionID = FileNetClientManager.getOnlineClient(data.clientID).id;
-    const senderClient = this.getClientFromID(senderConnectionID);
-    const client = this.getClientFromID(recipientConnectionID);
-    if (!client || !senderClient) {
-      return;
-    }
-    this.fileNet.emitToTide(client.tideID, "queue-incoming-transfer", {
-      ...data,
-      transferType: "send"
-    });
-    this.transferQueue.waitForData(data.transferID, () => this.currentTransferBarrierTransfers.get(data.transferID), [true, false], 100, 50000).then(accepted => {
-      console.log(`Transfer ${accepted ? "accepted!" : "denied."}`);
-      this.fileNet.emitToTide(senderClient.tideID, "transfer-barrier-response", accepted);
-      this.currentTransferBarrierTransfers.delete(data.transferID);
-      if (accepted) {
-        this.fileNet.emitToTide(client.tideID, "accepted-transfer", {
+    console.log(`[FileTide ~ Net] Checking ${data.recipientID}'s transfer barrier for transfer ~ ${data.transferID}...`);
+    try {
+      const recipientConnectionID = FileNetClientManager.getOnlineClient(data.recipientID).id;
+      const senderConnectionID = FileNetClientManager.getOnlineClient(data.clientID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      const client = this.getClientFromID(recipientConnectionID);
+      if (!client || !senderClient) {
+        return;
+      }
+      this.fileNet.emitToTide(client.tideID, "queue-incoming-transfer", {
+        ...data,
+        transferType: "send"
+      });
+      this.transferQueue.waitForData(data.transferID, () => this.currentTransferBarrierTransfers.get(data.transferID), [true, false], 100, 50000).then(accepted => {
+        console.log(`[FileTide ~ Net Barrier] ~ Transfer ${accepted ? "accepted!" : "denied."}`);
+        this.fileNet.emitToTide(senderClient.tideID, "transfer-barrier-response", accepted);
+        this.currentTransferBarrierTransfers.delete(data.transferID);
+        if (accepted) {
+          this.fileNet.emitToTide(client.tideID, "accepted-transfer", {
+            accepted,
+            ...data
+          });
+          return;
+        }
+        this.fileNet.emitToTide(client.tideID, "denied-transfer", {
           accepted,
           ...data
         });
+      }).catch(error => {
+        console.error(error.message);
+      });
+      return;
+    } catch (error) {
+      return;
+    }
+  }
+  handleSendCurrentTransferState(data) {
+    try {
+      const senderConnectionID = FileNetClientManager.getOnlineClient(data.senderID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      if (!senderClient) {
         return;
       }
-      this.fileNet.emitToTide(client.tideID, "denied-transfer", {
-        accepted,
-        ...data
+      this.fileNet.emitToTide(senderClient.tideID, "sent-current-transfer-state", data.transferState || {
+        transferredSize: 0,
+        lastChunkSent: 0
       });
-    }).catch(error => {
-      console.error(error.message);
-    });
-    return;
+    } catch (error) {
+      return;
+    }
+  }
+  handleGetClientTransferState(data) {
+    const filePath = data.filePath || data.path;
+    console.log(`[FileTide ~ Net] Getting ${data.recipientID}'s transfer state for ~ ${filePath}...`);
+    try {
+      const recipientConnectionID = FileNetClientManager.getOnlineClient(data.recipientID).id;
+      const client = this.getClientFromID(recipientConnectionID);
+      if (!client) {
+        return;
+      }
+      this.fileNet.emitToTide(client.tideID, "current-transfer-state", {
+        senderID: data.senderID,
+        filePath
+      });
+    } catch (error) {
+      return;
+    }
   }
   handleNotifyTransferBarrier(data, emitToChannel) {
     const accepted = data.accepted;
@@ -232,13 +342,40 @@ class FileMessager {
     this.currentTransferBarrierTransfers.set(transferID, accepted);
   }
   handleFileTransferStart(data, emitToChannel) {
-    console.log(`Alerting client of incoming transfer from ${data.senderID}...`);
-    const recipientConnectionID = FileNetClientManager.getOnlineClient(data.recipientID).id;
-    const client = this.getClientFromID(recipientConnectionID);
-    if (!client) {
+    console.log(`[FileTide ~ File Messager] ~ Alerting client of incoming transfer from ${data.senderID}...`);
+    try {
+      const recipientConnectionID = FileNetClientManager.getOnlineClient(data.recipientID).id;
+      const senderConnectionID = FileNetClientManager.getOnlineClient(data.senderID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      const client = this.getClientFromID(recipientConnectionID);
+      if (!client || !senderClient) {
+        return;
+      }
+      this.fileNet.emitToTide(client.tideID, "incoming-transfer", data);
+      this.fileNet.emitToTide(senderClient.tideID, "incoming-transfer", {
+        ...data,
+        currentSender: data.senderID
+      });
+      if (!this.currentTidingTransfers.get(senderID) || this.currentTidingTransfers.get(senderID).length <= 0) {
+        const tidingTransfers = new Map();
+        tidingTransfers.set(data.recipientID, {
+          recipientID: data.recipientID,
+          startedOn: new Date(),
+          size: data.fileSize,
+          path: data.path
+        });
+        this.currentTidingTransfers.set(data.senderID, tidingTransfers);
+        return;
+      }
+      this.currentTidingTransfers.get(senderID).set(data.recipientID, {
+        recipientID: data.recipientID,
+        startedOn: new Date(),
+        size: data.fileSize,
+        path: data.path
+      });
+    } catch (error) {
       return;
     }
-    this.fileNet.emitToTide(client.tideID, "incoming-transfer", data);
   }
   handleFileTransferProgress(data, emitToChannel) {
     console.log(`Progress: ${data.progress}`);
@@ -249,12 +386,12 @@ class FileMessager {
   handleClientsOnline(clientID) {
     const connectionDetails = FileNetClientManager.getOnlineClient(clientID);
     if (!connectionDetails) {
-      console.log(`Client ~ ${clientID} not authorized.`);
+      console.log(`[FileTide ~ Net Activity] ~ Client ~ ${clientID} not authorized.`);
       return;
     }
     const client = this.onlineClients.get(connectionDetails.id);
     if (!client) {
-      console.log(`Client ~ ${clientID} not found.`);
+      console.log(`[FileTide ~ Net Activity] ~ Client ~ ${clientID} not found.`);
       return;
     }
     this.fileNet.emitToTide(connectionDetails.tideID, "current-online-clients", Object.fromEntries(FileNetClientManager.getOnlineClients()));
@@ -276,8 +413,36 @@ class FileMessager {
   handleClientOffline(clientID) {
     FileNetClientManager.clearOnlineClient(clientID);
     this.fileNet.emit("current-clients", Object.fromEntries(FileNetClientManager.getOnlineClients()));
+    this.fileNet.emit("save-current-transfer-states", {
+      clientID
+    });
+    const tidingTransfers = this.currentTidingTransfers.get(clientID);
+    if (tidingTransfers && tidingTransfers.size > 0) {
+      for (let recipientID of tidingTransfers.keys()) {
+        this.fileNet.emit("save-current-transfer-states", {
+          clientID: recipientID
+        });
+      }
+    }
+    this.currentTidingTransfers.delete(clientID);
     return;
   }
+  handleClientTiding(data) {
+    try {
+      const senderConnectionID = FileNetClientManager.getOnlineClient(data.senderID).id;
+      const senderClient = this.getClientFromID(senderConnectionID);
+      if (!senderClient) {
+        return;
+      }
+      console.log(`[FileTide ~ Net Activity] ~ Received client ~ ${data.senderID} tiding.`);
+      this.fileNet.emitToTide(senderClient.tideID, "current-client-tiding", {
+        message: data.message
+      });
+      return;
+    } catch (error) {
+      return;
+    }
+  }
 }
 ;
 module.exports = FileMessager;