vendor/gopkg.in/cheggaaa/pb.v1/runecount.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

package pb

import (
	"github.com/mattn/go-runewidth"
	"regexp"
)

// Finds the control character sequences (like colors)
var ctrlFinder = regexp.MustCompile("\x1b\x5b[0-9]+\x6d")

func escapeAwareRuneCountInString(s string) int {
	n := runewidth.StringWidth(s)
	for _, sm := range ctrlFinder.FindAllString(s, -1) {
		n -= runewidth.StringWidth(sm)
	}
	return n
}