Added rendering stuff, updated go mod

2024-08-08 18:51:51 -06:00
parent f7eaa0d08d
commit d426892e6a
9 changed files with 447 additions and 3 deletions
@@ -0,0 +1,135 @@
+package ansii
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"golang.org/x/term"
+)
+
+type ANSI string
+
+const (
+	reset       ANSI = "\033[0m"
+	plain       ANSI = ""
+	bold        ANSI = "\033[1m"
+	underline   ANSI = "\033[4m"
+	black       ANSI = "\033[30m"
+	red         ANSI = "\033[31m"
+	green       ANSI = "\033[32m"
+	yellow      ANSI = "\033[33m"
+	blue        ANSI = "\033[34m"
+	purple      ANSI = "\033[35m"
+	cyan        ANSI = "\033[36m"
+	white       ANSI = "\033[37m"
+	blackBg     ANSI = "\033[40m"
+	redBg       ANSI = "\033[41m"
+	greenBg     ANSI = "\033[42m"
+	yellowBg    ANSI = "\033[43m"
+	blueBg      ANSI = "\033[44m"
+	purpleBg    ANSI = "\033[45m"
+	cyanBg      ANSI = "\033[46m"
+	whiteBg     ANSI = "\033[47m"
+	clearScreen ANSI = "\033[2J"
+	hideCursor  ANSI = "\033[?25l"
+	showCursor  ANSI = "\033[?25h"
+)
+
+type Offset struct {
+	X int
+	Y int
+}
+
+type style struct {
+	Reset     ANSI
+	Plain     ANSI
+	Bold      ANSI
+	Underline ANSI
+}
+
+type color struct {
+	Black  ANSI
+	Red    ANSI
+	Green  ANSI
+	Yellow ANSI
+	Blue   ANSI
+	Purple ANSI
+	Cyan   ANSI
+	White  ANSI
+}
+
+type screen struct {
+	ClearScreen ANSI
+	HideCursor  ANSI
+	ShowCursor  ANSI
+}
+
+type ascii struct {
+	Block string
+}
+
+func GetTermSize() (width int, height int) {
+	var fd int = int(os.Stdout.Fd())
+	width, height, err := term.GetSize(fd)
+	if err != nil {
+		fmt.Println(string(Screen.ClearScreen) + "Fatal: error getting terminal size.")
+		os.Exit(1)
+	}
+	return width, height
+}
+
+func MakeTermRaw() (*term.State, error) {
+	var fd int = int(os.Stdout.Fd())
+	return term.MakeRaw(fd)
+}
+
+func RestoreTerm(prev *term.State) error {
+	var fd int = int(os.Stdout.Fd())
+	return term.Restore(fd, prev)
+}
+
+func (s screen) PlaceCursor(offset Offset) ANSI {
+	return ANSI(fmt.Sprintf("\033[%d;%dH", offset.Y, offset.X))
+}
+
+var (
+	Styles = style{Bold: bold, Underline: underline, Reset: reset, Plain: plain}
+	Colors = color{Red: red, Green: green, Yellow: yellow, Blue: blue, Purple: purple, Cyan: cyan, White: white}
+	Screen = screen{ClearScreen: clearScreen, HideCursor: hideCursor, ShowCursor: showCursor}
+	Blocks = ascii{Block: "█"}
+)
+
+// Draws a box of dimensions `height` and `width` at `offset`.
+// The `offset` is the top left cell of the square.
+// Blocks that would be placed off screen are clipped.
+func DrawBox(builder *strings.Builder, offset Offset, height int, width int, style ANSI) {
+	builder.WriteString(string(style))
+	for hIdx := 0; hIdx < height; hIdx++ {
+
+		if hIdx == 0 || hIdx == height-1 {
+			for wIdx := 0; wIdx < width; wIdx++ {
+				DrawPixel(builder, Offset{X: offset.X + wIdx, Y: offset.Y + hIdx})
+			}
+		} else {
+			DrawPixel(builder, Offset{X: offset.X, Y: offset.Y + hIdx})
+			DrawPixel(builder, Offset{X: offset.X + width - 1, Y: offset.Y + hIdx})
+		}
+	}
+	builder.WriteString(string(Styles.Reset))
+	return
+}
+
+func DrawPixel(builder *strings.Builder, offset Offset) {
+	termWidth, termHeight := GetTermSize()
+	if offset.X > termWidth || offset.Y > termHeight || offset.X < 0 || offset.Y < 0 {
+		return
+	}
+	builder.WriteString(string(Screen.PlaceCursor(offset) + ANSI(Blocks.Block)))
+}
+
+func DrawPixelStyle(builder *strings.Builder, offset Offset, style ANSI) {
+	builder.WriteString(string(style))
+	DrawPixel(builder, offset)
+	builder.WriteString(string(Styles.Reset))
+}