typer/src/line_index.go

package main

import (
	"github.com/gdamore/tcell/v2"
	"strconv"
	"strings"
)

func drawLineIndex(window *Window) {
	screen := window.screen
	buffer := window.CurrentBuffer

	lineIndexStyle := tcell.StyleDefault.Background(CurrentStyle.LineIndexBg).Foreground(CurrentStyle.LineIndexFg)

	_, sizeY := screen.Size()

	y := 0
	if window.ShowTopMenu {
		y = 1
	}

	lineIndexSize := getLineIndexSize(window)

	for lineIndex := 1 + buffer.OffsetY; lineIndex <= strings.Count(buffer.Contents, "\n")+1 && lineIndex < sizeY+buffer.OffsetY; lineIndex++ {

		for x := 0; x < lineIndexSize; x++ {
			screen.SetContent(x, y, ' ', nil, lineIndexStyle)
		}

		text := strconv.Itoa(lineIndex)

		drawText(screen, lineIndexSize-len(text)-1, y, lineIndexSize, y, lineIndexStyle, text)
		y++
	}
}

func getLineIndexSize(window *Window) int {
	i := strings.Count(window.CurrentBuffer.Contents, "\n") + 1
	if i == 0 {
		return 4
	}
	count := 0
	for i != 0 {
		i /= 10
		count++
	}

	if count < 3 {
		count = 4
	} else {
		count += 1
	}

	return count
}