main.go

package main

import (
	"fmt"
	"os"
	"os/signal"
	"path/filepath"
	"syscall"

	flag "github.com/spf13/pflag"
	"github.com/JoshuaDoes/menuify"
	"github.com/JoshuaDoes/menuify/screens/ncurses"
)

var (
	workingDir         string //where the menu should treat as its working directory
	configFile         string //path to menu configuration
	keyCalibrationFile string //path to keyboard calibration, can be written for embedded devices or generated by first run calibrator

	menu   *menuify.Menu               //holds the context of the text menu engine
	screen *ncurses.MenuScreen_Ncurses //interfaces with the actual terminal window

	SOURCEIMG *WIM
	SOURCEIDX int
	SOURCES   []*WIM
)

func init() {
	//Apply all command-line flags
	flag.StringVar(&workingDir, "workingDir", "", "path to treat as working directory")
	flag.StringVar(&configFile, "menu", "", "path to menu configuration")
	flag.StringVar(&keyCalibrationFile, "keyCalibration", "", "path to keyboard calibration, generated by calibrator if not present")
	flag.Parse()

	if workingDir == "" {
		workingDir, _ = os.Getwd()
	}
	if configFile == "" {
		configFile = workingDir + "/menu.json"
	}
	if keyCalibrationFile == "" {
		keyCalibrationFile = workingDir + "/keyCalibration.json"
	}

	menu = menuify.NewMenu()
	if err := menu.Load(configFile); err != nil {
		panic(fmt.Sprintf("error loading menu config: %v", err))
	}

	screen = ncurses.NewMenuScreenNcurses(menu)
	if screen == nil {
		panic(fmt.Sprintf("error initializing screen, is nil?"))
	}
	menu.SetScreen(screen)

	SOURCES = make([]*WIM, 0)
}

func main() {
	if err := menu.Engine.Calibrate(keyCalibrationFile); err != nil {
		menuify.ScreenPrintf(screen, "Error calibrating: %v\n", err)
		return //Allows ncurses to gracefully exit
	}
	screen.Clear()

	wims, _ := WalkMatch("sources", "*.*")
	if len(wims) > 0 {
		for i := 0; i < len(wims); i++ {
			wiminfo, err := WIMInfo(wims[i])
			if err == nil { //We only want valid WIM and ESD files that don't fail to parse
				SOURCES = append(SOURCES, wiminfo)
			}
		}
	} else {
		menu.Engine.ErrorText("No install sources are available!", "")
		return
	}

	menu.Engine.Hook("install-select_wim", hookInstallSelectWim)
	menu.Engine.Hook("install-select_edition", hookInstallSelectEdition)
	menu.Engine.Hook("install-select_disk", hookInstallSelectDisk)
	menu.Engine.Hook("install-windows-confirmation", hookInstallWindowsConfirmation)
	menu.Engine.Hook("install-windows", hookInstallWindows)

	menu.Engine.Home()
	menu.Engine.BindKeys()

	sc := make(chan os.Signal, 1)
	signal.Notify(sc, syscall.SIGINT)
	<-sc
}

func WalkMatch(root, pattern string) ([]string, error) {
	var matches []string
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if info.IsDir() {
			return nil
		}
		if matched, err := filepath.Match(pattern, filepath.Base(path)); err != nil {
			return err
		} else if matched {
			matches = append(matches, path)
		}
		return nil
	})
	if err != nil {
		return nil, err
	}
	return matches, nil
}