GitHub/storybook
1
0
Fork 0
mirror of https://github.com/storybookjs/storybook.git synced 2025-04-06 05:11:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Clean up after unmount

Browse Source
This commit is contained in:
Michael Shilman 2021-04-29 22:01:11 +08:00
parent e52207d8d7
commit c8e3cbedaa
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
addons/actions/src/register.tsx
View File

@ -1,4 +1,4 @@
import React from 'react';
import React, { useState, useEffect } from 'react';
import { addons, types } from '@storybook/addons';
import { STORY_CHANGED } from '@storybook/core-events';
import ActionLogger from './containers/ActionLogger';
@ -7,11 +7,18 @@ import { ADDON_ID, EVENT_ID, PANEL_ID, PARAM_KEY } from './constants';
addons.register(ADDON_ID, (api) => {
addons.addPanel(PANEL_ID, {
title() {
const [actionsCount, setActionsCount] = React.useState(0);
api.on(EVENT_ID, () => {
setActionsCount(actionsCount + 1);
const [actionsCount, setActionsCount] = useState(0);
const onEvent = () => setActionsCount(actionsCount + 1);
const onChange = () => setActionsCount(0);
useEffect(() => {
api.on(EVENT_ID, onEvent);
api.on(STORY_CHANGED, onChange);
return () => {
api.off(EVENT_ID, onEvent);
api.off(STORY_CHANGED, onChange);
};
});
api.on(STORY_CHANGED, () => setActionsCount(0));
const suffix = actionsCount === 0 ? '' : ` (${actionsCount})`;
return `Actions${suffix}`;
},