Better upper case

Author: rjtokenring

internal/ota-api/dto.go

@@ -18,8 +18,11 @@
 package otaapi
 
 import (
+	"strings"
 	"time"
 
+	"unicode"
+
 	"github.com/arduino/arduino-cli/table"
 )
 
@@ -92,18 +95,7 @@ func (r OtaStatusByDeviceResponse) String() string {
 }
 
 func (o Ota) MapStatus() string {
-	switch o.Status {
-	case "in_progress":
-		return "In progress"
-	case "failed":
-		return "Failed"
-	case "pending":
-		return "Pending"
-	case "succeeded":
-		return "Succeeded"
-	default:
-		return o.Status
-	}
+	return upperCaseFirst(o.Status)
 }
 
 func (r Ota) Data() interface{} {
@@ -164,14 +156,24 @@ func (r OtaStatusPrint) String() string {
 		t = table.New()
 		t.SetHeader("Time", "Status", "Detail")
 		for _, s := range r.Details {
-			t.AddRow(formatHumanReadableTs(s.Timestamp), s.State, s.StateData)
+			t.AddRow(formatHumanReadableTs(s.Timestamp), upperCaseFirst(s.State), s.StateData)
 		}
 		output += "\nDetails:\n" + t.Render()
 	}
 
 	return output
 }
 
+func upperCaseFirst(s string) string {
+	if len(s) > 0 {
+		s = strings.ReplaceAll(s, "_", " ")
+		for i, v := range s {
+			return string(unicode.ToUpper(v)) + s[i+1:]
+		}
+	}
+	return ""
+}
+
 func formatHumanReadableTs(ts string) string {
 	if ts == "" {
 		return ""