是否可以将React组件传递给puppeteer?

时间:2017-12-30 14:18:29

标签: reactjs puppeteer

我有一个带有一些componentDidMount逻辑的React组件:

export default class MyComponent {
    componentDidMount() {
        // some changes to DOM done here by a library  
    }

    render() {
        return (
            <div>{props.data}</div>
        );
    }
}

是否可以通过道具传递此组件,以便componentDidMount()中的所有内容都被执行,以某种方式对于puppeteer进行截屏?这些方面的东西:

const browser = await puppeteer.launch({ headless: true });
const page = await browser.newPage();

const html = ReactDOMServer.renderToString(<MyComponent data='' />); <-- but this skips the componentDidMount logic
await page.setContent(html);
await page.screenshot({ path: 'screenshot.png' });

我知道我可以使用page.goto(),但是我有一些复杂的登录逻辑,我希望通过这样的快捷方式避免这种情况,而是直接将所有需要的道具传递给组件?

1 个答案:

答案 0 :(得分:5)

我回答了这个问题here。 我们在这里尝试一样。

安装babel,webpack和puppeteer包。

{
  "name": "react-puppeteer",
  "version": "1.0.0",
  "main": "index.js",
  "license": "MIT",
  "scripts": {
    "compile": "webpack",
    "build": "webpack -p",
    "start": "webpack && node pup.js"
  },
  "devDependencies": {
    "babel-core": "^6.26.0",
    "babel-loader": "^7.1.2",
    "babel-preset-env": "^1.6.1",
    "babel-preset-es2015": "^6.24.1",
    "babel-preset-react": "^6.24.1",
    "react": "^16.2.0",
    "react-dom": "^16.2.0",
    "webpack": "^3.10.0",
    "webpack-dev-middleware": "^2.0.3"
  },
  "dependencies": {
    "puppeteer": "^0.13.0"
  }
}

准备webpack配置,

const webpack = require('webpack');

const loaders = [
  {
    test: /\.jsx?$/,
    exclude: /node_modules/,
    loader: 'babel-loader',
    query: {
      presets: ['babel-preset-es2015', 'babel-preset-react'],
      plugins: []
    }
  }
];

module.exports = {
  entry: './entry.js',
  output: {
    path: __dirname,
    filename: 'bundle.js',
    libraryTarget: 'umd'
  },
  module: {
    loaders: loaders
  }
};

创建条目文件, 在此文件中,不是直接挂载元素,而是将其导出到窗口,以便我们以后可以访问它。

import React from 'react';
import { render } from 'react-dom';

class Hello extends React.Component {
  render() {
    return <h1>Hello, {this.props.name}</h1>;
  }
}

// accept a name for example and a domNode where to render
function renderIt(name, domNode) {
  render(<Hello name={name} />, domNode);
}

window.renderIt = renderIt;

当我们运行webpack时,它将生成一个bundle.js文件。我们可以在木偶上使用它。

他们已经在puppeteer上弃用了injectFile函数,所以我们将复活它。这是一个样本回购,你可以添加纱线。

https://github.com/entrptaher/puppeteer-inject-file

现在,让我们创建一个木偶操作脚本。

const puppeteer = require('puppeteer');
const injectFile = require('puppeteer-inject-file');

(async () => {
  const browser = await puppeteer.launch({ headless: false });
  const page = await browser.newPage();
  await page.goto('https://github.com');
  await injectFile(page, require.resolve('./bundle.js'));
  await page.evaluate(() => {
    renderIt("Someone", document.querySelector('div.jumbotron.jumbotron-codelines > div > div > div > h1'));
  });
  await page.screenshot({ path: 'example.png' });
  await browser.close();
})();

当我们运行时,我们得到以下结果,

enter image description here

如果我们添加componentDidMount()电话,我们也可以这样做。但是如果我们想要做更多修改,我们必须让puppeteer脚本等待在其他问题中多次讨论的那些。

假设我们现在有一个状态,一旦加载了组件就会返回一些东西。

class Hello extends React.Component {
  state = {
    jokes: null
  };

  componentDidMount() {
    const self = this;
    const jokesUrl = `http://api.icndb.com/jokes/random?firstName=John&amp;lastName=Doe`;
    fetch(jokesUrl)
      .then(data => data.json())
      .then(data => {
        self.setState({
          jokes: data.value.joke
        });
      });
  }

  render() {
    if(!!this.state.jokes){
      return <p id='quote'>{this.state.jokes}</p>
    }
    return <h1>Hello, {this.props.name}</h1>;
  }
}

在木偶戏上,我可以等待这样的元素,

  ...
  await injectFile(page, require.resolve('./bundle.js'));
  await page.evaluate(() => {
    renderIt("Someone", document.querySelector('div'));
  });
  await page.waitFor('p#quote');
  ...

我们可能需要babel-preset-stage-2但我会把它留给你。这是结果,

enter image description here

自己解决问题的其余部分:) ...