feat: add webshell

packages/webshell/webshell-node/src/socket/terminal.ts

@@ -0,0 +1,49 @@
+import * as pty from 'node-pty';
+import * as os from 'os';
+import { homedir } from 'os';
+import { Socket } from 'socket.io';
+
+const shell: string = os.platform() === 'win32' ? 'powershell.exe' : 'zsh';
+
+interface Option {
+  name: string;
+  cols?: number;
+  rows?: number;
+  cwd?: string;
+}
+
+let ptyContainers: { [key: string]: pty.IPty } = {};
+
+export const terminal = (socket: Socket) => {
+  socket.on('create', (option: Option) => {
+    let ptyProcess = pty.spawn(shell, ['--login'], {
+      name: 'xterm-color',
+      cols: option.cols || 80,
+      rows: option.rows || 24,
+      cwd: option.cwd || homedir(),
+      env: process.env,
+    }) as any;
+    console.log('create', option.name);
+    ptyProcess.on('data', (data: string) => socket.emit(option.name + '-output', data));
+    socket.on(option.name + '-input', (data: string) => ptyProcess.write(data));
+    socket.on(option.name + '-resize', (size: [number, number]) => {
+      ptyProcess.resize(size[0], size[1]);
+    });
+    socket.on(option.name + '-exit', () => {
+      console.log('exit', option.name)
+      ptyProcess.destroy();
+    });
+    socket.emit(option.name + '-pid', ptyProcess.pid);
+    ptyContainers[option.name] = ptyProcess;
+  });
+  socket.on('remove', (name: string) => {
+    socket.removeAllListeners(name + '-input');
+    socket.removeAllListeners(name + '-resize');
+    socket.removeAllListeners(name + '-exit');
+    if (name && ptyContainers[name] && ptyContainers[name].pid) {
+      const curentContainer = ptyContainers[name] as any;
+      curentContainer.destroy();
+      delete ptyContainers[name];
+    }
+  });
+};