export: align reanimator contract v2.7

internal/server/handlers.go

@@ -1510,6 +1510,69 @@ func (s *Server) handleCollectStart(w http.ResponseWriter, r *http.Request) {
 	_ = json.NewEncoder(w).Encode(job.toJobResponse("Collection job accepted"))
 }
 
+func (s *Server) handleCollectProbe(w http.ResponseWriter, r *http.Request) {
+	var req CollectRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		jsonError(w, "Invalid JSON body", http.StatusBadRequest)
+		return
+	}
+	if err := validateCollectRequest(req); err != nil {
+		jsonError(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	connector, ok := s.getCollector(req.Protocol)
+	if !ok {
+		jsonError(w, "Коннектор для протокола не зарегистрирован", http.StatusBadRequest)
+		return
+	}
+	prober, ok := connector.(collector.Prober)
+	if !ok {
+		jsonError(w, "Проверка подключения для протокола не поддерживается", http.StatusBadRequest)
+		return
+	}
+
+	ctx, cancel := context.WithTimeout(r.Context(), 20*time.Second)
+	defer cancel()
+
+	result, err := prober.Probe(ctx, toCollectorRequest(req))
+	if err != nil {
+		jsonError(w, "Проверка подключения не удалась: "+err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	message := "Связь с BMC установлена"
+	if result != nil {
+		switch {
+		case !result.HostPoweredOn && result.PowerControlAvailable:
+			message = "Связь с BMC установлена, host выключен. Можно включить перед сбором."
+		case !result.HostPoweredOn:
+			message = "Связь с BMC установлена, host выключен."
+		default:
+			message = "Связь с BMC установлена, host включен."
+		}
+	}
+
+	hostPowerState := ""
+	hostPoweredOn := false
+	powerControlAvailable := false
+	reachable := false
+	if result != nil {
+		reachable = result.Reachable
+		hostPowerState = strings.TrimSpace(result.HostPowerState)
+		hostPoweredOn = result.HostPoweredOn
+		powerControlAvailable = result.PowerControlAvailable
+	}
+
+	jsonResponse(w, CollectProbeResponse{
+		Reachable:             reachable,
+		Protocol:              req.Protocol,
+		HostPowerState:        hostPowerState,
+		HostPoweredOn:         hostPoweredOn,
+		PowerControlAvailable: powerControlAvailable,
+		Message:               message,
+	})
+}
+
 func (s *Server) handleCollectStatus(w http.ResponseWriter, r *http.Request) {
 	jobID := strings.TrimSpace(r.PathValue("id"))
 	if !isValidCollectJobID(jobID) {
@@ -1787,14 +1850,15 @@ func applyCollectSourceMetadata(result *models.AnalysisResult, req CollectReques
 
 func toCollectorRequest(req CollectRequest) collector.Request {
 	return collector.Request{
-		Host:     req.Host,
-		Protocol: req.Protocol,
-		Port:     req.Port,
-		Username: req.Username,
-		AuthType: req.AuthType,
-		Password: req.Password,
-		Token:    req.Token,
-		TLSMode:  req.TLSMode,
+		Host:             req.Host,
+		Protocol:         req.Protocol,
+		Port:             req.Port,
+		Username:         req.Username,
+		AuthType:         req.AuthType,
+		Password:         req.Password,
+		Token:            req.Token,
+		TLSMode:          req.TLSMode,
+		PowerOnIfHostOff: req.PowerOnIfHostOff,
 	}
 }