zyedidia.micro/cmd/micro/display/window.go

package display

import (
	"strconv"
	"unicode/utf8"

	runewidth "github.com/mattn/go-runewidth"
	"github.com/zyedidia/micro/cmd/micro/buffer"
	"github.com/zyedidia/micro/cmd/micro/config"
	"github.com/zyedidia/micro/cmd/micro/screen"
	"github.com/zyedidia/micro/cmd/micro/util"
	"github.com/zyedidia/tcell"
)

type Window interface {
	Display()
	Clear()
}

type BufWindow struct {
	// X and Y coordinates for the top left of the window
	X int
	Y int

	// Width and Height for the window
	Width  int
	Height int

	// Which line in the buffer to start displaying at (vertical scroll)
	StartLine int
	// Which visual column in the to start displaying at (horizontal scroll)
	StartCol int

	// Buffer being shown in this window
	Buf *buffer.Buffer

	sline *StatusLine
}

func NewBufWindow(x, y, width, height int, buf *buffer.Buffer) *BufWindow {
	w := new(BufWindow)
	w.X, w.Y, w.Width, w.Height, w.Buf = x, y, width, height, buf

	w.sline = NewStatusLine(w)

	return w
}

func (w *BufWindow) Clear() {
	for y := 0; y < w.Height; y++ {
		for x := 0; x < w.Width; x++ {
			screen.Screen.SetContent(w.X+x, w.Y+y, ' ', nil, config.DefStyle)
		}
	}
}

func (w *BufWindow) drawLineNum(lineNumStyle tcell.Style, softwrapped bool, maxLineNumLength int, vloc *buffer.Loc, bloc *buffer.Loc) {
	lineNum := strconv.Itoa(bloc.Y + 1)

	// Write the spaces before the line number if necessary
	for i := 0; i < maxLineNumLength-len(lineNum); i++ {
		screen.Screen.SetContent(w.X+vloc.X, w.Y+vloc.Y, ' ', nil, lineNumStyle)
		vloc.X++
	}
	// Write the actual line number
	for _, ch := range lineNum {
		if softwrapped {
			screen.Screen.SetContent(w.X+vloc.X, w.Y+vloc.Y, ' ', nil, lineNumStyle)
		} else {
			screen.Screen.SetContent(w.X+vloc.X, w.Y+vloc.Y, ch, nil, lineNumStyle)
		}
		vloc.X++
	}

	// Write the extra space
	screen.Screen.SetContent(w.X+vloc.X, w.Y+vloc.Y, ' ', nil, lineNumStyle)
	vloc.X++
}

// getStyle returns the highlight style for the given character position
// If there is no change to the current highlight style it just returns that
func (w *BufWindow) getStyle(style tcell.Style, bloc buffer.Loc, r rune) tcell.Style {
	if group, ok := w.Buf.Match(bloc.Y)[bloc.X]; ok {
		s := config.GetColor(group.String())
		return s
	}
	return style
}

func (w *BufWindow) showCursor(x, y int, main bool) {
	if main {
		screen.Screen.ShowCursor(x, y)
	} else {
		r, _, _, _ := screen.Screen.GetContent(x, y)
		screen.Screen.SetContent(x, y, r, nil, config.DefStyle.Reverse(true))
	}
}

// displayBuffer draws the buffer being shown in this window on the screen.Screen
func (w *BufWindow) displayBuffer() {
	b := w.Buf

	bufHeight := w.Height
	if b.Settings["statusline"].(bool) {
		bufHeight--
	}

	// TODO: Rehighlighting
	start := w.StartLine
	if b.Settings["syntax"].(bool) && b.SyntaxDef != nil {
		if start > 0 && b.Rehighlight(start-1) {
			b.Highlighter.ReHighlightLine(b, start-1)
			b.SetRehighlight(start-1, false)
		}

		b.Highlighter.ReHighlightStates(b, start)

		b.Highlighter.HighlightMatches(b, w.StartLine, w.StartLine+bufHeight)
	}

	lineNumStyle := config.DefStyle
	if style, ok := config.Colorscheme["line-number"]; ok {
		lineNumStyle = style
	}

	// We need to know the string length of the largest line number
	// so we can pad appropriately when displaying line numbers
	maxLineNumLength := len(strconv.Itoa(b.LinesNum()))

	tabsize := int(b.Settings["tabsize"].(float64))
	softwrap := b.Settings["softwrap"].(bool)

	// this represents the current draw position
	// within the current window
	vloc := buffer.Loc{0, 0}

	// this represents the current draw position in the buffer (char positions)
	bloc := buffer.Loc{w.StartCol, w.StartLine}

	activeC := w.Buf.GetActiveCursor()

	curStyle := config.DefStyle
	for vloc.Y = 0; vloc.Y < bufHeight; vloc.Y++ {
		vloc.X = 0
		if b.Settings["ruler"].(bool) {
			w.drawLineNum(lineNumStyle, false, maxLineNumLength, &vloc, &bloc)
		}

		line := b.LineBytes(bloc.Y)
		line, nColsBeforeStart := util.SliceVisualEnd(line, bloc.X, tabsize)

		draw := func(r rune, style tcell.Style) {
			if nColsBeforeStart <= 0 {
				if activeC.HasSelection() &&
					(bloc.GreaterEqual(activeC.CurSelection[0]) && bloc.LessThan(activeC.CurSelection[1]) ||
						bloc.LessThan(activeC.CurSelection[0]) && bloc.GreaterEqual(activeC.CurSelection[1])) {
					// The current character is selected
					style = config.DefStyle.Reverse(true)

					if s, ok := config.Colorscheme["selection"]; ok {
						style = s
					}

				}

				screen.Screen.SetContent(w.X+vloc.X, w.Y+vloc.Y, r, nil, style)
				vloc.X++
			}
			nColsBeforeStart--
		}

		totalwidth := bloc.X - nColsBeforeStart
		for len(line) > 0 {
			if activeC.X == bloc.X && activeC.Y == bloc.Y {
				w.showCursor(vloc.X, vloc.Y, true)
			}

			r, size := utf8.DecodeRune(line)
			curStyle = w.getStyle(curStyle, bloc, r)

			draw(r, curStyle)

			width := 0

			char := ' '
			switch r {
			case '\t':
				ts := tabsize - (totalwidth % tabsize)
				width = ts
			default:
				width = runewidth.RuneWidth(r)
				char = '@'
			}

			bloc.X++
			line = line[size:]

			// Draw any extra characters either spaces for tabs or @ for incomplete wide runes
			if width > 1 {
				for i := 1; i < width; i++ {
					draw(char, curStyle)
				}
			}
			totalwidth += width

			// If we reach the end of the window then we either stop or we wrap for softwrap
			if vloc.X >= w.Width {
				if !softwrap {
					break
				} else {
					vloc.Y++
					if vloc.Y >= bufHeight {
						break
					}
					vloc.X = 0
					// This will draw an empty line number because the current line is wrapped
					w.drawLineNum(lineNumStyle, true, maxLineNumLength, &vloc, &bloc)
				}
			}
		}
		if activeC.X == bloc.X && activeC.Y == bloc.Y {
			w.showCursor(vloc.X, vloc.Y, true)
		}
		bloc.X = w.StartCol
		bloc.Y++
		if bloc.Y >= b.LinesNum() {
			break
		}
	}
}

func (w *BufWindow) displayStatusLine() {
	w.sline.Display()
}

func (w *BufWindow) Display() {
	w.displayBuffer()
	w.displayStatusLine()
}