@steve02081504/virtual-console 0.0.1 → 0.0.3

package/main.mjs

@@ -6,6 +6,10 @@ import supportsAnsi from 'supports-ansi'
 import { FullProxy } from 'full-proxy'
 
 export const consoleAsyncStorage = new AsyncLocalStorage()
+const cleanupRegistry = new FinalizationRegistry(cleanupToken => {
+	const { stream, listener } = cleanupToken
+	stream.off?.('resize', listener)
+})
 
 /**
  * 创建一个虚拟控制台，用于捕获输出，同时可以选择性地将输出传递给真实的控制台。
@@ -86,8 +90,9 @@ export class VirtualConsole extends Console {
 					if (this.options.recordOutput)
 						this.outputs += chunk.toString()
 					if (this.options.realConsoleOutput)
-						targetStream.write(chunk, encoding)
-					callback()
+						targetStream.write(chunk, encoding, callback)
+					else
+						callback()
 				},
 			})
 
@@ -100,16 +105,25 @@ export class VirtualConsole extends Console {
 					hasColors: { get: () => targetStream.hasColors.bind(targetStream), configurable: true, enumerable: true },
 				})
 
-				targetStream.on?.('resize', () => {
-					virtualStream.emit('resize')
-				})
+				const virtualStreamRef = new WeakRef(virtualStream)
+
+				const resizeListener = () => {
+					virtualStreamRef.deref()?.emit('resize')
+				}
+
+				targetStream.on?.('resize', resizeListener)
+
+				cleanupRegistry.register(this, {
+					stream: targetStream,
+					listener: resizeListener,
+				}, this)
 			}
 
 			return virtualStream
 		}
 
-		this._stdout = createVirtualStream(this.#base_console._stdout || process.stdout)
-		this._stderr = createVirtualStream(this.#base_console._stderr || process.stderr)
+		this._stdout = createVirtualStream(this.#base_console?._stdout || process.stdout)
+		this._stderr = createVirtualStream(this.#base_console?._stderr || process.stderr)
 	}
 
 	/**

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@steve02081504/virtual-console",
-	"version": "0.0.1",
+	"version": "0.0.3",
 	"description": "A virtual console for capturing and manipulating terminal output.",
 	"main": "main.mjs",
 	"type": "module",