@balaji003/lantransfer 1.0.2 → 1.0.3

package/bin/cli.js

@@ -1,3 +1,3 @@
-#!/usr/bin/env node
-
-import '../server.js';
+#!/usr/bin/env node
+
+require('../server.js');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@balaji003/lantransfer",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "LAN File Transfer — peer-to-peer file sharing over local network. Zero dependencies.",
   "main": "server.js",
   "bin": {

package/public/index.html

@@ -65,6 +65,17 @@
     }
     .toggle.active .toggle-dot { background: var(--green); }
 
+    .header-right { display: flex; align-items: center; gap: 10px; }
+
+    .btn-shutdown {
+      display: flex; align-items: center; gap: 6px;
+      padding: 6px 16px; border-radius: 20px; border: 1px solid transparent;
+      background: rgba(248, 113, 113, 0.12); color: var(--red);
+      font-size: 13px; font-weight: 500; cursor: pointer;
+      transition: all 0.2s; user-select: none;
+    }
+    .btn-shutdown:hover { border-color: var(--red); background: rgba(248, 113, 113, 0.22); }
+
     /* ── Main layout ── */
     .main { flex: 1; display: flex; overflow: hidden; }
 
@@ -207,9 +218,14 @@
       <div class="logo">&#8652; LAN Transfer</div>
       <div class="device-info" id="device-info"></div>
     </div>
-    <div class="toggle" id="toggle" onclick="api('toggle')">
-      <div class="toggle-dot"></div>
-      <span id="toggle-text">Hidden</span>
+    <div class="header-right">
+      <div class="toggle" id="toggle" onclick="api('toggle')">
+        <div class="toggle-dot"></div>
+        <span id="toggle-text">Hidden</span>
+      </div>
+      <button class="btn-shutdown" onclick="doShutdown()" title="Stop the server">
+        &#x23FB; Stop Server
+      </button>
     </div>
   </div>
 
@@ -303,6 +319,13 @@
     api('respond', { requestId: requestId, accepted: accepted });
   }
 
+  function doShutdown() {
+    if (confirm('Stop the server? This will close LAN Transfer.')) {
+      api('shutdown');
+      document.body.innerHTML = '<div style="display:flex;align-items:center;justify-content:center;height:100vh;color:#6b6b8d;font-size:18px;">Server stopped. You can close this tab.</div>';
+    }
+  }
+
   // ──────────────────────────────────────────────────────────────────────────
   // Render
   // ──────────────────────────────────────────────────────────────────────────

package/server.js

@@ -56,6 +56,22 @@ function getLocalIP() {
   return '127.0.0.1';
 }
 
