webscanner/src/bridge.js

Automation scanner for web resources

const CRI = require('chrome-remote-interface');
const url = require('url');

/**
 * @param page
 * Puppeteer Page object or object with this structure: {host, port}
 * @return {Promise<unknown>}
 */
async function getChromeClient(page) {
  if (page.hostname && page.port) {
    return await CRI({ host: page.hostname, port: page.port });
  } else {
    const connection = page._client._connection._url;
    const { hostname, port } = url.parse(connection, true);
    return await CRI({ host: hostname, port });
  }
}

module.exports = {
  getChromeClient,
};