nix-amer/internal/clifmt/symbols.go

package clifmt

import (
	"fmt"
	"strings"
)

var titleIndex = 0
var alignOffset = 30

// Warn returns a red warning ASCII sign. If a string is given
// as argument, it will print it after the warning sign
func Warn(s ...string) string {
	if len(s) == 0 {
		return Color(31, "/!\\")
	}

	return fmt.Sprintf("%s  %s", Warn(), s[0])
}

// Info returns a blue info ASCII sign. If a string is given
// as argument, it will print it after the info sign
func Info(s ...string) string {
	if len(s) == 0 {
		return Color(34, "(!)")
	}

	return fmt.Sprintf("%s  %s", Info(), s[0])
}

// Title prints a formatted title (auto-indexed from local counted)
func Title(s string) {
	titleIndex++
	fmt.Printf("\n%s |%d| %s %s\n", Color(33, ">>", false), titleIndex, s, Color(33, "<<", false))

}

// Align prints strings with space padding to align line ends (fixed width)
// also crops the content to add '...' at the end if too long (must have a space
// at the end)
func Align(s string) {

	// 1. replace tabs with 4 spaces
	tabs := strings.Split(s, "\t")
	s = strings.Join(tabs, "    ")

	// 1. get real size
	size := getRealSize(s)

	if size > alignOffset-6 {

		for i, l := 0, len(s); i < l; i++ { // find when real size is right under
			next := fmt.Sprintf("%s\033[0m… ", s[0:i+1])
			if getRealSize(next) >= alignOffset-5 {
				s = next
				break
			}

		}
		size = getRealSize(s)
	}

	// 2. print string
	fmt.Printf("%s", s)

	// 3. print trailing spaces
	for i := size; i < alignOffset; i++ {
		fmt.Printf(" ")
	}
}

// getRealSize returns the real size escaping special characters
func getRealSize(s string) int {
	// 1. get actual size
	size := len(s)

	// 2. remove \033[XYm format characters
	size -= 7 * len(strings.Split(s, "\033"))

	// 3. compensate 1 char for \033[0m
	size += len(strings.Split(s, "\033[0m"))

	return size
}