@imenam/self-register-proxy 1.0.46 → 1.0.48

package/terminal-server.js

@@ -1,154 +1,154 @@
-const os = require('os');
-const { execSync } = require('child_process');
-const pty = require('node-pty');
-const { WebSocketServer } = require('ws');
-
-const DEFAULT_SHELL = os.platform() === 'win32' ? 'powershell.exe' : 'bash';
-
-class TerminalServer {
-  /**
-   * @param {import('http').Server} httpServer
-   * @param {object} [options]
-   * @param {string} [options.shell] - Shell to spawn (default: auto-detected)
-   * @param {string[]} [options.shellArgs] - Shell arguments (default: [])
-   * @param {object} [options.ptyOptions] - node-pty spawn options overrides
-   * @param {number} [options.ptyOptions.cols] - Initial columns (default: 80)
-   * @param {number} [options.ptyOptions.rows] - Initial rows (default: 24)
-   * @param {string} [options.ptyOptions.cwd] - Working directory (default: user home)
-   * @param {object} [options.ptyOptions.env] - Environment variables (default: process.env)
-   * @param {string} [options.command] - Command to execute automatically after spawn
-   * @param {string} [options.path] - WebSocket endpoint path (default: '/terminal')
-   * @param {object} [options.wsOptions] - WebSocketServer options overrides
-   */
-  constructor(httpServer, options = {}) {
-    this.shell = options.shell || DEFAULT_SHELL;
-    this.shellArgs = options.shellArgs || [];
-    this.command = options.command || '';
-    this.ptyOptions = {
-      name: 'xterm-color',
-      cols: 80,
-      rows: 24,
-      cwd: process.env.HOME || process.env.USERPROFILE,
-      env: process.env,
-      ...options.ptyOptions,
-    };
-
-    this.connections = new Set();
-
-    this.wss = new WebSocketServer({
-      server: httpServer,
-      path: options.path || '/terminal',
-      ...options.wsOptions,
-    });
-    this.wss.on('connection', (ws, req) => this._handleConnection(ws, req));
-  }
-
-  _handleConnection(ws, req) {
-    const cwd = this.ptyOptions.cwd;
-
-    let ptyProcess;
-    try {
-      ptyProcess = pty.spawn(this.shell, this.shellArgs, {
-        ...this.ptyOptions,
-        cwd,
-      });
-    } catch (err) {
-      console.error(`[TERMINAL] Failed to spawn PTY (cwd: ${cwd}):`, err.message);
-      ws.close(1011, 'Failed to spawn terminal');
-      return;
-    }
-
-    const connection = { ws, ptyProcess };
-    this.connections.add(connection);
-
-    ptyProcess.onData((data) => {
-      try {
-        ws.send(data);
-      } catch (e) {
-        // WebSocket already closed
-      }
-    });
-
-    ws.on('message', (data) => {
-      const message = data.toString();
-      try {
-        const msg = JSON.parse(message);
-        if (msg.type === 'resize') {
-          ptyProcess.resize(msg.cols, msg.rows);
-          return;
-        }
-      } catch (e) {
-        // Not JSON, treat as terminal input
-      }
-      ptyProcess.write(message);
-    });
-
-    ws.on('close', () => {
-      this._gracefulKill(ptyProcess);
-      this.connections.delete(connection);
-    });
-
-    // Exécuter automatiquement la commande configurée
-    if (this.command) {
-      // Petit délai pour laisser le shell s'initialiser
-      setTimeout(() => {
-        ptyProcess.write(this.command + '\r');
-        console.log(`[TERMINAL] Auto-executed command: ${this.command}`);
-      }, 500);
-    }
-  }
-
-  /**
-   * Arrêt gracieux : envoie Ctrl+C deux fois pour interrompre proprement
-   * le programme en cours, attend qu'il se termine, puis tue le shell.
-   * @param {import('node-pty').IPty} ptyProcess
-   */
-  _gracefulKill(ptyProcess) {
-    const pid = ptyProcess.pid;
-    console.log(`[TERMINAL] Graceful shutdown (PID: ${pid})...`);
-
-    // 1er Ctrl+C
-    try { ptyProcess.write('\x03'); } catch (_) { /* ignore */ }
-
-    setTimeout(() => {
-      // 2e Ctrl+C
-      try { ptyProcess.write('\x03'); } catch (_) { /* ignore */ }
-
-      setTimeout(() => {
-        // Tuer tout l'arbre de processus
-        this._killProcessTree(pid);
-      }, 1000);
-    }, 500);
-  }
-
-  /**
-   * Tue l'arbre de processus par PID.
-   * Sur Windows : taskkill /F /T /PID
-   * Sur Linux/macOS : kill du groupe de processus
-   * @param {number} pid
-   */
-  _killProcessTree(pid) {
-    try {
-      if (os.platform() === 'win32') {
-        execSync(`taskkill /F /T /PID ${pid}`, { stdio: 'ignore' });
-      } else {
-        process.kill(-pid, 'SIGKILL');
-      }
-      console.log(`[TERMINAL] Killed process tree (PID: ${pid})`);
-    } catch (e) {
-      // Le processus est peut-être déjà terminé
-      console.log(`[TERMINAL] Process tree already exited (PID: ${pid})`);
-    }
-  }
-
-  dispose() {
-    for (const { ws, ptyProcess } of this.connections) {
-      this._gracefulKill(ptyProcess);
-      ws.close();
-    }
-    this.connections.clear();
-    this.wss.close();
-  }
-}
-
-module.exports = { TerminalServer };
+const os = require('os');
+const { execSync } = require('child_process');
+const pty = require('node-pty');
+const { WebSocketServer } = require('ws');
+
+const DEFAULT_SHELL = os.platform() === 'win32' ? 'powershell.exe' : 'bash';
+
+class TerminalServer {
+  /**
+   * @param {import('http').Server} httpServer
+   * @param {object} [options]
+   * @param {string} [options.shell] - Shell to spawn (default: auto-detected)
+   * @param {string[]} [options.shellArgs] - Shell arguments (default: [])
+   * @param {object} [options.ptyOptions] - node-pty spawn options overrides
+   * @param {number} [options.ptyOptions.cols] - Initial columns (default: 80)
+   * @param {number} [options.ptyOptions.rows] - Initial rows (default: 24)
+   * @param {string} [options.ptyOptions.cwd] - Working directory (default: user home)
+   * @param {object} [options.ptyOptions.env] - Environment variables (default: process.env)
+   * @param {string} [options.command] - Command to execute automatically after spawn
+   * @param {string} [options.path] - WebSocket endpoint path (default: '/terminal')
+   * @param {object} [options.wsOptions] - WebSocketServer options overrides
+   */
+  constructor(httpServer, options = {}) {
+    this.shell = options.shell || DEFAULT_SHELL;
+    this.shellArgs = options.shellArgs || [];
+    this.command = options.command || '';
+    this.ptyOptions = {
+      name: 'xterm-color',
+      cols: 80,
+      rows: 24,
+      cwd: process.env.HOME || process.env.USERPROFILE,
+      env: process.env,
+      ...options.ptyOptions,
+    };
+
+    this.connections = new Set();
+
+    this.wss = new WebSocketServer({
+      server: httpServer,
+      path: options.path || '/terminal',
+      ...options.wsOptions,
+    });
+    this.wss.on('connection', (ws, req) => this._handleConnection(ws, req));
+  }
+
+  _handleConnection(ws, req) {
+    const cwd = this.ptyOptions.cwd;
+
+    let ptyProcess;
+    try {
+      ptyProcess = pty.spawn(this.shell, this.shellArgs, {
+        ...this.ptyOptions,
+        cwd,
+      });
+    } catch (err) {
+      console.error(`[TERMINAL] Failed to spawn PTY (cwd: ${cwd}):`, err.message);
+      ws.close(1011, 'Failed to spawn terminal');
+      return;
+    }
+
+    const connection = { ws, ptyProcess };
+    this.connections.add(connection);
+
+    ptyProcess.onData((data) => {
+      try {
+        ws.send(data);
+      } catch (e) {
+        // WebSocket already closed
+      }
+    });
+
+    ws.on('message', (data) => {
+      const message = data.toString();
+      try {
+        const msg = JSON.parse(message);
+        if (msg.type === 'resize') {
+          ptyProcess.resize(msg.cols, msg.rows);
+          return;
+        }
+      } catch (e) {
+        // Not JSON, treat as terminal input
+      }
+      ptyProcess.write(message);
+    });
+
+    ws.on('close', () => {
+      this._gracefulKill(ptyProcess);
+      this.connections.delete(connection);
+    });
+
+    // Exécuter automatiquement la commande configurée
+    if (this.command) {
+      // Petit délai pour laisser le shell s'initialiser
+      setTimeout(() => {
+        ptyProcess.write(this.command + '\r');
+        console.log(`[TERMINAL] Auto-executed command: ${this.command}`);
+      }, 500);
+    }
+  }
+
+  /**
+   * Arrêt gracieux : envoie Ctrl+C deux fois pour interrompre proprement
+   * le programme en cours, attend qu'il se termine, puis tue le shell.
+   * @param {import('node-pty').IPty} ptyProcess
+   */
+  _gracefulKill(ptyProcess) {
+    const pid = ptyProcess.pid;
+    console.log(`[TERMINAL] Graceful shutdown (PID: ${pid})...`);
+
+    // 1er Ctrl+C
+    try { ptyProcess.write('\x03'); } catch (_) { /* ignore */ }
+
+    setTimeout(() => {
+      // 2e Ctrl+C
+      try { ptyProcess.write('\x03'); } catch (_) { /* ignore */ }
+
+      setTimeout(() => {
+        // Tuer tout l'arbre de processus
+        this._killProcessTree(pid);
+      }, 1000);
+    }, 500);
+  }
+
+  /**
+   * Tue l'arbre de processus par PID.
+   * Sur Windows : taskkill /F /T /PID
+   * Sur Linux/macOS : kill du groupe de processus
+   * @param {number} pid
+   */
+  _killProcessTree(pid) {
+    try {
+      if (os.platform() === 'win32') {
+        execSync(`taskkill /F /T /PID ${pid}`, { stdio: 'ignore' });
+      } else {
+        process.kill(-pid, 'SIGKILL');
+      }
+      console.log(`[TERMINAL] Killed process tree (PID: ${pid})`);
+    } catch (e) {
+      // Le processus est peut-être déjà terminé
+      console.log(`[TERMINAL] Process tree already exited (PID: ${pid})`);
+    }
+  }
+
+  dispose() {
+    for (const { ws, ptyProcess } of this.connections) {
+      this._gracefulKill(ptyProcess);
+      ws.close();
+    }
+    this.connections.clear();
+    this.wss.close();
+  }
+}
+
+module.exports = { TerminalServer };