import type { DOMWindow } from 'jsdom'

import type { Environment } from '../../types/environment'

import type { JSDOMOptions } from '../../types/jsdom-options'

import { populateGlobal } from './utils'

function catchWindowErrors(window: Window) {

function catchWindowErrors(window: DOMWindow) {

let userErrorListenerCount = 0

function throwUnhandlerError(e: ErrorEvent) {

if (userErrorListenerCount === 0 && e.error != null) {

userAgent,

...restOptions,

})

const clearWindowErrors = catchWindowErrors(dom.window as any)

const clearAddEventListenerPatch = patchAddEventListener(dom.window)

const clearWindowErrors = catchWindowErrors(dom.window)

// TODO: browser doesn't expose Buffer, but a lot of dependencies use it

dom.window.Buffer = Buffer

return dom.getInternalVMContext()

},

teardown() {

clearAddEventListenerPatch()

clearWindowErrors()

dom.window.close()

dom = undefined as any

...restOptions,

})

const clearAddEventListenerPatch = patchAddEventListener(dom.window)

const { keys, originals } = populateGlobal(global, dom.window, {

bindFunctions: true,

})

return {

teardown(global) {

clearAddEventListenerPatch()

clearWindowErrors()

dom.window.close()

delete global.jsdom

}

},

}

function patchAddEventListener(window: DOMWindow) {

const JSDOMAbortSignal = window.AbortSignal

const JSDOMAbortController = window.AbortController

const originalAddEventListener = window.EventTarget.prototype.addEventListener

window.EventTarget.prototype.addEventListener = function addEventListener(

type: string,

callback: EventListenerOrEventListenerObject | null,

options?: AddEventListenerOptions | boolean,

) {

if (typeof options === 'object' && options.signal != null) {

const { signal, ...otherOptions } = options

// - this happens because AbortSignal is provided by Node.js,

// but jsdom APIs require jsdom's AbortSignal, while Node APIs

// (like fetch and Request) require a Node.js AbortSignal

// - disable narrow typing with "as any" because we need it later

if (!((signal as any) instanceof JSDOMAbortSignal)) {

const jsdomCompatOptions = Object.create(null)

Object.assign(jsdomCompatOptions, otherOptions)

// use jsdom-native abort controller instead and forward the

// previous one with `addEventListener`

const jsdomAbortController = new JSDOMAbortController()

signal.addEventListener('abort', () => {

jsdomAbortController.abort(signal.reason)

})

jsdomCompatOptions.signal = jsdomAbortController.signal

return originalAddEventListener.call(this, type, callback, jsdomCompatOptions)

}

}

return originalAddEventListener.call(this, type, callback, options)

}

return () => {

window.EventTarget.prototype.addEventListener = originalAddEventListener

}

}