Merge pull request olekukonko#18 from lixin9311/master

fix wrong CJK Character width.
w-haibara · Oct 29, 2015 · a5eefc2 · a5eefc2
2 parents 031e82a + 226b93e
commit a5eefc2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/util.go b/util.go
@@ -19,7 +19,15 @@ var (
 )
 
 func DisplayWidth(str string) int {
-	return utf8.RuneCountInString(ansi.ReplaceAllLiteralString(str, ""))
+	tmp := ansi.ReplaceAllLiteralString(str, "")
+	tmp_rune := []rune(tmp)
+	count := 0
+	for _, v := range tmp_rune {
+		if v > 128 {
+			count++
+		}
+	}
+	return utf8.RuneCountInString(tmp) + count
 }
 
 // Simple Condition for string