qawatake · qawatake · Mar 22, 2024
diff --git a/Makefile b/Makefile
@@ -0,0 +1,2 @@
+wire:
+	go run ./cmd/wire ./_tutorial/
diff --git a/_tutorial/main.go b/_tutorial/main.go
@@ -20,6 +20,8 @@ import (
 	"fmt"
 	"os"
 	"time"
+
+	"github.com/google/wire"
 )
 
 // Message is what greeters will use to greet guests.
@@ -81,3 +83,24 @@ func main() {
 	}
 	e.Start()
 }
+
+type X int
+type Y int
+type Z int
+
+type A struct{}
+
+type God struct {
+	X X
+	Y Y
+	Z Z
+}
+
+var set = wire.NewSet(
+	New,
+	wire.FieldsOf(new(God), "*"),
+)
+
+func New(X, Y, Z) A {
+	return A{}
+}
diff --git a/_tutorial/wire.go b/_tutorial/wire.go
@@ -26,3 +26,7 @@ func InitializeEvent(phrase string) (Event, error) {
 	wire.Build(NewEvent, NewGreeter, NewMessage)
 	return Event{}, nil
 }
+
+func InitializeX(g God) A {
+	panic(wire.Build(set))
+}
diff --git a/_tutorial/wire_gen.go b/_tutorial/wire_gen.go
diff --git a/internal/wire/parse.go b/internal/wire/parse.go
@@ -1038,6 +1038,33 @@ func processFieldsOf(fset *token.FileSet, info *types.Info, call *ast.CallExpr)
 	}
 
 	fields := make([]*Field, 0, len(call.Args)-1)
+	startExists := func() bool {
+		for i := 1; i < len(call.Args); i++ {
+			if b, ok := call.Args[i].(*ast.BasicLit); ok && b.Value == `"*"` {
+				return true
+			}
+		}
+		return false
+	}()
+	if startExists {
+		for i := 0; i < struc.NumFields(); i++ {
+			v := struc.Field(i)
+			out := []types.Type{v.Type()}
+			if isPtrToStruct {
+				// If the field is from a pointer to a struct, then
+				// wire.Fields also provides a pointer to the field.
+				out = append(out, types.NewPointer(v.Type()))
+			}
+			fields = append(fields, &Field{
+				Parent: structPtr.Elem(),
+				Name:   v.Name(),
+				Pkg:    v.Pkg(),
+				Pos:    v.Pos(),
+				Out:    out,
+			})
+		}
+		return fields, nil
+	}
 	for i := 1; i < len(call.Args); i++ {
 		v, err := checkField(call.Args[i], struc)
 		if err != nil {
@@ -1173,9 +1200,9 @@ func (pt ProvidedType) IsNil() bool {
 //
 //   - For a function provider, this is the first return value type.
 //   - For a struct provider, this is either the struct type or the pointer type
-// 	   whose element type is the struct type.
-// 	 - For a value, this is the type of the expression.
-// 	 - For an argument, this is the type of the argument.
+//     whose element type is the struct type.
+//   - For a value, this is the type of the expression.
+//   - For an argument, this is the type of the argument.
 func (pt ProvidedType) Type() types.Type {
 	return pt.t
 }