From 8db24e901601a5072bd5ec7806f30ca5f52d948d Mon Sep 17 00:00:00 2001
From: Alex Holliday <ajhollid@gmail.com>
Date: Mon, 25 Nov 2024 13:00:28 +0800
Subject: [PATCH] refactor AreaChart to take an array of data keys

---
 .../src/Components/Charts/AreaChart/index.jsx | 20 +++++++++++--------
 .../Pages/Infrastructure/Details/index.jsx    |  2 +-
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/Client/src/Components/Charts/AreaChart/index.jsx b/Client/src/Components/Charts/AreaChart/index.jsx
index d645779ad..e3c231ebc 100644
--- a/Client/src/Components/Charts/AreaChart/index.jsx
+++ b/Client/src/Components/Charts/AreaChart/index.jsx
@@ -45,6 +45,7 @@ import { useId } from "react";
  *   return (
  *     <CustomAreaChart
  *       data={data}
+ *       dataKeys={["cpu.usage_percent"]}
  *       xKey="createdAt"
  * 		 xDomain={["2023-01-01T00:00:00Z", "2023-01-01T02:00:00Z"]}
  *       yKey="cpu.usage_percent"
@@ -73,7 +74,7 @@ import { useId } from "react";
  */
 const CustomAreaChart = ({
 	data,
-	dataKey,
+	dataKeys,
 	xKey,
 	xDomain,
 	yKey,
@@ -123,12 +124,15 @@ const CustomAreaChart = ({
 					fill="transparent"
 					vertical={false}
 				/>
-				<Area
-					type="monotone"
-					dataKey={dataKey}
-					stroke={strokeColor}
-					fill={gradient === true ? `url(#${gradientId})` : fillColor}
-				/>
+				{dataKeys.map((dataKey) => (
+					<Area
+						key={dataKey}
+						type="monotone"
+						dataKey={dataKey}
+						stroke={strokeColor}
+						fill={gradient === true ? `url(#${gradientId})` : fillColor}
+					/>
+				))}
 				{customTooltip ? (
 					<Tooltip
 						cursor={{ stroke: theme.palette.border.light }}
@@ -145,7 +149,7 @@ const CustomAreaChart = ({
 
 CustomAreaChart.propTypes = {
 	data: PropTypes.array.isRequired,
-	dataKey: PropTypes.string.isRequired,
+	dataKeys: PropTypes.array.isRequired,
 	xTick: PropTypes.object, // Recharts takes an instance of component, so we can't pass the component itself
 	yTick: PropTypes.object, // Recharts takes an instance of component, so we can't pass the component itself
 	xKey: PropTypes.string.isRequired,
diff --git a/Client/src/Pages/Infrastructure/Details/index.jsx b/Client/src/Pages/Infrastructure/Details/index.jsx
index 204f442ac..d90b54ba6 100644
--- a/Client/src/Pages/Infrastructure/Details/index.jsx
+++ b/Client/src/Pages/Infrastructure/Details/index.jsx
@@ -460,7 +460,7 @@ const InfrastructureDetails = () => {
 								<AreaChart
 									height={areaChartHeight}
 									data={monitor?.checks ?? []}
-									dataKey={config.dataKey}
+									dataKeys={[config.dataKey]}
 									xKey="createdAt"
 									yKey={config.dataKey}
 									yDomain={config.yDomain}