ghostwire 1.0.0 → 2.0.0

package/bin/index.js

@@ -1,4 +1,4 @@
-#!/usr/bin/env node
+#!/usr/bin/env node
 import chalk from "chalk";
 import startServer from "../src/handler/server.js";
 import startClient from "../src/handler/client.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ghostwire",
-  "version": "1.0.0",
+  "version": "2.0.0",
   "description": "Terminal Chat with File Sharing",
   "main": "bin/index.js",
   "bin": {

package/src/handler/client.js

@@ -1,10 +1,10 @@
 import net from 'net';
 import chalk from 'chalk';
+import fs from 'fs';
+import path from 'path';
 import showBanner from '../lib/banner.js';
 
 export default function startClient(address) {
-    // Input might be "0.tcp.ngrok.io:12345" OR "localhost:3000"
-    // We strip the "tcp://" prefix if the user copied it directly from Ngrok
     const cleanAddress = address.replace('tcp://', '');
     const parts = cleanAddress.split(':');
 
@@ -61,13 +61,25 @@ export default function startClient(address) {
         }
 
         if (input) {
+            // --- FILE SHARING (SIDECAR) ---
+            if (input.startsWith('/share ')) {
+                const filePath = input.split('/share ')[1]?.trim();
+                handleFileUpload(filePath, HOST, PORT);
+            }
+            // --- FILE DOWNLOADING (SIDECAR) ---
+            else if (input.startsWith('/download ')) {
+                const fileName = input.split('/download ')[1]?.trim();
+                handleFileDownload(fileName, HOST, PORT);
+            }
+            // --- NORMAL CHAT ---
+            else {
+                const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
+                console.log(`${chalk.blue('┌──(')}${chalk.green.bold('YOU@ghostwire')}${chalk.blue(')-[')}${chalk.white(time)}${chalk.blue(']')}`);
+                console.log(`${chalk.blue('└─#')} ${chalk.white(input)}`);
 
-            const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
-            console.log(`${chalk.blue('┌──(')}${chalk.green.bold('YOU@ghostwire')}${chalk.blue(')-[')}${chalk.white(time)}${chalk.blue(']')}`);
-            console.log(`${chalk.blue('└─#')} ${chalk.white(input)}`);
-
-            // Send to server
-            client.write(input);
+                // Send to server
+                client.write(input);
+            }
         }
 
         // Draw new prompt
@@ -82,4 +94,80 @@ export default function startClient(address) {
     client.on('error', (err) => {
         console.log(chalk.red(`\n❌ Error: ${err.message}`));
     });
-};
+};
+
+// --- HELPER FUNCTIONS ---
+
+function handleFileUpload(filePath, host, port) {
+    if (!filePath) {
+        console.log(chalk.red('Usage: /share <path_to_file>'));
+        return;
+    }
+
+    // Check if file exists
+    // Resolve path relative to current working directory of the user
+    const absolutePath = path.resolve(process.cwd(), filePath);
+
+    if (!fs.existsSync(absolutePath)) {
+        console.log(chalk.red(`\n ❌ File not found: ${absolutePath}`));
+        return;
+    }
+
+    const filename = path.basename(absolutePath);
+    console.log(chalk.cyan(`\n 🚀 Starting upload: ${filename} (Sidecar Socket)...`));
+
+    const sidecar = new net.Socket();
+    sidecar.connect(port, host, () => {
+        // Handshake
+        sidecar.write(`SIDECAR_UPLOAD|${filename}\n`);
+
+        const fileStream = fs.createReadStream(absolutePath);
+        fileStream.pipe(sidecar);
+    });
+
+    sidecar.on('close', () => {
+        // console.log(chalk.green(`\n ✅ Upload finished: ${filename}`));
+        // We rely on server system message or just silent finish to not clutter too much
+    });
+
+    sidecar.on('error', (err) => {
+        console.log(chalk.red(`\n ❌ Upload failed: ${err.message}`));
+    });
+}
+
+function handleFileDownload(filename, host, port) {
+    if (!filename) {
+        console.log(chalk.red('Usage: /download <filename>'));
+        return;
+    }
+
+    // Ensure downloads folder
+    const dlDir = path.join(process.cwd(), 'downloads');
+    if (!fs.existsSync(dlDir)) {
+        fs.mkdirSync(dlDir, { recursive: true });
+    }
+
+    const destPath = path.join(dlDir, filename);
+    console.log(chalk.cyan(`\n 📥 requesting download: ${filename}...`));
+
+    const sidecar = new net.Socket();
+    sidecar.connect(port, host, () => {
+        sidecar.write(`SIDECAR_DOWNLOAD|${filename}\n`);
+    });
+
+    const fileStream = fs.createWriteStream(destPath);
+    sidecar.pipe(fileStream);
+
+    fileStream.on('finish', () => {
+        console.log(chalk.green(`\n ✅ File saved to: ${destPath}`));
+        sidecar.end();
+    });
+
+    sidecar.on('error', (err) => {
+        console.log(chalk.red(`\n ❌ Sidecar Error: ${err.message}`));
+    });
+
+    fileStream.on('error', (err) => {
+        console.log(chalk.red(`\n ❌ File Write Error: ${err.message}`));
+    });
+}

package/src/handler/server.js

@@ -1,67 +1,173 @@
 import net from 'net';
 import SendMessage from '../lib/sendMessage.js';
-import express from 'express';
 import chalk from 'chalk';
 import fs from 'fs';
 import path from 'path';
 import state from '../lib/state.js';
 import showBanner from '../lib/banner.js';
 
+// Ensure server storage exists
+const SERVER_STORAGE = path.join(process.cwd(), 'server_storage');
+if (!fs.existsSync(SERVER_STORAGE)) {
+  fs.mkdirSync(SERVER_STORAGE, { recursive: true });
+}
+
 export default function startServer() {
-  const CHAT_PORT = 8000;
-  const HTTP_PORT = 8001;
+  const PORT = 3000;
+
   const server = net.createServer((socket) => {
-    socket.name = `User-${state.sockets.length + 1}`; // Temporary name
-    socket.isRegistered = false;
-    state.sockets.push(socket);
+    // We listen for the first packet to determine connection type
+    socket.once('data', (initialData) => {
+      // Find newline delimiter
+      const newlineIndex = initialData.indexOf(10); // 0x0A is \n
 
-    // Welcome the user and ask for name
-    const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
-    socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white(`Welcome to Ghostwire!`)}\n`);
+      let headerStr = '';
+      let remainingData = null;
+
+      if (newlineIndex !== -1) {
+        headerStr = initialData.subarray(0, newlineIndex).toString().trim();
+        remainingData = initialData.subarray(newlineIndex + 1);
+      } else {
+        headerStr = initialData.toString().trim();
+      }
+
+      // --- FILE UPLOAD SIDECAR ---
+      // Protocol: SIDECAR_UPLOAD|<filename>\n
+      if (headerStr.startsWith('SIDECAR_UPLOAD|')) {
+        const parts = headerStr.split('|');
+        const filename = parts[1];
+        if (!filename) {
+          socket.end();
+          return;
+        }
 
-    socket.on('data', (data) => {
-      const msg = data.toString().trim();
-      if (!msg) return;
+        const safeFilename = path.basename(filename);
+        console.log(chalk.gray(`[FILE] Incoming upload: ${safeFilename}`));
 
-      if (!socket.isRegistered) {
-        socket.name = msg;
-        socket.isRegistered = true;
+        const writeStream = fs.createWriteStream(path.join(SERVER_STORAGE, safeFilename));
+
+        // Write the remaining validation data if any exists in the first chunk
+        if (remainingData && remainingData.length > 0) {
+          writeStream.write(remainingData);
+        }
 
-        const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
-        socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white('Name set to')} ${chalk.cyan.bold(socket.name)}\n`);
+        socket.pipe(writeStream);
 
-        SendMessage('SYSTEM', `${socket.name} joined the chat`);
+        writeStream.on('finish', () => {
+          console.log(chalk.green(`[FILE] Upload complete: ${safeFilename}`));
+          // Broadcast availability
+          SendMessage('SYSTEM', `New file shared: ${chalk.bold(safeFilename)} (Type /download ${safeFilename})`);
+          socket.end();
+        });
+
+        writeStream.on('error', (err) => {
+          console.error(chalk.red(`[FILE] Write error: ${err.message}`));
+          socket.end();
+        });
         return;
       }
 
-      if (msg.startsWith('/nick ')) {
-        const name = msg.split(' ')[1];
-        if (name) {
-          socket.name = name;
-          const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
-          socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white('Name changed to')} ${chalk.cyan.bold(name)}\n`);
+      // --- FILE DOWNLOAD SIDECAR ---
+      // Protocol: SIDECAR_DOWNLOAD|<filename>\n
+      if (headerStr.startsWith('SIDECAR_DOWNLOAD|')) {
+        const parts = headerStr.split('|');
+        const filename = parts[1];
+        if (!filename) {
+          socket.end();
+          return;
         }
-      } else {
-        SendMessage(socket.name, msg, socket);
+
+        const safeFilename = path.basename(filename);
+        const filePath = path.join(SERVER_STORAGE, safeFilename);
+
+        console.log(chalk.gray(`[FILE] download request: ${safeFilename}`));
+
+        if (fs.existsSync(filePath)) {
+          const readStream = fs.createReadStream(filePath);
+          readStream.pipe(socket);
+          readStream.on('error', (err) => {
+            console.log(chalk.red('[FILE] Read error'));
+          });
+          // Socket ends when stream ends
+        } else {
+          console.log(chalk.red(`[FILE] Not found: ${safeFilename}`));
+          socket.write('ERROR_NOT_FOUND');
+          socket.end();
+        }
+        return;
       }
-    });
-    // Handle Disconnects
-    socket.on('end', () => {
-      state.sockets = state.sockets.filter(s => s !== socket);
-      SendMessage('SYSTEM', `${socket.name} disconnected.`);
+
+      // --- STANDARD CHAT CLIENT ---
+      // Treat first packet as username/registration or just init
+      handleChatConnection(socket, initialData);
     });
 
-    socket.on('error', () => {
-      state.sockets = state.sockets.filter(s => s !== socket);
-      SendMessage('SYSTEM', `${socket.name} disconnected.`);
+    socket.on('error', (err) => {
+      // Silent handle - cleanliness
     });
   });
-  server.listen(CHAT_PORT, () => {
+
+  server.listen(PORT, () => {
     showBanner('GHOSTWIRE', 'HOST ACTIVE');
     console.log(chalk.gray('--------------------------------------------------'));
-    console.log(chalk.green(` ✅ TCP Chat running on Port ${CHAT_PORT}`));
+    console.log(chalk.green(` ✅ TCP Server (Chat + Files) running on Port ${PORT}`));
     console.log(chalk.gray('--------------------------------------------------'));
     console.log(chalk.yellow('\n ⚠  NEXT STEP: Start Ngrok ⚠'));
-    console.log(` To start ngrok, open new terminal and run: ${chalk.cyan.bold('ngrok tcp 8000')}`);
+    console.log(` To start ngrok, open new terminal and run: ${chalk.cyan.bold(`ngrok tcp ${PORT}`)}`);
+  });
+};
+
+function handleChatConnection(socket, initialData) {
+  socket.name = `User-${state.sockets.length + 1}`;
+  socket.isRegistered = false;
+  state.sockets.push(socket);
+
+  const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
+  socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white(`Welcome to Ghostwire!`)}\n`);
+
+  // Process the initial data (likely the username request or empty)
+  processChatData(socket, initialData);
+
+  // Continue listening
+  socket.on('data', (data) => {
+    processChatData(socket, data);
   });
-};
+
+  socket.on('end', () => {
+    state.sockets = state.sockets.filter(s => s !== socket);
+    if (socket.isRegistered) {
+      SendMessage('SYSTEM', `${socket.name} disconnected.`);
+    }
+  });
+
+  socket.on('error', () => {
+    state.sockets = state.sockets.filter(s => s !== socket);
+  });
+}
+
+function processChatData(socket, data) {
+  const msg = data.toString().trim();
+  if (!msg) return;
+
+  if (!socket.isRegistered) {
+    socket.name = msg;
+    socket.isRegistered = true;
+
+    const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
+    socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white('Name set to')} ${chalk.cyan.bold(socket.name)}\n`);
+
+    SendMessage('SYSTEM', `${socket.name} joined the chat`);
+    return;
+  }
+
+  if (msg.startsWith('/nick ')) {
+    const name = msg.split(' ')[1];
+    if (name) {
+      socket.name = name;
+      const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit', hour12: true }).toLowerCase();
+      socket.write(`${chalk.gray(`[${time}]`)} ${chalk.yellow.bold('⚡ SYSTEM:')} ${chalk.white('Name changed to')} ${chalk.cyan.bold(name)}\n`);
+    }
+  } else {
+    SendMessage(socket.name, msg, socket);
+  }
+}