From 6c31263e606d44bcf2ef4618d13521cd613d0a7f Mon Sep 17 00:00:00 2001
From: Henry Dollman <hank@henrygd.me>
Date: Sat, 12 Oct 2024 17:22:25 -0400
Subject: [PATCH] add bandwidth alerts

---
 beszel/internal/agent/system.go                    |  1 +
 beszel/internal/alerts/alerts.go                   |  2 ++
 beszel/internal/entities/system/system.go          |  1 +
 beszel/site/src/components/command-palette.tsx     |  2 +-
 .../src/components/systems-table/systems-table.tsx |  5 +++--
 beszel/site/src/components/table-alerts.tsx        | 14 ++++++++++++--
 6 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/beszel/internal/agent/system.go b/beszel/internal/agent/system.go
index 448bacf..4716dd2 100644
--- a/beszel/internal/agent/system.go
+++ b/beszel/internal/agent/system.go
@@ -214,6 +214,7 @@ func (a *Agent) getSystemStats() system.Stats {
 	a.systemInfo.MemPct = systemStats.MemPct
 	a.systemInfo.DiskPct = systemStats.DiskPct
 	a.systemInfo.Uptime, _ = host.Uptime()
+	a.systemInfo.Bandwidth = twoDecimals(systemStats.NetworkSent + systemStats.NetworkRecv)
 
 	return systemStats
 }
diff --git a/beszel/internal/alerts/alerts.go b/beszel/internal/alerts/alerts.go
index bfdc7c2..8f5eef9 100644
--- a/beszel/internal/alerts/alerts.go
+++ b/beszel/internal/alerts/alerts.go
@@ -57,6 +57,8 @@ func (am *AlertManager) HandleSystemAlerts(systemRecord *models.Record, systemIn
 			am.handleSlidingValueAlert(systemRecord, alertRecord, name, "%", systemInfo.MemPct)
 		case "Disk":
 			am.handleSlidingValueAlert(systemRecord, alertRecord, name+" usage", "%", systemInfo.DiskPct)
+		case "Bandwidth":
+			am.handleSlidingValueAlert(systemRecord, alertRecord, name, " MB/s", systemInfo.Bandwidth)
 		case "Temperature":
 			if temperatures == nil {
 				continue
diff --git a/beszel/internal/entities/system/system.go b/beszel/internal/entities/system/system.go
index 56ad70b..9829212 100644
--- a/beszel/internal/entities/system/system.go
+++ b/beszel/internal/entities/system/system.go
@@ -61,6 +61,7 @@ type Info struct {
 	Cpu           float64 `json:"cpu"`
 	MemPct        float64 `json:"mp"`
 	DiskPct       float64 `json:"dp"`
+	Bandwidth     float64 `json:"b"`
 	AgentVersion  string  `json:"v"`
 }
 
diff --git a/beszel/site/src/components/command-palette.tsx b/beszel/site/src/components/command-palette.tsx
index e9fb89d..f4b9010 100644
--- a/beszel/site/src/components/command-palette.tsx
+++ b/beszel/site/src/components/command-palette.tsx
@@ -143,7 +143,7 @@ export default function CommandPalette() {
 								}}
 							>
 								<DatabaseBackupIcon className="mr-2 h-4 w-4" />
-								<span>Database backups</span>
+								<span>Backups</span>
 								<CommandShortcut>Admin</CommandShortcut>
 							</CommandItem>
 							<CommandItem
diff --git a/beszel/site/src/components/systems-table/systems-table.tsx b/beszel/site/src/components/systems-table/systems-table.tsx
index 43a7881..3c1a405 100644
--- a/beszel/site/src/components/systems-table/systems-table.tsx
+++ b/beszel/site/src/components/systems-table/systems-table.tsx
@@ -164,12 +164,13 @@ export default function SystemsTable({ filter }: { filter?: string }) {
 			},
 			{
 				accessorKey: 'info.b',
-				size: 80,
+				size: 115,
 				header: ({ column }) => sortableHeader(column, 'Net', EthernetIcon),
 				cell: (info) => {
+					const val = (info.getValue() as number) || 0
 					return (
 						<span className="tabular-nums whitespace-nowrap pl-1">
-							{decimalString((info.getValue() as number) || 0, 2)} MB/s
+							{decimalString(val, val >= 100 ? 1 : 2)} MB/s
 						</span>
 					)
 				},
diff --git a/beszel/site/src/components/table-alerts.tsx b/beszel/site/src/components/table-alerts.tsx
index 1273e55..1df814e 100644
--- a/beszel/site/src/components/table-alerts.tsx
+++ b/beszel/site/src/components/table-alerts.tsx
@@ -48,7 +48,7 @@ export default function AlertsButton({ system }: { system: SystemRecord }) {
 					/>
 				</Button>
 			</DialogTrigger>
-			<DialogContent className="max-h-full overflow-auto">
+			<DialogContent className="max-h-full overflow-auto max-w-[35rem]">
 				<DialogHeader>
 					<DialogTitle className="text-xl">{system.name} alerts</DialogTitle>
 					<DialogDescription className="mb-1">
@@ -82,6 +82,14 @@ export default function AlertsButton({ system }: { system: SystemRecord }) {
 						title="Disk Usage"
 						description="Triggers when root usage exceeds a threshold."
 					/>
+					<AlertWithSlider
+						system={system}
+						alerts={systemAlerts}
+						name="Bandwidth"
+						title="Bandwidth"
+						description="Triggers when combined up/down exceeds a threshold."
+						unit=" MB/s"
+					/>
 					<AlertWithSlider
 						system={system}
 						alerts={systemAlerts}
@@ -152,6 +160,7 @@ function AlertWithSlider({
 	title,
 	description,
 	unit = '%',
+	max = 99,
 }: {
 	system: SystemRecord
 	alerts: AlertRecord[]
@@ -159,6 +168,7 @@ function AlertWithSlider({
 	title: string
 	description: string
 	unit?: string
+	max?: number
 }) {
 	const [pendingChange, setPendingChange] = useState(false)
 	const [liveValue, setLiveValue] = useState(80)
@@ -226,7 +236,7 @@ function AlertWithSlider({
 								setLiveValue(val[0])
 							}}
 							min={1}
-							max={99}
+							max={max}
 							// step={1}
 						/>
 					</Suspense>