chore: add UnionUnmarshaler for responses that are interfaces (#6)

stainless-app[bot] · stainless-app[bot] · commit d6fff8045ed9 · 2025-02-06T05:52:38.000Z
diff --git a/internal/apijson/registry.go b/internal/apijson/registry.go
@@ -29,3 +29,13 @@ func RegisterUnion(typ reflect.Type, discriminator string, variants ...UnionVari
 		unionVariants[variant.Type] = typ
 	}
 }
+
+// Useful to wrap a union type to force it to use [apijson.UnmarshalJSON] since you cannot define an
+// UnmarshalJSON function on the interface itself.
+type UnionUnmarshaler[T any] struct {
+	Value T
+}
+
+func (c *UnionUnmarshaler[T]) UnmarshalJSON(buf []byte) error {
+	return UnmarshalRoot(buf, &c.Value)
+}

Original file line number	Diff line number	Diff line change
`@@ -29,3 +29,13 @@ func RegisterUnion(typ reflect.Type, discriminator string, variants ...UnionVari`
`29`	`29`	`unionVariants[variant.Type] = typ`
`30`	`30`	`}`
`31`	`31`	`}`
	`32`	`+`
	`33`	`+// Useful to wrap a union type to force it to use [apijson.UnmarshalJSON] since you cannot define an`
	`34`	`+// UnmarshalJSON function on the interface itself.`
	`35`	`+type UnionUnmarshaler[T any] struct {`
	`36`	`+ Value T`
	`37`	`+}`
	`38`	`+`
	`39`	`+func (c *UnionUnmarshaler[T]) UnmarshalJSON(buf []byte) error {`
	`40`	`+ return UnmarshalRoot(buf, &c.Value)`
	`41`	`+}`