49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/cheggaaa/pb/v3"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"go.arsenm.dev/infinitime"
|
||
|
|
)
|
||
|
|
|
||
|
|
func resourcesLoad(c *cli.Context) error {
|
||
|
|
if c.Args().Len() == 0 {
|
||
|
|
return cli.Exit("Command load requires one argument.", 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create progress bar templates
|
||
|
|
rmTmpl := `Removing {{string . "filename"}}`
|
||
|
|
upTmpl := `Uploading {{string . "filename"}} {{counters . }} B {{bar . "|" "-" (cycle .) " " "|"}} {{percent . }} {{rtime . "%s"}}`
|
||
|
|
// Start full bar at 0 total
|
||
|
|
bar := pb.ProgressBarTemplate(rmTmpl).Start(0)
|
||
|
|
|
||
|
|
path, err := filepath.Abs(c.Args().Get(0))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
progCh, err := client.LoadResources(c.Context, path)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
for evt := range progCh {
|
||
|
|
if evt.Operation == infinitime.ResourceOperationRemoveObsolete {
|
||
|
|
bar.SetTemplateString(rmTmpl)
|
||
|
|
bar.Set("filename", evt.Name)
|
||
|
|
} else {
|
||
|
|
bar.SetTemplateString(upTmpl)
|
||
|
|
bar.Set("filename", evt.Name)
|
||
|
|
|
||
|
|
bar.SetTotal(evt.Total)
|
||
|
|
bar.SetCurrent(evt.Sent)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bar.Finish()
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|