From d2bfd4b0704c1487769f39ab95d37a8d45f909a2 Mon Sep 17 00:00:00 2001
From: Sofiya Pavlenko <sofiushko@yandex-team.ru>
Date: Tue, 14 Jan 2025 14:29:47 +0300
Subject: [PATCH] feat(Toc): render 6 levels of nested items

---
 src/components/Toc/Toc.scss                   | 23 -------
 src/components/Toc/Toc.tsx                    | 65 ++++---------------
 src/components/Toc/TocItem/TocItem.scss       | 10 ++-
 src/components/Toc/TocItem/TocItem.tsx        | 28 +++-----
 src/components/Toc/TocItem/index.ts           |  1 +
 .../Toc/TocSections/TocSections.scss          | 13 ++++
 .../Toc/TocSections/TocSections.tsx           | 62 ++++++++++++++++++
 src/components/Toc/TocSections/index.ts       |  1 +
 .../Toc/__stories__/Toc.stories.tsx           | 17 +++++
 src/components/Toc/__tests__/Toc.test.tsx     | 49 +++++++++++++-
 10 files changed, 167 insertions(+), 102 deletions(-)
 delete mode 100644 src/components/Toc/Toc.scss
 create mode 100644 src/components/Toc/TocItem/index.ts
 create mode 100644 src/components/Toc/TocSections/TocSections.scss
 create mode 100644 src/components/Toc/TocSections/TocSections.tsx
 create mode 100644 src/components/Toc/TocSections/index.ts

diff --git a/src/components/Toc/Toc.scss b/src/components/Toc/Toc.scss
deleted file mode 100644
index 14b59f6d35..0000000000
--- a/src/components/Toc/Toc.scss
+++ /dev/null
@@ -1,23 +0,0 @@
-@use '../variables';
-@use '../../../styles/mixins.scss';
-
-$block: '.#{variables.$ns}toc';
-
-#{$block} {
-    &__title {
-        @include mixins.text-body-2();
-
-        color: var(--g-color-text-primary);
-        margin-block-end: 12px;
-    }
-
-    &__sections,
-    &__subsections {
-        padding: 0;
-        margin: 0;
-
-        overflow: hidden auto;
-
-        list-style: none;
-    }
-}
diff --git a/src/components/Toc/Toc.tsx b/src/components/Toc/Toc.tsx
index 50aba8a882..c47f6f0684 100644
--- a/src/components/Toc/Toc.tsx
+++ b/src/components/Toc/Toc.tsx
@@ -1,17 +1,14 @@
 import * as React from 'react';
 
-import type {AriaLabelingProps, QAProps} from '../types';
+import type {QAProps} from '../types';
 import {block} from '../utils/cn';
-import {filterDOMProps} from '../utils/filterDOMProps';
 
-import {TocItem} from './TocItem/TocItem';
+import {TocSections} from './TocSections';
 import type {TocItem as TocItemType} from './types';
 
-import './Toc.scss';
-
 const b = block('toc');
 
-export interface TocProps extends AriaLabelingProps, QAProps {
+export interface TocProps extends QAProps {
     className?: string;
     items: TocItemType[];
     value?: string;
@@ -20,55 +17,17 @@ export interface TocProps extends AriaLabelingProps, QAProps {
 }
 
 export const Toc = React.forwardRef<HTMLElement, TocProps>(function Toc(props, ref) {
-    const {value: activeValue, items, className, onUpdate, onItemClick, qa, ...restProps} = props;
+    const {value: activeValue, items, className, onUpdate, qa, onItemClick} = props;
 
     return (
-        <nav
-            {...filterDOMProps(restProps, {labelable: true})}
-            className={b(null, className)}
-            ref={ref}
-            data-qa={qa}
-        >
-            <ul className={b('sections')}>
-                {items.map(({value, content, href, items: childrenItems}) => (
-                    <li key={value ?? href} aria-current={activeValue === value}>
-                        <TocItem
-                            content={content}
-                            value={value}
-                            href={href}
-                            active={activeValue === value}
-                            onClick={onUpdate}
-                            onItemClick={onItemClick}
-                        />
-                        {childrenItems && childrenItems.length > 0 && (
-                            <ul className={b('subsections')}>
-                                {childrenItems?.map(
-                                    ({
-                                        value: childrenValue,
-                                        content: childrenContent,
-                                        href: childrenHref,
-                                    }) => (
-                                        <li
-                                            key={childrenValue ?? childrenHref}
-                                            aria-current={activeValue === childrenValue}
-                                        >
-                                            <TocItem
-                                                content={childrenContent}
-                                                value={childrenValue}
-                                                href={childrenHref}
-                                                childItem={true}
-                                                active={activeValue === childrenValue}
-                                                onClick={onUpdate}
-                                                onItemClick={onItemClick}
-                                            />
-                                        </li>
-                                    ),
-                                )}
-                            </ul>
-                        )}
-                    </li>
-                ))}
-            </ul>
+        <nav className={b(null, className)} ref={ref} data-qa={qa}>
+            <TocSections
+                items={items}
+                value={activeValue}
+                onUpdate={onUpdate}
+                depth={1}
+                onItemClick={onItemClick}
+            />
         </nav>
     );
 });
diff --git a/src/components/Toc/TocItem/TocItem.scss b/src/components/Toc/TocItem/TocItem.scss
index eeda96d709..f123b39523 100644
--- a/src/components/Toc/TocItem/TocItem.scss
+++ b/src/components/Toc/TocItem/TocItem.scss
@@ -33,9 +33,13 @@ $block: '.#{variables.$ns}toc-item';
             }
         }
 
