add method to disconnect process; auto disconnect; close #3

Author: drinking-code

src/__test__/subProcessIPCLoopback.ts

@@ -3,6 +3,7 @@ export default class subProcessIPCLoopback {
     private readonly originalProcessOn;
     private subProcessListener?: NodeJS.MessageListener;
     private mainProcessListener?: NodeJS.MessageListener;
+    private destroyListener?: NodeJS.ExitListener;
 
     constructor() {
         this.originalProcessSend = process.send
@@ -27,8 +28,15 @@ export default class subProcessIPCLoopback {
             this.mainProcessListener(message, process.send)
     }
 
-    public on(event: 'message', listener: NodeJS.MessageListener): this {
-        this.subProcessListener = listener
+    public on(event: 'message' | 'close', listener: NodeJS.MessageListener | NodeJS.ExitListener): this {
+        switch (event) {
+            case "message":
+                this.subProcessListener = listener as NodeJS.MessageListener
+                break
+            case 'close':
+                this.destroyListener = listener as NodeJS.ExitListener
+                break
+        }
         return this
     }
 
@@ -37,5 +45,7 @@ export default class subProcessIPCLoopback {
         process.on = this.originalProcessOn
         this.mainProcessListener = undefined
         this.subProcessListener = undefined
+        if (this.destroyListener)
+            this.destroyListener(0)
     }
 }

src/main.ts

@@ -2,7 +2,7 @@ import {ChildProcess} from 'child_process'
 import initChild from './init-child.js'
 import IPOSMessaging, {iposMessagingMessage, iposMessagingType} from './messaging.js'
 import intercept from './intercept.js'
-import {classes, deSerialize, SerializableType, SerializedType} from "./serialize";
+import {classes, deSerialize, SerializableType, SerializedType} from './serialize.js'
 
 export default class IPOS {
     private readonly fields: Map<string, any>
@@ -154,6 +154,9 @@ export default class IPOS {
         if (!process.send)
             throw new Error(`Process must have an ipc channel. Activate by passing "stdio: [<stdin>, <stdout>, <stderr>, 'ipc']" as an option.`)
         const messaging = new IPOSMessaging(process)
+        process.on('close', () => {
+            this.removeProcess(process)
+        })
 
         let registered = false, resolve: Function
         const promise: Promise<void> = new Promise(res => resolve = res)
@@ -173,6 +176,14 @@ export default class IPOS {
         return promise
     }
 
+    public removeProcess(process: ChildProcess): boolean {
+        const messaging = this.processMessagingMap.get(process)
+        if (!messaging) return false
+        messaging.destroy()
+        this.processMessagingMap.delete(process)
+        return true
+    }
+
     private syncProcess(process: ChildProcess): Promise<void> {
         let resolve: Function
         const promise: Promise<void> = new Promise(res => resolve = res)

src/messaging.ts

@@ -92,4 +92,9 @@ export default class IPOSMessaging {
     listenForAll(callback: iposMessagingCallback) {
         this.listenForType('any', callback)
     }
+
+    destroy() {
+        this.listeners.clear()
+        this.nonIPOSListeners.clear()
+    }
 }