if pos < w: add empty chars

mattn · Sep 16, 2022 · b18c354 · b18c354
1 parent 669734a
commit b18c354
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/runewidth.go b/runewidth.go
@@ -2,6 +2,7 @@ package runewidth
 
 import (
 	"os"
+	"strings"
 
 	"github.com/rivo/uniseg"
 )
@@ -233,7 +234,12 @@ func (c *Condition) TrimPrefix(s string, w int, prefix string) string {
 			}
 		}
 		if width+chWidth > w {
-			pos, _ = g.Positions()
+			if width < w {
+				_, pos = g.Positions()
+				prefix += strings.Repeat(" ", w-width)
+			} else {
+				pos, _ = g.Positions()
+			}
 
 			break
 		}

diff --git a/runewidth_test.go b/runewidth_test.go
@@ -392,6 +392,8 @@ var trimprefixtests = []struct {
 	{"あいうえお", 6, "...", "...えお"},
 	{"あいうえお", 10, "", ""},
 	{"あいうえお", 10, "...", "..."},
+	{"あいうえお", 5, "", " えお"},
+	{"Aあいうえお", 5, "", "うえお"},
 }
 
 func TestTrimPrefix(t *testing.T) {