76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/theater/picomap/lib/picoserial"
|
|
"github.com/theater/picomap/lib/picotool"
|
|
"github.com/theater/picomap/lib/wire"
|
|
)
|
|
|
|
func main() {
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
buildDir := filepath.Join(wd, "build")
|
|
|
|
if err := run(buildDir); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func run(buildDir string) error {
|
|
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)
|
|
}
|
|
|
|
dev, err := picoserial.FindDevice()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if dev != "" {
|
|
fmt.Printf("Sending 'b' to %s to enter BOOTSEL mode...\n", dev)
|
|
resp, err := picoserial.SendByteAndRead(dev, 'b', 2*time.Second)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(resp) > 0 {
|
|
msg, err := wire.DecodeMessage(resp)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "warning: failed to decode response: %v\n", err)
|
|
} else {
|
|
switch msg.(type) {
|
|
case *wire.RebootingBootsel:
|
|
fmt.Println("Device confirmed reboot into BOOTSEL mode.")
|
|
default:
|
|
fmt.Printf("Unexpected response type: %T\n", msg)
|
|
}
|
|
}
|
|
}
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
|
|
uf2 := filepath.Join(buildDir, "picomap.uf2")
|
|
fmt.Println("Loading firmware...")
|
|
if err := picotool.Load(uf2); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println("Rebooting...")
|
|
_ = picotool.Reboot()
|
|
|
|
fmt.Println("Done.")
|
|
return nil
|
|
}
|