v/state.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"os"
	"path"
	"strings"
)

// Persistent state used by the CLI to track runtime information
// between calls.
type State struct {
	GlobalVersion string `json:"globalVersion"`
}

func GetStatePath(pathSegments ...string) string {
	home, _ := os.UserHomeDir()
	userDefinedRoot, found := os.LookupEnv("V_ROOT")

	root := path.Join(home, ".v")

	if found {
		root = userDefinedRoot
	}
	allSegments := []string{root}
	allSegments = append(allSegments, pathSegments...)
	return path.Join(allSegments...)
}

func ReadState() State {
	c, _ := ioutil.ReadFile(GetStatePath("state.json"))

	state := State{}

	json.Unmarshal(c, &state)

	return state
}

func WriteState(version string) {
	state := State{GlobalVersion: version}

	d, _ := json.Marshal(state)
	ioutil.WriteFile(GetStatePath("state.json"), d, 0750)
}

func GetAvailableVersions() []string {
	entries, _ := os.ReadDir(GetStatePath("runtimes"))

	versions := []string{}

	for _, d := range entries {
		versions = append(versions, strings.TrimPrefix(d.Name(), "py-"))
	}

	return versions
}