Add Selenium downloadFile command

packages/wdio-protocols/src/protocols/selenium.ts

@@ -22,6 +22,49 @@ export default {
             },
         },
     },
+    '/session/:sessionId/se/files': {
+        GET: {
+            command: 'getDownloadableFiles',
+            description:
+                'List files from remote machine available for download.',
+            ref: 'https://www.seleniumhq.org/',
+            parameters: [],
+            returns: {
+                type: 'Object',
+                name: 'names',
+                description:
+                    'Object containing a list of downloadable files on remote machine.',
+            },
+        },
+        POST: {
+            command: 'download',
+            description:
+                'Download a file from remote machine on which the browser is running.',
+            ref: 'https://www.seleniumhq.org/',
+            parameters: [
+                {
+                    name: 'name',
+                    type: 'string',
+                    description:
+                        'Name of the file to be downloaded',
+                    required: true,
+                },
+            ],
+            returns: {
+                type: 'Object',
+                name: 'data',
+                description:
+                    'Object containing downloaded file name and its content',
+            },
+        },
+        DELETE: {
+            command: 'deleteDownloadableFiles',
+            description:
+                'Remove all downloadable files from remote machine on which the browser is running.',
+            ref: 'https://www.seleniumhq.org/',
+            parameters: [],
+        },
+    },
     '/grid/api/hub/': {
         GET: {
             isHubCommand: true,

packages/webdriverio/package.json

@@ -83,6 +83,7 @@
     "grapheme-splitter": "^1.0.2",
     "import-meta-resolve": "^4.0.0",
     "is-plain-obj": "^4.1.0",
+    "jszip": "^3.10.1",
     "lodash.clonedeep": "^4.5.0",
     "lodash.zip": "^4.2.0",
     "minimatch": "^9.0.0",

packages/webdriverio/src/commands/browser.ts

@@ -7,6 +7,7 @@ export * from './browser/custom$$.js'
 export * from './browser/custom$.js'
 export * from './browser/debug.js'
 export * from './browser/deleteCookies.js'
+export * from './browser/downloadFile.js'
 export * from './browser/emulate.js'
 export * from './browser/execute.js'
 export * from './browser/executeAsync.js'

packages/webdriverio/src/commands/browser/downloadFile.ts

@@ -0,0 +1,102 @@
+import fs from 'node:fs'
+import path from 'node:path'
+import JSZip from 'jszip'
+import logger from '@wdio/logger'
+
+const log = logger('webdriverio')
+
+/**
+ *
+ * Download a file from the remote computer running Selenium node to local file system
+ * by using the [`downloadFile`](https://webdriver.io/docs/api/selenium#downloadFile) command.
+ *
+ * :::info
+ * Note that this command is only supported if you use a
+ * [Selenium Grid](https://www.selenium.dev/documentation/en/grid/) with Chrome, Edge or Firefox.
+ * :::
+ *
+ * <example>
+ :downloadFile.js
+ it('should download a file', async () => {
+        // capabilities: [
+        //  {
+        //      browserName: 'chrome',
+        //      se:downloadsEnabled': true
+        //  }]
+
+        await browser.url('https://www.selenium.dev/selenium/web/downloads/download.html')
+
+        await $('#file-1').click()
+
+        await browser.waitUntil(async function () {
+            return (await browser.getDownloadableFiles()).names.includes('file_1.txt')
+        }, {timeout: 5000})
+
+        const files = await browser.getDownloadableFiles()
+
+        const downloaded = await browser.downloadFile(files.names[0], process.cwd())
+
+        await browser.deleteDownloadableFiles()
+    })
+ * </example>
+ *
+ * @alias browser.downloadFile
+ * @param {string} fileName remote path to file
+ * @param {string} targetDirectory target location on local computer
+ * @type utility
+ * @uses protocol/download
+ *
+ */
+export async function downloadFile(
+    this: WebdriverIO.Browser,
+    fileName: string,
+    targetDirectory: string
+): Promise<object> {
+    /**
+     * parameter check
+     */
+    if (typeof fileName !== 'string' || typeof targetDirectory !== 'string') {
+        throw new Error('number or type of arguments don\'t agree with downloadFile command')
+    }
+
+    /**
+     * check if command is available
+     */
+    if (typeof this.download !== 'function') {
+        throw new Error(`The downloadFile command is not available in ${(this.capabilities as WebdriverIO.Capabilities).browserName} and only available when using Selenium Grid`)
+    }
+
+    const response = await this.download(fileName)
+    const base64Content = response.contents
+
+    if (!targetDirectory.endsWith('/')) {
+        targetDirectory += '/'
+    }
+
+    fs.mkdirSync(targetDirectory, { recursive: true })
+    const zipFilePath = path.join(targetDirectory, `${fileName}.zip`)
+    fs.writeFileSync(zipFilePath, Buffer.from(base64Content, 'base64'))
+
+    const zipData = fs.readFileSync(zipFilePath)
+    const filesData: string[] = []
+
+    try {
+        const zip = await JSZip.loadAsync(zipData)
+        const keys = Object.keys(zip.files)
+
+        // Iterate through each file in the zip archive
+        for (let i = 0; i < keys.length; i++) {
+            const fileData = await zip.files[keys[i]].async('nodebuffer')
+            const dir = path.resolve(targetDirectory, keys[i])
+            fs.writeFileSync(dir, fileData)
+            log.info(`File extracted: ${keys[i]}`)
+            filesData.push(dir)
+        }
+    } catch (error) {
+        log.error('Error unzipping file:', error)
+    }
+
+    return Promise.resolve({
+        files: filesData
+    })
+}