Remove context.TODO() and use real contexts

Author: Bruce Hill

pkg/cmd/build.go

@@ -493,7 +493,7 @@ func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 	if cmd.Bool("wait") {
 		waitGroup := Info("Waiting for latest build to complete...")
 
-		build, err = waitForBuildCompletion(context.TODO(), cc.client, build, &waitGroup)
+		build, err = waitForBuildCompletion(ctx, cc.client, build, &waitGroup)
 		if err != nil {
 			return err
 		}
@@ -503,7 +503,7 @@ func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 
 		if shouldPull {
 			pullGroup := Info("Downloading build outputs...")
-			if err := pullBuildOutputs(context.TODO(), cc.client, *build, targetPaths, &pullGroup); err != nil {
+			if err := pullBuildOutputs(ctx, cc.client, *build, targetPaths, &pullGroup); err != nil {
 				pullGroup.Error("Failed to download outputs: %v", err)
 			} else {
 				pullGroup.Success("Successfully downloaded all outputs")

pkg/cmd/dev.go

@@ -211,7 +211,7 @@ func (m BuildModel) downloadTarget(target stainless.Target) tea.Cmd {
 			Output:  "git",
 		}
 		outputRes, err := m.cc.client.Builds.TargetOutputs.Get(
-			context.TODO(),
+			m.ctx,
 			params,
 		)
 		if err != nil {