Update Struct.String() function to not quote starlark.String custom c…

…onstructor (#449)
google · Jan 12, 2023 · fae38c8 · fae38c8
1 parent dcdc0f0
commit fae38c8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/starlarkstruct/struct.go b/starlarkstruct/struct.go
@@ -132,11 +132,12 @@ func (s *Struct) ToStringDict(d starlark.StringDict) {
 
 func (s *Struct) String() string {
 	buf := new(strings.Builder)
-	if s.constructor == Default {
+	switch constructor := s.constructor.(type) {
+	case starlark.String:
 		// NB: The Java implementation always prints struct
 		// even for Bazel provider instances.
-		buf.WriteString("struct") // avoid String()'s quotation
-	} else {
+		buf.WriteString(constructor.GoString()) // avoid String()'s quotation
+	default:
 		buf.WriteString(s.constructor.String())
 	}
 	buf.WriteByte('(')