jetbrains: run warmup as last task

Co-authored-by: Anton Kosyakov <[email protected]>
Co-authored-by: Victor Nogueira <[email protected]>

Author: 3 people

components/ide-service/pkg/server/testdata/ideconfig_happypath.golden

@@ -4,4 +4,4 @@
         "web_image": "eu.gcr.io/gitpod-core-dev/build/ide/code:commit-d6329814c2aa34c414574fd0d1301447d6fe82c9"
     },
     "Err": ""
-}
+}

components/ide-service/pkg/server/testdata/ideconfig_happypath_resolve.golden

@@ -10,4 +10,4 @@
         "web_image": "eu.gcr.io/gitpod-core-dev/build/ide/code:commit-d6329814c2aa34c414574fd0d1301447d6fe82c9"
     },
     "Err": ""
-}
+}

components/ide-service/pkg/server/testdata/ideconfig_idetype_not_correct.golden

@@ -161,6 +161,22 @@ func main() {
 	}
 
 	if launchCtx.warmup {
+		ctx, cancel := context.WithCancel(context.Background())
+
+	out:
+		for {
+			log.Debug("launcher: wait for tasks to finish before running warmup")
+			err := waitForTasksToFinish(ctx, cancel)
+			if err != nil {
+				log.WithError(err).Warn("launcher: failed to observe tasks completion")
+			}
+			select {
+			case <-ctx.Done():
+				break out
+			case <-time.After(1 * time.Second):
+			}
+		}
+
 		launch(launchCtx)
 		return
 	}
@@ -853,3 +869,42 @@ func resolveProjectContextDir(launchCtx *LaunchContext) string {
 
 	return launchCtx.projectDir
 }
+
+func waitForTasksToFinish(ctx context.Context, cancel context.CancelFunc) error {
+	conn, err := dial(ctx)
+	if err != nil {
+		return err
+	}
+	client := supervisor.NewStatusServiceClient(conn)
+	tasksResponse, err := client.TasksStatus(ctx, &supervisor.TasksStatusRequest{Observe: true})
+	if err != nil {
+		return xerrors.Errorf("failed get tasks status client: %w", err)
+	}
+	for {
+		resp, err := tasksResponse.Recv()
+		if err != nil {
+			return err
+		}
+
+		var runningTasksCounter int
+		for _, task := range resp.Tasks {
+			if task.State != 2 && task.Presentation.Name != "GITPOD_JB_WARMUP_TASK" {
+				runningTasksCounter++
+			}
+		}
+		if runningTasksCounter == 0 {
+			cancel()
+		}
+
+		return nil
+	}
+
+}
+
+func dial(ctx context.Context) (*grpc.ClientConn, error) {
+	supervisorConn, err := grpc.DialContext(ctx, util.GetSupervisorAddress(), grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		err = xerrors.Errorf("failed connecting to supervisor: %w", err)
+	}
+	return supervisorConn, err
+}