forked from Elara6331/itd
49 lines
992 B
Go
49 lines
992 B
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"go.arsenm.dev/itd/internal/types"
|
|
)
|
|
|
|
// DFUProgress stores the progress of a DFU upfate
|
|
type DFUProgress types.DFUProgress
|
|
|
|
// UpgradeType indicates the type of upgrade to be performed
|
|
type UpgradeType uint8
|
|
|
|
// Type of DFU upgrade
|
|
const (
|
|
UpgradeTypeArchive UpgradeType = iota
|
|
UpgradeTypeFiles
|
|
)
|
|
|
|
// FirmwareUpgrade initiates a DFU update and returns the progress channel
|
|
func (c *Client) FirmwareUpgrade(upgType UpgradeType, files ...string) (<-chan DFUProgress, error) {
|
|
err := json.NewEncoder(c.conn).Encode(types.Request{
|
|
Type: types.ReqTypeFwUpgrade,
|
|
Data: types.ReqDataFwUpgrade{
|
|
Type: int(upgType),
|
|
Files: files,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c.dfuProgressCh = make(chan types.Response, 5)
|
|
|
|
out := make(chan DFUProgress, 5)
|
|
go func() {
|
|
for res := range c.dfuProgressCh {
|
|
progress, err := decodeDFUProgress(res.Value)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
out <- progress
|
|
}
|
|
}()
|
|
|
|
return out, nil
|
|
}
|