main.go

package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"

	"git.cmplx.dev/yoko/adoptium"
	"git.cmplx.dev/yoko/cmd"
	"git.cmplx.dev/yoko/globals"
	"git.cmplx.dev/yoko/util"
	"github.com/mitchellh/colorstring"
	"github.com/mitchellh/go-homedir"
	"github.com/spf13/cobra"
)

func main() {

	var root = cobra.Command{Use: "yoko"}

	downloadJDKs := root.PersistentFlags().Bool("install-jdks", false, "always get requrired jdks")
	debug := root.PersistentFlags().Bool("debug", false, "run in debug mode")

	_ = debug

	root.PersistentPreRun = func(cmd *cobra.Command, args []string) {

		home, err := homedir.Dir()

		if *debug {
			home, _ = filepath.Abs("./fakehome")
		}

		if err != nil {
			log.Fatal(err)
		}

		var yokoHome = filepath.Join(home, ".yoko")
		globals.UserCacheDir = yokoHome

		jdks := []int{
			8,  // 1.6.2 - 1.17
			17, // 1.17  - 1.20.5
			21, // 1.20.5 +
		}

		missing := []int{}

		for _, v := range jdks {
			var jdkPath = filepath.Join(yokoHome, fmt.Sprintf("java%d", v))
			if _, err := os.Stat(jdkPath); err != nil {
				missing = append(missing, v)
			} else {
				globals.JDKPaths[v] = jdkPath
			}
		}

		if len(missing) > 0 {
			var missingStr = strings.Trim(strings.Join(strings.Split(fmt.Sprint(missing), " "), ", "), "[]")
			if !*downloadJDKs && !util.Confirm(fmt.Sprintf("[red]Required JDKs (%s) missing, Download?", missingStr)) {
				util.Abort()
			}

			colorstring.Println("[cyan]Downloading required JDKs...")

			for _, v := range missing {
				var jdkPath = filepath.Join(yokoHome, fmt.Sprintf("java%d", v))
				adoptium.GetJDK(v, jdkPath)
				globals.JDKPaths[v] = jdkPath
			}

		}

	}

	root.AddCommand(cmd.InitCmd, cmd.HydrateCmd, cmd.BuildCmd, cmd.TestCmd, cmd.CleanCmd)

	root.Execute()
}