Merge info/load/test CLIs into unified picomap subcommand CLI
This commit is contained in:
Ian Gulliver
326
cmd/picomap/main.go
Normal file
326
cmd/picomap/main.go
Normal file
@@ -0,0 +1,326 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"net"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/theater/picomap/lib/client"
|
||||
"github.com/theater/picomap/lib/picotool"
|
||||
)
|
||||
|
||||
func main() {
|
||||
if len(os.Args) < 2 {
|
||||
slog.Error("usage: picomap <info|load|test> [args...]")
|
||||
os.Exit(1)
|
||||
}
|
||||
cmd := os.Args[1]
|
||||
args := os.Args[2:]
|
||||
|
||||
var err error
|
||||
switch cmd {
|
||||
case "info":
|
||||
err = cmdInfo()
|
||||
case "load":
|
||||
err = cmdLoad(args)
|
||||
case "test":
|
||||
err = cmdTest(args)
|
||||
default:
|
||||
slog.Error("unknown command", "cmd", cmd)
|
||||
os.Exit(1)
|
||||
}
|
||||
if err != nil {
|
||||
slog.Error("fatal", "err", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
type deviceResult struct {
|
||||
dev string
|
||||
info *client.ResponseInfo
|
||||
err error
|
||||
}
|
||||
|
||||
func cmdInfo() error {
|
||||
devs, err := client.ListSerial()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if len(devs) == 0 {
|
||||
return fmt.Errorf("no devices found")
|
||||
}
|
||||
|
||||
results := make([]deviceResult, len(devs))
|
||||
var wg sync.WaitGroup
|
||||
for i, dev := range devs {
|
||||
results[i].dev = dev
|
||||
log := slog.With("dev", dev)
|
||||
wg.Go(func() {
|
||||
log.Info("connecting")
|
||||
c, err := client.NewSerial(dev, 2*time.Second)
|
||||
if err != nil {
|
||||
results[i].err = err
|
||||
return
|
||||
}
|
||||
log.Info("requesting info")
|
||||
info, err := c.Info()
|
||||
c.Close()
|
||||
if err != nil {
|
||||
results[i].err = err
|
||||
return
|
||||
}
|
||||
log.Info("got info", "firmware", info.FirmwareName)
|
||||
results[i].info = info
|
||||
})
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
for _, r := range results {
|
||||
if r.err != nil {
|
||||
slog.Error("device error", "dev", r.dev, "err", r.err)
|
||||
continue
|
||||
}
|
||||
slog.Info("device",
|
||||
"dev", r.dev,
|
||||
"board_id", hex.EncodeToString(r.info.BoardID[:]),
|
||||
"mac", net.HardwareAddr(r.info.MAC[:]).String(),
|
||||
"ip", net.IP(r.info.IP[:]).String(),
|
||||
"firmware", r.info.FirmwareName)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func boardSerial(id [8]byte) string {
|
||||
return fmt.Sprintf("%02X%02X%02X%02X%02X%02X%02X%02X",
|
||||
id[0], id[1], id[2], id[3], id[4], id[5], id[6], id[7])
|
||||
}
|
||||
|
||||
type deviceInfo struct {
|
||||
dev string
|
||||
serial string
|
||||
uf2 string
|
||||
name string
|
||||
}
|
||||
|
||||
func buildFirmware(buildDir string) error {
|
||||
slog.Info("configuring")
|
||||
cmake := exec.Command("cmake", "-S", filepath.Join(filepath.Dir(buildDir)), "-B", buildDir)
|
||||
cmake.Stdout = os.Stdout
|
||||
cmake.Stderr = os.Stderr
|
||||
if err := cmake.Run(); err != nil {
|
||||
return fmt.Errorf("cmake failed: %w", err)
|
||||
}
|
||||
|
||||
slog.Info("building")
|
||||
cmd := exec.Command("make", "-C", buildDir)
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
if err := cmd.Run(); err != nil {
|
||||
return fmt.Errorf("build failed: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func cmdLoad(args []string) error {
|
||||
target := "all"
|
||||
if len(args) > 0 {
|
||||
target = args[0]
|
||||
}
|
||||
|
||||
wd, err := os.Getwd()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
buildDir := filepath.Join(wd, "firmware", "build")
|
||||
|
||||
if err := buildFirmware(buildDir); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
devs, err := client.ListSerial()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
allTargets := []struct {
|
||||
name string
|
||||
uf2 string
|
||||
}{
|
||||
{"picomap", filepath.Join(buildDir, "picomap.uf2")},
|
||||
{"picomap_test", filepath.Join(buildDir, "picomap_test.uf2")},
|
||||
}
|
||||
|
||||
var targets []struct {
|
||||
name string
|
||||
uf2 string
|
||||
}
|
||||
switch target {
|
||||
case "all":
|
||||
targets = allTargets
|
||||
case "picomap":
|
||||
targets = allTargets[:1]
|
||||
case "picomap_test":
|
||||
targets = allTargets[1:]
|
||||
default:
|
||||
return fmt.Errorf("unknown target %q", target)
|
||||
}
|
||||
|
||||
if len(devs) < len(targets) {
|
||||
return fmt.Errorf("need %d device(s), found %d", len(targets), len(devs))
|
||||
}
|
||||
|
||||
devices := make([]deviceInfo, len(targets))
|
||||
errs := make([]error, len(targets))
|
||||
|
||||
var wg sync.WaitGroup
|
||||
for i := range targets {
|
||||
log := slog.With("dev", devs[i])
|
||||
wg.Go(func() {
|
||||
log.Info("connecting for info")
|
||||
c, err := client.NewSerial(devs[i], 2*time.Second)
|
||||
if err != nil {
|
||||
errs[i] = err
|
||||
return
|
||||
}
|
||||
info, err := c.Info()
|
||||
c.Close()
|
||||
if err != nil {
|
||||
errs[i] = err
|
||||
return
|
||||
}
|
||||
devices[i] = deviceInfo{
|
||||
dev: devs[i],
|
||||
serial: boardSerial(info.BoardID),
|
||||
uf2: targets[i].uf2,
|
||||
name: targets[i].name,
|
||||
}
|
||||
log.Info("got info", "serial", devices[i].serial, "firmware", info.FirmwareName)
|
||||
})
|
||||
}
|
||||
wg.Wait()
|
||||
for i, err := range errs {
|
||||
if err != nil {
|
||||
return fmt.Errorf("[%s] info: %w", devs[i], err)
|
||||
}
|
||||
}
|
||||
|
||||
for i := range devices {
|
||||
log := slog.With("serial", devices[i].serial)
|
||||
wg.Go(func() {
|
||||
log.Info("sending PICOBOOT")
|
||||
c, err := client.NewSerial(devices[i].dev, 2*time.Second)
|
||||
if err != nil {
|
||||
errs[i] = err
|
||||
return
|
||||
}
|
||||
err = c.PICOBOOT()
|
||||
c.Close()
|
||||
if err != nil {
|
||||
errs[i] = fmt.Errorf("PICOBOOT %s: %w", devices[i].serial, err)
|
||||
return
|
||||
}
|
||||
log.Info("PICOBOOT sent")
|
||||
})
|
||||
}
|
||||
wg.Wait()
|
||||
for i, err := range errs {
|
||||
if err != nil {
|
||||
return fmt.Errorf("[%s] %w", devices[i].serial, err)
|
||||
}
|
||||
}
|
||||
|
||||
uf2s := make([]string, len(targets))
|
||||
for i := range targets {
|
||||
uf2s[i] = targets[i].uf2
|
||||
}
|
||||
|
||||
for i := range devices {
|
||||
log := slog.With("serial", devices[i].serial)
|
||||
wg.Go(func() {
|
||||
log.Info("loading", "uf2", devices[i].name)
|
||||
errs[i] = picotool.Load(devices[i].uf2, devices[i].serial, 10*time.Second)
|
||||
if errs[i] == nil {
|
||||
log.Info("loaded", "uf2", devices[i].name)
|
||||
}
|
||||
})
|
||||
}
|
||||
wg.Wait()
|
||||
for i, err := range errs {
|
||||
if err != nil {
|
||||
return fmt.Errorf("[%s] load: %w", devices[i].serial, err)
|
||||
}
|
||||
}
|
||||
|
||||
slog.Info("done")
|
||||
return nil
|
||||
}
|
||||
|
||||
func cmdTest(args []string) error {
|
||||
if len(args) < 1 {
|
||||
return fmt.Errorf("usage: picomap test <name>")
|
||||
}
|
||||
name := args[0]
|
||||
|
||||
devs, err := client.ListSerial()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var testDev string
|
||||
for _, dev := range devs {
|
||||
log := slog.With("dev", dev)
|
||||
log.Info("connecting for info")
|
||||
c, err := client.NewSerial(dev, 2*time.Second)
|
||||
if err != nil {
|
||||
log.Warn("connect error", "err", err)
|
||||
continue
|
||||
}
|
||||
info, err := c.Info()
|
||||
c.Close()
|
||||
if err != nil {
|
||||
log.Warn("info error", "err", err)
|
||||
continue
|
||||
}
|
||||
log.Info("got info", "firmware", info.FirmwareName)
|
||||
if info.FirmwareName == "picomap_test" {
|
||||
testDev = dev
|
||||
break
|
||||
}
|
||||
}
|
||||
if testDev == "" {
|
||||
return fmt.Errorf("no picomap_test device found")
|
||||
}
|
||||
|
||||
log := slog.With("dev", testDev)
|
||||
log.Info("running test", "name", name)
|
||||
|
||||
c, err := client.NewSerial(testDev, 10*time.Second)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer c.Close()
|
||||
|
||||
result, err := c.Test(name)
|
||||
if err != nil {
|
||||
return fmt.Errorf("remote: %w", err)
|
||||
}
|
||||
|
||||
for _, msg := range result.Messages {
|
||||
log.Info("remote", "msg", msg)
|
||||
}
|
||||
|
||||
if result.Pass {
|
||||
log.Info("PASS")
|
||||
} else {
|
||||
log.Error("FAIL")
|
||||
os.Exit(1)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user