chore(Icons): fix code after review

Author: Petr Krotov

src/blocks/Icons/__stories__/Icons.stories.tsx

@@ -14,62 +14,92 @@ export default {
     component: Icons,
 } as Meta;
 
-const DefaultTemplate: StoryFn<IconsBlockModel> = (args) => {
-    const transformedArgs = blockTransform(args) as IconsBlockProps;
-    return (
-        <div style={{padding: '64px'}}>
-            <IconsBlock {...transformedArgs} />
-        </div>
-    );
-};
-
-const WithDescriptionTemplate: StoryFn<IconsBlockModel> = (args) => {
-    const transformedArgs = blockTransform(args) as IconsBlockProps;
-    return (
-        <div style={{padding: '64px'}}>
-            <IconsBlock {...transformedArgs} />
-        </div>
-    );
-};
+const DefaultTemplate: StoryFn<IconsBlockModel> = (args) => (
+    <div style={{padding: '64px'}}>
+        <IconsBlock {...(blockTransform(args) as IconsBlockProps)} />
+    </div>
+);
 
-const SizeTemplate: StoryFn<IconsBlockModel> = (args) => (
+const SizeTemplate: StoryFn<Record<string, IconsBlockModel>> = (args) => (
     <React.Fragment>
-        <DefaultTemplate title="Size S" {...args} size="s" />
-        <DefaultTemplate title="Size M" {...args} size="m" />
-        <DefaultTemplate title="Size L" {...args} size="l" />
+        {Object.entries(args)
+            .map(([key, item]) => {
+                const transformed = blockTransform(item) as IconsBlockProps;
+                return (
+                    <div key={key} style={{padding: '64px'}}>
+                        <IconsBlock {...transformed} />
+                    </div>
+                );
+            })
+            .filter(Boolean)}
     </React.Fragment>
 );
 
-const ColSizeTemplate: StoryFn<IconsBlockModel> = (args) => {
-    const transformedArgs12 = blockTransform({
-        ...args,
+const ColSizeTemplate: StoryFn<Record<string, IconsBlockModel>> = (args) => (
+    <div style={{padding: '64px'}}>
+        {Object.entries(args)
+            .map(([key, item]) => {
+                const transformed = blockTransform(item) as IconsBlockProps;
+                return <IconsBlock key={key} {...transformed} />;
+            })
+            .filter(Boolean)}
+    </div>
+);
+
+export const Default = DefaultTemplate.bind({});
+export const Size = SizeTemplate.bind({});
+export const WithText = DefaultTemplate.bind({});
+export const HeaderColSize = ColSizeTemplate.bind({});
+
+Default.args = data.default.content as IconsBlockModel;
+
+WithText.args = data.withDescription.content as IconsBlockModel;
+
+const SIZES: Record<string, IconsBlockModel> = {
+    size_s: {
+        ...data.size.content,
+        title: 'Size S',
+        size: 's',
+    } as IconsBlockModel,
+    size_m: {
+        ...data.size.content,
+        title: 'Size M',
+        size: 'm',
+    } as IconsBlockModel,
+    size_l: {
+        ...data.size.content,
+        title: 'Size L',
+        size: 'l',
+    } as IconsBlockModel,
+};
+
+Size.args = SIZES;
+Size.parameters = {
+    controls: {
+        include: Object.keys(SIZES),
+    },
+};
+
+const COL_SIZES: Record<string, IconsBlockModel> = {
+    col_12: {
+        ...data.withDescription.content,
         title: 'ColSize 12',
-    }) as IconsBlockProps;
-    const transformedArgs8 = blockTransform({
-        ...args,
+    } as IconsBlockModel,
+    col_8: {
+        ...data.withDescription.content,
         title: 'ColSize 8',
         colSizes: {all: 8},
-    }) as IconsBlockProps;
-    const transformedArgs4 = blockTransform({
-        ...args,
+    } as IconsBlockModel,
+    col_4: {
+        ...data.withDescription.content,
         title: 'ColSize 4',
         colSizes: {all: 4},
-    }) as IconsBlockProps;
-    return (
-        <div style={{padding: '64px'}}>
-            <IconsBlock {...transformedArgs12} />
-            <IconsBlock {...transformedArgs8} />
-            <IconsBlock {...transformedArgs4} />
-        </div>
-    );
+    } as IconsBlockModel,
 };
 
-export const Default = DefaultTemplate.bind([]);
-export const Size = SizeTemplate.bind([]);
-export const WithText = WithDescriptionTemplate.bind({});
-export const HeaderColSize = ColSizeTemplate.bind({});
-
-Default.args = data.default.content as IconsBlockProps;
-Size.args = data.size.content as Omit<IconsBlockProps, 'size'>;
-WithText.args = data.withDescription.content as IconsBlockProps;
-HeaderColSize.args = data.withDescription.content as IconsBlockProps;
+HeaderColSize.args = COL_SIZES;
+HeaderColSize.parameters = {
+    controls: {
+        include: Object.keys(COL_SIZES),
+    },
+};