Merge pull request #3 from henryhale/fix/issue-2

fix/issue 2

Author: Henry Hale

source/index.ts

@@ -226,15 +226,18 @@ export default class ViteShell implements Shell {
             return Promise.reject(SHELL_INACTIVE);
         }
 
-        // flush the output stream and then activate it
-        this.#output.reset();
-
         // incase there's a currently executing command requiring user input
         if (this.#input.isBusy) {
             this.#input.insert(line);
             return Promise.resolve();
+        } else {
+            // otherwise clear the input buffer and any input callback
+            this.#input.reset();
         }
 
+        // flush the output stream and then activate it
+        this.#output.reset();
+
         // check if the line contains characters
         if (typeof line !== "string" || !line.trim()) {
             this.#prompt();
@@ -253,6 +256,7 @@ export default class ViteShell implements Shell {
 
         // reset abort token for reuse
         this.#abortSignal.reset();
+        this.#abortSignal.onAbort(() => this.#input.reset());
 
         // setup a nodejs-like process object
         const process = createProcessContext(

source/streams/input.ts

@@ -41,7 +41,8 @@ export default class InputStream implements StandardInput {
         });
     }
 
-    public clear(): void {
+    public reset(): void {
         this.buffer.splice(0);
+        this.extractor = undefined;
     }
 }

tests/input.test.ts

@@ -13,7 +13,7 @@ describe("Input stream interface", () => {
         expect(stdin).toHaveProperty("isBusy");
         expect(stdin).toHaveProperty("insert");
         expect(stdin).toHaveProperty("readline");
-        expect(stdin).toHaveProperty("clear");
+        expect(stdin).toHaveProperty("reset");
     });
 
     test("usage", () => {