1 files changed, 101 insertions, 0 deletions
diff --git a/remote/test/puppeteer/packages/puppeteer-core/src/bidi/JSHandle.ts b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/JSHandle.ts
new file mode 100644
index 0000000000..7104601553
--- /dev/null
+++ b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/JSHandle.ts
@@ -0,0 +1,101 @@
+/**
+ * @license
+ * Copyright 2023 Google Inc.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import type * as Bidi from 'chromium-bidi/lib/cjs/protocol/protocol.js';
+
+import type {ElementHandle} from '../api/ElementHandle.js';
+import {JSHandle} from '../api/JSHandle.js';
+import {UnsupportedOperation} from '../common/Errors.js';
+
+import {BidiDeserializer} from './Deserializer.js';
+import type {BidiRealm} from './Realm.js';
+import type {Sandbox} from './Sandbox.js';
+import {releaseReference} from './util.js';
+
+/**
+ * @internal
+ */
+export class BidiJSHandle<T = unknown> extends JSHandle<T> {
+  #disposed = false;
+  readonly #sandbox: Sandbox;
+  readonly #remoteValue: Bidi.Script.RemoteValue;
+
+  constructor(sandbox: Sandbox, remoteValue: Bidi.Script.RemoteValue) {
+    super();
+    this.#sandbox = sandbox;
+    this.#remoteValue = remoteValue;
+  }
+
+  context(): BidiRealm {
+    return this.realm.environment.context();
+  }
+
+  override get realm(): Sandbox {
+    return this.#sandbox;
+  }
+
+  override get disposed(): boolean {
+    return this.#disposed;
+  }
+
+  override async jsonValue(): Promise<T> {
+    return await this.evaluate(value => {
+      return value;
+    });
+  }
+
+  override asElement(): ElementHandle<Node> | null {
+    return null;
+  }
+
+  override async dispose(): Promise<void> {
+    if (this.#disposed) {
+      return;
+    }
+    this.#disposed = true;
+    if ('handle' in this.#remoteValue) {
+      await releaseReference(
+        this.context(),
+        this.#remoteValue as Bidi.Script.RemoteReference
+      );
+    }
+  }
+
+  get isPrimitiveValue(): boolean {
+    switch (this.#remoteValue.type) {
+      case 'string':
+      case 'number':
+      case 'bigint':
+      case 'boolean':
+      case 'undefined':
+      case 'null':
+        return true;
+
+      default:
+        return false;
+    }
+  }
+
+  override toString(): string {
+    if (this.isPrimitiveValue) {
+      return 'JSHandle:' + BidiDeserializer.deserialize(this.#remoteValue);
+    }
+
+    return 'JSHandle@' + this.#remoteValue.type;
+  }
+
+  override get id(): string | undefined {
+    return 'handle' in this.#remoteValue ? this.#remoteValue.handle : undefined;
+  }
+
+  remoteValue(): Bidi.Script.RemoteValue {
+    return this.#remoteValue;
+  }
+
+  override remoteObject(): never {
+    throw new UnsupportedOperation('Not available in WebDriver BiDi');
+  }
+}