+/** Compute subnet broadcast address from local IP and netmask. */
+function getBroadcastAddresses() {
+  const addrs = [];
+  for (const ifaces of Object.values(os.networkInterfaces())) {
+    for (const iface of ifaces) {
+      if (iface.family === 'IPv4' && !iface.internal && iface.netmask) {
+        const ipParts = iface.address.split('.').map(Number);
+        const maskParts = iface.netmask.split('.').map(Number);
+        const broadcast = ipParts.map((octet, i) => (octet | (~maskParts[i] & 255))).join('.');
+        addrs.push(broadcast);
+      }
+    }
+  }
+  return addrs.length > 0 ? addrs : ['255.255.255.255'];
+}
+
 function serializeState() {
   return JSON.stringify({
     isDiscoverable,
@@ -193,6 +209,13 @@ const server = http.createServer(async (req, res) => {
         pendingRequests.delete(data.requestId);
       }
     }
+    else if (req.url === '/api/shutdown') {
+      res.writeHead(200, { 'Content-Type': 'text/plain' });
+      res.end('ok');
+      console.log('\n  Server stopped by user.');
+      setTimeout(() => process.exit(0), 200);
+      return;
+    }
 
     res.writeHead(200, { 'Content-Type': 'text/plain' });
     res.end('ok');
@@ -256,11 +279,13 @@ udp.on('error', err => {
 
 udp.bind(DISCOVERY_PORT);
 
-// Broadcaster — send beacon every 3 s
+// Broadcaster — send beacon every 3 s (to subnet broadcast addresses)
 setInterval(() => {
   if (!isDiscoverable) return;
   const msg = JSON.stringify({ device_name: deviceName, tcp_port: TRANSFER_PORT });
-  udp.send(msg, DISCOVERY_PORT, '255.255.255.255', () => {});
+  for (const addr of getBroadcastAddresses()) {
+    udp.send(msg, DISCOVERY_PORT, addr, () => {});
+  }
 }, BROADCAST_INTERVAL);
 
 // Prune stale peers
@@ -478,6 +503,55 @@ function formatSize(bytes) {
   return (bytes / 1073741824).toFixed(1) + ' GB';
 }
 
+// ──────────────────────────────────────────────────────────────────────────────
+// Windows Firewall check
+// ──────────────────────────────────────────────────────────────────────────────
+function checkWindowsFirewall() {
+  if (process.platform !== 'win32') return;
+
+  const ruleName = 'LAN Transfer (lantransfer)';
+  const nodeExe = process.execPath;
+
+  // Check if a firewall rule already exists for this app
+  exec(`netsh advfirewall firewall show rule name="${ruleName}"`, { windowsHide: true }, (err, stdout) => {
+    if (!err && stdout.includes(ruleName)) {
+      // Rule exists, check if it's enabled
+      if (stdout.includes('Enabled:') && stdout.includes('Yes')) {
+        console.log('  Firewall:  Allowed');
+        return;
+      }
+    }
+
+    // No rule found — prompt the user to allow it
+    console.log('');
+    console.log('  !! Firewall rule not found for LAN Transfer.');
+    console.log('  !! Device discovery requires UDP/TCP access through Windows Firewall.');
+    console.log('  !! Attempting to add firewall rule (requires Admin)...');
+    console.log('');
+
+    // Try to add the rule via elevated PowerShell — this triggers a UAC prompt
+    const addCmd = [
+      `netsh advfirewall firewall add rule name="${ruleName}" dir=in action=allow protocol=UDP localport=${DISCOVERY_PORT} program="${nodeExe}" enable=yes`,
+      `netsh advfirewall firewall add rule name="${ruleName}" dir=in action=allow protocol=TCP localport=${TRANSFER_PORT} program="${nodeExe}" enable=yes`,
+    ].join(' & ');
+
+    const psCmd = `powershell -Command "Start-Process cmd -ArgumentList '/c ${addCmd.replace(/"/g, '\\"')}' -Verb RunAs -Wait"`;
+
+    exec(psCmd, { windowsHide: false, timeout: 60_000 }, (err2) => {
+      if (err2) {
+        console.log('  !! Could not add firewall rule. Discovery may not work.');
+        console.log('  !! To fix manually:');
+        console.log('  !!   1. Open Windows Security > Firewall & network protection');
+        console.log('  !!   2. Click "Allow an app through firewall"');
+        console.log('  !!   3. Add Node.js and allow Private + Public networks');
+        console.log('');
+      } else {
+        console.log('  Firewall:  Rule added successfully! Discovery should work now.');
+      }
+    });
+  });
+}
+
 // ──────────────────────────────────────────────────────────────────────────────
 // Start
 // ──────────────────────────────────────────────────────────────────────────────
@@ -498,6 +572,9 @@ server.listen(HTTP_PORT, () => {
   console.log(`  UI:        http://localhost:${HTTP_PORT}`);
   console.log('');
 
+  // Check firewall on Windows
+  checkWindowsFirewall();
+
   // Auto-open browser
   const url = `http://localhost:${HTTP_PORT}`;
   if (process.platform === 'win32') exec(`start "" "${url}"`);