From b5ed7cd555b71c4057ffdde7e5e4cf341745ab11 Mon Sep 17 00:00:00 2001
From: Henry Dollman <hank@henrygd.me>
Date: Wed, 6 Nov 2024 13:52:35 -0500
Subject: [PATCH] fix display of values under 1 GB in disk usage chart (closes
 #261)

---
 beszel/site/src/components/charts/disk-chart.tsx | 5 +++++
 beszel/site/src/components/routes/system.tsx     | 8 ++------
 beszel/site/src/lib/utils.ts                     | 2 +-
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/beszel/site/src/components/charts/disk-chart.tsx b/beszel/site/src/components/charts/disk-chart.tsx
index 7533022..3617811 100644
--- a/beszel/site/src/components/charts/disk-chart.tsx
+++ b/beszel/site/src/components/charts/disk-chart.tsx
@@ -27,6 +27,11 @@ export default memo(function DiskChart({
 	const { yAxisWidth, updateYAxisWidth } = useYAxisWidth()
 	const { _ } = useLingui()
 
+	// round to nearest GB
+	if (diskSize >= 100) {
+		diskSize = Math.round(diskSize)
+	}
+
 	if (chartData.systemStats.length === 0) {
 		return null
 	}
diff --git a/beszel/site/src/components/routes/system.tsx b/beszel/site/src/components/routes/system.tsx
index 11c2997..c7dfb9b 100644
--- a/beszel/site/src/components/routes/system.tsx
+++ b/beszel/site/src/components/routes/system.tsx
@@ -413,11 +413,7 @@ export default function SystemDetail({ name }: { name: string }) {
 					)}
 
 					<ChartCard empty={dataEmpty} grid={grid} title={t`Disk Usage`} description={t`Usage of root partition`}>
-						<DiskChart
-							chartData={chartData}
-							dataKey="stats.du"
-							diskSize={Math.round(systemStats.at(-1)?.stats.d ?? NaN)}
-						/>
+						<DiskChart chartData={chartData} dataKey="stats.du" diskSize={systemStats.at(-1)?.stats.d ?? NaN} />
 					</ChartCard>
 
 					<ChartCard
@@ -497,7 +493,7 @@ export default function SystemDetail({ name }: { name: string }) {
 										<DiskChart
 											chartData={chartData}
 											dataKey={`stats.efs.${extraFsName}.du`}
-											diskSize={Math.round(systemStats.at(-1)?.stats.efs?.[extraFsName].d ?? NaN)}
+											diskSize={systemStats.at(-1)?.stats.efs?.[extraFsName].d ?? NaN}
 										/>
 									</ChartCard>
 									<ChartCard
diff --git a/beszel/site/src/lib/utils.ts b/beszel/site/src/lib/utils.ts
index 6872c0b..9cbbb9e 100644
--- a/beszel/site/src/lib/utils.ts
+++ b/beszel/site/src/lib/utils.ts
@@ -272,7 +272,7 @@ export const getSizeAndUnit = (n: number, isGigabytes = true) => {
 	} else if (sizeInGB >= 1) {
 		return { v: sizeInGB, u: " GB" }
 	}
-	return { v: n, u: " MB" }
+	return { v: isGigabytes ? sizeInGB * 1_000 : n, u: " MB" }
 }
 
 export const chartMargin = { top: 12 }