ansi.go

package main

import (
	"fmt"
	"strconv"
	"strings"
)

/*
Useful Reading:
	https://en.wikipedia.org/wiki/ANSI_escape_code#Colors
	https://github.com/fatih/color
*/

const esc = "\x1b"

type ansiCode int

const (
	reset ansiCode = iota
	bold
	faint
	italic
	underline
)

const (
	black ansiCode = iota + 30
	red
	green
	yellow
	blue
	magenta
	cyan
	white
)

func ansi(codes ...ansiCode) string {
	codeStrings := make([]string, len(codes))
	for i, v := range codes {
		codeStrings[i] = strconv.Itoa(int(v))
	}
	delimitedCodes := strings.Join(codeStrings, ";")
	return fmt.Sprintf("%s[%sm", esc, delimitedCodes)
}

func color(s string, codes ...ansiCode) string {
	return fmt.Sprintf("%s%v%s", ansi(codes...), s, ansi(reset))
}