package updater

import (
	"encoding/json"
	"fmt"
	"context"
	"net/http"
	"net/url"
	"strings"

	coreerr "forge.lthn.ai/core/go-log"
)

// GenericUpdateInfo holds the information from a latest.json file.
// This file is expected to be at the root of a generic HTTP update server.
type GenericUpdateInfo struct {
	Version string `json:"version"` // The version number of the update.
	URL     string `json:"url"`     // The URL to download the update from.
}

// GetLatestUpdateFromURL fetches and parses a latest.json file from a base URL.
// The server at the baseURL should host a 'latest.json' file that contains
// the version and download URL for the latest update.
//
// Example of latest.json:
//
//	{
//	  "version": "1.2.3",
//	  "url": "https://your-server.com/path/to/release-asset"
//	}
func GetLatestUpdateFromURL(baseURL string) (*GenericUpdateInfo, error) {
	u, err := url.Parse(baseURL)
	if err != nil {
		return nil, coreerr.E("GetLatestUpdateFromURL", "invalid base URL", err)
	}

	// Append latest.json to the path
	u.Path = strings.TrimSuffix(u.Path, "/") + "/latest.json"

	req, err := newAgentRequest(context.Background(), "GET", u.String())
	if err != nil {
		return nil, coreerr.E("GetLatestUpdateFromURL", "failed to create update check request", err)
	}

	resp, err := NewHTTPClient().Do(req)
	if err != nil {
		return nil, coreerr.E("GetLatestUpdateFromURL", "failed to fetch latest.json", err)
	}
	defer func() { _ = resp.Body.Close() }()

	if resp.StatusCode != http.StatusOK {
		return nil, coreerr.E("GetLatestUpdateFromURL", fmt.Sprintf("failed to fetch latest.json: status code %d", resp.StatusCode), nil)
	}

	var info GenericUpdateInfo
	if err := json.NewDecoder(resp.Body).Decode(&info); err != nil {
		return nil, coreerr.E("GetLatestUpdateFromURL", "failed to parse latest.json", err)
	}

	if info.Version == "" || info.URL == "" {
		return nil, coreerr.E("GetLatestUpdateFromURL", "invalid latest.json content: version or url is missing", nil)
	}

	return &info, nil
}