Remove type augmentation application from charts codebase

Author: bernardobelchior

packages/x-charts/src/LineChart/useLinePlotData.ts

@@ -8,13 +8,7 @@ import { getCurveFactory } from '../internals/getCurve';
 import { ChartsXAxisProps, ChartsYAxisProps } from '../models';
 import { getValueToPositionMapper, useLineSeriesContext, useXAxes, useYAxes } from '../hooks';
 import { DEFAULT_X_AXIS_KEY } from '../constants';
-import { SeriesId } from '../models/seriesType/common';
-
-declare module '@mui/x-charts/typeAugmentation' {
-  interface ChartsEnabledFeature {
-    colorCallback: true;
-  }
-}
+import { SeriesColorProp, SeriesId } from '../models/seriesType/common';
 
 interface LinePlotDataPoint {
   d: string;
@@ -134,7 +128,8 @@ export function useLinePlotData(
           .y((d) => yScale(d.y[1])!);
 
         const d = linePath.curve(getCurveFactory(curve))(d3Data) || '';
-        const seriesColor = series[seriesId].color;
+        // FIXME: V9: remove this cast as it will no longer be necessary
+        const seriesColor = series[seriesId].color as SeriesColorProp<number | null>;
         linePlotData.push({
           color: typeof seriesColor === 'function' ? seriesColor(null) : seriesColor,
           gradientId,

packages/x-charts/src/ScatterChart/ScatterPlot.tsx

@@ -7,12 +7,7 @@ import { useXAxes, useYAxes } from '../hooks';
 import { useZAxes } from '../hooks/useZAxis';
 import { seriesConfig as scatterSeriesConfig } from './seriesConfig';
 import { BatchScatter } from './BatchScatter';
-
-declare module '@mui/x-charts/typeAugmentation' {
-  interface ChartsEnabledFeature {
-    colorCallback: true;
-  }
-}
+import { SeriesColorProp } from '../models/seriesType/common';
 
 export interface ScatterPlotSlots extends ScatterSlots {
   scatter?: React.JSXElementConstructor<ScatterProps>;
@@ -80,6 +75,9 @@ function ScatterPlot(props: ScatterPlotProps) {
       {seriesOrder.map((seriesId) => {
         const { id, xAxisId, yAxisId, zAxisId, color } = series[seriesId];
 
+        // FIXME: V9: remove this cast as it will no longer be necessary
+        const seriesColor = color as SeriesColorProp<number | null>;
+
         const colorGetter = scatterSeriesConfig.colorProcessor(
           series[seriesId],
           xAxis[xAxisId ?? defaultXAxisId],
@@ -93,7 +91,7 @@ function ScatterPlot(props: ScatterPlotProps) {
             key={id}
             xScale={xScale}
             yScale={yScale}
-            color={typeof color === 'function' ? color(null) : color}
+            color={typeof seriesColor === 'function' ? seriesColor(null) : seriesColor}
             colorGetter={colorGetter}
             series={series[seriesId]}
             onItemClick={onItemClick}