@indiccoder/mentis-cli 1.0.3

package/src/repl/PersistentShell.ts

@@ -0,0 +1,85 @@
+import { spawn, ChildProcessWithoutNullStreams } from 'child_process';
+import os from 'os';
+
+export class PersistentShell {
+    private process: ChildProcessWithoutNullStreams | null = null;
+    private buffer: string = '';
+    private delimiter: string = 'MENTIS_SHELL_DELIMITER';
+    private resolveCallback: ((output: string) => void) | null = null;
+    private rejectCallback: ((error: Error) => void) | null = null;
+
+    constructor() {
+        // Lazy init: Do not spawn here.
+    }
+
+    private initialize() {
+        if (this.process) return;
+
+        const shell = os.platform() === 'win32' ? 'powershell.exe' : 'bash';
+        const args = os.platform() === 'win32' ? ['-NoLogo', '-NoExit', '-Command', '-'] : [];
+
+        this.process = spawn(shell, args, {
+            stdio: ['pipe', 'pipe', 'pipe']
+        });
+
+        this.process.stdout.on('data', (data) => this.handleOutput(data));
+        this.process.stderr.on('data', (data) => this.handleOutput(data));
+
+        this.process.on('error', (err) => {
+            console.error('Shell process error:', err);
+        });
+
+        this.process.on('exit', (code) => {
+            if (code !== 0) {
+                console.warn(`Shell process exited with code ${code}`);
+            }
+            this.process = null; // Reset on exit
+        });
+    }
+
+    private handleOutput(data: Buffer) {
+        const chunk = data.toString();
+        this.buffer += chunk;
+
+        if (this.buffer.includes(this.delimiter)) {
+            const output = this.buffer.replace(this.delimiter, '').trim();
+            if (this.resolveCallback) {
+                this.resolveCallback(output);
+                this.resolveCallback = null;
+                this.rejectCallback = null;
+            }
+            this.buffer = '';
+        }
+    }
+
+    public async execute(command: string): Promise<string> {
+        this.initialize(); // Ensure shell is running
+
+        if (this.resolveCallback) {
+            throw new Error('Shell is busy execution another command.');
+        }
+
+        if (!this.process) {
+            throw new Error('Failed to initialize shell process.');
+        }
+
+        return new Promise((resolve, reject) => {
+            this.resolveCallback = resolve;
+            this.rejectCallback = reject;
+            this.buffer = '';
+
+            const cleanCommand = command.replace(/\n/g, '; ');
+
+            const fullCommand = `${cleanCommand}; echo "${this.delimiter}"\n`;
+
+            this.process?.stdin.write(fullCommand);
+        });
+    }
+
+    public kill() {
+        if (this.process) {
+            this.process.kill();
+            this.process = null;
+        }
+    }
+}