PV reclaim policy coloring (#73)

* PV reclaim policy coloring * Updated CHANGELOG * Added more colors
applejag · Dec 13, 2023 · ba115a8 · ba115a8
1 parent b37c242
commit ba115a8
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -22,6 +22,8 @@ This project tries to follow [SemVer 2.0.0](https://semver.org/).
   where `NotReady,SchedulingDisabled` would have red `NotReady`
   and orange `SchedulingDisabled`. (#69)
 
+- Added coloring on PV/StorageClass reclaim policy and PV/PVC status. (#73)
+
 - Changed to Go 1.21.5 to resolve vulnerability [GO-2023-2182](https://pkg.go.dev/vuln/GO-2023-2182),
   for denial of service in net/http, and [GO-2023-2185](https://pkg.go.dev/vuln/GO-2023-2185)
   for insecure parsing of Windows paths with a `\??\` prefix. (#72)

diff --git a/pkg/klock/color.go b/pkg/klock/color.go
@@ -126,7 +126,10 @@ func StatusStyle(status string) lipgloss.Style {
 		"Evicted",
 		"FailedScheduling",
 		"Error",
-		"ErrImagePull":
+		"ErrImagePull",
+
+		// PVC status
+		"Lost":
 		return StyleStatusError
 	case
 		// from https://github.com/kubernetes/kubernetes/blob/master/pkg/kubelet/events/event.go
@@ -160,7 +163,14 @@ func StatusStyle(status string) lipgloss.Style {
 		"ContainerCreating",
 		"PodInitializing",
 		"Terminating",
-		"Warning":
+		"Warning",
+
+		// PV reclaim policy
+		"Delete",
+
+		// PVC status
+		"Available",
+		"Released":
 		return StyleStatusWarning
 	case
 		"Running",
@@ -173,7 +183,13 @@ func StatusStyle(status string) lipgloss.Style {
 		"Normal",
 		"VolumeResizeSuccessful",
 		"FileSystemResizeSuccessful",
-		"Ready":
+		"Ready",
+
+		// PV reclaim policy
+		"Retain",
+
+		// PVC status
+		"Bound":
 		return StyleStatusOK
 	}
 	// some ok status, not colored:

diff --git a/pkg/klock/klock.go b/pkg/klock/klock.go
@@ -524,6 +524,14 @@ func (p *Printer) parseCell(cell any, row metav1.TableRow, eventType watch.Event
 			}
 		}
 		return cell
+	case p.apiVersion == "storage.k8s.io/v1" && p.kind == "StorageClass" && columnNameLower == "reclaimpolicy":
+		return StatusColumn(cellStr)
+	case p.apiVersion == "v1" && p.kind == "PersistentVolume" && columnNameLower == "reclaim policy":
+		return StatusColumn(cellStr)
+	case p.apiVersion == "v1" && p.kind == "PersistentVolume" && columnNameLower == "status":
+		return StatusColumn(cellStr)
+	case p.apiVersion == "v1" && p.kind == "PersistentVolumeClaim" && columnNameLower == "status":
+		return StatusColumn(cellStr)
 	// Only parse fraction (e.g "1/2") if the resources was not deleted,
 	// so we don't have colored fraction on a grayed-out row.
 	case eventType != watch.Deleted: