fix(doc): restart dev server on config changes

Author: chenjiahan

packages/rstack/src/rspressConfig.ts

@@ -1,3 +1,4 @@
+import type { WatchFiles } from '@rsbuild/core';
 import type { UserConfig } from '@rspress/core';
 import { loadRstackConfig, type Configs } from './config.ts';
 
@@ -13,6 +14,30 @@ const resolveRspressConfig = async (configs: Configs): Promise<UserConfig> => {
 };
 
 export default async (): Promise<UserConfig> => {
-  const { configs } = await loadRstackConfig();
-  return resolveRspressConfig(configs);
+  const { configs, filePath, dependencies } = await loadRstackConfig();
+  const config = await resolveRspressConfig(configs);
+
+  if (!filePath) {
+    return config;
+  }
+
+  const watchFiles = config.builderConfig?.dev?.watchFiles;
+  const watchConfig: WatchFiles = {
+    paths: [filePath, ...dependencies],
+    type: 'restart',
+  };
+
+  return {
+    ...config,
+    builderConfig: {
+      ...config.builderConfig,
+      dev: {
+        ...config.builderConfig?.dev,
+        watchFiles: [
+          ...(watchFiles ? (Array.isArray(watchFiles) ? watchFiles : [watchFiles]) : []),
+          watchConfig,
+        ],
+      },
+    },
+  };
 };

packages/rstack/tests/config/define-doc/index.test.ts

@@ -1,4 +1,6 @@
-import { getDistFiles, getFileContent } from '@rstackjs/test-utils';
+import { writeFile } from 'node:fs/promises';
+import path from 'node:path';
+import { getDistFiles, getFileContent, getRandomPort } from '@rstackjs/test-utils';
 import { test } from '#test-helpers';
 
 const expectedText = 'define.doc works';
@@ -13,3 +15,79 @@ test('should build docs with define.doc config', async ({ prepareDist, execCli,
 
   expect(output).toContain(expectedText);
 }, 30_000);
+
+test('should restart doc dev server when Rstack config changes', async ({
+  execCliAsync,
+  logHelper,
+}) => {
+  const configFile = path.join(import.meta.dirname, 'test-temp-rstack.config.ts');
+  const userWatchFile = path.join(import.meta.dirname, 'test-temp-user-watch.txt');
+
+  const writeConfig = (title: string) =>
+    writeFile(
+      configFile,
+      `import { define } from 'rstack';
+
+define.doc({
+  root: 'docs',
+  title: '${title}',
+  builderConfig: {
+    dev: {
+      watchFiles: {
+        paths: ${JSON.stringify(userWatchFile)},
+        type: 'restart',
+      },
+    },
+  },
+});
+`,
+    );
+
+  await writeFile(userWatchFile, 'initial\n');
+  await writeConfig('before config change');
+
+  execCliAsync(`doc --config test-temp-rstack.config.ts --port ${await getRandomPort()}`);
+  await logHelper.expectBuildEnd();
+  logHelper.clearLogs();
+
+  await writeConfig('after config change');
+
+  await logHelper.expectLog('restarting server as test-temp-rstack.config.ts changed');
+  await logHelper.expectBuildEnd();
+  logHelper.clearLogs();
+
+  await writeFile(userWatchFile, 'changed\n');
+
+  await logHelper.expectLog('restarting server as test-temp-user-watch.txt changed');
+  await logHelper.expectBuildEnd();
+}, 30_000);
+
+test('should restart doc dev server when an imported config file changes', async ({
+  execCliAsync,
+  logHelper,
+}) => {
+  const configFile = path.join(import.meta.dirname, 'test-temp-import.config.ts');
+  const importedFile = path.join(import.meta.dirname, 'test-temp-imported.ts');
+
+  await writeFile(importedFile, "export const title = 'before import change';\n");
+  await writeFile(
+    configFile,
+    `import { define } from 'rstack';
+import { title } from './test-temp-imported.ts';
+
+define.doc({
+  root: 'docs',
+  title,
+});
+`,
+  );
+
+  execCliAsync(`doc --config test-temp-import.config.ts --port ${await getRandomPort()}`);
+  await logHelper.expectBuildEnd();
+  logHelper.clearLogs();
+
+  await writeFile(importedFile, "export const title = 'after import change';\n");
+
+  await logHelper.expectLog('restarting server as test-temp-imported.ts changed');
+  await logHelper.expectBuildEnd();
+}, 30_000);