index.js 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. const fs = require('fs');
  2. const path = require('path');
  3. const args = require('minimist')(process.argv.slice(2));
  4. const headless = args.headless || 'true';
  5. const url = args.url || 'true';
  6. const token = args.token || '';
  7. const filePath = args.path || 'index.pdf';
  8. const puppeteer = require('puppeteer');
  9. const saveScreenshot = async (url) => {
  10. if (!url) {
  11. return;
  12. }
  13. // 启动浏览器
  14. const browser = await puppeteer.launch({
  15. args: ['--no-sandbox'],
  16. headless: headless === 'true',
  17. devtools: false
  18. });
  19. // 打开页面
  20. const page = await browser.newPage();
  21. // 设置浏览器视窗
  22. page.setViewport({
  23. width: 1123,
  24. height: 1080,
  25. });
  26. // 地址栏输入网页地址
  27. await page.goto(url, {
  28. // 等界面加载完
  29. waitUntil: 'networkidle0'
  30. });
  31. await page.addStyleTag({ path: './style.css' });
  32. await page.pdf({
  33. path: filePath,
  34. printBackground: true,
  35. displayHeaderFooter: false,
  36. margin: {
  37. top: '1.27cm',
  38. bottom: '1.27cm',
  39. left: '1.27cm',
  40. right: '1.27cm'
  41. },
  42. format: 'A4'
  43. });
  44. // 关闭浏览器
  45. await browser.close();
  46. };
  47. saveScreenshot(`${url}`);