wehub-resource-sync
108 lines
2.4 KiB
Go
108 lines
2.4 KiB
Go
|
|
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
|
|
"service-operation/operations"
|
|
"service-operation/types"
|
|
)
|
|
|
|
func (h *OperationHandler) HandleOperation(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
var req types.OperationRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
|
http.Error(w, "Invalid JSON payload", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
if req.Host == "" && req.URL == "" {
|
|
http.Error(w, "Host or URL is required", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
// Set defaults
|
|
if req.Count <= 0 {
|
|
req.Count = h.config.DefaultCount
|
|
}
|
|
if req.Count > h.config.MaxCount {
|
|
req.Count = h.config.MaxCount
|
|
}
|
|
|
|
if req.Timeout <= 0 {
|
|
req.Timeout = int(h.config.DefaultTimeout.Seconds())
|
|
}
|
|
if time.Duration(req.Timeout)*time.Second > h.config.MaxTimeout {
|
|
req.Timeout = int(h.config.MaxTimeout.Seconds())
|
|
}
|
|
|
|
timeout := time.Duration(req.Timeout) * time.Second
|
|
var result *types.OperationResult
|
|
var err error
|
|
|
|
switch req.Type {
|
|
case types.OperationPing:
|
|
pingOp := operations.NewPingOperation(timeout)
|
|
result, err = pingOp.Execute(req.Host, req.Count)
|
|
|
|
case types.OperationDNS:
|
|
dnsOp := operations.NewDNSOperation(timeout)
|
|
query := req.Query
|
|
if query == "" {
|
|
query = "A"
|
|
}
|
|
result, err = dnsOp.Execute(req.Host, query)
|
|
|
|
case types.OperationTCP:
|
|
if req.Port <= 0 {
|
|
http.Error(w, "Port is required for TCP operations", http.StatusBadRequest)
|
|
return
|
|
}
|
|
tcpOp := operations.NewTCPOperation(timeout)
|
|
result, err = tcpOp.Execute(req.Host, req.Port)
|
|
|
|
case types.OperationHTTP:
|
|
httpOp := operations.NewHTTPOperation(timeout)
|
|
url := req.URL
|
|
if url == "" {
|
|
url = req.Host
|
|
}
|
|
method := req.Method
|
|
if method == "" {
|
|
method = "GET"
|
|
}
|
|
result, err = httpOp.Execute(url, method)
|
|
|
|
case types.OperationSSL:
|
|
sslOp := operations.NewSSLOperation(timeout)
|
|
result, err = sslOp.Execute(req.Host)
|
|
|
|
default:
|
|
http.Error(w, "Invalid operation type", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
result = &types.OperationResult{
|
|
Type: req.Type,
|
|
Host: req.Host,
|
|
Port: req.Port,
|
|
Success: false,
|
|
Error: err.Error(),
|
|
}
|
|
}
|
|
|
|
// Save metrics to PocketBase if available
|
|
if h.pbClient != nil && h.pbClient.IsAuthenticated() {
|
|
go h.saveMetricsToPocketBase(result, req.ServiceID)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(result)
|
|
} |