@trap_stevo/filetide 0.0.39 → 0.0.41

package/dist/cjs/FileMessager.js

@@ -1,5 +1,8 @@
 "use strict";
 
+const {
+  OmniMap
+} = require("@trap_stevo/legendarybuilderpronodejs-utilities/HUDStructs");
 const IoTide = require("@trap_stevo/iotide");
 const {
   FileMessagerConfigManager
@@ -24,23 +27,27 @@ 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();
+    this.onlineClients = new OmniMap();
     return;
   }
   onConnect(socket) {
-    console.log(`Client connected: ${socket.id}`);
+    console.log(`Client connected ~ ${socket.id}`);
     this.onlineClients.set(socket.id, socket);
     return;
   }
   onDisconnect(connectionRes, socket) {
-    console.log(`Client disconnected: ${socket.id}`);
+    console.log(`Client disconnected ~ ${socket.id}`);
     this.onlineClients.delete(socket.id);
     return;
   }
   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("get-client", this.handleGetClient.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));
@@ -48,9 +55,11 @@ class FileMessager {
     this.fileNet.on("transfer-complete", this.handleFileTransferComplete.bind(this));
     this.fileNet.on("request-transfer-from-client", this.handleRequestTransferFromClient.bind(this));
     this.fileNet.on("client-to-client-transfer", this.handleClientToClientTransfer.bind(this));
+    this.fileNet.on("check-client-online", this.handleCheckClientOnline.bind(this));
     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) {
@@ -102,6 +111,7 @@ class FileMessager {
       });
     }
     this.fileNet.emitToTide(client.tideID, "transfer-progress", {
+      senderName: senderData.userID,
       senderID: senderData.id,
       fileName,
       fileChunk,
@@ -115,7 +125,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
       });
     }
   }
@@ -131,20 +147,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 {
@@ -152,81 +174,188 @@ 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,
-      chunkSize,
-      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, chunkSize, 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;
+    }
+  }
+  handleGetClient(data) {
+    console.log(`[FileTide ~ Net] Getting client ~ ${data.clientID}...`);
+    try {
+      const senderConnectionID = FileNetClientManager.getOnlineClient(data.senderID).id;
+      const clientConnectionID = FileNetClientManager.getOnlineClient(data.clientID).id;
+      const senderClient = this.getClientFromID(clientConnectionID);
+      const client = this.getClientFromID(clientConnectionID);
+      if (!client || !senderClient) {
+        return;
+      }
+      console.log(`[FileTide ~ Net] Got client ~ ${data.clientID}!`);
+      this.fileNet.emitToTide(senderClient.tideID, "retrieved-current-client", client);
+    } catch (error) {
+      return;
+    }
   }
   handleNotifyTransferBarrier(data, emitToChannel) {
     const accepted = data.accepted;
@@ -234,13 +363,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}`);
@@ -248,22 +404,51 @@ class FileMessager {
   handleFileTransferComplete(data) {
     console.log(`File transfer completed: ${data.fileName}`);
   }
+  handleCheckClientOnline(data) {
+    console.log(`[FileTide ~ Net Activity] ~ Checking if client ~ ${data.clientID} online...`);
+    try {
+      const client = FileNetClientManager.getOnlineClient(data.clientID);
+      if (!client) {
+        this.fileNet.emit("retrieved-client-online", {
+          online: false,
+          client: null
+        });
+        return;
+      }
+      console.log(`[FileTide ~ Net Activity] ~ Client ~ ${data.clientID} online!`);
+      this.fileNet.emit("retrieved-client-online", {
+        online: true,
+        client
+      });
+    } catch (error) {
+      this.fileNet.emit("retrieved-client-online", {
+        online: false,
+        client: null
+      });
+      return;
+    }
+  }
   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()));
     return;
   }
   handleClientOnline(clientID, tideID, pClientID, connectionID, activeTransferBarrier) {
-    if (!this.onlineClients.has(connectionID)) {
+    if (!this.onlineClients.contains(connectionID)) {
+      return;
+    }
+    if (FileNetClientManager.getOnlineClient(clientID) && this.onlineClients.get(connectionID)) {
+      clientID = `${clientID}_${connectionID.slice(0, 4)}`;
+      this.fileNet.emitToTide(tideID, "current-client-id-already-online", clientID);
       return;
     }
     FileNetClientManager.addOnlineClient(clientID, pClientID, tideID, connectionID);
@@ -275,11 +460,49 @@ class FileMessager {
     this.fileNet.emit("current-clients", Object.fromEntries(FileNetClientManager.getOnlineClients()));
     return;
   }
-  handleClientOffline(clientID) {
-    FileNetClientManager.clearOnlineClient(clientID);
-    this.fileNet.emit("current-clients", Object.fromEntries(FileNetClientManager.getOnlineClients()));
+  handleClientOffline(clientData) {
+    console.log(`Client ~ ${clientData.clientID} | connection origin ~ ${clientData.connectionID} going offline...`);
+    try {
+      const clientDetails = FileNetClientManager.getOnlineClient(clientData.clientID);
+      if (!clientDetails || clientDetails.id !== clientData.connectionID) {
+        console.log(!clientDetails ? `[FileTide ~ Net Activity] ~ Client ~ ${clientData.clientID} already offline.` : `[FileTide ~ Net Activity] ~ Unauthorized offline signal from connection ID ~ ${clientData.connectionID}.`);
+        return;
+      }
+      FileNetClientManager.clearOnlineClient(clientData.clientID);
+      this.fileNet.emit("current-clients", Object.fromEntries(FileNetClientManager.getOnlineClients()));
+      this.fileNet.emit("save-current-transfer-states", {
+        clientID: clientData.clientID
+      });
+      const tidingTransfers = this.currentTidingTransfers.get(clientData.clientID);
+      if (tidingTransfers && tidingTransfers.size > 0) {
+        for (let recipientID of tidingTransfers.keys()) {
+          this.fileNet.emit("save-current-transfer-states", {
+            clientID: recipientID
+          });
+        }
+      }
+      this.currentTidingTransfers.delete(clientData.clientID);
+    } catch (error) {
+      console.log(`[FileTide ~ Net Activity] ~ Did not process client ~ ${clientData} offline signal ~ ${error}`);
+    }
     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;