Provide ways to pass in plain objects and boolean values about quickjs-emscripten HOT 8 CLOSED

import { QuickJSHandle, QuickJSVm } from 'quickjs-emscripten'

const QuickJSVmScopeSymbol = Symbol('QuickJSVmScope')

/**
 * 为 QuickJSVm 添加局部作用域，局部作用域的所有方法调用不再需要手动释放内存
 * @param vm
 * @param handle
 */
export function withScope<F extends (vm: QuickJSVm) => any>(
  vm: QuickJSVm,
  handle: F,
): {
  value: ReturnType<F>
  dispose(): void
} {
  let disposes: (() => void)[] = []

  function wrap(handle: QuickJSHandle) {
    disposes.push(() => handle.alive && handle.dispose())
    return handle
  }

  //避免多层代理
  const isProxy = !!Reflect.get(vm, QuickJSVmScopeSymbol)
  function dispose() {
    if (isProxy) {
      Reflect.get(vm, QuickJSVmScopeSymbol)()
      return
    }
    disposes.forEach((dispose) => dispose())
    //手动释放闭包变量的内存
    disposes.length = 0
  }
  const value = handle(
    isProxy
      ? vm
      : new Proxy(vm, {
          get(
            target: QuickJSVm,
            p: keyof QuickJSVm | typeof QuickJSVmScopeSymbol,
          ): any {
            if (p === QuickJSVmScopeSymbol) {
              return dispose
            }
            //锁定所有方法的 this 值为 QuickJSVm 对象而非 Proxy 对象
            const res = Reflect.get(target, p, target)
            if (
              p.startsWith('new') ||
              ['getProp', 'unwrapResult'].includes(p)
            ) {
              return (...args: any[]): QuickJSHandle => {
                return wrap(Reflect.apply(res, target, args))
              }
            }
            if (['evalCode', 'callFunction'].includes(p)) {
              return (...args: any[]) => {
                const res = (target[p] as any)(...args)
                disposes.push(() => {
                  const handle = res.error ?? res.value
                  handle.alive && handle.dispose()
                })
                return res
              }
            }
            if (typeof res === 'function') {
              return (...args: any[]) => {
                return Reflect.apply(res, target, args)
              }
            }
            return res
          },
        }),
  )

  return { value, dispose }
}

withScope(vm, (vm) => {
  const _hello = vm.newFunction('hello', () => {})
  const _object = vm.newObject()
  vm.setProp(_object, 'hello', _hello)
  vm.setProp(_object, 'name', vm.newString('liuli'))
  expect(vm.dump(vm.getProp(_object, 'hello'))).not.toBeNull()
  vm.setProp(vm.global, 'VM_GLOBAL', _object)
}).dispose()