package version

import (
	"cmp"
	"fmt"
	"strconv"
	"strings"
	"time"
)

// From https://github.com/superfly/flyctl/blob/0dff860a878e2b280f2f53ce2aaf21ce39d800c2/internal/version/version.go
// This code is subject to Apache-2.0 license as per flyctl project

type InvalidVersionError struct {
	val     string
	message string
}

func (e *InvalidVersionError) Error() string {
	return fmt.Sprintf("invalid version %q: %s", e.val, e.message)
}

func New(t time.Time, channel string, buildNum int) Version {
	return Version{
		Major:   t.Year(),
		Minor:   int(t.Month()),
		Patch:   t.Day(),
		Channel: channel,
		Build:   buildNum,
	}
}

type Version struct {
	Major     int
	Minor     int
	Patch     int
	Build     int
	Channel   string
	BuildMeta string
}

func (v Version) String() string {
	return v.baseString() + v.suffixString() + v.buildSuffixString()
}

func (v Version) baseString() string {
	return fmt.Sprintf("%d.%d.%d", v.Major, v.Minor, v.Patch)
}

func (v Version) suffixString() string {
	// TODO[md]: remove this when we're done with the semver to calver migration
	if !IsCalVer(v) && !isDev(v) {
		// version is bumped on every release -- no channel or build on stable
		if v.Channel == "stable" || v.Channel == "" {
			return ""
		}

		if v.Build > 0 {
			return fmt.Sprintf("-%s-%d", v.Channel, v.Build)
		}

		return fmt.Sprintf("-%s", v.Channel)
	}

	if v.Channel != "" && v.Build != 0 {
		return fmt.Sprintf("-%s.%d", v.Channel, v.Build)
	}

	if v.Channel != "" && v.Build == 0 {
		return fmt.Sprintf("-%s", v.Channel)
	}

	return ""
}

func (v Version) buildSuffixString() string {
	if v.BuildMeta != "" {
		return fmt.Sprintf("+%s", v.BuildMeta)
	}
	return ""
}

// flag to indicate which side of the semver to calver migration we're on. drop when we're done
func IsCalVer(v Version) bool {
	return v.Major > 2022
}

func isDev(v Version) bool {
	return v.Major == 0 && v.Minor == 0 && v.Patch == 0 && v.Channel == "dev"
}

func (v Version) Equal(other Version) bool {
	return Compare(v, other) == 0
}

func (v Version) Newer(other Version) bool {
	return Compare(v, other) == 1
}

func (v Version) Older(other Version) bool {
	return Compare(v, other) == -1
}

func ChannelFromCalverOrSemver(v Version) string {
	if IsCalVer(v) {
		return v.Channel
	}
	return "stable"
}

// Compare returns
//
//	-1 if x is less than y,
//	 0 if x equals y,
//	+1 if x is greater than y.
//
// Versions with channels are considered less than versions without channels, as
// per semver spec. If both versions have channels, they are compared as strings.
// A channel of "stable" is greater than any other channel.
func Compare(x Version, y Version) int {
	if ret := cmp.Compare(x.Major, y.Major); ret != 0 {
		return ret
	}

	if ret := cmp.Compare(x.Minor, y.Minor); ret != 0 {
		return ret
	}

	if ret := cmp.Compare(x.Patch, y.Patch); ret != 0 {
		return ret
	}

	// in semver, if one version has a prerel and the other doesn't, the one without is newer
	if x.Channel == "" && y.Channel != "" {
		return 1
	} else if y.Channel == "" && x.Channel != "" {
		return -1
	} else {
		if ret := strings.Compare(x.Channel, y.Channel); ret != 0 {
			return ret
		}
	}

	if ret := cmp.Compare(x.Build, y.Build); ret != 0 {
		return ret
	}

	return 0
}

func (v Version) dateFromVersion() time.Time {
	return time.Date(v.Major, time.Month(v.Minor), v.Patch, 0, 0, 0, 0, time.UTC)
}

func (v Version) SignificantlyBehind(latest Version) bool {
	// both versions are calver, use date comparison. >28 days is old
	if IsCalVer(latest) && IsCalVer(v) {
		latestDate := latest.dateFromVersion()
		currentDate := v.dateFromVersion()
		return latestDate.Sub(currentDate) >= 28*24*time.Hour
	}

	// latest is calver, current is not. consider out of date if latest is >30 days old
	if IsCalVer(latest) && !IsCalVer(v) {
		latestDate := latest.dateFromVersion()
		return time.Until(latestDate) >= 28*24*time.Hour
	}

	// both are old format, consider 5 patches old
	if latest.Major > v.Major {
		return true
	}
	if latest.Minor > v.Minor {
		return true
	}
	if latest.Patch > v.Patch+5 {
		return true
	}
	return false
}

