storybook-builder-rsbuild 0.0.9 → 0.0.10

package/dist/index.d.ts

@@ -29,6 +29,10 @@ type BuilderOptions = {
      * Enable Rspack's lazy compilation (experimental).
      */
     lazyCompilation?: boolean;
+    /**
+     * Which environment to use from the Rsbuild config.
+     */
+    environment?: string;
 };
 interface BuilderResult extends BuilderResult$1 {
     stats?: Stats;

package/dist/index.js

@@ -372,14 +372,46 @@ var iframe_rsbuild_config_default = async (options) => {
   if (!options.cache) {
     throw new Error("Cache is required");
   }
+  let contentFromConfig = {};
   const { content } = await (0, import_core.loadConfig)({
     cwd: workingDir,
     path: rsbuildConfigPath
   });
-  content.source ??= {};
-  content.source.entry = {};
+  const { environments, ...withoutEnv } = content;
+  if (content.environments) {
+    const envCount = Object.keys(content.environments).length;
+    if (envCount === 0) {
+      contentFromConfig = withoutEnv;
+    } else if (envCount === 1) {
+      contentFromConfig = (0, import_core.mergeRsbuildConfig)(
+        withoutEnv,
+        content.environments[0]
+      );
+    } else {
+      const userEnv = builderOptions.environment;
+      if (typeof userEnv !== "string") {
+        throw new Error(
+          "You must specify an environment when there are multiple environments in the Rsbuild config."
+        );
+      }
+      if (Object.keys(content.environments).includes(userEnv)) {
+        contentFromConfig = (0, import_core.mergeRsbuildConfig)(
+          withoutEnv,
+          content.environments[userEnv]
+        );
+      } else {
+        throw new Error(
+          `The specified environment "${userEnv}" is not found in the Rsbuild config.`
+        );
+      }
+    }
+  } else {
+    contentFromConfig = content;
+  }
+  contentFromConfig.source ??= {};
+  contentFromConfig.source.entry = {};
   const resourceFilename = isProd ? "static/media/[name].[contenthash:8][ext]" : "static/media/[path][name][ext]";
-  const merged = (0, import_core.mergeRsbuildConfig)(content, {
+  const merged = (0, import_core.mergeRsbuildConfig)(contentFromConfig, {
     output: {
       cleanDistPath: false,
       dataUriLimit: {

package/dist/index.mjs

@@ -315,14 +315,46 @@ var iframe_rsbuild_config_default = async (options) => {
   if (!options.cache) {
     throw new Error("Cache is required");
   }
+  let contentFromConfig = {};
   const { content } = await loadConfig({
     cwd: workingDir,
     path: rsbuildConfigPath
   });
-  content.source ??= {};
-  content.source.entry = {};
+  const { environments, ...withoutEnv } = content;
+  if (content.environments) {
+    const envCount = Object.keys(content.environments).length;
+    if (envCount === 0) {
+      contentFromConfig = withoutEnv;
+    } else if (envCount === 1) {
+      contentFromConfig = mergeRsbuildConfig(
+        withoutEnv,
+        content.environments[0]
+      );
+    } else {
+      const userEnv = builderOptions.environment;
+      if (typeof userEnv !== "string") {
+        throw new Error(
+          "You must specify an environment when there are multiple environments in the Rsbuild config."
+        );
+      }
+      if (Object.keys(content.environments).includes(userEnv)) {
+        contentFromConfig = mergeRsbuildConfig(
+          withoutEnv,
+          content.environments[userEnv]
+        );
+      } else {
+        throw new Error(
+          `The specified environment "${userEnv}" is not found in the Rsbuild config.`
+        );
+      }
+    }
+  } else {
+    contentFromConfig = content;
+  }
+  contentFromConfig.source ??= {};
+  contentFromConfig.source.entry = {};
   const resourceFilename = isProd ? "static/media/[name].[contenthash:8][ext]" : "static/media/[path][name][ext]";
-  const merged = mergeRsbuildConfig(content, {
+  const merged = mergeRsbuildConfig(contentFromConfig, {
     output: {
       cleanDistPath: false,
       dataUriLimit: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "storybook-builder-rsbuild",
-  "version": "0.0.9",
+  "version": "0.0.10",
   "description": "Rsbuild builder for Storybook",
   "keywords": [
     "storybook",