From f818a3525b08c4d25e64acad3394c02c06e6ac80 Mon Sep 17 00:00:00 2001
From: Samuel Lorch <sam@soontm.de>
Date: Sun, 14 Jul 2024 04:15:25 +0200
Subject: [PATCH] Adjust more colors

---
 server/stats.go | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/server/stats.go b/server/stats.go
index 5e9cc63..3d870e3 100644
--- a/server/stats.go
+++ b/server/stats.go
@@ -36,8 +36,9 @@ type PieIntValue struct {
 	Value int
 }
 
-const SUCCESS_CHART_COLOR = "#7cffb2"
-const FAILED_CHART_COLOR = "#ff6e76"
+const CHART_COLOR_SUCCESS = "#7cffb2"
+const CHART_COLOR_FAILED = "#ff6e76"
+const CHART_COLOR_UNKNOWN = "#fddd60"
 
 func generatePie(name string, data []opts.PieData) ChartData {
 	pie := charts.NewPie()
@@ -51,13 +52,17 @@ func generatePie(name string, data []opts.PieData) ChartData {
 		}))
 
 	for i := range data {
-		if data[i].Name == "Success" {
+		if data[i].Name == "Success" || data[i].Name == "Healthy" {
 			data[i].ItemStyle = &opts.ItemStyle{
-				Color: SUCCESS_CHART_COLOR,
+				Color: CHART_COLOR_SUCCESS,
 			}
-		} else if data[i].Name == "Failed" {
+		} else if data[i].Name == "Failed" || data[i].Name == "Damaged" {
 			data[i].ItemStyle = &opts.ItemStyle{
-				Color: FAILED_CHART_COLOR,
+				Color: CHART_COLOR_FAILED,
+			}
+		} else if data[i].Name == "Unknown" || data[i].Name == "None" {
+			data[i].ItemStyle = &opts.ItemStyle{
+				Color: CHART_COLOR_UNKNOWN,
 			}
 		}
 	}
@@ -243,14 +248,14 @@ func generateStats(ctx context.Context) ([]ChartData, error) {
 			successBarData = append(successBarData, opts.BarData{
 				Value: v.Count,
 				ItemStyle: &opts.ItemStyle{
-					Color: SUCCESS_CHART_COLOR,
+					Color: CHART_COLOR_SUCCESS,
 				},
 			})
 		} else if v.Status == constants.TASK_STATUS_FAILED {
 			failedBarData = append(failedBarData, opts.BarData{
 				Value: v.Count,
 				ItemStyle: &opts.ItemStyle{
-					Color: FAILED_CHART_COLOR,
+					Color: CHART_COLOR_FAILED,
 				},
 			})
 		}