godoxy-yusing/internal/idlewatcher/handle_http_debug.go

//go:build !production

package idlewatcher

import (
	"math/rand/v2"
	"net/http"
	"time"

	idlewatcher "github.com/yusing/godoxy/internal/idlewatcher/types"
	"github.com/yusing/godoxy/internal/types"
	gevents "github.com/yusing/goutils/events"
)

func DebugHandler(rw http.ResponseWriter, r *http.Request) {
	w := &Watcher{
		events: gevents.NewHistory(),
		cfg: &types.IdlewatcherConfig{
			IdlewatcherProviderConfig: types.IdlewatcherProviderConfig{
				Docker: &types.DockerConfig{
					ContainerName: "test",
				},
			},
		},
	}

	switch r.URL.Path {
	case idlewatcher.LoadingPageCSSPath:
		serveStaticContent(rw, http.StatusOK, "text/css", cssBytes)
	case idlewatcher.LoadingPageJSPath:
		serveStaticContent(rw, http.StatusOK, "application/javascript", jsBytes)
	case idlewatcher.WakeEventsPath:
		go w.handleWakeEventsSSE(rw, r)
		ticker := time.NewTicker(1 * time.Second)
		defer ticker.Stop()
		eventTypes := []WakeEventType{
			WakeEventStarting,
			WakeEventWakingDep,
			WakeEventDepReady,
			WakeEventContainerWoke,
			WakeEventWaitingReady,
			WakeEventError,
			WakeEventReady,
		}
		messages := []string{
			"Starting",
			"Waking dependency",
			"Dependency ready",
			"Container woke",
			"Waiting for ready",
			"Error",
			"Ready",
		}

		for {
			select {
			case <-r.Context().Done():
				return
			case <-ticker.C:
				idx := rand.IntN(len(eventTypes))
				w.sendEvent(eventTypes[idx], messages[idx], nil)
			}
		}
	default:
		w.writeLoadingPage(rw)
	}
}