halfwidth.go

package autocorrect

import (
	"regexp"
	"strings"
)

var (
	charWidthMap = map[rune]rune{
		'ａ': 'a', 'ｂ': 'b', 'ｃ': 'c', 'ｄ': 'd', 'ｅ': 'e', 'ｆ': 'f', 'ｇ': 'g', 'ｈ': 'h', 'ｉ': 'i', 'ｊ': 'j', 'ｋ': 'k', 'ｌ': 'l', 'ｍ': 'm', 'ｎ': 'n', 'ｏ': 'o', 'ｐ': 'p', 'ｑ': 'q', 'ｒ': 'r', 'ｓ': 's', 'ｔ': 't', 'ｕ': 'u', 'ｖ': 'v', 'ｗ': 'w', 'ｘ': 'x', 'ｙ': 'y', 'ｚ': 'z', 'Ａ': 'A', 'Ｂ': 'B', 'Ｃ': 'C', 'Ｄ': 'D', 'Ｅ': 'E', 'Ｆ': 'F', 'Ｇ': 'G', 'Ｈ': 'H', 'Ｉ': 'I', 'Ｊ': 'J', 'Ｋ': 'K', 'Ｌ': 'L', 'Ｍ': 'M', 'Ｎ': 'N', 'Ｏ': 'O', 'Ｐ': 'P', 'Ｑ': 'Q', 'Ｒ': 'R', 'Ｓ': 'S', 'Ｔ': 'T', 'Ｕ': 'U', 'Ｖ': 'V', 'Ｗ': 'W', 'Ｘ': 'X', 'Ｙ': 'Y', 'Ｚ': 'Z', '１': '1', '２': '2', '３': '3', '４': '4', '５': '5', '６': '6', '７': '7', '８': '8', '９': '9', '０': '0', '　': ' ',
	}

	halfTimeRe = regexp.MustCompile(`(\d)(：)(\d)`)
)

func halfwidth(text string) string {
	runes := []rune{}
	for _, char := range text {
		newChar := charWidthMap[char]
		if newChar != 0 {
			runes = append(runes, charWidthMap[char])
		} else {
			runes = append(runes, char)
		}
	}

	out := string(runes)

	// Fix 12：00 -> 12:00
	out = halfTimeRe.ReplaceAllStringFunc(out, func(part string) string {
		return strings.Replace(part, "：", ":", 1)
	})

	return out
}