cybuzuma/itd
1
0
Fork 0
forked from Elara6331/itd
Code Issues Pull Requests Projects Releases Wiki Activity

Switch from custom socket API to rpcx

Browse Source
This commit is contained in:
Elara
2022-04-22 17:12:30 -07:00
parent d318c584da
commit 0cdf8a4bed
24 changed files with 1569 additions and 1311 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/itctl/firmware.go
View File

@@ -8,7 +8,6 @@ import (
"github.com/cheggaaa/pb/v3"
"github.com/urfave/cli/v2"
"go.arsenm.dev/itd/api"
"go.arsenm.dev/itd/internal/types"
)
func fwUpgrade(c *cli.Context) error {
@@ -19,11 +18,11 @@ func fwUpgrade(c *cli.Context) error {
// Get relevant data struct
if c.String("archive") != "" {
// Get archive data struct
upgType = types.UpgradeTypeArchive
upgType = api.UpgradeTypeArchive
files = []string{c.String("archive")}
} else if c.String("init-packet") != "" && c.String("firmware") != "" {
// Get files data struct
upgType = types.UpgradeTypeFiles
upgType = api.UpgradeTypeFiles
files = []string{c.String("init-packet"), c.String("firmware")}
} else {
return cli.Exit("Upgrade command requires either archive or init packet and firmware.", 1)
@@ -43,9 +42,9 @@ func fwUpgrade(c *cli.Context) error {
// Set total bytes in progress bar
bar.SetTotal(event.Total)
// Set amount of bytes received in progress bar
bar.SetCurrent(event.Received)
bar.SetCurrent(int64(event.Received))
// If transfer finished, break
if event.Sent == event.Total {
if int64(event.Sent) == event.Total {
break
}
}