Avoid top-level await

storybookjs · Jun 22, 2022 · 7d828e3 · 7d828e3
1 parent fca14d2
commit 7d828e3
Showing 1 changed file with 19 additions and 13 deletions.
diff --git a/packages/builder-vite/codegen-iframe-script.ts b/packages/builder-vite/codegen-iframe-script.ts
@@ -22,18 +22,20 @@ export async function generateIframeScriptCode(options: ExtendedOptions) {
     // is loaded. That way our client-apis can assume the existence of the API+store
     import { configure } from '${frameworkImportPath}';
 
-    import {
+    import * as clientApi from "@storybook/client-api";
+    import { logger } from '@storybook/client-logger';
+    ${absoluteFilesToImport(configEntries, 'config')}
+    import * as preview from '${virtualPreviewFile}';
+    import { configStories } from '${virtualStoriesFile}';
+
+    const {
       addDecorator,
       addParameters,
       addLoader,
       addArgTypesEnhancer,
       addArgsEnhancer,
-      setGlobalRender
-    } from '@storybook/client-api';
-    import { logger } from '@storybook/client-logger';
-    ${absoluteFilesToImport(configEntries, 'config')}
-    import * as preview from '${virtualPreviewFile}';
-    import { configStories } from '${virtualStoriesFile}';
+      setGlobalRender,
+    } = clientApi;
 
     const configs = [${importArray('config', configEntries.length).concat('preview.default').join(',')}].filter(Boolean)
 
@@ -49,17 +51,21 @@ export async function generateIframeScriptCode(options: ExtendedOptions) {
         const value = config[key];
         switch (key) {
           case 'args': {
-            if (typeof addArgs !== 'undefined') {
-              return addArgs(value);
+            if (typeof clientApi.addArgs !== "undefined") {
+              return clientApi.addArgs(value);
             } else {
-              return logger.warn('Could not add global args. Please open an issue in storybookjs/builder-vite.');
+              return logger.warn(
+                "Could not add global args. Please open an issue in storybookjs/builder-vite."
+              );
             }
           }
           case 'argTypes': {
-            if (typeof addArgTypes !== 'undefined') {
-              return addArgTypes(value);
+            if (typeof clientApi.addArgTypes !== "undefined") {
+              return clientApi.addArgTypes(value);
             } else {
-              return logger.warn('Could not add global argTypes. Please open an issue in storybookjs/builder-vite.');
+              return logger.warn(
+                "Could not add global argTypes. Please open an issue in storybookjs/builder-vite."
+              );
             }
           }
           case 'decorators': {