在使用 puppeteer 进行脚本评估之前注入 HTML

Question

我想使用 puppeteer 将一些 HTML 注入页面上的特定元素。

必须在执行任何 JavaScript 之前注入 HTML。

我认为有两种方法可以做到这一点：

1. 注入 HTML 使用 page.evaluateOnNewDocument

此函数是“在创建文档后调用”，但我无法从中访问 DOM 元素。例如：

const puppeteer = require('puppeteer');

(async () => {
  const browser = await puppeteer.launch();
  const page = await browser.newPage();

  page.on('console', consoleObj => console.log(consoleObj.text()));

  await page.evaluateOnNewDocument(
    () => {
      const content = document.querySelector('html');
      console.log(content);
    }
  );

  await page.goto(process.argv[2]);

  await browser.close();
})();

当我访问页面时，此脚本仅输出换行符。

2. 使用page.setJavaScriptEnabled防止的JavaScript之前，我注入HTML执行。由于按照文档不过，这并没有开始执行的JavaScript我重新打开它后。例如：

我的脚本看起来像这样：

const fs = require('fs');
const puppeteer = require('puppeteer');

(async () => {
  const browser = await puppeteer.launch();
  const page = await browser.newPage();

  const html = fs.readFileSync('./example.html', 'utf8');

  await page.setJavaScriptEnabled(false)
  await page.goto(process.argv[2]);
  await page.evaluate(
    content => {
      const pageEl = document.querySelector('div.page');
      let node = document.createElement('div');
      node.innerHTML = content;
      pageEl.appendChild(node);
    }, html
  );
  await page.setJavaScriptEnabled(true)

  await browser.close();
})();

另外，它也有可能做这样的事情这样，虽然这似乎对于什么是一个相当简单的要求过于复杂。

有没有更简单的方法来做到这一点，我忽略了？

干杯

Answer 1

看来这实际上是一个非常受欢迎的请求，我也许应该在发布问题之前进行更彻底的搜索。

尽管如此，我还是选择了 aslushnikov这里提出的解决方案。

下面的代码只是我为了测试这个想法而编写的，我确信还有很大的改进空间。

我做了一个简单的函数来执行 XHR：

const requestPage = async (url) => {
  return new Promise(function (resolve, reject) {
    let xhr = new XMLHttpRequest();
    xhr.open('GET', url);
    xhr.setRequestHeader('Ignore-Intercept', 'Value');
    xhr.onload = function () {
      if (this.status >= 200 && this.status < 300) {
        const response = {};
        xhr.getAllResponseHeaders()
          .trim()
          .split(/[\r\n]+/)
          .map(value => value.split(/: /))
          .forEach(keyValue => {
              response[keyValue[0].trim()] = keyValue[1].trim();
          });
        resolve({ ...response, body: xhr.response });
      } else {
        reject({
            status: this.status,
            statusText: xhr.statusText
        });
      }
    };
    xhr.onerror = function () {
      reject({
          status: this.status,
          statusText: xhr.statusText
      });
    };
    xhr.send();
  });
};

然后我将这个函数暴露给页面。

然后，我使用此函数执行 XHR，而不是允许请求继续，并将结果用作对请求的响应。

await page.setRequestInterception(true);
page.on('request', async (request) => {
  if (
    request.url() === url
    && (
      typeof request.headers()['access-control-request-headers'] === 'undefined'
      || !request.headers()['access-control-request-headers'].match(/ignore-intercept/gi)
    ) && typeof request.headers()['ignore-intercept'] === 'undefined'
  ) {
    const response = await page.evaluate(`requestPage('${url}')`);
    response.body += "hello";
    request.respond(response);
  } else {
    request.continue();
  }
});

await page.goto(`data:text/html,<iframe style='width:100%; height:100%' src=${url}></iframe>`);

page.evaluate令人烦恼的是，除非所需的页面位于 iframe 中，否则似乎无法使用。（因此await page.goto(`data:text/html....