-        &_child {
-            #{$class}__section-link {
-                padding-inline-start: 25px;
+        @for $i from 1 through 6 {
+            $item-padding: 12px * $i;
+
+            &_depth_#{$i} {
+                #{$class}__section-link {
+                    padding-inline-start: $item-padding;
+                }
             }
         }
 
diff --git a/src/components/Toc/TocItem/TocItem.tsx b/src/components/Toc/TocItem/TocItem.tsx
index 0d8308e3c0..54e2a65b37 100644
--- a/src/components/Toc/TocItem/TocItem.tsx
+++ b/src/components/Toc/TocItem/TocItem.tsx
@@ -1,6 +1,6 @@
 'use client';
 
-import * as React from 'react';
+import type * as React from 'react';
 
 import {useActionHandlers} from '../../../hooks';
 import {block} from '../../utils/cn';
@@ -13,26 +13,14 @@ const b = block('toc-item');
 export interface TocItemProps extends TocItemType {
     childItem?: boolean;
     active?: boolean;
-    onClick?: (value: string) => void;
-    onItemClick?: (event: React.MouseEvent) => void;
+    onClick?: (event: React.MouseEvent) => void;
+    depth: number;
 }
 
 export const TocItem = (props: TocItemProps) => {
-    const {active = false, childItem = false, content, href, value, onClick, onItemClick} = props;
+    const {active = false, childItem = false, content, href, onClick, depth} = props;
 
-    const handleClick = React.useCallback(
-        (event: React.MouseEvent) => {
-            onItemClick?.(event);
-            if (value === undefined || !onClick) {
-                return;
-            }
-
-            onClick(value);
-        },
-        [onClick, onItemClick, value],
-    );
-
-    const {onKeyDown} = useActionHandlers(handleClick);
+    const {onKeyDown} = useActionHandlers(onClick);
 
     const item =
         href === undefined ? (
@@ -40,16 +28,16 @@ export const TocItem = (props: TocItemProps) => {
                 role="button"
                 tabIndex={0}
                 className={b('section-link')}
-                onClick={handleClick}
+                onClick={onClick}
                 onKeyDown={onKeyDown}
             >
                 {content}
             </div>
         ) : (
-            <a href={href} onClick={handleClick} className={b('section-link')}>
+            <a href={href} onClick={onClick} className={b('section-link')}>
                 {content}
             </a>
         );
 
-    return <div className={b('section', {child: childItem, active})}>{item}</div>;
+    return <div className={b('section', {child: childItem, depth, active})}>{item}</div>;
 };
diff --git a/src/components/Toc/TocItem/index.ts b/src/components/Toc/TocItem/index.ts
new file mode 100644
index 0000000000..74cdc8aad0
--- /dev/null
+++ b/src/components/Toc/TocItem/index.ts
@@ -0,0 +1 @@
+export * from './TocItem';
diff --git a/src/components/Toc/TocSections/TocSections.scss b/src/components/Toc/TocSections/TocSections.scss
new file mode 100644
index 0000000000..7db9c78472
--- /dev/null
+++ b/src/components/Toc/TocSections/TocSections.scss
@@ -0,0 +1,13 @@
+@use '../../variables';
+@use '../../../../styles/mixins.scss';
+
+$block: '.#{variables.$ns}toc';
+
+#{$block}__sections {
+    padding: 0;
+    margin: 0;
+
+    overflow: hidden auto;
+
+    list-style: none;
+}
diff --git a/src/components/Toc/TocSections/TocSections.tsx b/src/components/Toc/TocSections/TocSections.tsx
new file mode 100644
index 0000000000..f8ebf88495
--- /dev/null
+++ b/src/components/Toc/TocSections/TocSections.tsx
@@ -0,0 +1,62 @@
+import * as React from 'react';
+
+import {block} from '../../utils/cn';
+import {TocItem} from '../TocItem';
+import type {TocItem as TocItemType} from '../types';
+
+import './TocSections.scss';
+
+const b = block('toc');
+
+export interface TocSectionsProps {
+    items: TocItemType[];
+    value?: string;
+    onUpdate?: (value: string) => void;
+    depth?: number;
+    childItem?: boolean;
+    onItemClick?: (event: React.MouseEvent) => void;
+}
+
+export const TocSections = React.forwardRef<HTMLElement, TocSectionsProps>(
+    function TocSections(props) {
+        const {value: activeValue, items, onUpdate, childItem, depth = 1, onItemClick} = props;
+
+        if (depth > 6) {
+            return null;
+        }
+
+        return (
+            <ul className={b('sections')}>
+                {items.map(({value, content, href, items: childrenItems}) => (
+                    <li key={value ?? href} aria-current={activeValue === value}>
+                        <TocItem
+                            content={content}
+                            href={href}
+                            active={activeValue === value}
+                            onClick={(event: React.MouseEvent) => {
+                                onItemClick?.(event);
+
+                                if (value === undefined || !onUpdate) {
+                                    return;
+                                }
+
+                                onUpdate?.(value);
+                            }}
+                            childItem={childItem}
+                            depth={depth}
+                        />
+                        {childrenItems && childrenItems.length > 0 && (
+                            <TocSections
+                                items={childrenItems}
+                                onUpdate={onUpdate}
+                                childItem
+                                depth={depth + 1}
+                                value={activeValue}
+                            />
+                        )}
+                    </li>
+                ))}
+            </ul>
+        );
+    },
+);
diff --git a/src/components/Toc/TocSections/index.ts b/src/components/Toc/TocSections/index.ts
new file mode 100644
index 0000000000..b1600b6f47
--- /dev/null
+++ b/src/components/Toc/TocSections/index.ts
@@ -0,0 +1 @@
+export * from './TocSections';
diff --git a/src/components/Toc/__stories__/Toc.stories.tsx b/src/components/Toc/__stories__/Toc.stories.tsx
index 21081b2c68..ab8d9b6b4b 100644
--- a/src/components/Toc/__stories__/Toc.stories.tsx
+++ b/src/components/Toc/__stories__/Toc.stories.tsx
@@ -53,10 +53,27 @@ Default.args = {
                 {
                     value: 'control',
                     content: 'Disk controls',
+                    items: [
+                        {
+                            value: 'floppy',
+                            content: 'Floppy',
+                        },
+                        {
+                            value: 'hard',
+                            content: 'Hard',
+                            items: [],
+                        },
+                    ],
                 },
                 {
                     value: 'snapshots',
                     content: 'Disk snapshots',
+                    items: [
+                        {
+                            value: 'standard',
+                            content: 'Standard',
+                        },
+                    ],
                 },
             ],
         },
