|
1 | 1 | import type { FC } from 'react'; |
2 | | -import React from 'react'; |
| 2 | +import React, { useMemo, useState } from 'react'; |
3 | 3 |
|
4 | | -import { type API_LeafEntry, Addon_TypesEnum } from '@storybook/core/types'; |
| 4 | +import { Addon_TypesEnum } from '@storybook/core/types'; |
5 | 5 |
|
6 | | -import { Consumer } from '@storybook/core/manager-api'; |
7 | | -import type { API, Combo } from '@storybook/core/manager-api'; |
8 | | - |
9 | | -import memoize from 'memoizerific'; |
| 6 | +import { useChannel, useStorybookApi, useStorybookState } from '@storybook/core/manager-api'; |
10 | 7 |
|
| 8 | +import { STORY_PREPARED } from '../../core-events'; |
11 | 9 | import { AddonPanel } from '../components/panel/Panel'; |
12 | 10 |
|
13 | | -const createPanelActions = memoize(1)((api) => ({ |
14 | | - onSelect: (panel: string) => api.setSelectedPanel(panel), |
15 | | - toggleVisibility: () => api.togglePanel(), |
16 | | - togglePosition: () => api.togglePanelPosition(), |
17 | | -})); |
| 11 | +const Panel: FC<any> = (props) => { |
| 12 | + const api = useStorybookApi(); |
| 13 | + const state = useStorybookState(); |
| 14 | + const [story, setStory] = useState(api.getCurrentStoryData()); |
18 | 15 |
|
19 | | -const getPanels = memoize(1)((api: API, story: API_LeafEntry) => { |
20 | | - const allPanels = api.getElements(Addon_TypesEnum.PANEL); |
| 16 | + useChannel( |
| 17 | + { |
| 18 | + [STORY_PREPARED]: () => { |
| 19 | + setStory(api.getCurrentStoryData()); |
| 20 | + }, |
| 21 | + }, |
| 22 | + [] |
| 23 | + ); |
21 | 24 |
|
22 | | - if (!allPanels || !story || story.type !== 'story') { |
23 | | - return allPanels; |
24 | | - } |
| 25 | + const { parameters, type } = story ?? {}; |
25 | 26 |
|
26 | | - const { parameters } = story; |
| 27 | + const panelActions = useMemo( |
| 28 | + () => ({ |
| 29 | + onSelect: (panel: string) => api.setSelectedPanel(panel), |
| 30 | + toggleVisibility: () => api.togglePanel(), |
| 31 | + togglePosition: () => api.togglePanelPosition(), |
| 32 | + }), |
| 33 | + [api] |
| 34 | + ); |
27 | 35 |
|
28 | | - const filteredPanels: typeof allPanels = {}; |
29 | | - Object.entries(allPanels).forEach(([id, panel]) => { |
30 | | - const { paramKey }: any = panel; |
31 | | - if (paramKey && parameters && parameters[paramKey] && parameters[paramKey].disable) { |
32 | | - return; |
33 | | - } |
34 | | - if ( |
35 | | - panel.disabled === true || |
36 | | - (typeof panel.disabled === 'function' && panel.disabled(parameters)) |
37 | | - ) { |
38 | | - return; |
| 36 | + const panels = useMemo(() => { |
| 37 | + const allPanels = api.getElements(Addon_TypesEnum.PANEL); |
| 38 | + |
| 39 | + if (!allPanels || type !== 'story') { |
| 40 | + return allPanels; |
39 | 41 | } |
40 | | - filteredPanels[id] = panel; |
41 | | - }); |
42 | | - |
43 | | - return filteredPanels; |
44 | | -}); |
45 | | - |
46 | | -const mapper = ({ state, api }: Combo) => ({ |
47 | | - panels: getPanels(api, api.getCurrentStoryData()), |
48 | | - selectedPanel: api.getSelectedPanel(), |
49 | | - panelPosition: state.layout.panelPosition, |
50 | | - actions: createPanelActions(api), |
51 | | - shortcuts: api.getShortcutKeys(), |
52 | | -}); |
53 | | - |
54 | | -const Panel: FC<any> = (props) => ( |
55 | | - <Consumer filter={mapper}>{(customProps) => <AddonPanel {...props} {...customProps} />}</Consumer> |
56 | | -); |
| 42 | + |
| 43 | + const filteredPanels: typeof allPanels = {}; |
| 44 | + Object.entries(allPanels).forEach(([id, p]) => { |
| 45 | + const { paramKey }: any = p; |
| 46 | + if (paramKey && parameters && parameters[paramKey] && parameters[paramKey].disable) { |
| 47 | + return; |
| 48 | + } |
| 49 | + if (p.disabled === true || (typeof p.disabled === 'function' && p.disabled(parameters))) { |
| 50 | + return; |
| 51 | + } |
| 52 | + filteredPanels[id] = p; |
| 53 | + }); |
| 54 | + |
| 55 | + return filteredPanels; |
| 56 | + }, [api, type, parameters]); |
| 57 | + |
| 58 | + return ( |
| 59 | + <AddonPanel |
| 60 | + panels={panels} |
| 61 | + selectedPanel={api.getSelectedPanel()} |
| 62 | + panelPosition={state.layout.panelPosition} |
| 63 | + actions={panelActions} |
| 64 | + shortcuts={api.getShortcutKeys()} |
| 65 | + {...props} |
| 66 | + /> |
| 67 | + ); |
| 68 | +}; |
57 | 69 |
|
58 | 70 | export default Panel; |
0 commit comments