fix: correct stats.color type to include fine-grained options (#11797)

* fix: correct  type

* fix: lint

Author: chenjiahan

package.json

@@ -33,7 +33,7 @@
     "lint:js": "pnpm run lint-ci:js --write",
     "lint-ci:js": "biome check --diagnostic-level=warn --no-errors-on-unmatched --max-diagnostics=none --error-on-warnings",
     "lint:rs": "node ./scripts/check_rust_dependency.cjs",
-    "lint:type": "rslint --config rslint.json --max-warnings=2601",
+    "lint:type": "rslint --config rslint.json --max-warnings=2602",
     "build:binding:dev": "pnpm --filter @rspack/binding run build:dev",
     "build:binding:debug": "pnpm --filter @rspack/binding run build:debug",
     "build:binding:ci": "pnpm --filter @rspack/binding run build:ci",

packages/rspack/etc/core.api.md

@@ -6582,6 +6582,7 @@ declare namespace rspackExports {
         Loader,
         SnapshotOptions,
         CacheOptions,
+        StatsColorOptions,
         StatsOptions,
         StatsValue,
         RspackPluginInstance,
@@ -7327,6 +7328,16 @@ type StatsChunkGroup = KnownStatsChunkGroup & Record<string, any>;
 // @public (undocumented)
 type StatsChunkOrigin = KnownStatsChunkOrigin & Record<string, any>;
 
+// @public (undocumented)
+export type StatsColorOptions = {
+    bold?: string;
+    cyan?: string;
+    green?: string;
+    magenta?: string;
+    red?: string;
+    yellow?: string;
+};
+
 // @public (undocumented)
 export type StatsCompilation = KnownStatsCompilation & Record<string, any>;
 
@@ -7404,7 +7415,7 @@ export type StatsOptions = {
     warningsCount?: boolean;
     errors?: boolean;
     errorsCount?: boolean;
-    colors?: boolean;
+    colors?: boolean | StatsColorOptions;
     hash?: boolean;
     version?: boolean;
     reasons?: boolean;

packages/rspack/src/config/adapter.ts

@@ -801,6 +801,6 @@ function getRawNode(node: Node): RawOptions["node"] {
 function getRawStats(stats: StatsValue): RawOptions["stats"] {
 	const statsOptions = normalizeStatsPreset(stats);
 	return {
-		colors: statsOptions.colors ?? false
+		colors: Boolean(statsOptions.colors)
 	};
 }

packages/rspack/src/config/types.ts

@@ -1751,6 +1751,33 @@ type ModuleFilterTypes =
 	| ModuleFilterItemTypes
 	| ModuleFilterItemTypes[];
 
+export type StatsColorOptions = {
+	/**
+	 * Custom color for bold text.
+	 */
+	bold?: string;
+	/**
+	 * Custom color for cyan text.
+	 */
+	cyan?: string;
+	/**
+	 * Custom color for green text.
+	 */
+	green?: string;
+	/**
+	 * Custom color for magenta text.
+	 */
+	magenta?: string;
+	/**
+	 * Custom color for red text.
+	 */
+	red?: string;
+	/**
+	 * Custom color for yellow text.
+	 */
+	yellow?: string;
+};
+
 /** Options for stats */
 export type StatsOptions = {
 	/**
@@ -1810,7 +1837,7 @@ export type StatsOptions = {
 	 * Enables or disables the use of colors in the output.
 	 * @default false
 	 */
-	colors?: boolean;
+	colors?: boolean | StatsColorOptions;
 	/**
 	 * Enables or disables the display of the hash.
 	 * @default true

packages/rspack/src/stats/DefaultStatsPrinterPlugin.ts

@@ -9,6 +9,7 @@
  */
 
 import type { Compiler } from "../Compiler";
+import type { StatsColorOptions } from "../config";
 import { compareIds } from "../util/comparators";
 import { formatSize } from "../util/SizeFormatHelpers";
 import type { StatsPrinter, StatsPrinterContext } from "./StatsPrinter";
@@ -1197,7 +1198,7 @@ const SIMPLE_ELEMENT_JOINERS: Record<
 	moduleTraceDependency: joinOneLine
 };
 
-const AVAILABLE_COLORS = {
+const AVAILABLE_COLORS: Record<keyof StatsColorOptions, string> = {
 	bold: "\u001b[1m",
 	yellow: "\u001b[1m\u001b[33m",
 	red: "\u001b[1m\u001b[31m",
@@ -1359,8 +1360,13 @@ export class DefaultStatsPrinterPlugin {
 						"DefaultStatsPrinterPlugin",
 						// @ts-expect-error
 						(compilation: StatsCompilation, context) => {
-							for (const color of Object.keys(AVAILABLE_COLORS)) {
+							const colorNames = Object.keys(
+								AVAILABLE_COLORS
+							) as (keyof typeof AVAILABLE_COLORS)[];
+
+							for (const color of colorNames) {
 								let start: string | undefined;
+
 								if (options.colors) {
 									if (
 										typeof options.colors === "object" &&
@@ -1372,6 +1378,7 @@ export class DefaultStatsPrinterPlugin {
 											AVAILABLE_COLORS[color as keyof typeof AVAILABLE_COLORS];
 									}
 								}
+
 								if (start) {
 									context[color] = (str: string) =>
 										`${start}${
@@ -1386,12 +1393,14 @@ export class DefaultStatsPrinterPlugin {
 									context[color] = (str: string) => str;
 								}
 							}
+
 							for (const format of Object.keys(AVAILABLE_FORMATS)) {
 								// @ts-expect-error
 								context[format] = (content, ...args) =>
 									// @ts-expect-error
 									AVAILABLE_FORMATS[format](content, context, ...args);
 							}
+
 							context.timeReference = compilation.time;
 						}
 					);