add prefix for enum type

ktr0731 · Jan 10, 2022 · 4620275 · 4620275
1 parent fbe7341
commit 4620275
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/fill/proto/interactive_filler.go b/fill/proto/interactive_filler.go
@@ -152,6 +152,7 @@ func (r *resolver) resolveField(f *desc.FieldDescriptor) error {
 			)
 			return msgr.resolve()
 		case descriptorpb.FieldDescriptorProto_TYPE_ENUM:
+			fmt.Printf("%s%s%s\n", r.color.ForegroundANSI(), f.GetName(), "\033[0m")
 			return r.resolveEnum(f.GetEnumType())
 		case descriptorpb.FieldDescriptorProto_TYPE_DOUBLE:
 			converter = func(v string) (interface{}, error) { return strconv.ParseFloat(v, 64) }

diff --git a/prompt/prompt.go b/prompt/prompt.go
@@ -2,6 +2,7 @@
 package prompt
 
 import (
+	"fmt"
 	"io"
 	"os"
 
@@ -53,6 +54,15 @@ func (c *Color) NextVal() Color {
 	return (*c + 1) % 16
 }
 
+func (c Color) ForegroundANSI() string {
+	return fmt.Sprintf("\033[%sm", foregroundANSIColors[c])
+}
+
+var foregroundANSIColors = map[Color][]byte{
+	ColorInitial: {'3', '2'},
+	ColorBlue:    {'9', '4'},
+}
+
 type Prompt interface {
 	// Input reads keyboard input.
 	// If ctrl+d is entered, Input returns io.EOF.