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/resolve_ws_config_imagebuild.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/resolve_ws_config_prebuild_code.golden

@@ -14,4 +14,4 @@
         "tasks": "[{\"init\":\"echo 'warming up stable release of intellij...'\\nJETBRAINS_BACKEND_QUALIFIER=stable /ide-desktop/jb-launcher warmup intellij\\n\\necho 'warming up stable release of goland...'\\nJETBRAINS_BACKEND_QUALIFIER=stable /ide-desktop/jb-launcher warmup goland\\n\\necho 'warming up latest release of goland...'\\nJETBRAINS_BACKEND_QUALIFIER=latest /ide-desktop/jb-launcher warmup goland\\n\\necho 'warming up latest release of phpstorm...'\\nJETBRAINS_BACKEND_QUALIFIER=latest /ide-desktop/jb-launcher warmup phpstorm\",\"name\":\"GITPOD_JB_WARMUP_TASK\"}]\n"
     },
     "Err": ""
-}
+}

components/ide-service/pkg/server/testdata/resolve_ws_config_prebuild_empty.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/resolve_ws_config_prebuild_intellij.golden

@@ -12,4 +12,4 @@
         "tasks": "[{\"init\":\"echo 'warming up stable release of intellij...'\\nJETBRAINS_BACKEND_QUALIFIER=stable /ide-desktop/jb-launcher warmup intellij\\n\\necho 'warming up latest release of intellij...'\\nJETBRAINS_BACKEND_QUALIFIER=latest /ide-desktop/jb-launcher warmup intellij\",\"name\":\"GITPOD_JB_WARMUP_TASK\"}]\n"
     },
     "Err": ""
-}
+}

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

@@ -14,4 +14,4 @@
         "tasks": "[{\"init\":\"echo 'warming up stable release of intellij...'\\nJETBRAINS_BACKEND_QUALIFIER=stable /ide-desktop/jb-launcher warmup intellij\\n\\necho 'warming up stable release of goland...'\\nJETBRAINS_BACKEND_QUALIFIER=stable /ide-desktop/jb-launcher warmup goland\\n\\necho 'warming up latest release of goland...'\\nJETBRAINS_BACKEND_QUALIFIER=latest /ide-desktop/jb-launcher warmup goland\\n\\necho 'warming up latest release of phpstorm...'\\nJETBRAINS_BACKEND_QUALIFIER=latest /ide-desktop/jb-launcher warmup phpstorm\",\"name\":\"GITPOD_JB_WARMUP_TASK\"}]\n"
     },
     "Err": ""
-}
+}

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

@@ -10,4 +10,4 @@
         "referer_ide": "intellij"
     },
     "Err": ""
-}
+}

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

@@ -10,4 +10,4 @@
         "referer_ide": "goland"
     },
     "Err": ""
-}
+}

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

@@ -10,4 +10,4 @@
         "referer_ide": "intellij"
     },
     "Err": ""
-}
+}

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

@@ -16,4 +16,4 @@
         "referer_ide": "intellij"
     },
     "Err": ""
-}
+}

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

@@ -16,4 +16,4 @@
         "referer_ide": "intellij"
     },
     "Err": ""
-}
+}

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

@@ -16,4 +16,4 @@
         "referer_ide": "goland"
     },
     "Err": ""
-}
+}

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

@@ -15,4 +15,4 @@
         ]
     },
     "Err": ""
-}
+}

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

@@ -15,4 +15,4 @@
         ]
     },
     "Err": ""
-}
+}

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

@@ -15,4 +15,4 @@
         ]
     },
     "Err": ""
-}
+}

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

@@ -15,4 +15,4 @@
         ]
     },
     "Err": ""
-}
+}

components/ide/jetbrains/launcher/main.go

@@ -161,6 +161,27 @@ func main() {
 	}
 
 	if launchCtx.warmup {
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		for {
+			tasks, err := getTasksList(ctx)
+			if err != nil {
+				log.Fatalf("cannot get task list: %s", err)
+			}
+
+			var runningTasksCounter int
+			for _, task := range tasks {
+				if task.State != 2 && task.Presentation.Name != "GITPOD_JB_WARMUP_TASK" {
+					runningTasksCounter++
+				}
+			}
+			if runningTasksCounter == 0 {
+				break
+			}
+			time.Sleep(1 * time.Second)
+		}
+
 		launch(launchCtx)
 		return
 	}
@@ -853,3 +874,28 @@ func resolveProjectContextDir(launchCtx *LaunchContext) string {
 
 	return launchCtx.projectDir
 }
+
+func getTasksList(ctx context.Context) ([]*supervisor.TaskStatus, error) {
+	conn, err := dial(ctx)
+	if err != nil {
+		return nil, err
+	}
+	client := supervisor.NewStatusServiceClient(conn)
+	respClient, err := client.TasksStatus(ctx, &supervisor.TasksStatusRequest{Observe: false})
+	if err != nil {
+		return nil, xerrors.Errorf("failed get tasks status client: %w", err)
+	}
+	resp, err := respClient.Recv()
+	if err != nil {
+		return nil, xerrors.Errorf("failed receive data: %w", err)
+	}
+	return resp.GetTasks(), 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
+}