simon
d24b0cb5c3
JSON configs describe network and node MACs; scenarios run command sequences with expect checks. Share UART client API across CLI and tests. Co-authored-by: Cursor <cursoragent@cursor.com>
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"powerpod/gotool/pb"
|
|
)
|
|
|
|
func (s *serialPort) getVersion() (*pb.VersionResponse, error) {
|
|
payload, err := s.exchange(byte(pb.MessageType_VERSION), "VERSION")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var msg pb.UartMessage
|
|
if err := proto.Unmarshal(payload[1:], &msg); err != nil {
|
|
return nil, fmt.Errorf("decode: %w", err)
|
|
}
|
|
if msg.GetType() != pb.MessageType_VERSION {
|
|
return nil, fmt.Errorf("unexpected type %v", msg.GetType())
|
|
}
|
|
ver := msg.GetVersionResponse()
|
|
if ver == nil {
|
|
return nil, fmt.Errorf("missing version_response")
|
|
}
|
|
return ver, nil
|
|
}
|
|
|
|
func (s *serialPort) listClients() ([]*pb.ClientInfo, error) {
|
|
payload, err := s.exchange(byte(pb.MessageType_CLIENT_INFO), "CLIENT_INFO")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var msg pb.UartMessage
|
|
if err := proto.Unmarshal(payload[1:], &msg); err != nil {
|
|
return nil, fmt.Errorf("decode: %w", err)
|
|
}
|
|
if msg.GetType() != pb.MessageType_CLIENT_INFO {
|
|
return nil, fmt.Errorf("unexpected type %v", msg.GetType())
|
|
}
|
|
info := msg.GetClientInfoResponse()
|
|
if info == nil {
|
|
return nil, fmt.Errorf("missing client_info_response")
|
|
}
|
|
return info.GetClients(), nil
|
|
}
|
|
|
|
func (s *serialPort) accelDeadzone(req *pb.AccelDeadzoneRequest) (*pb.AccelDeadzoneResponse, error) {
|
|
msg := &pb.UartMessage{
|
|
Type: pb.MessageType_ACCEL_DEADZONE,
|
|
Payload: &pb.UartMessage_AccelDeadzoneRequest{
|
|
AccelDeadzoneRequest: req,
|
|
},
|
|
}
|
|
body, err := proto.Marshal(msg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("encode: %w", err)
|
|
}
|
|
payload := append([]byte{byte(pb.MessageType_ACCEL_DEADZONE)}, body...)
|
|
respPayload, err := s.exchangePayload(payload, "ACCEL_DEADZONE")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var respMsg pb.UartMessage
|
|
if err := proto.Unmarshal(respPayload[1:], &respMsg); err != nil {
|
|
return nil, fmt.Errorf("decode: %w", err)
|
|
}
|
|
r := respMsg.GetAccelDeadzoneResponse()
|
|
if r == nil {
|
|
return nil, fmt.Errorf("missing accel_deadzone_response")
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
func (s *serialPort) espnowUnicastTest(clientID, seq uint32) (*pb.EspNowUnicastTestResponse, error) {
|
|
req := &pb.EspNowUnicastTestRequest{ClientId: clientID, Seq: seq}
|
|
msg := &pb.UartMessage{
|
|
Type: pb.MessageType_ESPNOW_UNICAST_TEST,
|
|
Payload: &pb.UartMessage_EspnowUnicastTestRequest{
|
|
EspnowUnicastTestRequest: req,
|
|
},
|
|
}
|
|
body, err := proto.Marshal(msg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("encode: %w", err)
|
|
}
|
|
payload := append([]byte{byte(pb.MessageType_ESPNOW_UNICAST_TEST)}, body...)
|
|
respPayload, err := s.exchangePayload(payload, "ESPNOW_UNICAST_TEST")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var respMsg pb.UartMessage
|
|
if err := proto.Unmarshal(respPayload[1:], &respMsg); err != nil {
|
|
return nil, fmt.Errorf("decode: %w", err)
|
|
}
|
|
r := respMsg.GetEspnowUnicastTestResponse()
|
|
if r == nil {
|
|
return nil, fmt.Errorf("missing espnow_unicast_test_response")
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
func (s *serialPort) GetVersion() (*pb.VersionResponse, error) { return s.getVersion() }
|
|
|
|
func (s *serialPort) ListClients() ([]*pb.ClientInfo, error) { return s.listClients() }
|
|
|
|
func (s *serialPort) AccelDeadzone(req *pb.AccelDeadzoneRequest) (*pb.AccelDeadzoneResponse, error) {
|
|
return s.accelDeadzone(req)
|
|
}
|
|
|
|
func (s *serialPort) EspnowUnicastTest(clientID, seq uint32) (*pb.EspNowUnicastTestResponse, error) {
|
|
return s.espnowUnicastTest(clientID, seq)
|
|
}
|