func (v Version) Increment(t time.Time) Version {
	if !IsCalVer(v) {
		return Version{
			Major:   v.Major,
			Minor:   v.Minor,
			Patch:   v.Patch + 1,
			Build:   0,
			Channel: v.Channel,
		}
	}

	buildNum := 0
	if v.Major == t.Year() && v.Minor == int(t.Month()) && v.Patch == t.Day() {
		buildNum = v.Build
	}
	buildNum++
	return New(t, v.Channel, buildNum)
}

func Parse(version string) (Version, error) {
	version = strings.TrimPrefix(version, "v")

	out := Version{}

	parts := strings.SplitN(version, "-", 2)
	// versionStr contains "MAJOR.MINOR.PATCH"
	versionStr := parts[0]
	suffixStr := ""
	// if parts has a length of 2, suffixStr contains "CHANNEL.BUILD" or "CHANNEL-BUILD" (latter is old format)
	// suffix may also contain "+BUILD.META"
	if len(parts) == 2 {
		suffixStr = parts[1]
	}

	// split versionStr into "MAJOR", "MINOR", "PATCH"
	parts = strings.SplitN(versionStr, ".", 3)

	// version must have 3 parts
	if len(parts) != 3 {
		return Version{}, &InvalidVersionError{version, "must begin with YEAR.MONTH.DAY or MAJOR.MINOR.PATCH"}
	}

	// only reject zero padding on calver strings
	if parts[0] != "0" {
		// if any part is zero padded, return an error
		for _, part := range parts {
			if part[0] == '0' {
				return Version{}, &InvalidVersionError{version, "date parts cannot be zero padded"}
			}
		}
	}

	// if any part is not an integer, return an error
	if x, err := strconv.Atoi(parts[0]); err != nil {
		return Version{}, &InvalidVersionError{version, err.Error()}
	} else {
		out.Major = x
	}
	if x, err := strconv.Atoi(parts[1]); err != nil {
		return Version{}, &InvalidVersionError{version, err.Error()}
	} else {
		out.Minor = x
	}
	if x, err := strconv.Atoi(parts[2]); err != nil {
		return Version{}, &InvalidVersionError{version, err.Error()}
	} else {
		out.Patch = x
	}

	if suffixStr != "" {
		// if suffix contains a "+", put everything after the plux into BuildMeta and remove it from suffixStr
		parts = strings.Split(suffixStr, "+")
		if len(parts) == 2 {
			out.BuildMeta = parts[1]
			suffixStr = parts[0]
		}

		// handle old v0.1.xxx format first, which separated channel and build with a dash
		// old channels began with either "pre-", or "beta-"
		if !IsCalVer(out) && (strings.HasPrefix(suffixStr, "pre-") || strings.HasPrefix(suffixStr, "beta-")) {
			parts = strings.SplitN(suffixStr, "-", 2)
		} else {
			// handle new calver format, which separates channel and build with a dot
			if pos := strings.LastIndex(suffixStr, "."); pos >= 0 {
				parts = []string{suffixStr[:pos], suffixStr[pos+1:]}
			} else {
				parts = []string{suffixStr}
			}
		}

		out.Channel = parts[0]

		// handle `-channel.build` suffix
		if len(parts) == 2 {
			if x, err := strconv.Atoi(parts[1]); err != nil {
				// if build is not an integer, return an error
				return Version{}, &InvalidVersionError{version, err.Error()}
			} else {
				out.Build = x
			}
		} else {
			// if no build was given, default to zero
			out.Build = 0
		}
	} else {
		// if no suffix was given, default to no channel and zero build
		out.Build = 0
		out.Channel = ""
	}

	return out, nil
}

func (v Version) MarshalJSON() ([]byte, error) {
	return []byte(fmt.Sprintf("\"%s\"", v.String())), nil
}

func (v *Version) UnmarshalJSON(data []byte) error {
	// Ignore null, like in the main JSON package.
	if string(data) == "null" || string(data) == `""` {
		return nil
	}
	str, err := strconv.Unquote(string(data))
	if err != nil {
		return err
	}
	decodedVer, err := Parse(str)
	if err != nil {
		return err
	}
	*v = decodedVer
	return nil
}
