package main import ( "fmt" "path/filepath" "time" "github.com/cheggaaa/pb/v3" "github.com/urfave/cli/v2" "go.arsenm.dev/itd/api" "go.arsenm.dev/logger/log" ) func fwUpgrade(c *cli.Context) error { resources := c.String("resources") if resources != "" { absRes, err := filepath.Abs(resources) if err != nil { return err } err = resLoad(c.Context, []string{absRes}) if err != nil { log.Error("Resource loading has returned an error. This can happen if your current version of InfiniTime doesn't support BLE FS. Try updating without resource loading, and then load them after using the `itctl res load` command.").Send() return err } } start := time.Now() var upgType api.UpgradeType var files []string // Get relevant data struct if c.String("archive") != "" { // Get archive data struct upgType = api.UpgradeTypeArchive files = []string{c.String("archive")} } else if c.String("init-packet") != "" && c.String("firmware") != "" { // Get files data struct 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) } progress, err := client.FirmwareUpgrade(c.Context, upgType, abs(files)...) if err != nil { return err } // Create progress bar template barTmpl := `{{counters . }} B {{bar . "|" "-" (cycle .) " " "|"}} {{percent . }} {{rtime . "%s"}}` // Start full bar at 0 total bar := pb.ProgressBarTemplate(barTmpl).Start(0) // Create new scanner of connection for event := range progress { if event.Err != nil { return event.Err } // Set total bytes in progress bar bar.SetTotal(event.Total) // Set amount of bytes received in progress bar bar.SetCurrent(int64(event.Received)) // If transfer finished, break if int64(event.Sent) == event.Total { break } } // Finish progress bar bar.Finish() fmt.Printf("Transferred %d B in %s.\n", bar.Total(), time.Since(start)) fmt.Println("Remember to validate the new firmware in the InfiniTime settings.") return nil } func fwVersion(c *cli.Context) error { version, err := client.Version(c.Context) if err != nil { return err } fmt.Println(version) return nil } func abs(paths []string) []string { for index, path := range paths { newPath, err := filepath.Abs(path) if err != nil { continue } paths[index] = newPath } return paths }