Health check doesn't wait for everything to start

pkg/api/controllers/provider/health.go

@@ -0,0 +1,37 @@
+// Copyright 2024 Daytona Platforms Inc.
+// SPDX-License-Identifier: Apache-2.0
+
+package provider
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/daytonaio/daytona/pkg/api/controllers/provider/dto"
+	"github.com/daytonaio/daytona/pkg/server"
+	"github.com/gin-gonic/gin"
+)
+
+// HealthCheck      godoc
+//
+//	@Tags			provider
+//	@Summary		Provider health check
+//	@Description	Provider health check
+//	@Success		200
+//	@Router			/provider/health [get]
+//	@id				HealthCheck
+func HealthCheck(ctx *gin.Context) {
+	var req dto.Provider
+	err := ctx.BindJSON(&req)
+	if err != nil {
+		ctx.AbortWithError(http.StatusBadRequest, fmt.Errorf("invalid request body: %w", err))
+		return
+	}
+	server := server.GetInstance(nil)
+	exist := server.ProviderManager.IsInitialized(req.Name)
+	if !exist {
+		ctx.AbortWithError(http.StatusInternalServerError, fmt.Errorf("failed to initialize provider: %w", err))
+		return
+	}
+	ctx.Status(200)
+}

pkg/api/server.go

@@ -236,6 +236,7 @@ func (a *ApiServer) Start() error {
 		providerController.GET("/", provider.ListProviders)
 		providerController.POST("/:provider/uninstall", provider.UninstallProvider)
 		providerController.GET("/:provider/target-manifest", provider.GetTargetManifest)
+		providerController.GET("/health", provider.HealthCheck)
 	}
 
 	containerRegistryController := protected.Group("/container-registry")

pkg/provider/manager/manager.go

@@ -46,6 +46,7 @@ type IProviderManager interface {
 	TerminateProviderProcesses(providersBasePath string) error
 	UninstallProvider(name string) error
 	Purge() error
+	IsInitialized(providerName string) bool
 }
 
 type ProviderManagerConfig struct {
@@ -335,3 +336,11 @@ func (m *ProviderManager) dispenseProvider(client *plugin.Client, name string) (
 
 	return &provider, nil
 }
+
+func (m *ProviderManager) IsInitialized(providerName string) bool {
+	pluginRef, exists := m.pluginRefs[providerName]
+	if !exists {
+		return false
+	}
+	return pluginRef != nil
+}