picomap/cmd/load/main.go

package main

import (
	"fmt"
	"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 {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
	buildDir := filepath.Join(wd, "firmware", "build")

	if err := run(buildDir); err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}

func build(buildDir string) error {
	fmt.Println("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)
	}

	fmt.Println("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)

	var wg sync.WaitGroup
	for i := range 2 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			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)
		}()
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("info %s: %w", devs[i], err)
		}
	}

	fmt.Println("Sending PICOBOOT requests...")
	for i := range 2 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			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", devs[i], err)
			}
		}()
	}
	wg.Wait()
	for _, err := range errs {
		if err != nil {
			return err
		}
	}

	for i := range 2 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			errs[i] = picotool.WaitForBootsel(serials[i], 10*time.Second)
		}()
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("wait %s: %w", serials[i], err)
		}
	}

	uf2s := []string{
		filepath.Join(buildDir, "picomap.uf2"),
		filepath.Join(buildDir, "picomap_test.uf2"),
	}

	fmt.Println("Loading firmware...")
	for i := range 2 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			errs[i] = picotool.Load(uf2s[i], serials[i])
		}()
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("load %s: %w", serials[i], err)
		}
	}

	fmt.Println("Done.")
	return nil
}