add screenshot command to api

Author: toddtarsi

packages/selenium-ide/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@seleniumhq/selenium-ide",
-  "version": "4.0.0-alpha.56",
+  "version": "4.0.0-alpha.57",
   "private": true,
   "description": "Selenium IDE electron app",
   "author": "Todd <[email protected]>",
@@ -106,7 +106,7 @@
     "@seleniumhq/code-export-ruby-rspec": "^4.0.0-alpha.1",
     "side-code-export": "^4.0.0-alpha.11",
     "@seleniumhq/get-driver": "^4.0.0-alpha.1",
-    "@seleniumhq/side-api": "^4.0.0-alpha.34",
+    "@seleniumhq/side-api": "^4.0.0-alpha.35",
     "@seleniumhq/side-model": "^4.0.0-alpha.4",
     "@seleniumhq/side-runtime": "^4.0.0-alpha.31",
     "dnd-core": "^16.0.1",

packages/selenium-ide/src/main/session/controllers/Driver/index.ts

@@ -151,4 +151,10 @@ export default class DriverController extends BaseController {
     }
     return null
   }
+
+  async takeScreenshot(): Promise<string> {
+    const driver = this.session.playback.playbacks?.[0]?.executor?.driver
+    if (driver) return await driver.takeScreenshot()
+    return ''
+  }
 }

packages/side-api/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@seleniumhq/side-api",
-  "version": "4.0.0-alpha.34",
+  "version": "4.0.0-alpha.35",
   "private": false,
   "description": "Selenium IDE API command shapes and such",
   "author": "Todd Tarsi <[email protected]>",

packages/side-api/src/commands/driver/index.ts

@@ -3,19 +3,22 @@ import type { Shape as ListBrowsers } from './listBrowsers'
 import type { Shape as SelectBrowser } from './selectBrowser'
 import type { Shape as StartProcess } from './startProcess'
 import type { Shape as StopProcess } from './stopProcess'
+import type { Shape as TakeScreenshot } from './takeScreenshot'
 
 import * as download from './download'
 import * as listBrowsers from './listBrowsers'
 import * as selectBrowser from './selectBrowser'
 import * as startProcess from './startProcess'
 import * as stopProcess from './stopProcess'
+import * as takeScreenshot from './takeScreenshot'
 
 export const commands = {
   download,
   listBrowsers,
   selectBrowser,
   startProcess,
   stopProcess,
+  takeScreenshot,
 }
 
 /**
@@ -28,4 +31,5 @@ export type Shape = {
   selectBrowser: SelectBrowser
   startProcess: StartProcess
   stopProcess: StopProcess
+  takeScreenshot: TakeScreenshot
 }

packages/side-api/src/commands/driver/takeScreenshot.ts

@@ -0,0 +1,4 @@
+/**
+ * Return a base64 encoded screenshot of the current window.
+ */
+export type Shape = () => Promise<string>