picomap/cmd/load/main.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
}