v/python/version.go

package python

import (
	"errors"
	"io/ioutil"
	"os"
	"path"
	"strings"
	exec "v/exec"
	state "v/state"
)

func VersionStringToStruct(version string) VersionTag {
	splitVersion := strings.Split(version, ".")

	return VersionTag{Major: splitVersion[0], Minor: splitVersion[1], Patch: splitVersion[2]}
}

func ValidateVersion(version string) error {
	if splitVersion := strings.Split(version, "."); len(splitVersion) != 3 {
		return errors.New("Invalid version string. Expected format 'a.b.c'.")
	}

	return nil
}

type SelectedVersion struct {
	Version string
	Source  string
}

func ListInstalledVersions() ([]string, error) {
	if ensureErr := state.EnsureStatePath("runtimes"); ensureErr != nil {
		return []string{}, ensureErr
	}

	runtimesDir := state.GetStatePath("runtimes", "python")
	entries, err := os.ReadDir(runtimesDir)

	if os.IsNotExist(err) {
		return []string{}, nil
	}

	if err != nil {
		return []string{}, err
	}

	installedVersions := []string{}

	for _, d := range entries {
		installedVersions = append(installedVersions, d.Name())
	}

	return installedVersions, nil
}

// SearchForPythonVersionFile crawls up to the system root to find any
// .python-version file that could set the current version.
func SearchForPythonVersionFile() (SelectedVersion, bool) {
	currentPath, _ := os.Getwd()
	var versionFound string
	for {
		content, err := ioutil.ReadFile(path.Join(currentPath, ".python-version"))

		if err == nil {
			versionFound = strings.TrimSpace(string(content))
			break
		}

		nextPath := path.Dir(currentPath)

		if currentPath == nextPath {
			break
		}

		currentPath = nextPath
	}

	if versionFound == "" {
		return SelectedVersion{}, false
	}

	return SelectedVersion{Version: versionFound, Source: path.Join(currentPath, ".python-version")}, true
}

// DetermineSelectedPythonVersion returns the Python runtime version that should be
// used according to v.
//
// First, v will look in the current directory and all its parents for a .python-version
// file that would indicate which version is preferred. If none are found, the global
// user-defined version (via `v use <version>`) is used. If there is none, the system
// Python version is used.
func DetermineSelectedPythonVersion(currentState state.State) (SelectedVersion, error) {
	pythonFileVersion, pythonFileVersionFound := SearchForPythonVersionFile()

	if pythonFileVersionFound {
		return pythonFileVersion, nil
	}

	if len(currentState.GlobalVersion) != 0 {
		return SelectedVersion{Version: currentState.GlobalVersion, Source: state.GetStatePath("state.json")}, nil
	}

	systemVersion, _ := DetermineSystemPython()
	return SelectedVersion{Source: "system", Version: systemVersion}, nil
}

// DetermineSystemPython returns the unshimmed Python version and path.
// It assumes that /bin/python is where system Python lives.
func DetermineSystemPython() (string, string) {
	versionOut, _ := exec.RunCommand([]string{"/bin/python", "--version"}, state.GetStatePath())
	detectedVersion, _ := strings.CutPrefix(versionOut, "Python")
	return strings.TrimSpace(detectedVersion), "/bin/python"
}

// Gets all install versions.
func GetAvailableVersions() []string {
	entries, _ := os.ReadDir(state.GetStatePath("runtimes", "python"))

	versions := []string{}

	for _, d := range entries {
		versions = append(versions, d.Name())
	}

	return versions
}