FilesExpand file tree

squash-edit-commit-msg

/

truncate.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

75 lines (64 loc) · 1.25 KB

squash-edit-commit-msg

/

truncate.go

Copy path

Top

File metadata and controls

• Code
• Blame

75 lines (64 loc) · 1.25 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

package text

import (

runewidth "github.com/mattn/go-runewidth"

"github.com/rivo/uniseg"

)

const (

ellipsisWidth = 3

minWidthForEllipsis = 5

)

// DisplayWidth calculates what the rendered width of a string may be

func DisplayWidth(s string) int {

g := uniseg.NewGraphemes(s)

w := 0

for g.Next() {

w += graphemeWidth(g)

}

return w

}

// Truncate shortens a string to fit the maximum display width

func Truncate(maxWidth int, s string) string {

w := DisplayWidth(s)

if w <= maxWidth {

return s

}

useEllipsis := false

if maxWidth >= minWidthForEllipsis {

useEllipsis = true

maxWidth -= ellipsisWidth

}

g := uniseg.NewGraphemes(s)

r := ""

rWidth := 0

for {

g.Next()

gWidth := graphemeWidth(g)

if rWidth+gWidth <= maxWidth {

r += g.Str()

rWidth += gWidth

continue

} else {

break

}

}

if useEllipsis {

r += "..."

}

if rWidth < maxWidth {

r += " "

}

return r

}

// graphemeWidth calculates what the rendered width of a grapheme may be

func graphemeWidth(g *uniseg.Graphemes) int {

// If grapheme spans one rune use that rune width

// If grapheme spans multiple runes use the first non-zero rune width

w := 0

for _, r := range g.Runes() {

w = runewidth.RuneWidth(r)

if w > 0 {

break

}

}

return w

}