import type {FlutterDriver} from '../driver.js';
import type {IsolateSocket} from '../sessions/isolate_socket.js';

export const getScreenshot = async function (this: FlutterDriver) {
  const response = (await (this.socket as IsolateSocket).call(`_flutter.screenshot`)) as any;
  return response.screenshot;
};