chore(cli): Migrate argo logs to use API client. See #2116

api/argo-server/swagger.json

@@ -5308,6 +5308,9 @@
  "properties": {
  "content": {
  "type": "string"
+ },
+ "podName": {
+ "type": "string"
  }
  }
  },

cmd/argo/commands/common.go

@@ -118,6 +118,15 @@ func InitWorkflowClient(ns ...string) v1alpha1.WorkflowInterface {
  return wfClient
 }
 
+func ansiColorCode(s string) int {
+ i := 0
+ for _, c := range s {
+ i += int(c)
+ }
+ colors := []int{FgRed, FgGreen, FgYellow, FgBlue, FgMagenta, FgCyan, FgWhite}
+ return colors[i%len(colors)]
+}
+
 // ansiFormat wraps ANSI escape codes to a string to format the string to a desired color.
 // NOTE: we still apply formatting even if there is no color formatting desired.
 // The purpose of doing this is because when we apply ANSI color escape sequences to our

cmd/argo/commands/common_test.go

@@ -0,0 +1,15 @@
+package commands
+
+import (
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+)
+
+func Test_ansiFormatCode(t *testing.T) {
+ // check we get a nice range of colours
+ assert.Equal(t, FgYellow, ansiColorCode("foo"))
+ assert.Equal(t, FgGreen, ansiColorCode("bar"))
+ assert.Equal(t, FgYellow, ansiColorCode("baz"))
+ assert.Equal(t, FgRed, ansiColorCode("qux"))
+}