From e3cdcdc826696299289840e2f7301cd382e750e0 Mon Sep 17 00:00:00 2001
From: Sebastian Malton <sebastian@malton.name>
Date: Thu, 9 Mar 2023 15:00:42 -0500
Subject: [PATCH] Expand folding of workload status phases

Signed-off-by: Sebastian Malton <sebastian@malton.name>
---
 .../renderer/utils/fold-pod-status-phase.ts   | 18 --------
 .../utils/fold-workload-status-phase.ts       | 43 +++++++++++++++++++
 2 files changed, 43 insertions(+), 18 deletions(-)
 delete mode 100644 packages/core/src/renderer/utils/fold-pod-status-phase.ts
 create mode 100644 packages/core/src/renderer/utils/fold-workload-status-phase.ts

diff --git a/packages/core/src/renderer/utils/fold-pod-status-phase.ts b/packages/core/src/renderer/utils/fold-pod-status-phase.ts
deleted file mode 100644
index 038dfc911e..0000000000
--- a/packages/core/src/renderer/utils/fold-pod-status-phase.ts
+++ /dev/null
@@ -1,18 +0,0 @@
-/**
- * Copyright (c) OpenLens Authors. All rights reserved.
- * Licensed under MIT License. See LICENSE in root directory for more information.
- */
-
-import { PodStatusPhase } from "../../common/k8s-api/endpoints";
-
-export const foldPodStatusPhase = (previous: "failed" | "pending" | "running", current: PodStatusPhase): "failed" | "pending" | "running" => {
-  if (previous === "failed" || current === PodStatusPhase.FAILED) {
-    return "failed";
-  }
-
-  if (previous === "pending" || current === PodStatusPhase.PENDING) {
-    return "pending";
-  }
-
-  return "running";
-};
diff --git a/packages/core/src/renderer/utils/fold-workload-status-phase.ts b/packages/core/src/renderer/utils/fold-workload-status-phase.ts
new file mode 100644
index 0000000000..401752030d
--- /dev/null
+++ b/packages/core/src/renderer/utils/fold-workload-status-phase.ts
@@ -0,0 +1,43 @@
+/**
+ * Copyright (c) OpenLens Authors. All rights reserved.
+ * Licensed under MIT License. See LICENSE in root directory for more information.
+ */
+
+import type { WorkloadStatus } from "../components/+workloads-overview/overview-workload-status";
+import type { WorkloadStatusPhase } from "../components/+workloads-overview/workloads/workload-injection-token";
+
+export const foldWorkloadStatusPhase = (previous: WorkloadStatus, current: WorkloadStatusPhase): WorkloadStatus => {
+  if (previous === "failed" || current === "Failed") {
+    return "failed";
+  }
+
+  if (previous === "evicted" || current === "Evicted") {
+    return "evicted";
+  }
+
+  if (previous === "pending" || current === "Pending") {
+    return "pending";
+  }
+
+  if (previous === "suspended" || current === "Suspended") {
+    return "suspended";
+  }
+
+  if (previous === "running" || current === "Running") {
+    return "running";
+  }
+
+  if (previous === "scheduled" || current === "Scheduled") {
+    return "scheduled";
+  }
+
+  if (previous === "succeeded" || current === "Succeeded") {
+    return "succeeded";
+  }
+
+  if (previous === "terminated" || current === "Terminated") {
+    return "terminated";
+  }
+
+  return "unknown";
+};