一个 Storybook 插件,通过 puppeteer 保存您故事的屏幕截图图像!

在 Github 上查看

Storycap

DEMO

npm

一个 Storybook 插件,保存您故事的屏幕截图图像 :camera: 通过 Puppeteer

Storycap 爬取您的 Storybook 并截取屏幕截图图像。它主要负责视觉测试(例如 reg-suit)所需的图像生成。

特性

  • :camera: 通过 Puppeteer 截取每个故事的屏幕截图。
  • :zap: 速度极快。
  • :package: 零配置。
  • :rocket: 提供灵活的屏幕截图拍摄选项。
  • :tada: 独立于任何 UI 框架(React、Angular、Vue 等...)。

安装

$ npm install storycap

或者

$ npm install storycap puppeteer

安装 puppeteer 是可选的。请参阅 Chromium 版本 以获取更多详细信息。

入门

Storycap 以两种模式运行。一种是“简单”模式,另一种是“托管”模式。

在简单模式下,您无需配置 Storybook。您只需要提供 Storybook 的 URL,例如

$ npx storycap https://127.0.0.1:9001

您可以通过 --serverCmd 选项启动服务器。

$ storycap --serverCmd "start-storybook -p 9001" https://127.0.0.1:9001

当然,您可以使用预构建的 Storybook

$ build-storybook -o dist-storybook
$ storycap --serverCmd "npx http-server dist-storybook -p 9001" https://127.0.0.1:9001

此外,Storycap 可以爬取已构建并托管的 Storybook 页面

$ storycap https://next--storybookjs.netlify.app/vue-kitchen-sink/

托管模式

设置 Storybook

如果您想控制故事的捕获方式(时间或大小等),请使用托管模式。

首先,将 storycap 添加到您的 Storybook 配置文件中

/* .storybook/main.js */

module.exports = {
  stories: ['../src/**/*.stories.@(js|mdx)'],
  addons: [
    '@storybook/addon-actions',
    '@storybook/addon-links',
    'storycap', // <-- Add storycap
  ],
};

接下来,使用 withScreenshot 装饰器来告诉 Storycap 如何捕获您的故事。

/* .storybook/preview.js */

import { withScreenshot } from 'storycap';

export const decorators = [
  withScreenshot, // Registration the decorator is required
];

export const parameters = {
  // Global parameter is optional.
  screenshot: {
    // Put global screenshot parameters(e.g. viewport)
  },
};

[!注意] 您可以使用 addParametersscreenshot 键设置屏幕截图的配置。

设置您的故事(可选)

并且您可以在特定故事文件中通过 parameters 覆盖全局屏幕截图选项。

import React from 'react';
import MyComponent from './MyComponent';

export default {
  title: 'MyComponent',
  component: MyComponent,
  parameters: {
    screenshot: {
      delay: 200,
    },
  },
};

export const Normal = {};

export const Small = {
  args: {
    text: 'small',
  },
  parameters: {
    screenshot: {
      viewport: 'iPhone 5',
    },
  },
};

运行 storycap 命令

$ npx start-storybook -p 9009
$ npx storycap https://127.0.0.1:9009

或者您可以通过 --serverCmd 选项使用单行命令执行

$ npx storycap https://127.0.0.1:9009 --serverCmd "start-storybook -p 9009"

API

withScreenshot

一个 Storybook 装饰器,通知 Storycap 捕获故事。

类型 ScreenshotOptions

ScreenshotOptions 对象作为 addParameters 参数或 withScreenshot 参数的 screenshot 键的值可用。

