package version

import (
	"context"

	"github.com/google/go-github/github"
	"oss.terrastruct.com/cmdlog"
)

// Pre-built binaries will have version set during build time.
var Version = "master (built from source)"

func CheckVersion(ctx context.Context, logger *cmdlog.Logger) {
	logger.Info.Printf("D2 version: %s\n", Version)

	if Version == "master (built from source)" {
		return
	}

	logger.Info.Printf("Checking for updates...")
	latest, err := getLatestVersion(ctx)
	if err != nil {
		logger.Debug.Printf("Error reaching Github for latest version: %s", err.Error())
	} else if Version != "master" && Version != latest {
		logger.Info.Printf("A new version of D2 is available: %s -> %s", Version, latest)
	}
}

func getLatestVersion(ctx context.Context) (string, error) {
	client := github.NewClient(nil)
	rep, _, err := client.Repositories.GetLatestRelease(ctx, "terrastruct", "d2")

	if err != nil {
		return "", err
	}

	return *rep.TagName, nil
}