feat(calendar): add support for custom color scale (#703)

plouc · Oct 19, 2019 · 484d308 · 484d308
1 parent 050f0a9
commit 484d308
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/calendar/src/enhance.js b/packages/calendar/src/enhance.js
@@ -22,15 +22,16 @@ const commonEnhancers = [
     withTheme(),
     withDimensions(),
     withPropsOnChange(
-        ['data', 'minValue', 'maxValue', 'colors'],
-        ({ data, minValue, maxValue, colors }) => {
+        ['data', 'minValue', 'maxValue', 'colors', 'colorScale'],
+        ({ data, minValue, maxValue, colors, colorScale }) => {
+            if (colorScale) return { colorScale }
             const domain = computeDomain(data, minValue, maxValue)
 
-            const colorScale = scaleQuantize()
+            const defaultColorScale = scaleQuantize()
                 .domain(domain)
                 .range(colors)
 
-            return { colorScale }
+            return { colorScale: defaultColorScale }
         }
     ),
     withPropsOnChange(