148 lines
2.9 KiB
Go
148 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/theater/picomap/lib/client"
|
|
"github.com/theater/picomap/lib/picotool"
|
|
)
|
|
|
|
func main() {
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
slog.Error("fatal", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
buildDir := filepath.Join(wd, "firmware", "build")
|
|
|
|
if err := run(buildDir); err != nil {
|
|
slog.Error("fatal", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func build(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 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])
|
|
}
|
|
|
|
func run(buildDir string) error {
|
|
if err := build(buildDir); err != nil {
|
|
return err
|
|
}
|
|
|
|
devs, err := client.ListSerial()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(devs) < 2 {
|
|
return fmt.Errorf("expected 2 devices, found %d", len(devs))
|
|
}
|
|
|
|
serials := make([]string, 2)
|
|
errs := make([]error, 2)
|
|
uf2Names := []string{"picomap", "picomap_test"}
|
|
|
|
var wg sync.WaitGroup
|
|
for i := range 2 {
|
|
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
|
|
}
|
|
serials[i] = boardSerial(info.BoardID)
|
|
log.Info("got info", "serial", serials[i], "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 2 {
|
|
log := slog.With("serial", serials[i])
|
|
wg.Go(func() {
|
|
log.Info("sending PICOBOOT")
|
|
c, err := client.NewSerial(devs[i], 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", serials[i], err)
|
|
return
|
|
}
|
|
log.Info("PICOBOOT sent")
|
|
})
|
|
}
|
|
wg.Wait()
|
|
for _, err := range errs {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
uf2s := []string{
|
|
filepath.Join(buildDir, "picomap.uf2"),
|
|
filepath.Join(buildDir, "picomap_test.uf2"),
|
|
}
|
|
|
|
for i := range 2 {
|
|
log := slog.With("serial", serials[i])
|
|
wg.Go(func() {
|
|
log.Info("loading", "uf2", uf2Names[i])
|
|
errs[i] = picotool.Load(uf2s[i], serials[i], 10*time.Second)
|
|
if errs[i] == nil {
|
|
log.Info("loaded", "uf2", uf2Names[i])
|
|
}
|
|
})
|
|
}
|
|
wg.Wait()
|
|
for i, err := range errs {
|
|
if err != nil {
|
|
return fmt.Errorf("[%s] load: %w", serials[i], err)
|
|
}
|
|
}
|
|
|
|
slog.Info("done")
|
|
return nil
|
|
}
|