simon
a0f4a81a55
Expose OTA_SLAVE_PROGRESS on the master, track per-slave state during distribution, run ESP-NOW OTA in a background task so the host can poll while slaves update, and show master/slave progress in the dashboard with table layout and faster WebSocket refresh during uploads. Co-authored-by: Cursor <cursoragent@cursor.com>
85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
const defaultBaud = 921600
|
|
|
|
func usage() {
|
|
fmt.Fprintf(os.Stderr, "usage: gotool [-port /dev/ttyUSB0] <command>\n")
|
|
fmt.Fprintf(os.Stderr, " test uses uart.master from bench config when -port is omitted\n\n")
|
|
fmt.Fprintf(os.Stderr, "commands:\n")
|
|
fmt.Fprintf(os.Stderr, " version firmware version and git hash\n")
|
|
fmt.Fprintf(os.Stderr, " clients registered ESP-NOW slaves on the master\n")
|
|
fmt.Fprintf(os.Stderr, " deadzone get/set accelerometer deadzone (LSB)\n")
|
|
fmt.Fprintf(os.Stderr, " unicast-test send ESP-NOW unicast test to one slave\n")
|
|
fmt.Fprintf(os.Stderr, " test run automated scenario (see testdata/)\n")
|
|
fmt.Fprintf(os.Stderr, " serve web dashboard (Bootstrap + WebSocket)\n")
|
|
fmt.Fprintf(os.Stderr, " ota UART OTA upload (A/B partitions)\n")
|
|
fmt.Fprintf(os.Stderr, " ota-progress query per-slave ESP-NOW OTA progress on master\n\n")
|
|
flag.PrintDefaults()
|
|
}
|
|
|
|
func main() {
|
|
portName := flag.String("port", "", "serial port (e.g. /dev/ttyUSB0)")
|
|
baud := flag.Int("baud", defaultBaud, "UART baud rate")
|
|
flag.Parse()
|
|
|
|
if flag.NArg() < 1 {
|
|
usage()
|
|
os.Exit(2)
|
|
}
|
|
|
|
cmd := flag.Arg(0)
|
|
|
|
var runErr error
|
|
switch cmd {
|
|
case "test", "autotest":
|
|
runErr = runTest(*portName, *baud, flag.Args()[1:])
|
|
case "serve", "web", "dashboard":
|
|
if *portName == "" {
|
|
fmt.Fprintf(os.Stderr, "command %q requires -port\n\n", cmd)
|
|
usage()
|
|
os.Exit(2)
|
|
}
|
|
runErr = runServe(*portName, *baud, flag.Args()[1:])
|
|
case "version", "clients", "client-info", "deadzone", "accel-deadzone", "unicast-test", "unicast_test", "ota", "ota-progress", "ota_progress":
|
|
if *portName == "" {
|
|
fmt.Fprintf(os.Stderr, "command %q requires -port\n\n", cmd)
|
|
usage()
|
|
os.Exit(2)
|
|
}
|
|
sp, err := openSerial(*portName, *baud)
|
|
if err != nil {
|
|
log.Fatalf("open serial: %v", err)
|
|
}
|
|
defer sp.Close()
|
|
switch cmd {
|
|
case "version":
|
|
runErr = runVersion(sp)
|
|
case "clients", "client-info":
|
|
runErr = runClients(sp)
|
|
case "deadzone", "accel-deadzone":
|
|
runErr = runDeadzone(sp, flag.Args()[1:])
|
|
case "unicast-test", "unicast_test":
|
|
runErr = runUnicastTest(sp, flag.Args()[1:])
|
|
case "ota":
|
|
runErr = runOTA(sp, flag.Args()[1:])
|
|
case "ota-progress", "ota_progress":
|
|
runErr = runOtaProgress(sp, flag.Args()[1:])
|
|
}
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "unknown command %q\n\n", cmd)
|
|
usage()
|
|
os.Exit(2)
|
|
}
|
|
|
|
if runErr != nil {
|
|
log.Fatal(runErr)
|
|
}
|
|
}
|