From 27863dcea8379e70ecd20ebb6519428118657ea3 Mon Sep 17 00:00:00 2001
From: CanadaHonk <oj@oojmed.com>
Date: Fri, 30 Dec 2022 22:18:46 +0000
Subject: [PATCH] inject: add Window.close() api

---
 src/launcher/inject.js | 7 ++++++-
 src/launcher/start.js  | 2 +-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/launcher/inject.js b/src/launcher/inject.js
index 047c672..c8ae486 100644
--- a/src/launcher/inject.js
+++ b/src/launcher/inject.js
@@ -1,6 +1,6 @@
 import IPCApi from '../lib/ipc.js';
 
-export default async (CDP, injectionType = 'browser', { browserName }) => {
+export default async (CDP, proc, injectionType = 'browser', { browserName } = { browserName: 'unknown' }) => {
   let pageLoadCallback = () => {}, onWindowMessage = () => {};
   CDP.onMessage(msg => {
     if (msg.method === 'Runtime.bindingCalled' && msg.params.name === '_gluonSend') onWindowMessage(JSON.parse(msg.params.payload));
@@ -62,6 +62,11 @@ export default async (CDP, injectionType = 'browser', { browserName }) => {
 
     cdp: {
       send: (method, params) => CDP.sendMessage(method, params, sessionId)
+    },
+
+    close: () => {
+      CDP.close();
+      proc.kill();
     }
   };
 };
\ No newline at end of file
diff --git a/src/launcher/start.js b/src/launcher/start.js
index 2143561..f130630 100644
--- a/src/launcher/start.js
+++ b/src/launcher/start.js
@@ -33,5 +33,5 @@ export default async (browserPath, args, transport, extra) => {
       break;
   }
 
-  return await InjectInto(CDP, transport === 'stdio' ? 'browser' : 'target', extra);
+  return await InjectInto(CDP, proc, transport === 'stdio' ? 'browser' : 'target', extra);
 };
\ No newline at end of file