interface ScreenshotOptions {
  delay?: number;                           // default 0 msec
  waitAssets?: boolean;                     // default true
  waitFor?: string | () => Promise<void>;   // default ""
  fullPage?: boolean;                       // default true
  hover?: string;                           // default ""
  focus?: string;                           // default ""
  click?: string;                           // default ""
  skip?: boolean;                           // default false
  viewport?: Viewport;
  viewports?: string[] | { [variantName]: Viewport };
  variants?: Variants;
  waitImages?: boolean;                     // default true
  omitBackground?: boolean;                 // default false
  captureBeyondViewport?: boolean;          // default true
  clip?: { x: number; y: number; width: number; height: number } | null; // default null
}
  • delay:捕获前的等待时间 [毫秒]。
  • waitAssets:如果设置为 true,Storycap 会等待故事请求的所有资源(例如 <img> 或 CSS 背景图像)完成。
  • waitFor:如果您设置一个返回 Promise 的函数,Storycap 会等待 promise 解析。您也可以设置一个返回 Promise 的全局函数的名称。
  • fullPage:如果设置为 true,Storycap 会捕获故事的整个页面。
  • focus:如果设置有效的 CSS 选择器字符串,Storycap 会在聚焦与选择器匹配的元素后捕获。
  • hover:如果设置有效的 CSS 选择器字符串,Storycap 会在悬停与选择器匹配的元素后捕获。
  • click:如果设置有效的 CSS 选择器字符串,Storycap 会在单击与选择器匹配的元素后捕获。
  • skip:如果设置为 true,Storycap 会取消捕获相应的 story。
  • viewportviewports:请参阅下面的 Viewport 类型部分。
  • variants:请参阅下面的 Variants 类型部分。
  • waitImages:已弃用。请使用 waitAssets。如果设置为 true,Storycap 会等待 story 中的 <img> 加载完成。
  • omitBackground:如果设置为 true,Storycap 会省略页面的背景,从而允许生成透明的屏幕截图。请注意,storybook 主题也需要是透明的。
  • captureBeyondViewport:如果设置为 true,Storycap 会捕获视口之外的屏幕截图。另请参阅 Puppeteer API 文档
  • clip:如果设置,Storycap 将仅捕获由 x/y/width/height 限制的屏幕部分。

类型 Variants

Variants 用于从 1 个故事生成 多个 PNG

type Variants = {
  [variantName: string]: {
    extends?: string | string[]; // default: ""
    delay?: number;
    waitAssets?: boolean;
    waitFor?: string | () => Promise<void>;
    fullPage?: boolean;
    hover?: string;
    focus?: string;
    click?: string;
    skip?: boolean;
    viewport?: Viewport;
    waitImages?: boolean;
    omitBackground?: boolean;
    captureBeyondViewport?: boolean;
    clip?: { x: number; y: number; width: number; height: number } | null;
  };
};
  • extends:如果设置其他变体的名称(或它们的名称数组),则此变体扩展其他变体的选项。并且此变体将生成一个后缀为 _${parentVariantName}_${thisVariantName} 的 PNG 文件。

类型 Viewport

ViewportPuppeteer 视口接口 兼容。

type Viewport =
  | string
  | {
      width: number; // default: 800
      height: number; // default: 600
      deviceScaleFactor: ?number; // default: 1,
      isMobile?: boolean; // default: false,
      hasTouch?: boolean; // default: false,
      isLandscape?: boolean; // default: false,
    };

[!注意] 如果设置字符串,则应选择有效的 设备名称

Viewport 值在 viewports 字段中可用,例如

addParameters({
  screenshot: {
    viewports: {
      large: {
        width: 1024,
        height: 768,
      },
      small: {
        width: 375,
        height: 668,
      },
      xsmall: {
        width: 320,
        height: 568,
      },
    },
  },
});

函数 isScreenshot

function isScreenshot(): boolean;

返回当前进程是否在 Storycap 浏览器中运行。这对于仅在 Storycap 中更改故事的行为(例如禁用 JavaScript 动画)很有用。

命令行选项

usage: storycap [options] storybook_url

