wit: prepare for filtering interface contents

Author: ydnar

wit/wit.go

@@ -321,23 +321,15 @@ func (*Interface) WITKind() string { return "interface" }
 //
 // [WIT]: https://github.com/WebAssembly/component-model/blob/main/design/mvp/WIT.md
 func (i *Interface) WIT(ctx Node, name string) string {
+	// filter, _ := ctx.(*witFilter)
+
 	if i.Name != nil && name == "" {
 		name = *i.Name
 	}
 
 	var b strings.Builder
 
 	switch ctx := ctx.(type) {
-	case *Package:
-		b.WriteString(i.Docs.WIT(ctx, ""))
-		if i.Stability != nil {
-			b.WriteString(i.Stability.WIT(ctx, ""))
-			b.WriteRune('\n')
-		}
-		b.WriteString("interface ")
-		b.WriteString(escape(name))
-		b.WriteRune(' ')
-
 	case worldImport:
 		rname := relativeName(i, ctx.Package)
 		if rname != "" {
@@ -361,13 +353,26 @@ func (i *Interface) WIT(ctx Node, name string) string {
 		b.WriteString("export ")
 		b.WriteString(escape(name))
 		b.WriteString(": interface ")
+
+	default: // e.g. *Package
+		b.WriteString(i.Docs.WIT(ctx, ""))
+		if i.Stability != nil {
+			b.WriteString(i.Stability.WIT(ctx, ""))
+			b.WriteRune('\n')
+		}
+		b.WriteString("interface ")
+		b.WriteString(escape(name))
+		b.WriteRune(' ')
 	}
 
 	b.WriteRune('{')
 	n := 0
 
 	// Emit use statements first
 	i.TypeDefs.All()(func(name string, td *TypeDef) bool {
+		// if filter.filter(td) {
+		// 	return true
+		// }
 		if td.Root().Owner == td.Owner {
 			return true // Skip declarations
 		}
@@ -382,6 +387,9 @@ func (i *Interface) WIT(ctx Node, name string) string {
 
 	// Declarations
 	i.TypeDefs.All()(func(name string, td *TypeDef) bool {
+		// if filter.filter(td) {
+		// 	return true
+		// }
 		if td.Root().Owner != td.Owner {
 			return true // Skip use statements
 		}
@@ -396,6 +404,9 @@ func (i *Interface) WIT(ctx Node, name string) string {
 
 	// Functions
 	i.Functions.All()(func(name string, f *Function) bool {
+		// if filter.filter(f) {
+		// 	return true
+		// }
 		if !f.IsFreestanding() {
 			return true
 		}
@@ -1126,9 +1137,9 @@ func (p *Package) WIT(ctx Node, name string) string {
 		}
 		b.WriteRune('\n')
 		if multi {
-			b.WriteString(indent(face.WIT(p, name)))
+			b.WriteString(indent(face.WIT(ctx, name)))
 		} else {
-			b.WriteString(face.WIT(p, name))
+			b.WriteString(face.WIT(ctx, name))
 		}
 		b.WriteRune('\n')
 		i++
@@ -1140,9 +1151,9 @@ func (p *Package) WIT(ctx Node, name string) string {
 		}
 		b.WriteRune('\n')
 		if multi {
-			b.WriteString(indent(w.WIT(filter, name)))
+			b.WriteString(indent(w.WIT(ctx, name)))
 		} else {
-			b.WriteString(w.WIT(filter, name))
+			b.WriteString(w.WIT(ctx, name))
 		}
 		b.WriteRune('\n')
 		i++