diff --git a/src/components/Toc/__tests__/Toc.test.tsx b/src/components/Toc/__tests__/Toc.test.tsx
index 430b750b47..36b8d64c72 100644
--- a/src/components/Toc/__tests__/Toc.test.tsx
+++ b/src/components/Toc/__tests__/Toc.test.tsx
@@ -4,6 +4,7 @@ import userEvent from '@testing-library/user-event';
 
 import {render, screen} from '../../../../test-utils/utils';
 import {Toc} from '../Toc';
+const titles = ['l1Title', 'l2Title', 'l3Title', 'l4Title', 'l5Title', 'l6Title', 'l7Title'];
 
 const defaultItems = [
     {
@@ -18,12 +19,42 @@ const defaultItems = [
     },
     {
         value: 'thirdItem',
-        content: 'Third item',
+        content: titles[0],
         items: [
             {
                 value: 'firstChildItem',
-                content: 'First child item',
-                items: [],
+                content: titles[1],
+                items: [
+                    {
+                        value: 'firstChildItem-depth3',
+                        content: titles[2],
+                        items: [
+                            {
+                                value: 'firstChildItem-depth4',
+                                content: titles[3],
+                                items: [
+                                    {
+                                        value: 'firstChildItem-depth5',
+                                        content: titles[4],
+                                        items: [
+                                            {
+                                                value: 'firstChildItem-depth6',
+                                                content: titles[5],
+                                                items: [
+                                                    {
+                                                        value: 'firstChildItem-depth7',
+                                                        content: titles[6],
+                                                        items: [],
+                                                    },
+                                                ],
+                                            },
+                                        ],
+                                    },
+                                ],
+                            },
+                        ],
+                    },
+                ],
             },
             {
                 value: 'secondChildItem',
@@ -160,4 +191,16 @@ describe('Toc', () => {
 
         expect(currentItem.textContent).toBe(content);
     });
+
+    test('should render 6 levels', async () => {
+        const value = defaultItems[0].value;
+
+        render(<Toc value={value} items={defaultItems} qa={qaId} />);
+
+        for (let i = 0; i <= 5; i++) {
+            expect(screen.getByText(titles[i])).toBeVisible();
+        }
+
+        expect(screen.queryByText(titles[6])).not.toBeInTheDocument();
+    });
 });