VsCode WebView to Comlink Endpoint

index.ts

import * as vscode from "vscode"
import * as Comlink from "comlink"

type MessageEvent = Parameters<typeof listener>[0]

function webviewEndpoint(webview: vscode.Webview): Comlink.Endpoint {
	const listeners = new Map<any, vscode.Disposable>()

	return {
    // NO OWNERSHIPT TRANSFERS POSSIBLE
		postMessage(message) {
			webview.postMessage(message)
		},

    // we just assume the "type" is always "message". Comlink also does this with the node adapter
		addEventListener(_, listener) {
			const disposeable = webview.onDidReceiveMessage((data) => {
				if ("handleEvent" in listener) {
					listener.handleEvent({ data } as MessageEvent)
				} else {
					listener({ data } as MessageEvent)
				}
			})

			listeners.set(listener, disposeable)
		},

		removeEventListener(_, listener) {
			const disposeable = listeners.get(listener)
			if (disposeable) {
				disposeable.dispose()
				listeners.delete(listener)
			}
		},
	}
}