fix(storybook): add focus on story iframe

gravity-ui · Sep 29, 2023 · e83e8ce · e83e8ce
1 parent 3352042
commit e83e8ce
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/.storybook/focus-addon/register.tsx b/.storybook/focus-addon/register.tsx
@@ -0,0 +1,11 @@
+import {addons, } from '@storybook/addons';
+
+addons.register('initialFocus', () => {
+    addons.getChannel().on('currentStoryWasSet', () => focusIframe());
+})
+
+const focusIframe = () => {
+    const previewIframe: HTMLInputElement | null = document.querySelector('#storybook-preview-iframe');
+
+    previewIframe?.focus?.();
+  }