Options:
      --help                       Show help                                                                   [boolean]
      --version                    Show version number                                                         [boolean]
  -o, --outDir                     Output directory.                               [string] [default: "__screenshots__"]
  -p, --parallel                   Number of browsers to screenshot.                               [number] [default: 4]
  -f, --flat                       Flatten output filename.                                   [boolean] [default: false]
  -i, --include                    Including stories name rule.                                    [array] [default: []]
  -e, --exclude                    Excluding stories name rule.                                    [array] [default: []]
      --delay                      Waiting time [msec] before screenshot for each story.           [number] [default: 0]
  -V, --viewport                   Viewport.                                              [array] [default: ["800x600"]]
      --disableCssAnimation        Disable CSS animation and transition.                       [boolean] [default: true]
      --disableWaitAssets          Disable waiting for requested assets                       [boolean] [default: false]
      --trace                      Emit Chromium trace files per screenshot.                  [boolean] [default: false]
      --silent                                                                                [boolean] [default: false]
      --verbose                                                                               [boolean] [default: false]
      --forwardConsoleLogs         Forward in-page console logs to the user's console.        [boolean] [default: false]
      --serverCmd                  Command line to launch Storybook server.                       [string] [default: ""]
      --serverTimeout              Timeout [msec] for starting Storybook server.               [number] [default: 60000]
      --shard                      The sharding options for this run. In the format <shardNumber>/<totalShards>.
                                   <shardNumber> is a number between 1 and <totalShards>. <totalShards> is the total
                                   number of computers working.                                [string] [default: "1/1"]
      --captureTimeout             Timeout [msec] for capture a story.                          [number] [default: 5000]
      --captureMaxRetryCount       Number of count to retry to capture.                            [number] [default: 3]
      --metricsWatchRetryCount     Number of count to retry until browser metrics stable.       [number] [default: 1000]
      --viewportDelay              Delay time [msec] between changing viewport and capturing.    [number] [default: 300]
      --reloadAfterChangeViewport  Whether to reload after viewport changed.                  [boolean] [default: false]
      --stateChangeDelay           Delay time [msec] after changing element's state.               [number] [default: 0]
      --listDevices                List available device descriptors.                         [boolean] [default: false]
  -C, --chromiumChannel            Channel to search local Chromium. One of "puppeteer", "canary", "stable", "*"
                                                                                                 [string] [default: "*"]
      --chromiumPath               Executable Chromium path.                                      [string] [default: ""]
      --puppeteerLaunchConfig      JSON string of launch config for Puppeteer.
               [string] [default: "{ "args": ["--no-sandbox", "--disable-setuid-sandbox", "--disable-dev-shm-usage"] }"]

Examples:
  storycap https://127.0.0.1:9009
  storycap https://127.0.0.1:9009 -V 1024x768 -V 320x568
  storycap https://127.0.0.1:9009 -i "some-kind/a-story"
  storycap http://example.com/your-storybook -e "**/default" -V iPad
  storycap --serverCmd "start-storybook -p 3000" https://127.0.0.1:3000

从 1 个故事生成多个 PNG

默认情况下,storycap 从 1 个故事生成 1 个屏幕截图图像。如果您想要为 1 个故事生成多个 PNG(例如视口、元素状态变化等),请使用 variants

基本用法

例如

import React from 'react';
import MyButton from './MyButton';

export default {
  title: 'MyButton',
  component: MyButton,
};

export const Normal = {
  parameters: {
    screenshot: {
      variants: {
        hovered: {
          hover: 'button.my-button',
        },
      },
    },
  },
};

上述配置将生成 2 个 PNG

  • MyButton/normal.png
  • MyButton/normal_hovered.png

上述示例中的变体键 hovered 用作生成的 PNG 文件名的后缀。并且几乎所有 ScreenshotOptions 字段都可作为变体值的字段使用。

注意:variants 本身和 viewports 不允许作为变体的字段。

变体组合

您可以通过 extends 字段组合多个变体。

export const Normal = {
  parameters: {
    screenshot: {
      variants: {
        small: {
          viewport: 'iPhone 5',
        },
        hovered: {
          extends: 'small',
          hover: 'button.my-button',
        },
      },
    },
  },
};

上述示例将生成以下内容

  • MyButton/normal.png(默认)
  • MyButton/normal_small.png(来自 small 变体)
  • MyButton/normal_hovered.png(来自 hovered 变体)
  • MyButton/normal_small_hovered.png(来自 hoveredsmall 变体)

[!注意] 您可以使用 viewports 选项的键扩展一些视口,因为 viewports 字段在内部扩展为变体。

跨多台计算机并行化

要跨多台计算机并行处理更多故事,可以使用 shard 参数。

