From 96eacf3867c51987975b4ec5d7b2c8241d12718c Mon Sep 17 00:00:00 2001
From: Jacco van den Berg <jaccoberg2281@gmail.com>
Date: Tue, 22 Nov 2022 21:36:37 +0100
Subject: [PATCH] Allow color plugin to work with custom dataset controllers

---
 src/plugins/plugin.colors.ts | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/plugins/plugin.colors.ts b/src/plugins/plugin.colors.ts
index 8286c27c569..14292ca6cbf 100644
--- a/src/plugins/plugin.colors.ts
+++ b/src/plugins/plugin.colors.ts
@@ -1,3 +1,4 @@
+import {DoughnutController, PolarAreaController} from '../index.js';
 import type {Chart, ChartConfiguration, ChartDataset} from '../types.js';
 
 export interface ColorsPluginOptions {
@@ -49,17 +50,17 @@ function colorizePolarAreaDataset(dataset: ChartDataset, i: number) {
   return i;
 }
 
-function getColorizer(chartType: string) {
+function getColorizer(chart: Chart) {
   let i = 0;
 
-  return (dataset: ChartDataset) => {
-    const type = dataset.type || chartType;
+  return (dataset: ChartDataset, datasetIndex: number) => {
+    const controller = chart.getDatasetMeta(datasetIndex).controller;
 
-    if (type === 'doughnut' || type === 'pie') {
+    if (controller instanceof DoughnutController) {
       i = colorizeDoughnutDataset(dataset, i);
-    } else if (type === 'polarArea') {
+    } else if (controller instanceof PolarAreaController) {
       i = colorizePolarAreaDataset(dataset, i);
-    } else if (type) {
+    } else if (controller) {
       i = colorizeDefaultDataset(dataset, i);
     }
   };
@@ -101,7 +102,7 @@ export default {
       return;
     }
 
-    const colorizer = getColorizer(type);
+    const colorizer = getColorizer(chart);
 
     datasets.forEach(colorizer);
   }