From 104e68de169e3a3a026d37ebb90b29e8c5884ccb Mon Sep 17 00:00:00 2001
From: Khalah Jones-Golden <khasan222@gmail.com>
Date: Mon, 9 May 2016 09:41:53 -0400
Subject: [PATCH] Chart js work

---
 src/ui/public/vislib/styles/_legend.less |  4 ++++
 src/ui/public/visualize/visualize.html   |  8 +++++---
 src/ui/public/visualize/visualize.js     | 23 +++++++++++++++++------
 3 files changed, 26 insertions(+), 9 deletions(-)

diff --git a/src/ui/public/vislib/styles/_legend.less b/src/ui/public/vislib/styles/_legend.less
index ddd2ae39d7785..fcfa67cae294d 100644
--- a/src/ui/public/vislib/styles/_legend.less
+++ b/src/ui/public/vislib/styles/_legend.less
@@ -20,6 +20,10 @@ visualize-legend {
   flex-direction: row;
   padding-top: 5px;
 
+  &.fixed-width {
+    width: 200px;
+  }
+
   .header {
     cursor: pointer;
     width: 15px;
diff --git a/src/ui/public/visualize/visualize.html b/src/ui/public/visualize/visualize.html
index 9a19cbbdf527d..895457a142d21 100644
--- a/src/ui/public/visualize/visualize.html
+++ b/src/ui/public/visualize/visualize.html
@@ -13,8 +13,10 @@ <h4>No results found</h4>
     class="visualize-chart"></div>
   <visualize-legend></visualize-legend>
 </div>
-<canvas id="canvas-chart">
-</canvas>
-<div id="chart-legend"></div>
+<div class="visualize-chart">
+  <canvas id="canvas-chart" class="visualize-chart">
+  </canvas>
+  <div id="chart-legend" class="fixed-width legend-col-wrapper"></div>
+</div>
 <!-- <pre>{{chartData | json}}</pre> -->
 <visualize-spy ng-if="vis.type.requiresSearch && showSpyPanel"></visualize-spy>
diff --git a/src/ui/public/visualize/visualize.js b/src/ui/public/visualize/visualize.js
index 22a246b1906e1..7f7b01b0986b2 100644
--- a/src/ui/public/visualize/visualize.js
+++ b/src/ui/public/visualize/visualize.js
@@ -33,6 +33,7 @@ uiModules
       }
       return '#' + hexStr;
     }
+    const isPieChart = (chartType === 'pie');
     function decodeBucketData(buckets, aggregations) {
       const chartDatasetConfigs = {};
       if (!buckets) { return chartDatasetConfigs; }
@@ -64,17 +65,21 @@ uiModules
       decodeBucket(buckets[0], aggregations[buckets[0].id]);
       const arrDatasets = buckets.map(bucket => { return chartDatasetConfigs[bucket.id]; });
       // Make some colors for all of the data points.
+      // This need to be different, instead of looking at all the colors
+      // you should look at RGB separate and limit the number from there
+      // then multiply to get your result
       arrDatasets.forEach(set => {
-        const colorOffset = 10000;
+        const allTheColors = Math.pow(16, 6);
+        const colorOffset = allTheColors / 8;
         let numDataPoints = set.data.length;
-        const maxColors = Math.pow(16,6) - (colorOffset * 2);
+        const maxColors = allTheColors - (colorOffset * 2);
         const difference = maxColors / numDataPoints;
         let currColor = colorOffset;
-        do {
+        while (numDataPoints-- > 0) {
           set.backgroundColor.push(makeColor(currColor));
           currColor += difference;
-        } while (--numDataPoints > 0);
-        if (chartType !== 'pie') {
+        }
+        if (!isPieChart) {
           set.backgroundColor = set.backgroundColor[0];
         }
       });
@@ -86,7 +91,7 @@ uiModules
       const aggConfigMap = aggConfigs.byId;
       const decodedData = decodeBucketData(aggConfigs.bySchemaGroup.buckets, esResp.aggregations);
       const flattenedLabels = _.reduce(decodedData, (prev, dataset) => {
-        return prev.concat(dataset.labels);
+        return _.uniq(prev.concat(dataset.labels));
       }, []);
       if (myChart) { // Not a fan of this, i should be using update
         myChart.destroy();
@@ -98,6 +103,12 @@ uiModules
           datasets: decodedData
         },
         options: {
+          legendCallback: function (chartObj) {
+            const multipleBuckets = aggConfigs.bySchemaGroup.buckets.length > 1;
+            const legendItems = multipleBuckets || isPieChart ? flattenedLabels : [aggConfigs.bySchemaGroup.metrics[0]._opts.type];
+            const itemsHtmlArr = legendItems.map(item => { return '<li>' + item + '</li>'; });
+            return '<ul>' + itemsHtmlArr.join('') + '</ul>';
+          },
           legend: {
             display: false
           },