add legend to zoomed in plot

iterative · May 31, 2022 · fd81d2a · fd81d2a
1 parent 8de768e
commit fd81d2a
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/webview/src/plots/components/ZoomablePlot.tsx b/webview/src/plots/components/ZoomablePlot.tsx
@@ -31,7 +31,17 @@ export const ZoomablePlot: React.FC<ZoomablePlotOwnProps> = ({
     }
   }, [plotProps, id, renderZoomedInPlot])
 
-  const handleOnClick = () => renderZoomedInPlot(plotProps, id)
+  const handleOnClick = () => {
+    const zoomedPlotWithLegend = { ...plotProps } as VegaLiteProps & {
+      spec: { encoding?: { color?: { legend?: { disable?: boolean } } } }
+    }
+    if (
+      zoomedPlotWithLegend.spec.encoding?.color?.legend?.disable !== undefined
+    ) {
+      delete zoomedPlotWithLegend.spec.encoding.color.legend.disable
+    }
+    return renderZoomedInPlot(zoomedPlotWithLegend, id)
+  }
 
   return (
     <button className={styles.zoomablePlot} onClick={handleOnClick}>