feat(Modal): focus to modal container if controllable element doesn't focused

packages/vkui/src/components/ModalCard/ModalCard.e2e-playground.tsx

@@ -94,7 +94,12 @@ export const ModalCardPlayground = (props: ComponentPlaygroundProps) => {
     <ComponentPlayground {...props} propSets={propSets} AppWrapper={AppWrapper}>
       {(props: ModalCardProps) => (
         <div style={{ height: 500, transform: 'translateZ(0)' }}>
-          <ModalRoot activeModal={props.nav}>
+          <ModalRoot
+            activeModal={props.nav}
+            // Note: с включенным фокусом ломаются скриншоты на движке Webkit из-за фокуса сразу
+            // на несколько окон
+            noFocusToDialog
+          >
             <ModalCard {...props} />
           </ModalRoot>
         </div>

packages/vkui/src/components/ModalCard/ModalCard.module.css

@@ -10,6 +10,10 @@
   align-items: flex-end;
 }
 
+.ModalCard:focus {
+  outline: none;
+}
+
 .ModalCard__in {
   width: 100%;
   margin-left: auto;

packages/vkui/src/components/ModalCard/ModalCard.test.tsx

@@ -5,7 +5,11 @@ import { ConfigProvider } from '../ConfigProvider/ConfigProvider';
 import { ModalCard } from './ModalCard';
 
 describe('ModalCard', () => {
-  baselineComponent((p) => <ModalCard nav="id" {...p} />);
+  baselineComponent((p) => <ModalCard nav="id" {...p} />, {
+    // TODO [a11y]: "ARIA dialog and alertdialog nodes should have an accessible name (aria-dialog-name)"
+    //              https://dequeuniversity.com/rules/axe/4.5/aria-dialog-name?application=axeAPI
+    a11y: false,
+  });
 
   test('testid for modal card content', () => {
     const { rerender } = render(<ModalCard nav="id" />);

packages/vkui/src/components/ModalCard/ModalCard.tsx

@@ -1,6 +1,7 @@
 import * as React from 'react';
 import { classNames } from '@vkontakte/vkjs';
 import { useAdaptivityWithJSMediaQueries } from '../../hooks/useAdaptivityWithJSMediaQueries';
+import { useExternRef } from '../../hooks/useExternRef';
 import { usePlatform } from '../../hooks/usePlatform';
 import { getNavId, NavIdProps } from '../../lib/getNavId';
 import { warnOnce } from '../../lib/warnOnce';
@@ -34,17 +35,26 @@ export const ModalCard = ({
   id,
   size,
   modalDismissButtonTestId,
+  getRootRef,
   ...restProps
 }: ModalCardProps) => {
   const { isDesktop } = useAdaptivityWithJSMediaQueries();
   const platform = usePlatform();
 
   const modalContext = React.useContext(ModalRootContext);
   const { refs } = useModalRegistry(getNavId({ nav, id }, warn), ModalType.CARD);
+  const rootRef = useExternRef(getRootRef, refs.modalElement);
+
+  const contextValue = React.useMemo(() => ({ labelId: `${id}-label` }), [id]);
 
   return (
     <RootComponent
       {...restProps}
+      getRootRef={rootRef}
+      tabIndex={-1}
+      role="dialog"
+      aria-modal="true"
+      aria-labelledby={contextValue.labelId}
       id={id}
       baseClassName={classNames(
         styles['ModalCard'],

packages/vkui/src/components/ModalPage/ModalPage.module.css

@@ -7,6 +7,10 @@
   pointer-events: none;
 }
 
+.ModalPage:focus {
+  outline: none;
+}
+
 .ModalPage--desktop {
   display: flex;
   justify-content: center;

packages/vkui/src/components/ModalPage/ModalPage.tsx

@@ -1,6 +1,7 @@
 import * as React from 'react';
 import { classNames } from '@vkontakte/vkjs';
 import { useAdaptivityWithJSMediaQueries } from '../../hooks/useAdaptivityWithJSMediaQueries';
+import { useExternRef } from '../../hooks/useExternRef';
 import { useId } from '../../hooks/useId';
 import { useOrientationChange } from '../../hooks/useOrientationChange';
 import { usePlatform } from '../../hooks/usePlatform';
@@ -96,6 +97,7 @@ export const ModalPage = ({
   height,
   modalContentTestId,
   modalDismissButtonTestId,
+  getRootRef,
   ...restProps
 }: ModalPageProps) => {
   const generatingId = useId();
@@ -118,13 +120,16 @@ export const ModalPage = ({
 
   const modalContext = React.useContext(ModalRootContext);
   const { refs } = useModalRegistry(getNavId({ nav, id }, warn), ModalType.PAGE);
+  const rootRef = useExternRef(getRootRef, refs.modalElement);
 
   const contextValue = React.useMemo(() => ({ labelId: `${id}-label` }), [id]);
 
   return (
     <ModalPageContext.Provider value={contextValue}>
       <RootComponent
         {...restProps}
+        getRootRef={rootRef}
+        tabIndex={-1}
         role="dialog"
         aria-modal="true"
         aria-labelledby={contextValue.labelId}

packages/vkui/src/components/ModalRoot/ModalRoot.test.tsx

@@ -179,4 +179,75 @@ describe.each([
     );
     expect(screen.queryByTestId('modal-mask')).toBeTruthy();
   });
+
+  describe('focus', () => {
+    let modalPageRef: React.RefObject<HTMLDivElement> = React.createRef<HTMLDivElement>();
+    let modalCardRef: React.RefObject<HTMLDivElement> = React.createRef<HTMLDivElement>();
+    let modalPageWithInputRef: React.RefObject<HTMLDivElement> = React.createRef<HTMLDivElement>();
+    let inputInnerModalPageRef: React.RefObject<HTMLInputElement> =
+      React.createRef<HTMLInputElement>();
+    let modals: React.ReactElement[] = [];
+
+    beforeEach(() => {
+      modalPageRef = React.createRef<HTMLDivElement>();
+      modalCardRef = React.createRef<HTMLDivElement>();
+      modalPageWithInputRef = React.createRef<HTMLDivElement>();
+      inputInnerModalPageRef = React.createRef<HTMLInputElement>();
+      modals = [
+        <ModalPage id="modal-page" key="1" getRootRef={modalPageRef} />,
+        <ModalCard id="modal-card" key="2" getRootRef={modalCardRef} />,
+        <ModalPage id="modal-page-with-input" key="3" getRootRef={modalPageWithInputRef}>
+          <input ref={inputInnerModalPageRef} autoFocus />
+        </ModalPage>,
+      ];
+    });
+
+    it('should focus on modal container if controllable element does not exist or is not focused', () => {
+      const component = render(<ModalRoot activeModal="modal-page">{modals}</ModalRoot>, {
+        baseElement: document.documentElement,
+      });
+      runAllTimers();
+
+      expect(modalPageRef.current).toHaveFocus();
+
+      component.rerender(<ModalRoot activeModal="modal-card">{modals}</ModalRoot>);
+      runAllTimers();
+
+      expect(modalCardRef.current).toHaveFocus();
+    });
+
+    it('should focus on controllable element instead modal if it is focused', () => {
+      render(<ModalRoot activeModal="modal-page-with-input">{modals}</ModalRoot>, {
+        baseElement: document.documentElement,
+      });
+      runAllTimers();
+
+      expect(modalPageWithInputRef.current).not.toHaveFocus();
+      expect(inputInnerModalPageRef.current).toHaveFocus();
+    });
+
+    it('should focus on controllable element only (if it is focused)', () => {
+      const component = render(
+        <ModalRoot activeModal="modal-page" noFocusToDialog>
+          {modals}
+        </ModalRoot>,
+        {
+          baseElement: document.documentElement,
+        },
+      );
+      runAllTimers();
+
+      expect(modalPageRef.current).not.toHaveFocus();
+
+      component.rerender(
+        <ModalRoot activeModal="modal-page-with-input" noFocusToDialog>
+          {modals}
+        </ModalRoot>,
+      );
+      runAllTimers();
+
+      expect(modalPageWithInputRef.current).not.toHaveFocus();
+      expect(inputInnerModalPageRef.current).toHaveFocus();
+    });
+  });
 });

packages/vkui/src/components/ModalRoot/ModalRoot.tsx

@@ -107,14 +107,15 @@ class ModalRootTouchComponent extends React.Component<
 
     // transition phase 3: animate entering modal
     if (this.props.enteringModal && this.props.enteringModal !== prevProps.enteringModal) {
-      const { enteringModal } = this.props;
-      const enteringState = this.props.getModalState(enteringModal);
+      const enteringState = this.props.getModalState(this.props.enteringModal);
       this.props.onEnter();
       this.waitTransitionFinish(enteringState, () => {
-        if (enteringState?.innerElement) {
-          enteringState.innerElement.style.transitionDelay = '';
+        if (enteringState) {
+          if (enteringState.innerElement) {
+            enteringState.innerElement.style.transitionDelay = '';
+          }
+          this.onEntered(enteringState);
         }
-        this.props.onEntered(enteringModal);
       });
 
       if (enteringState?.innerElement) {
@@ -219,6 +220,18 @@ class ModalRootTouchComponent extends React.Component<
     }
   };
 
+  onEntered({ id, modalElement }: ModalsStateEntry) {
+    if (
+      !this.props.noFocusToDialog &&
+      modalElement &&
+      !modalElement.contains(this.document.activeElement)
+    ) {
+      modalElement.focus();
+    }
+
+    this.props.onEntered(id);
+  }
+
   closeModal(id: string) {
     // Сбрасываем состояния, которые могут помешать закрытию модального окна
     this.setState({ touchDown: false });
@@ -599,12 +612,6 @@ class ModalRootTouchComponent extends React.Component<
                 return (
                   <FocusTrap
                     key={key}
-                    getRootRef={(e) => {
-                      const modalState = this.props.getModalState(modalId);
-                      if (modalState) {
-                        modalState.modalElement = e;
-                      }
-                    }}
                     onClose={this.props.onExit}
                     timeout={this.timeout}
                     className={classNames(

packages/vkui/src/components/ModalRoot/ModalRootDesktop.tsx

@@ -20,6 +20,7 @@ const warn = warnOnce('ModalRoot');
 export const ModalRootDesktop = ({
   activeModal: activeModalProp,
   children,
+  noFocusToDialog = false,
   onOpen,
   onOpened,
   onClose,
@@ -39,7 +40,7 @@ export const ModalRootDesktop = ({
     getModalState,
     enteringModal,
     onEnter,
-    onEntered,
+    onEntered: onEnteredProp,
     onExited,
     history,
     delayEnter,
@@ -92,6 +93,14 @@ export const ModalRootDesktop = ({
     });
   };
 
+  const onEntered = ({ id, modalElement }: ModalsStateEntry) => {
+    if (!noFocusToDialog && modalElement && !modalElement.contains(document!.activeElement)) {
+      modalElement.focus();
+    }
+
+    onEnteredProp(id);
+  };
+
   const openModal = () => {
     if (!enteringModal || !prevProps) {
       return;
@@ -103,16 +112,10 @@ export const ModalRootDesktop = ({
     // Анимация открытия модального окна
     if (!prevProps.exitingModal) {
       requestAnimationFrame(() => {
-        if (enteringModal === enteringModal) {
-          waitTransitionFinish(
-            enteringState?.innerElement,
-            () => onEntered(enteringModal),
-            timeout,
-          );
+        if (enteringModal === enteringModal && enteringState) {
+          waitTransitionFinish(enteringState.innerElement, () => onEntered(enteringState), timeout);
           animateModalOpacity(enteringState, true);
-          if (enteringState) {
-            setMaskOpacity(enteringState, 1);
-          }
+          setMaskOpacity(enteringState, 1);
         }
       });
 
@@ -128,7 +131,9 @@ export const ModalRootDesktop = ({
       }
     });
 
-    onEntered(enteringModal);
+    if (enteringState) {
+      onEntered(enteringState);
+    }
   };
 
   const closeModal = (id: string) => {

packages/vkui/src/components/ModalRoot/types.ts

@@ -99,6 +99,11 @@ export interface ModalRootProps {
    * `data-testid` для маски
    */
   modalOverlayTestId?: string;
+
+  /**
+   * Отключает фокус на контейнер диалогового окна при открытии.
+   */
+  noFocusToDialog?: boolean;
 }
 
 export interface ModalRootWithDOMProps extends HasPlatform, ModalRootProps, DOMContextInterface {