feat: refactor Calendar stories

Author: TitanKuzmich

packages/plasma-b2c/src/components/Calendar/Calendar.stories.tsx

@@ -1,304 +1,37 @@
-import React, { ComponentProps, useCallback, useRef, useState } from 'react';
-import type { StoryObj, Meta } from '@storybook/react-vite';
-import { action } from 'storybook/actions';
-import { disableProps, InSpacingDecorator } from '@salutejs/plasma-sb-utils';
+import React, { ComponentProps } from 'react';
+import type { Meta } from '@storybook/react';
+import { getCalendarStories } from '@salutejs/plasma-sb-utils';
 
 import { Tabs } from '../Tabs/Tabs';
 import { TabItem } from '../Tabs/TabItem';
 import { H3 } from '../Typography';
 
-import { Calendar, CalendarBase, CalendarBaseRange, CalendarDouble, CalendarDoubleRange } from '.';
+import { config } from './CalendarBase.config';
+import { Calendar, CalendarBase, CalendarBaseRange, CalendarDouble, CalendarDoubleRange } from './Calendar';
 
-const onChangeValue = action('onChangeValue');
+type ButtonProps = ComponentProps<typeof Calendar>;
 
-const meta: Meta<CalendarProps> = {
-    title: 'Data Entry/Calendar',
-    decorators: [InSpacingDecorator],
+const { meta: META, Default, Range } = getCalendarStories({
     component: Calendar,
-    argTypes: {
-        min: {
-            control: {
-                type: 'date',
-            },
-        },
-        max: {
-            control: {
-                type: 'date',
-            },
-        },
-        ...disableProps([
-            'size',
-            'value',
-            'onChangeValue',
-            'date',
-            'eventList',
-            'disabledList',
-            'eventMonthList',
-            'disabledMonthList',
-            'eventQuarterList',
-            'disabledQuarterList',
-            'eventYearList',
-            'disabledYearList',
-            'type',
-            'view',
-            'onChangeStartOfRange',
-        ]),
-    },
-};
-
-export default meta;
-
-type CalendarProps = ComponentProps<typeof Calendar>;
-type CalendarBaseProps = ComponentProps<typeof CalendarBase>;
-type CalendarBaseRangeProps = ComponentProps<typeof CalendarBaseRange>;
-const defaultMinDate = new Date(2016, 6, 1);
-const defaultMaxDate = new Date(2030, 11, 24);
-
-const getBaseEvents = (type: 'days' | 'months' | 'quarters' | 'years', datesNumber = 2) => {
-    const baseDate = {
-        day: 10,
-        monthIndex: 6,
-        year: 2023,
-    };
-
-    const events = [...new Array(datesNumber)].map((_, index) => {
-        const eventNumber = index + 1;
-        const day = type === 'days' ? baseDate.day + index : 1;
-        const month =
-            // eslint-disable-next-line no-nested-ternary
-            type === 'months' || type === 'quarters'
-                ? baseDate.monthIndex + index
-                : type === 'days'
-                ? baseDate.monthIndex
-                : 0;
-        const year = type === 'years' ? baseDate.year + index : baseDate.year;
-
-        return [...new Array(eventNumber)].map(() => {
-            return { date: new Date(year, month, day), color: 'var(--text-warning)' };
-        });
-    });
-
-    return events.flat();
-};
-
-const StoryDefault = (args: CalendarProps) => {
-    const { isRange, isDouble, min, max, date, includeEdgeDates, size, ...rest } = args;
-    const [value, setValue] = useState(new Date(2023, 10, 16));
-    const [valueRange, setValueRange] = useState<[Date, Date?]>([new Date(2023, 10, 16), new Date(2023, 10, 23)]);
-
-    const handleOnChange = useCallback((newValue: Date) => {
-        setValue(newValue);
-        onChangeValue(newValue);
-    }, []);
-    const handleOnRangeChange = useCallback((newValue: [Date, Date?]) => {
-        setValueRange(newValue);
-    }, []);
-
-    return (
-        <div style={{ height: '100vh' }}>
-            <Calendar
-                size={size}
-                isRange={isRange}
-                isDouble={isDouble}
-                date={date}
-                includeEdgeDates={includeEdgeDates}
-                min={min}
-                max={max}
-                {...rest}
-                value={(isRange ? valueRange : value) as Date & [Date, Date?]}
-                onChangeValue={
-                    (isRange ? handleOnRangeChange : handleOnChange) as (value: Date | [Date, Date?]) => void
-                }
-            />
-        </div>
-    );
-};
-
-export const Default: StoryObj<CalendarProps> = {
-    args: {
-        min: new Date(2023, 10, 1),
-        max: new Date(2023, 11, 24),
-        isDouble: false,
-        isRange: false,
-        includeEdgeDates: false,
-        size: 's',
-        stretched: false,
+    componentConfig: config,
+    additionalComponents: {
+        Tabs,
+        TabItem,
+        Heading: H3,
+        CalendarBase,
+        CalendarBaseRange,
+        CalendarDouble,
+        CalendarDoubleRange,
     },
-    render: (args) => <StoryDefault {...args} />,
-};
-
-const StoryBase = (args: CalendarBaseProps & { displayDouble: boolean }) => {
-    const { min, max, includeEdgeDates, size, displayDouble, ...rest } = args;
-    const [value, setValue] = useState(new Date(2023, 6, 7));
-    const [currentState, setCurrentState] = useState('Days');
-
-    const handleOnChange = useCallback((newValue: Date, dateInfo?: unknown) => {
-        setValue(newValue);
-        onChangeValue(newValue, dateInfo);
-    }, []);
-
-    const eventList = useRef(getBaseEvents('days', 5));
-    const eventMonthList = useRef(getBaseEvents('months', 5));
-    const eventQuarterList = useRef(getBaseEvents('quarters'));
-    const eventYearList = useRef(getBaseEvents('years'));
+    defaultArgs: { enableEventTooltip: false, size: 's' },
+    disablePropsList: ['eventTooltipSize', 'enableEventTooltip'],
+});
 
-    const disabledDays = [...new Array(5)].map((_, day) => ({
-        date: new Date(2023, 6, 23 + day),
-    }));
-
-    const getCalendarComponent = ({ type, ...rest }) => {
-        return displayDouble ? (
-            <CalendarDouble
-                size={size}
-                value={value}
-                includeEdgeDates={includeEdgeDates}
-                type={type}
-                onChangeValue={handleOnChange}
-                min={min}
-                max={max}
-                {...rest}
-            />
-        ) : (
-            <CalendarBase
-                size={size}
-                value={value}
-                includeEdgeDates={includeEdgeDates}
-                type={type}
-                onChangeValue={handleOnChange}
-                min={min}
-                max={max}
-                {...rest}
-            />
-        );
-    };
-
-    const calendarMap = {
-        Days: getCalendarComponent({ type: 'Days', eventList, disabledList: disabledDays, ...rest }),
-        Months: getCalendarComponent({ type: 'Months', eventMonthList, ...rest }),
-        Quarters: getCalendarComponent({ type: 'Quarters', eventQuarterList, ...rest }),
-        Years: getCalendarComponent({ type: 'Years', eventYearList, ...rest }),
-    };
-
-    return (
-        <>
-            <H3>Тип календаря</H3>
-            <Tabs view="divider" size="s" hasDivider>
-                {Object.keys(calendarMap).map((state, i) => (
-                    <TabItem
-                        key={`item:${i}`}
-                        view="divider"
-                        selected={state === currentState}
-                        onClick={() => setCurrentState(state)}
-                        size="s"
-                    >
-                        {state}
-                    </TabItem>
-                ))}
-            </Tabs>
-            <div style={{ marginTop: '1rem', height: '100vh' }}>{calendarMap[currentState]}</div>
-        </>
-    );
-};
-
-export const Base: StoryObj<CalendarBaseProps & { displayDouble: boolean }> = {
-    argTypes: {
-        ...disableProps(['isRange', 'isDouble']),
-    },
-    args: {
-        size: 's',
-        min: defaultMinDate,
-        max: defaultMaxDate,
-        includeEdgeDates: false,
-        displayDouble: false,
-        stretched: false,
-    },
-    render: (args) => <StoryBase {...args} />,
+const meta: Meta<ButtonProps> = {
+    ...META,
+    title: 'Data Entry/Calendar',
 };
 
-const StoryRange = (args: CalendarBaseRangeProps & { displayDouble: boolean }) => {
-    const { min, max, includeEdgeDates, size, displayDouble, ...rest } = args;
-    const [values, setValue] = useState<[Date, Date?]>([new Date(2023, 6, 1), new Date(2023, 6, 16)]);
-    const [currentState, setCurrentState] = useState('Days');
-
-    const handleOnChange = (newValue: [Date, Date?], dateInfo) => {
-        onChangeValue(newValue, dateInfo);
-        setValue(newValue);
-    };
-
-    const eventList = useRef(getBaseEvents('days', 5));
-    const eventMonthList = useRef(getBaseEvents('months', 5));
-    const eventQuarterList = useRef(getBaseEvents('quarters'));
-    const eventYearList = useRef(getBaseEvents('years'));
-
-    const disabledDays = [...new Array(5)].map((_, day) => ({
-        date: new Date(2023, 6, 23 + day),
-    }));
-
-    const getCalendarComponent = ({ type, ...rest }) => {
-        return displayDouble ? (
-            <CalendarDoubleRange
-                size={size}
-                min={min}
-                max={max}
-                includeEdgeDates={includeEdgeDates}
-                type={type}
-                {...rest}
-                value={values}
-                onChangeValue={handleOnChange}
-            />
-        ) : (
-            <CalendarBaseRange
-                size={size}
-                min={min}
-                max={max}
-                includeEdgeDates={includeEdgeDates}
-                type={type}
-                {...rest}
-                value={values}
-                onChangeValue={handleOnChange}
-            />
-        );
-    };
-
-    const calendarMap = {
-        Days: getCalendarComponent({ type: 'Days', eventList, disabledList: disabledDays, ...rest }),
-        Months: getCalendarComponent({ type: 'Months', eventMonthList, ...rest }),
-        Quarters: getCalendarComponent({ type: 'Quarters', eventQuarterList, ...rest }),
-        Years: getCalendarComponent({ type: 'Years', eventYearList, ...rest }),
-    };
-
-    return (
-        <>
-            <H3>Тип календаря</H3>
-            <Tabs view="divider" size="s" hasDivider>
-                {Object.keys(calendarMap).map((state, i) => (
-                    <TabItem
-                        key={`item:${i}`}
-                        view="divider"
-                        selected={state === currentState}
-                        onClick={() => setCurrentState(state)}
-                        size="s"
-                    >
-                        {state}
-                    </TabItem>
-                ))}
-            </Tabs>
-            <div style={{ marginTop: '1rem', height: '100vh' }}>{calendarMap[currentState]}</div>
-        </>
-    );
-};
+export default meta;
 
-export const Range: StoryObj<CalendarBaseRangeProps & { displayDouble: boolean }> = {
-    argTypes: {
-        ...disableProps(['isRange', 'isDouble']),
-    },
-    args: {
-        size: 's',
-        min: defaultMinDate,
-        max: defaultMaxDate,
-        includeEdgeDates: false,
-        displayDouble: false,
-        stretched: false,
-    },
-    render: (args) => <StoryRange {...args} />,
-};
+export { Default, Range };