From 7b58f17fc3575e7253e6417a0220fbe103e6cd29 Mon Sep 17 00:00:00 2001
From: Alan Hamlett <alan.hamlett@gmail.com>
Date: Tue, 25 Jun 2024 20:25:16 +0200
Subject: [PATCH] Add Supporting category

---
 cmd/params/params_test.go      | 1 +
 cmd/root.go                    | 2 +-
 pkg/heartbeat/category.go      | 7 +++++++
 pkg/heartbeat/category_test.go | 1 +
 4 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/cmd/params/params_test.go b/cmd/params/params_test.go
index 5b928c4d..846d4079 100644
--- a/cmd/params/params_test.go
+++ b/cmd/params/params_test.go
@@ -65,6 +65,7 @@ func TestLoadParams_Category(t *testing.T) {
 		"planning":       heartbeat.PlanningCategory,
 		"researching":    heartbeat.ResearchingCategory,
 		"running tests":  heartbeat.RunningTestsCategory,
+		"supporting":     heartbeat.SupportingCategory,
 		"translating":    heartbeat.TranslatingCategory,
 		"writing docs":   heartbeat.WritingDocsCategory,
 		"writing tests":  heartbeat.WritingTestsCategory,
diff --git a/cmd/root.go b/cmd/root.go
index 64c647c1..71f56079 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -56,7 +56,7 @@ func setFlags(cmd *cobra.Command, v *viper.Viper) {
 		"",
 		"Category of this heartbeat activity. Can be \"coding\","+
 			" \"building\", \"indexing\", \"debugging\", \"learning\","+
-			" \"meeting\", \"planning\", \"researching\", \"communicating\","+
+			" \"meeting\", \"planning\", \"researching\", \"communicating\", \"supporting\" "+
 			" \"advising\", \"running tests\", \"writing tests\", \"manual testing\","+
 			" \"writing docs\", \"code reviewing\", \"browsing\","+
 			" \"translating\", or \"designing\". Defaults to \"coding\".",
diff --git a/pkg/heartbeat/category.go b/pkg/heartbeat/category.go
index 13d6abbf..04be3250 100644
--- a/pkg/heartbeat/category.go
+++ b/pkg/heartbeat/category.go
@@ -39,6 +39,8 @@ const (
 	ResearchingCategory
 	// RunningTestsCategory means user is currently running tests.
 	RunningTestsCategory
+	// SupportingCategory means user is doing customer support.
+	SupportingCategory
 	// TranslatingCategory means user is currently translating.
 	TranslatingCategory
 	// WritingDocsCategory means user is currently writing docs.
@@ -63,6 +65,7 @@ const (
 	planningCategoryString      = "planning"
 	researchingCategoryString   = "researching"
 	runningTestsCategoryString  = "running tests"
+	supportingCategoryString    = "supporting"
 	translatingCategoryString   = "translating"
 	writingDocsCategoryString   = "writing docs"
 	writingTestsCategoryString  = "writing tests"
@@ -101,6 +104,8 @@ func ParseCategory(s string) (Category, error) {
 		return ResearchingCategory, nil
 	case runningTestsCategoryString:
 		return RunningTestsCategory, nil
+	case supportingCategoryString:
+		return SupportingCategory, nil
 	case translatingCategoryString:
 		return TranslatingCategory, nil
 	case writingDocsCategoryString:
@@ -169,6 +174,8 @@ func (c Category) String() string {
 		return researchingCategoryString
 	case RunningTestsCategory:
 		return runningTestsCategoryString
+	case SupportingCategory:
+		return supportingCategoryString
 	case TranslatingCategory:
 		return translatingCategoryString
 	case WritingDocsCategory:
diff --git a/pkg/heartbeat/category_test.go b/pkg/heartbeat/category_test.go
index 50fb03e4..176c2992 100644
--- a/pkg/heartbeat/category_test.go
+++ b/pkg/heartbeat/category_test.go
@@ -27,6 +27,7 @@ func categoryTests() map[string]heartbeat.Category {
 		"planning":       heartbeat.PlanningCategory,
 		"researching":    heartbeat.ResearchingCategory,
 		"running tests":  heartbeat.RunningTestsCategory,
+		"supporting":     heartbeat.SupportingCategory,
 		"translating":    heartbeat.TranslatingCategory,
 		"writing docs":   heartbeat.WritingDocsCategory,
 		"writing tests":  heartbeat.WritingTestsCategory,