wit, testdata: add Variant.WIT()

bytecodealliance · Sep 25, 2023 · f489548 · f489548
1 parent 4879541
commit f489548
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 8 deletions.
diff --git a/testdata/types.wit.json.golden.wit b/testdata/types.wit.json.golden.wit
@@ -41,11 +41,25 @@ interface types {
     flags t30 {}
     flags t31 {a, b, c}
     flags t32 {a, b, c}
-    /* TODO(t33) */
-    /* TODO(t34) */
-    /* TODO(t35) */
-    /* TODO(t36) */
-    /* TODO(t37) */
+    variant t33 {
+        a
+    }
+    variant t34 {
+        a,
+        b
+    }
+    variant t35 {
+        a,
+        b
+    }
+    variant t36 {
+        a,
+        b(u32)
+    }
+    variant t37 {
+        a,
+        b(/* TODO() */)
+    }
     type t4 = u64
     /* TODO(t41) */
     /* TODO(t42) */

diff --git a/testdata/worlds-with-types.wit.json.golden.wit b/testdata/worlds-with-types.wit.json.golden.wit
@@ -20,7 +20,9 @@ world the-test {
     record a {
         x: u32
     }
-    /* TODO(b) */
+    variant b {
+        c(a)
+    }
     import foo: func(a: a) -> b
     export bar: func(a: a) -> b
 }
diff --git a/wit/wit.go b/wit/wit.go
@@ -239,8 +239,6 @@ func (f *Flags) WIT(ctx Node, name string) string {
 	return b.String()
 }
 
-const wrapFlags = 3
-
 func (f *Flag) WIT(_ Node, _ string) string {
 	// TODO: docs
 	return f.Name
@@ -259,6 +257,36 @@ func (t *Tuple) WIT(ctx Node, _ string) string {
 	return b.String()
 }
 
+func (v *Variant) WIT(ctx Node, name string) string {
+	var b strings.Builder
+	b.WriteString("variant ")
+	b.WriteString(name)
+	b.WriteString(" {")
+	if len(v.Cases) > 0 {
+		b.WriteRune('\n')
+		for i := range v.Cases {
+			if i > 0 {
+				b.WriteString(",\n")
+			}
+			b.WriteString(indent(v.Cases[i].WIT(ctx, "")))
+		}
+		b.WriteRune('\n')
+	}
+	b.WriteRune('}')
+	return b.String()
+}
+
+func (c *Case) WIT(_ Node, _ string) string {
+	var b strings.Builder
+	b.WriteString(c.Name)
+	if c.Type != nil {
+		b.WriteRune('(')
+		b.WriteString(c.Type.WIT(c, ""))
+		b.WriteRune(')')
+	}
+	return b.String()
+}
+
 func (l *List) WIT(ctx Node, _ string) string {
 	return "list<" + l.Type.WIT(l, "") + ">"
 }