itd/cmd/itctl/firmware.go

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
}