Method AgentHandler

apiserver/controllers/agent.go:18–61

(w http.ResponseWriter, r *http.Request)

)

func (a *APIController) AgentHandler(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	instance, err := auth.InstanceParams(ctx)
	if err != nil {
		w.WriteHeader(http.StatusForbidden)
		slog.ErrorContext(ctx, "failed to authenticate instance")
		return
	}

	conn, err := a.upgrader.Upgrade(w, r, nil)
	if err != nil {
		slog.With(slog.Any("error", err)).ErrorContext(ctx, "error upgrading to websockets")
		return
	}

	slog.DebugContext(ctx, "new agent connected", "agent_name", instance.Name)
	agent, err := agent.NewAgent(ctx, conn, instance, a.r)
	if err != nil {
		slog.With(slog.Any("error", err)).ErrorContext(ctx, "failed to create agent")
		return
	}
	defer func() {
		slog.DebugContext(ctx, "stopping agent", "agent_name", instance.Name)
		agent.Stop()
	}()

	if err := agent.Start(); err != nil {
		slog.ErrorContext(ctx, "failed to start agent loop", "error", err, "agent_name", instance.Name)
		handleError(ctx, w, err)
		return
	}

	if err := a.agentHub.RegisterAgent(agent); err != nil {
		handleError(ctx, w, err)
		return
	}
	defer a.agentHub.UnregisterAgent(instance.Name)

	select {
	case <-agent.Done():
	case <-ctx.Done():
	}
	slog.InfoContext(ctx, "connection closed", "agent_name", instance.Name)
}

func (a *APIController) AgentShellHandler(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()