OpenFlare/internal/apps/openflare/uptimekuma/sync_helpers.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package uptimekuma

import (
	"context"
	"errors"
	"fmt"
	"log/slog"
	"strings"
	"time"

	"github.com/Rain-kl/Wavelet/internal/model"
)

const monitorListWaitTimeout = 5 * time.Second

// validateKumaConfig 验证 kumaConfig 配置完整性
func validateKumaConfig(config *kumaConfig) error {
	if strings.TrimSpace(config.URL) == "" {
		return errors.New("uptime Kuma URL is not configured")
	}
	if strings.TrimSpace(config.Username) == "" {
		return errors.New("uptime Kuma username is not configured")
	}
	if strings.TrimSpace(config.Password) == "" {
		return errors.New("uptime Kuma password is not configured")
	}
	return nil
}

func connectAndLoginUptimeKuma(kumaURL, kumaUsername, kumaPassword string) (*SocketIOClient, error) {
	slog.Debug("Connecting to Uptime Kuma socket endpoint", "url", kumaURL)
	client := NewSocketIOClient(kumaURL)
	if err := client.Connect(); err != nil {
		slog.Error("Failed to connect to Uptime Kuma endpoint", "url", kumaURL, "error", err)
		return nil, fmt.Errorf("failed to connect to Uptime Kuma: %w", err)
	}

	slog.Debug("Sending login request to Uptime Kuma", "username", kumaUsername)
	loginAck, err := client.Emit("login", map[string]string{
		"username": kumaUsername,
		"password": kumaPassword,
	})
	if err != nil {
		client.Close()
		slog.Error("Failed to send login request to Uptime Kuma", "username", kumaUsername, "error", err)
		return nil, fmt.Errorf("login request failed: %w", err)
	}

	var loginResult struct {
		Ok bool `json:"ok"`
	}
	if err := ParseAckResponse(loginAck, &loginResult); err != nil || !loginResult.Ok {
		client.Close()
		slog.Error("Uptime Kuma login verification failed", "username", kumaUsername, "error", err)
		return nil, fmt.Errorf("login failed: %w", err)
	}
	slog.Debug("Successfully logged into Uptime Kuma", "username", kumaUsername)

	slog.Debug("Waiting for monitor list push from Uptime Kuma")
	select {
	case <-client.GetMonitorListChan():
		slog.Debug("Received monitor list from Uptime Kuma")
	case <-time.After(monitorListWaitTimeout):
		client.Close()
		slog.Error("Timeout waiting for Uptime Kuma monitorList push event")
		return nil, errors.New("timeout waiting for monitorList event from Uptime Kuma")
	}
	return client, nil
}

func syncRouteMonitors(ctx context.Context, client *SocketIOClient, expectedRoutes []*model.ProxyRoute, existingMonitors map[string]Monitor, openFlareTagID int, config *kumaConfig) map[string]bool {
	expectedSitesMap := make(map[string]bool, len(expectedRoutes))
	for _, route := range expectedRoutes {
		expectedSitesMap[route.SiteName] = true
		targetURL, urlErr := routeMonitorURL(ctx, route)
		if urlErr != nil {
			slog.Error("Failed to resolve monitor URL", "name", route.SiteName, "error", urlErr)
			continue
		}

		existing, exists := existingMonitors[route.SiteName]
		if !exists {
			if err := createMonitor(client, route.SiteName, targetURL, openFlareTagID, config); err != nil {
				slog.Error("Failed to add monitor to Uptime Kuma", "name", route.SiteName, "error", err)
			}
			continue
		}
		if monitorNeedsUpdate(existing, targetURL, config) {
			if err := updateMonitor(client, existing.ID, route.SiteName, targetURL, config); err != nil {
				slog.Error("Failed to edit monitor in Uptime Kuma", "name", route.SiteName, "error", err)
			}
		}
	}
	return expectedSitesMap
}

func removeStaleMonitors(client *SocketIOClient, existingMonitors map[string]Monitor, expectedSitesMap map[string]bool) {
	for name, monitor := range existingMonitors {
		if expectedSitesMap[name] {
			continue
		}
		slog.Info("Deleting monitor in Uptime Kuma", "name", name, "monitorID", monitor.ID)
		deleteAck, err := client.Emit("deleteMonitor", monitor.ID)
		if err != nil {
			slog.Error("Failed to delete monitor in Uptime Kuma", "name", name, "monitorID", monitor.ID, "error", err)
			continue
		}
		if err := ParseAckResponse(deleteAck, nil); err != nil {
			slog.Error("Failed to parse delete monitor result", "name", name, "monitorID", monitor.ID, "error", err)
		}
	}
}