utils.go

package utils

import (
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

type firmware struct {
	Path     string
	Name     string
	extra    string
	IsLoader bool
}

type combo struct {
	match  string
	loader string
}

func isPreferred(path string) bool {
	if path == "" {
		return false
	}
	return true
}

func GetCompatibleWith(name string, rootPath string) map[string][]firmware {

	files := make(map[string][]firmware)

	knownBoards := make(map[string]combo)
	knownBoards["mkr1000"] = combo{match: "(WINC1500)*(3a0)", loader: "WINC1500/FirmwareUpdater.mkr1000.ino.bin"}
	knownBoards["mkrwifi1010"] = combo{match: "NINA_W102.bin", loader: "NINA/FirmwareUpdater.mkrwifi1010.ino.bin"}
	knownBoards["nano_33_iot"] = combo{match: "NINA_W102.bin", loader: "NINA/FirmwareUpdater.nano_33_iot.ino.bin"}
	knownBoards["mkrvidor4000"] = combo{match: "NINA_W102.bin", loader: "NINA/FirmwareUpdater.mkrvidor4000.ino.bin"}
	knownBoards["uno2018"] = combo{match: "NINA_W102-Uno_WiFi_Rev2.bin", loader: "NINA/FirmwareUpdater.unowifirev2.without_bl.ino.hex"}
	knownBoards["mkrnb1500"] = combo{match: "SARA", loader: "SARA/SerialSARAPassthrough*"}
	knownBoards["nanorp2040connect"] = combo{match: "NINA_W102-Nano_RP2040_Connect", loader: "NINA/FirmwareUpdater.nanorp2040connect.ino.elf"}

	listAll := false

	if knownBoards[strings.ToLower(name)].match == "" {
		listAll = true
	}
	exePath := rootPath
	if exePath == "" {
		exePath, _ = os.Executable()
	}
	root := filepath.Dir(exePath)
	root = filepath.Join(root, "firmwares")
	loader := regexp.MustCompile(knownBoards[name].loader)
	fw := regexp.MustCompile(knownBoards[name].match)

	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if info.IsDir() {
			return nil
		}
		unixPath := filepath.ToSlash(path)
		parts := strings.Split(unixPath, "/")
		fancyName := parts[len(parts)-3] + " " + parts[len(parts)-2]
		f := firmware{
			Path:     unixPath,
			Name:     fancyName,
			IsLoader: loader.MatchString(unixPath) && !listAll,
		}
		folder := filepath.Dir(path)
		if strings.HasPrefix(f.Name, "firmwares") && !f.IsLoader {
			return nil
		}
		if listAll && !strings.HasPrefix(f.Name, "firmwares") {
			files[folder] = append(files[folder], f)
		} else if !listAll && (fw.MatchString(path) || f.IsLoader) {
			files[folder] = append(files[folder], f)
		}
		return nil
	})

	// check files and add information to fw.Name in case of name clashing
	for k := range files {
		for i := range files[k] {
			for j := range files[k] {
				if files[k][i].Name == files[k][j].Name && i != j {
					files[k][i].extra = filepath.Base(files[k][i].Path)
				}
			}
		}
	}
	for k := range files {
		for i := range files[k] {
			if files[k][i].extra != "" {
				files[k][i].Name = files[k][i].Name + " (" + files[k][i].extra + ")"
			}
		}
	}

	if err != nil {
		return files
	}
	return files
}