feat: fix builds:target-outputs --pull

Author: yjp20

pkg/cmd/buildtargetoutput.go

@@ -34,9 +34,9 @@ var buildsTargetOutputsRetrieve = cli.Command{
 			Usage: "Branch name (defaults to main if not provided)",
 			Value: "main",
 		},
-		&cli.StringFlag{
+		&cli.StringSliceFlag{
 			Name:  "target",
-			Usage: "SDK language target name",
+			Usage: "SDK language target name(s). Can be specified multiple times.",
 		},
 		&cli.StringFlag{
 			Name: "type",
@@ -67,30 +67,52 @@ func handleBuildsTargetOutputsRetrieve(ctx context.Context, cmd *cli.Command) er
 		buildID = latestBuild.ID
 	}
 
-	params := stainless.BuildTargetOutputGetParams{
-		BuildID: buildID,
-	}
-	var resBytes []byte
-	res, err := client.Builds.TargetOutputs.Get(
-		ctx,
-		params,
-		option.WithMiddleware(debugMiddleware(cmd.Bool("debug"))),
-		option.WithResponseBodyInto(&resBytes),
-	)
-	if err != nil {
-		return err
-	}
+	wc := getWorkspace(ctx)
+	downloadPaths, targets, _ := parseTargetPaths(wc, cmd.StringSlice("target"))
 
-	json := gjson.Parse(string(resBytes))
-	format := cmd.Root().String("format")
-	transform := cmd.Root().String("transform")
-	if err := ShowJSON("builds:target_outputs retrieve", json, format, transform); err != nil {
-		return err
+	if len(targets) == 0 {
+		return fmt.Errorf("at least one target must be specified")
 	}
 
-	group := console.Info("Downloading output")
-	if cmd.Bool("pull") {
-		return build.PullOutput(res.Output, res.URL, res.Ref, cmd.String("branch"), "", group)
+	outputType := cmd.String("type")
+	outputFormat := cmd.String("output")
+	isPull := cmd.Bool("pull")
+
+	for _, target := range targets {
+		params := stainless.BuildTargetOutputGetParams{
+			BuildID: buildID,
+			Target:  stainless.BuildTargetOutputGetParamsTarget(target),
+			Type:    stainless.BuildTargetOutputGetParamsType(outputType),
+			Output:  stainless.BuildTargetOutputGetParamsOutput(outputFormat),
+		}
+		res, err := client.Builds.TargetOutputs.Get(
+			ctx,
+			params,
+			option.WithMiddleware(debugMiddleware(cmd.Bool("debug"))),
+		)
+		if err != nil {
+			return fmt.Errorf("failed to get output for target %s: %v", target, err)
+		}
+
+		if !isPull {
+			json := gjson.Parse(res.RawJSON())
+			format := cmd.Root().String("format")
+			transform := cmd.Root().String("transform")
+			if err := ShowJSON("builds:target_outputs retrieve", json, format, transform); err != nil {
+				return err
+			}
+		}
+
+		if isPull {
+			group := console.Info(fmt.Sprintf("Downloading %s", target))
+
+			// Get target output path from downloadPaths (which includes workspace config)
+			targetDir := downloadPaths[target]
+
+			if err := build.PullOutput(res.Output, res.URL, res.Ref, cmd.String("branch"), targetDir, group); err != nil {
+				return fmt.Errorf("failed to pull %s: %v", target, err)
+			}
+		}
 	}
 
 	return nil