Fix: Add support for maintenance in badges

This commit is contained in:
Nelson Chan 2023-01-03 21:45:55 +08:00
parent b8e8c1b9db
commit 942b55ca03
2 changed files with 34 additions and 3 deletions

View File

@ -5,6 +5,7 @@ const badgeConstants = {
naColor: "#999", naColor: "#999",
defaultUpColor: "#66c20a", defaultUpColor: "#66c20a",
defaultDownColor: "#c2290a", defaultDownColor: "#c2290a",
defaultMaintenanceColor: "#1747f5",
defaultPingColor: "blue", // as defined by badge-maker / shields.io defaultPingColor: "blue", // as defined by badge-maker / shields.io
defaultStyle: "flat", defaultStyle: "flat",
defaultPingValueSuffix: "ms", defaultPingValueSuffix: "ms",

View File

@ -111,8 +111,10 @@ router.get("/api/badge/:id/status", cache("5 minutes"), async (request, response
label, label,
upLabel = "Up", upLabel = "Up",
downLabel = "Down", downLabel = "Down",
maintenanceLabel = "Maintenance",
upColor = badgeConstants.defaultUpColor, upColor = badgeConstants.defaultUpColor,
downColor = badgeConstants.defaultDownColor, downColor = badgeConstants.defaultDownColor,
maintenanceColor = badgeConstants.defaultMaintenanceColor,
style = badgeConstants.defaultStyle, style = badgeConstants.defaultStyle,
value, // for demo purpose only value, // for demo purpose only
} = request.query; } = request.query;
@ -139,11 +141,39 @@ router.get("/api/badge/:id/status", cache("5 minutes"), async (request, response
badgeValues.color = badgeConstants.naColor; badgeValues.color = badgeConstants.naColor;
} else { } else {
const heartbeat = await Monitor.getPreviousHeartbeat(requestedMonitorId); const heartbeat = await Monitor.getPreviousHeartbeat(requestedMonitorId);
const state = overrideValue !== undefined ? overrideValue : heartbeat.status === 1; const state = overrideValue !== undefined ? overrideValue : heartbeat.status;
badgeValues.label = label ? label : ""; badgeValues.label = label ? label : "";
badgeValues.color = state ? upColor : downColor; switch (state) {
badgeValues.message = label ?? state ? upLabel : downLabel; case 1:
badgeValues.color = upColor;
break;
case 3:
badgeValues.color = maintenanceColor;
break;
case 0:
badgeValues.color = downColor;
break;
default:
badgeValues.color = badgeConstants.naColor;
}
if (label !== undefined) {
badgeValues.message = label;
} else {
switch (state) {
case 1:
badgeValues.message = upLabel;
break;
case 3:
badgeValues.message = maintenanceLabel;
break;
case 0:
badgeValues.message = downLabel;
break;
default:
badgeValues.message = "N/A";
}
}
} }
// build the svg based on given values // build the svg based on given values