table_unicode.go

package tablewriter

import (
	"errors"

	"github.com/jwalton/gchalk"
)

type UnicodeLineStyle int

const (
	Regular UnicodeLineStyle = iota
	Thick
	Double
)

const (
	symsRR = "─│┌┐└┘├┤┬┴┼"
	symsTT = "━┃┏┓┗┛┣┫┳┻╋"
	symsDD = "═║╔╗╚╝╠╣╦╩╬"
	symsRT = "─┃┎┒┖┚┠┨┰┸╂"
	symsTR = "━│┍┑┕┙┝┥┯┷┿"
	symsRD = "─║╓╖╙╜╟╢╥╨╫"
	symsDR = "═│╒╕╘╛╞╡╤╧╪"
)

func simpleSyms(center, row, column string) []string {
	return []string{row, column, center, center, center, center, center, center, center, center, center}
}

// Use unicode box drawing symbols to achieve the specified line styles.
// Note that combinations of thick and double lines are not supported.
// Will return an error in case of unsupported combinations.
func (t *Table) SetUnicodeHV(horizontal, vertical UnicodeLineStyle) error {
	return t.SetUnicodeHVC(horizontal, vertical, gchalk.Ansi256(15))
}

// Set Unicode Table with Colored borders
func (t *Table) SetUnicodeHVC(horizontal, vertical UnicodeLineStyle, color gchalk.ColorFn) error {
	var syms string
	switch {
	case horizontal == Regular && vertical == Regular:
		syms = symsRR
	case horizontal == Thick && vertical == Thick:
		syms = symsTT
	case horizontal == Double && vertical == Double:
		syms = symsDD
	case horizontal == Regular && vertical == Thick:
		syms = symsRT
	case horizontal == Thick && vertical == Regular:
		syms = symsTR
	case horizontal == Regular && vertical == Double:
		syms = symsRD
	case horizontal == Double && vertical == Regular:
		syms = symsDR
	default:
		return errors.New("Unsupported combination of unicode line styles")
	}
	t.syms = make([]string, 0, 11)
	for _, sym := range []rune(syms) {
		t.syms = append(t.syms, color(string(sym)))
	}
	return nil
}