From 3f9a454ea05d9b6d260d655d939072ffdfb4e1c4 Mon Sep 17 00:00:00 2001
From: Marta Bondyra <marta.bondyra@elastic.co>
Date: Mon, 28 Oct 2024 11:03:01 +0100
Subject: [PATCH] wip

---
 dev_docs/tutorials/advanced_settings.mdx                  | 1 -
 src/plugins/vis_types/vislib/public/vislib/lib/data.js    | 8 ++------
 .../vis_types/vislib/public/vislib/lib/vis_config.js      | 4 ++--
 src/plugins/vis_types/vislib/public/vislib/vis.js         | 8 +-------
 4 files changed, 5 insertions(+), 16 deletions(-)

diff --git a/dev_docs/tutorials/advanced_settings.mdx b/dev_docs/tutorials/advanced_settings.mdx
index 7c684e1c75c1c..6789e9d6af7ad 100644
--- a/dev_docs/tutorials/advanced_settings.mdx
+++ b/dev_docs/tutorials/advanced_settings.mdx
@@ -108,7 +108,6 @@ export class ChartsPlugin implements Plugin<ChartsPluginSetup, ChartsPluginStart
     };
   }
 }
-
 ```
 
 ### Server side usage
diff --git a/src/plugins/vis_types/vislib/public/vislib/lib/data.js b/src/plugins/vis_types/vislib/public/vislib/lib/data.js
index 94c39c95930af..83c6167d40923 100644
--- a/src/plugins/vis_types/vislib/public/vislib/lib/data.js
+++ b/src/plugins/vis_types/vislib/public/vislib/lib/data.js
@@ -45,15 +45,11 @@ class D3MappableObject {
  * @param attr {Object|*} Visualization options
  */
 export class Data {
-  constructor(data, uiState, createColorLookupFunction) {
+  constructor(data, uiState) {
     this.uiState = uiState;
-    this.createColorLookupFunction = createColorLookupFunction;
     this.data = this.copyDataObj(data);
     this.type = this.getDataType();
     this.labels = this._getLabels(this.data);
-    this.color = this.labels
-      ? createColorLookupFunction(this.labels, uiState.get('vis.colors'))
-      : undefined;
     this._normalizeOrdered();
   }
 
@@ -385,7 +381,7 @@ export class Data {
     const defaultColors = this.uiState.get('vis.defaultColors');
     const overwriteColors = this.uiState.get('vis.colors');
     const colors = defaultColors ? _.defaults({}, overwriteColors, defaultColors) : overwriteColors;
-    return this.createColorLookupFunction(this.getLabels(), colors);
+    return (value) => colors[value];
   }
 
   /**
diff --git a/src/plugins/vis_types/vislib/public/vislib/lib/vis_config.js b/src/plugins/vis_types/vislib/public/vislib/lib/vis_config.js
index bb25a8ddaada4..28d0d0c67c277 100644
--- a/src/plugins/vis_types/vislib/public/vislib/lib/vis_config.js
+++ b/src/plugins/vis_types/vislib/public/vislib/lib/vis_config.js
@@ -25,8 +25,8 @@ const DEFAULT_VIS_CONFIG = {
 };
 
 export class VisConfig {
-  constructor(visConfigArgs, data, uiState, el, createColorLookupFunction) {
-    this.data = new Data(data, uiState, createColorLookupFunction);
+  constructor(visConfigArgs, data, uiState, el) {
+    this.data = new Data(data, uiState);
 
     const visType = visTypes[visConfigArgs.type];
     const typeDefaults = visType(visConfigArgs, this.data);
diff --git a/src/plugins/vis_types/vislib/public/vislib/vis.js b/src/plugins/vis_types/vislib/public/vislib/vis.js
index 80f7393d69af2..35241d999f2f2 100644
--- a/src/plugins/vis_types/vislib/public/vislib/vis.js
+++ b/src/plugins/vis_types/vislib/public/vislib/vis.js
@@ -41,13 +41,7 @@ export class Vis extends EventEmitter {
   initVisConfig(data, uiState) {
     this.data = data;
     this.uiState = uiState;
-    this.visConfig = new VisConfig(
-      this.visConfigArgs,
-      this.data,
-      this.uiState,
-      this.element,
-      this.charts.legacyColors.createColorLookupFunction.bind(this.charts.legacyColors)
-    );
+    this.visConfig = new VisConfig(this.visConfigArgs, this.data, this.uiState, this.element);
   }
 
   /**