shard 参数是格式为 <shardNumber>/<totalShards> 的字符串。<shardNumber> 是 1 到 <totalShards>(含)之间的数字。<totalShards> 是运行执行的计算机总数。

例如,使用 --shard 1/1 运行将被视为单台计算机上的默认行为。两台计算机分别运行 --shard 1/2--shard 2/2 将在两台计算机之间拆分故事。

故事以循环方式分布在分片中,并按其 ID 排序。如果一系列“彼此靠近”的故事的截图速度比其他故事慢,则应将它们均匀分布。

提示

使用 Docker 运行

使用 regviz/node-xcb

或创建您自己的 Docker 基础镜像,例如

FROM node:18

RUN apt-get update -y \
    && apt-get install -yq \
    ca-certificates \
    fonts-liberation \
    git \
    libayatana-appindicator3-1 \
    libasound2 \
    libatk-bridge2.0-0 \
    libatk1.0-0 \
    libc6 \
    libcairo2 \
    libcups2 \
    libdbus-1-3 \
    libexpat1 \
    libfontconfig1 \
    libgbm1 \
    libgcc1 \
    libglib2.0-0 \
    libgtk-3-0 \
    libnspr4 \
    libnss3 \
    libpango-1.0-0 \
    libpangocairo-1.0-0 \
    libstdc++6 \
    libx11-6 \
    libx11-xcb1 \
    libxcb1 \
    libxcomposite1 \
    libxcursor1 \
    libxdamage1 \
    libxext6 \
    libxfixes3 \
    libxi6 \
    libxrandr2 \
    libxrender1 \
    libxss1 \
    libxtst6 \
    lsb-release \
    wget \
    xdg-utils

完全控制截图时机

有时您可能希望完全管理执行屏幕截图的时间。如果您认为有必要,请使用 waitFor 选项。此参数接受返回 Promise 的函数或函数名称,该函数应指向返回 Promise 的全局函数。

示例 1

例如,您可以使用 @storybook/test 包提供的 screen 函数等待特定 HTML 元素出现。当元素延迟呈现时,这很有用。

/* MyComponent.stories.js */

import { screen } from '@storybook/test';

export const MyStory = {
  screenshot: {
    waitFor: async () => {
      await screen.findByRole('link');
    },
  },
};

示例 2

另一个示例,以下设置告诉 storycap 等待 fontLoading 解析

<!-- ./storybook/preview-head.html -->
<link rel="preload" href="/some-heavy-asset.woff" as="font" onload="this.setAttribute('loaded', 'loaded')" />
<script>
  function fontLoading() {
    const loaded = () => !!document.querySelector('link[rel="preload"][loaded="loaded"]');
    if (loaded()) return Promise.resolve();
    return new Promise((resolve, reject) => {
      const id = setInterval(() => {
        if (!loaded()) return;
        clearInterval(id);
        resolve();
      }, 50);
    });
  }
</script>
/* .storybook/preview.js */

import { withScreenshot } from 'storycap';

export const decorators = [withScreenshot];

export const parameters = {
  screenshot: {
    waitFor: 'fontLoading',
  },
};

Chromium 版本

从 v3.0.0 开始,Storycap 不再直接使用 Puppeteer。相反,Storycap 按以下顺序搜索 Chromium 二进制文件

  1. 已安装的 Puppeteer 包(如果您显式安装了)
  2. 本地安装的 Canary Chrome
  3. 本地安装的稳定版 Chrome

您可以使用 --chromiumChannel 选项更改搜索通道,或使用 --chromiumPath 选项设置可执行的 Chromium 文件路径。

Storybook 兼容性

Storybook 版本

Storycap 已通过以下版本测试

  • 简单模式
    • Storybook v7.x
    • Storybook v8.x
  • 托管模式
    • Storybook v7.x
    • Storybook v8.x

另请参阅 examples 目录中的包。

UI 框架

Storycap(使用简单模式和托管模式)与特定 UI 框架(例如 React、Angular、Vue.js 等...)无关。因此,您可以将其与您喜欢的框架一起使用 :smile: 。

工作原理

Storycap 使用 Puppeteer 访问已启动的页面。

贡献

请参阅 CONTRIBUTING.md

许可证

MIT © reg-viz