align.go

package main

import (
	"strings"

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

// MaxStringWidth は最も長い見た目の文字列の長さを返す。
func MaxStringWidth(lines []string) (max int) {
	for _, v := range lines {
		l := runewidth.StringWidth(v)
		if max < l {
			max = l
		}
	}
	return max
}

// AlignLeft は文字列を左寄せする。
// 左寄せは見た目上の文字列の長さで揃える。
// length = -1のときは、引数文字列の最長の長さに合わせる。
// padは埋める文字列を指定する。埋める文字が見た目上でマルチバイトの場合は
// たとえlengthが奇数でも+1して偶数になるように調整する。
func AlignLeft(lines []string, length int, pad string) []string {
	if length == 0 || len(lines) < 1 {
		return lines
	}

	// 空白埋めする文字列がマルチバイト文字かどうか
	padWidtn := runewidth.StringWidth(pad)
	padIsMultiByteString := padWidtn == 2

	// -1のときは文字列の長さをalignの長さにする
	// パッディングの長さと、処理対象の文字列のより長い方を揃える数値に指定
	maxWidth := MaxStringWidth(lines)
	if length == -1 {
		length = maxWidth
	} else if length < maxWidth {
		length = maxWidth
	}

	// マルチバイト文字を使うときは長さを偶数に揃える
	if padIsMultiByteString && length%2 != 0 {
		length++
	}

	ret := []string{}
	for _, line := range lines {
		l := runewidth.StringWidth(line)
		diff := length - l
		if diff%2 != 0 {
			line += " "
			diff--
		}
		// Repeatするときにマルチバイト文字を使うときは2分の1にする
		if padIsMultiByteString {
			diff /= 2
		}
		s := line + strings.Repeat(pad, diff)
		ret = append(ret, s)
	}
	return ret
}

// AlignCenter は文字列を中央寄せする。
// 中央寄せは見た目上の文字列の長さで揃える。
// length = -1のときは、引数文字列の最長の長さに合わせる。
// padは埋める文字列を指定する。
func AlignCenter(lines []string, length int, pad string) []string {
	if length == 0 || len(lines) < 1 {
		return lines
	}

	// 空白埋めする文字列がマルチバイト文字かどうか
	padWidtn := runewidth.StringWidth(pad)
	padIsMultiByteString := padWidtn == 2

	// -1のときは文字列の長さをalignの長さにする
	// パッディングの長さと、処理対象の文字列のより長い方を揃える数値に指定
	maxWidth := MaxStringWidth(lines)
	if length == -1 {
		length = maxWidth
	} else if length < maxWidth {
		length = maxWidth
	}

	ret := []string{}
	for _, line := range lines {
		l := runewidth.StringWidth(line)
		diff := length - l
		if diff%2 != 0 {
			line += " "
			diff--
		}

		var s string
		if padIsMultiByteString {
			div, mod := diff/4, diff%4
			s = line
			switch mod {
			case 1:
				s = " " + s
			case 2:
				s = " " + s + " "
			case 3:
				s = pad + s + " "
			}
			p := strings.Repeat(pad, div)
			s = p + s + p
		} else {
			div, mod := diff/2, diff%2
			s = line
			if 0 < mod {
				s = " " + s
			}
			p := strings.Repeat(pad, div)
			s = p + s + p
		}
		ret = append(ret, s)
	}
	return ret
}

// AlignRight は文字列を右寄せする。
// 右寄せは見た目上の文字列の長さで揃える。
// length = -1のときは、引数文字列の最長の長さに合わせる。
// padは埋める文字列を指定する。埋める文字が見た目上でマルチバイトの場合は
// たとえlengthが奇数でも+1して偶数になるように調整する。
func AlignRight(lines []string, length int, pad string) []string {
	if length == 0 || len(lines) < 1 {
		return lines
	}

	// 空白埋めする文字列がマルチバイト文字かどうか
	padWidtn := runewidth.StringWidth(pad)
	padIsMultiByteString := padWidtn == 2

	// -1のときは文字列の長さをalignの長さにする
	// パッディングの長さと、処理対象の文字列のより長い方を揃える数値に指定
	maxWidth := MaxStringWidth(lines)
	if length == -1 {
		length = maxWidth
	} else if length < maxWidth {
		length = maxWidth
	}

	// マルチバイト文字を使うときは長さを偶数に揃える
	if padIsMultiByteString && length%2 != 0 {
		length++
	}

	ret := []string{}
	for _, line := range lines {
		l := runewidth.StringWidth(line)
		diff := length - l
		if diff%2 != 0 {
			line = " " + line
			diff--
		}
		// Repeatするときにマルチバイト文字を使うときは2分の1にする
		if padIsMultiByteString {
			diff /= 2
		}
		s := strings.Repeat(pad, diff) + line
		ret = append(ret, s)
	}
	return ret
}

// AlignVerticalTop は垂直上方向に位置揃えする
func AlignVerticalTop(lines []string, height int) []string {
	if height <= len(lines) {
		return lines
	}

	var ret []string
	for _, line := range lines {
		ret = append(ret, line)
	}
	for i := 0; i < height-len(lines); i++ {
		ret = append(ret, "")
	}

	return ret
}

// AlignVerticalCenter は垂直中央方向に位置揃えする
func AlignVerticalCenter(lines []string, height int) []string {
	if height <= len(lines) {
		return lines
	}

	diffLen := height - len(lines)
	halfDiff := diffLen / 2

	var ret []string
	for i := 0; i < halfDiff; i++ {
		ret = append(ret, "")
	}
	for _, line := range lines {
		ret = append(ret, line)
	}
	for i := 0; i < halfDiff; i++ {
		ret = append(ret, "")
	}

	// 追加の必要な行数が奇数の場合は、下側に余った1行を追加する
	if diffLen%2 == 1 {
		ret = append(ret, "")
	}

	return ret
}

// AlignVerticalBottom は垂直下方向に位置揃えする
func AlignVerticalBottom(lines []string, height int) []string {
	if height <= len(lines) {
		return lines
	}

	var ret []string
	for i := 0; i < height-len(lines); i++ {
		ret = append(ret, "")
	}
	for _, line := range lines {
		ret = append(ret, line)
	}

	return ret
}