External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-18 04:55:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use custom formatter for percentage, ensure fixed string size is used

Browse Source
This commit is contained in:
Manuel Bua 2020-07-25 22:15:45 +02:00
parent d1c8527f1f
commit 65d7246b2f
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
v2/internal/progress/progress.go
View File

@ -132,7 +132,7 @@ func (p *Progress) setupProgressbar(name string, total int64, priority int) *mpb
mpb.PrependDecorators(
decor.Name(name, decor.WCSyncSpaceR),
decor.CountersNoUnit(color.BrightBlue(" %d/%d").String(), decor.WCSyncSpace),
decor.NewPercentage(color.Bold("%d").String(), decor.WCSyncSpace),
customFormattedPercentage(color.Bold("%3.0f%%").String(), decor.WCSyncSpace),
),
mpb.AppendDecorators(
decor.AverageSpeed(0, color.BrightBlue("%.2f r/s ").String(), decor.WCSyncSpace),
@ -142,6 +142,26 @@ func (p *Progress) setupProgressbar(name string, total int64, priority int) *mpb
)
}
// Helper function to calculate percentage
func computePercentage(total, current int64, width int) float64 {
if total <= 0 {
return 0
}
if current >= total {
return float64(width)
}
return float64(int64(width)*current) / float64(total)
}
// Percentage decorator with custom formatting
func customFormattedPercentage(format string, wcc ...decor.WC) decor.Decorator {
f := func(s decor.Statistics) string {
p := computePercentage(s.Total, s.Current, 100)
return fmt.Sprintf(format, p)
}
return decor.Any(f, wcc...)
}
// Starts capturing stdout and stderr instead of producing visual output that may interfere with the progress bars.
func (p *Progress) StartStdCapture() {
p.stdCaptureMutex.Lock()