package api import ( "context" "go.arsenm.dev/itd/internal/rpc" ) type DFUProgress struct { Sent int64 Received int64 Total int64 Err error } func (c *Client) FirmwareUpgrade(ctx context.Context, upgType UpgradeType, files ...string) (chan DFUProgress, error) { progressCh := make(chan DFUProgress, 5) fc, err := c.client.FirmwareUpgrade(ctx, &rpc.FirmwareUpgradeRequest{ Type: rpc.FirmwareUpgradeRequest_Type(upgType), Files: files, }) if err != nil { return nil, err } go fsRecvToChannel[rpc.DFUProgress](fc, progressCh, func(evt *rpc.DFUProgress, err error) DFUProgress { return DFUProgress{ Sent: evt.Sent, Received: evt.Recieved, Total: evt.Total, Err: err, } }) return progressCh, nil }