package cmd

import (
	"fmt"
	"log/slog"

	"github.com/portainer/kubesolo-os/update/pkg/image"
	"github.com/portainer/kubesolo-os/update/pkg/partition"
)

// Check queries the update server for available updates and compares
// against the currently running version.
func Check(args []string) error {
	opts := parseOpts(args)

	if opts.ServerURL == "" {
		return fmt.Errorf("--server is required (no default update server configured)")
	}

	// Get current version from active partition
	env := opts.NewBootEnv()
	activeSlot, err := env.ActiveSlot()
	if err != nil {
		return fmt.Errorf("reading active slot: %w", err)
	}

	partInfo, err := partition.GetSlotPartition(activeSlot)
	if err != nil {
		return fmt.Errorf("finding active partition: %w", err)
	}

	mountPoint := "/tmp/kubesolo-check-" + activeSlot
	if err := partition.MountReadOnly(partInfo.Device, mountPoint); err != nil {
		return fmt.Errorf("mounting active partition: %w", err)
	}
	defer partition.Unmount(mountPoint)

	currentVersion, err := partition.ReadVersion(mountPoint)
	if err != nil {
		slog.Warn("could not read current version", "error", err)
		currentVersion = "unknown"
	}

	// Check update server
	client := image.NewClient(opts.ServerURL, "")
	meta, err := client.CheckForUpdate()
	if err != nil {
		return fmt.Errorf("checking for update: %w", err)
	}

	fmt.Printf("Current version: %s (slot %s)\n", currentVersion, activeSlot)
	fmt.Printf("Latest version:  %s\n", meta.Version)

	if meta.Version == currentVersion {
		fmt.Println("Status: up to date")
	} else {
		fmt.Println("Status: update available")
		if meta.ReleaseNotes != "" {
			fmt.Printf("Release notes: %s\n", meta.ReleaseNotes)
		}
	}

	return nil
}