|
1 | | -import React from 'react'; |
2 | | -// @ts-ignore |
| 1 | +import React, {useState, useRef} from 'react'; |
| 2 | + |
3 | 3 | import {customClassSwitcher} from '~/core'; |
| 4 | +import {AccordionContext} from '../contexts/AccordionContext'; |
| 5 | + |
| 6 | +const COMPONENT_NAME = 'Accordion'; |
4 | 7 |
|
5 | 8 | export type AccordionRootProps = { |
6 | 9 | children: React.ReactNode; |
7 | 10 | customRootClass?: string; |
8 | 11 | } |
9 | 12 |
|
10 | 13 | const AccordionRoot= ({children, customRootClass}: AccordionRootProps) => { |
11 | | - const rootClass = customClassSwitcher(customRootClass, 'Accordion'); |
| 14 | + const accordionRef = useRef(null); |
| 15 | + const [activeItem, setActiveItem] = useState(null); |
| 16 | + const [focusItem, setFocusItem] = useState(null); |
| 17 | + const rootClass = customClassSwitcher(customRootClass, COMPONENT_NAME); |
| 18 | + |
| 19 | + |
| 20 | + const getActiveItemId = () => { |
| 21 | + let elem = accordionRef?.current; |
| 22 | + // get children that have data-state open |
| 23 | + if (focusItem) { |
| 24 | + elem = focusItem; |
| 25 | + } else { |
| 26 | + elem = elem?.querySelector('[data-state="open"]'); |
| 27 | + } |
| 28 | + |
| 29 | + return elem; |
| 30 | + }; |
| 31 | + |
| 32 | + const focusNextItem = () => { |
| 33 | + const elem = getActiveItemId(); |
| 34 | + const nextElem = elem.nextElementSibling; |
| 35 | + setFocusItem(nextElem); |
| 36 | + // get button |
| 37 | + const button = nextElem.querySelector('button'); |
| 38 | + // focus button |
| 39 | + button?.focus(); |
| 40 | + }; |
| 41 | + const focusPrevItem = () => { |
| 42 | + const elem = getActiveItemId(); |
| 43 | + const prevElem = elem.previousElementSibling; |
| 44 | + setFocusItem(prevElem); |
| 45 | + // get button |
| 46 | + const button = prevElem.querySelector('button'); |
| 47 | + // focus button |
| 48 | + button?.focus(); |
| 49 | + }; |
| 50 | + |
12 | 51 | return ( |
13 | | - <span className={`${rootClass}-root`}> |
14 | | - {children} |
15 | | - </span> |
| 52 | + <AccordionContext.Provider |
| 53 | + value={{ |
| 54 | + rootClass: rootClass, |
| 55 | + activeItem, |
| 56 | + setActiveItem, |
| 57 | + focusNextItem, |
| 58 | + focusPrevItem, |
| 59 | + focusItem, |
| 60 | + setFocusItem, |
| 61 | + |
| 62 | + }}> |
| 63 | + <div className={`${rootClass}-root`} ref={accordionRef} > |
| 64 | + {children} |
| 65 | + </div> |
| 66 | + </AccordionContext.Provider> |
| 67 | + |
16 | 68 | ); |
17 | 69 | }; |
18 | 70 |
|
|
0 commit comments