feat: migrate state plugin

web-infra-dev · Dec 12, 2024 · 950a386 · 950a386
1 parent 0e1b9d9
commit 950a386
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 36 deletions.
diff --git a/packages/runtime/plugin-runtime/src/cli/index.ts b/packages/runtime/plugin-runtime/src/cli/index.ts
@@ -31,8 +31,8 @@ export const runtimePlugin = (params?: {
   usePlugins: params?.plugins || [
     ssrPlugin(),
     routerPlugin(),
-    statePlugin() as any,
-    documentPlugin(),
+    statePlugin(),
+    documentPlugin() as any,
   ],
   setup: api => {
     api.checkEntryPoint(({ path, entry }) => {

diff --git a/packages/runtime/plugin-runtime/src/state/cli/index.ts b/packages/runtime/plugin-runtime/src/state/cli/index.ts
@@ -1,48 +1,46 @@
-import type { AppTools, CliPlugin } from '@modern-js/app-tools';
+import type { AppTools, CliPluginFuture } from '@modern-js/app-tools';
 import { createRuntimeExportsUtils, getEntryOptions } from '@modern-js/utils';
 
 const PLUGIN_IDENTIFIER = 'state';
 
-export const statePlugin = (): CliPlugin<AppTools> => ({
+export const statePlugin = (): CliPluginFuture<AppTools> => ({
   name: '@modern-js/plugin-state',
 
   required: ['@modern-js/runtime'],
 
   setup: api => {
-    return {
-      _internalRuntimePlugins({ entrypoint, plugins }) {
-        const { entryName, isMainEntry } = entrypoint;
-        const userConfig = api.useResolvedConfigContext();
-        const { packageName, metaName } = api.useAppContext();
+    api._internalRuntimePlugins(({ entrypoint, plugins }) => {
+      const { entryName, isMainEntry } = entrypoint;
+      const userConfig = api.useResolvedConfigContext();
+      const { packageName, metaName } = api.useAppContext();
 
-        const stateConfig = getEntryOptions(
-          entryName,
-          isMainEntry,
-          userConfig.runtime,
-          userConfig.runtimeByEntries,
-          packageName,
-        )?.state;
-        if (stateConfig) {
-          plugins.push({
-            name: PLUGIN_IDENTIFIER,
-            path: `@${metaName}/runtime/model`,
-            config: typeof stateConfig === 'boolean' ? {} : stateConfig,
-          });
-        }
-        return { entrypoint, plugins };
-      },
-      addRuntimeExports() {
-        const { internalDirectory, metaName } = api.useAppContext();
+      const stateConfig = getEntryOptions(
+        entryName,
+        isMainEntry,
+        userConfig.runtime,
+        userConfig.runtimeByEntries,
+        packageName,
+      )?.state;
+      if (stateConfig) {
+        plugins.push({
+          name: PLUGIN_IDENTIFIER,
+          path: `@${metaName}/runtime/model`,
+          config: typeof stateConfig === 'boolean' ? {} : stateConfig,
+        });
+      }
+      return { entrypoint, plugins };
+    });
+    api.addRuntimeExports(() => {
+      const { internalDirectory, metaName } = api.useAppContext();
 
-        const pluginsExportsUtils = createRuntimeExportsUtils(
-          internalDirectory,
-          'plugins',
-        );
-        pluginsExportsUtils.addExport(
-          `export { default as state } from '@${metaName}/runtime/model'`,
-        );
-      },
-    };
+      const pluginsExportsUtils = createRuntimeExportsUtils(
+        internalDirectory,
+        'plugins',
+      );
+      pluginsExportsUtils.addExport(
+        `export { default as state } from '@${metaName}/runtime/model'`,
+      );
+    });
   },
 });