mirror of
https://github.com/celogeek/go-comic-converter.git
synced 2025-05-24 15:52:38 +02:00
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
/*
|
|
create a progress bar with custom settings.
|
|
*/
|
|
package epubprogress
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/schollz/progressbar/v3"
|
|
)
|
|
|
|
type Options struct {
|
|
Quiet bool
|
|
Json bool
|
|
Max int
|
|
Description string
|
|
CurrentJob int
|
|
TotalJob int
|
|
}
|
|
|
|
type EpubProgress interface {
|
|
Add(num int) error
|
|
Close() error
|
|
}
|
|
|
|
func New(o Options) EpubProgress {
|
|
if o.Quiet {
|
|
return progressbar.DefaultSilent(int64(o.Max))
|
|
}
|
|
|
|
if o.Json {
|
|
return newEpubProgressJson(o)
|
|
}
|
|
|
|
fmtJob := fmt.Sprintf("%%0%dd", len(fmt.Sprint(o.TotalJob)))
|
|
fmtDesc := fmt.Sprintf("[%s/%s] %%-15s", fmtJob, fmtJob)
|
|
return progressbar.NewOptions(o.Max,
|
|
progressbar.OptionSetWriter(os.Stderr),
|
|
progressbar.OptionThrottle(65*time.Millisecond),
|
|
progressbar.OptionOnCompletion(func() {
|
|
fmt.Fprint(os.Stderr, "\n")
|
|
}),
|
|
progressbar.OptionSetDescription(fmt.Sprintf(fmtDesc, o.CurrentJob, o.TotalJob, o.Description)),
|
|
progressbar.OptionSetWidth(60),
|
|
progressbar.OptionShowCount(),
|
|
progressbar.OptionSetRenderBlankState(true),
|
|
progressbar.OptionEnableColorCodes(true),
|
|
progressbar.OptionSetTheme(progressbar.Theme{
|
|
Saucer: "[green]=[reset]",
|
|
SaucerHead: "[green]>[reset]",
|
|
SaucerPadding: " ",
|
|
BarStart: "[",
|
|
BarEnd: "]",
|
|
}),
|
|
)
|
|
}
|