➕ implement disconnected event for Browser (#1)

src/api/browser.ts

@@ -1,35 +1,54 @@
+import EventEmitter from 'events'
 import { TSend } from '../protocol'
 import createPage from './page'
 
-const createBrowser = (send: TSend, onEnd: () => void) => ({
-  close: async (): Promise<void> => {
-    await send('Marionette:Quit')
-    onEnd()
-  },
+const createBrowser = (send: TSend) => {
+  const eventEmitter = new EventEmitter()
 
-  disconnect: async (): Promise<void> => {
-    await send('WebDriver:DeleteSession')
-    onEnd()
-  },
+  return {
+    on: (event: string, callback: () => void) => {
+      eventEmitter.on(event, callback)
+    },
 
-  newPage: async () => {
-    await send('WebDriver:ExecuteScript', {
-      script: 'window.open()'
-    })
+    once: (event: string, callback: () => void) => {
+      eventEmitter.once(event, callback)
+    },
 
-    const windows = await send('WebDriver:GetWindowHandles')
+    off: (event: string, callback: () => void) => {
+      eventEmitter.off(event, callback)
+    },
 
-    await send('WebDriver:SwitchToWindow', {
-      focus: true,
-      name: windows[windows.length - 1]
-    })
+    close: async (): Promise<void> => {
+      await send('Marionette:Quit')
 
-    return createPage(send)
-  },
+      eventEmitter.emit('disconnected')
+    },
 
-  pages: (): Promise<number[]> => {
-    return send('WebDriver:GetWindowHandles')
+    disconnect: async (): Promise<void> => {
+      await send('WebDriver:DeleteSession')
+
+      eventEmitter.emit('disconnected')
+    },
+
+    newPage: async () => {
+      await send('WebDriver:ExecuteScript', {
+        script: 'window.open()'
+      })
+
+      const windows = await send('WebDriver:GetWindowHandles')
+
+      await send('WebDriver:SwitchToWindow', {
+        focus: true,
+        name: windows[windows.length - 1]
+      })
+
+      return createPage(send)
+    },
+
+    pages: (): Promise<number[]> => {
+      return send('WebDriver:GetWindowHandles')
+    }
   }
-})
+}
 
 export default createBrowser

src/api/foxr.ts

@@ -20,7 +20,11 @@ const foxr = {
 
     await send('WebDriver:NewSession', { capabilities: {} })
 
-    return createBrowser(send, disconnect)
+    const browser = await createBrowser(send)
+
+    browser.once('disconnected', disconnect)
+
+    return browser
   }
 }