text_model.go

package gutter

import (
	"sort"
	"unicode"
)

type textModel struct {
	clientID        float64
	word            []rune
	selectionBase   int
	selectionExtent int
	notifyState     func()
}

// Modifier keys from glfw
const (
	ModNone         int = 0
	ModShift        int = 1
	ModControl      int = 2
	ModShiftControl int = 3
	ModAlt          int = 4
	ModSuper        int = 8
)

func (state *textModel) isSelected() bool {
	return state.selectionBase != state.selectionExtent
}

func (state *textModel) addChar(char []rune) {
	state.RemoveSelectedText()
	newWord := make([]rune, 0, len(char)+len(state.word))
	newWord = append(newWord, state.word[:state.selectionBase]...)
	newWord = append(newWord, char...)
	newWord = append(newWord, state.word[state.selectionBase:]...)

	state.word = newWord

	state.selectionBase += len(char)
	state.selectionExtent = state.selectionBase
	state.notifyState()
}

func (state *textModel) MoveCursorHome(mods int) {
	state.selectionBase = 0
	if mods != ModShift {
		state.selectionExtent = state.selectionBase
	}
	state.notifyState()
}

func (state *textModel) MoveCursorEnd(mods int) {
	state.selectionBase = len(state.word)
	if mods != ModShift {
		state.selectionExtent = state.selectionBase
	}
	state.notifyState()
}

func (state *textModel) MoveCursorLeft(mods int) {
	if mods == ModShiftControl || mods == ModControl {
		state.selectionBase = indexStartLeadingWord(state.word, state.selectionBase)
	} else if state.selectionBase > 0 {

		if mods != ModShift && state.isSelected() {
			state.selectionBase, _, _ = state.GetSelectedText()
		} else {
			state.selectionBase--
		}
	}

	if mods == ModNone || mods == ModControl {
		state.selectionExtent = state.selectionBase
	}

	state.notifyState()
}

func (state *textModel) MoveCursorRight(mods int) {
	if mods == ModShiftControl || mods == ModControl {
		state.selectionBase = indexEndForwardWord(state.word, state.selectionBase)
	} else if state.selectionBase < len(state.word) {

		if mods != ModShift && state.isSelected() {
			_, state.selectionBase, _ = state.GetSelectedText()
		} else {
			state.selectionBase++
		}
	}

	if mods == ModNone || mods == ModControl {
		state.selectionExtent = state.selectionBase
	}

	state.notifyState()
}

func (state *textModel) SelectAll() {
	state.selectionBase = 0
	state.selectionExtent = len(state.word)
	state.notifyState()
}

func (state *textModel) Delete(mods int) {
	if state.RemoveSelectedText() {
		state.notifyState()
		return
	}

	if state.selectionBase < len(state.word) {
		state.word = append(state.word[:state.selectionBase], state.word[state.selectionBase+1:]...)
		state.notifyState()
	}
}

func (state *textModel) Backspace(mods int) {
	if state.RemoveSelectedText() {
		state.notifyState()
		return
	}

	if len(state.word) > 0 && state.selectionBase > 0 {
		if mods == ModControl {
			deleteUpTo := indexStartLeadingWord(state.word, state.selectionBase)
			state.word = append(state.word[:deleteUpTo], state.word[state.selectionBase:]...)
			state.selectionBase = deleteUpTo
			state.selectionExtent = deleteUpTo
			state.notifyState()
		} else {
			state.word = append(state.word[:state.selectionBase-1], state.word[state.selectionBase:]...)
			if state.selectionBase > 0 {
				state.selectionBase--
			}
			state.selectionExtent = state.selectionBase
			state.notifyState()
		}
	}

}

// RemoveSelectedText do nothing if no text is selected
// return true if the state has been updated
func (state *textModel) RemoveSelectedText() bool {
	if state.isSelected() {
		selectionIndexStart, selectionIndexEnd, _ := state.GetSelectedText()
		state.word = append(state.word[:selectionIndexStart], state.word[selectionIndexEnd:]...)
		state.selectionBase = selectionIndexStart
		state.selectionExtent = selectionIndexStart
		state.selectionExtent = state.selectionBase
		state.notifyState()
		return true
	}
	return false

}

// GetSelectedText return
// (left index of the selection, right index of the selection,
// the content of the selection)
func (state *textModel) GetSelectedText() (int, int, string) {
	selectionIndex := []int{state.selectionBase, state.selectionExtent}
	sort.Ints(selectionIndex)
	return selectionIndex[0],
		selectionIndex[1],
		string(state.word[selectionIndex[0]:selectionIndex[1]])
}

// Helpers
func indexStartLeadingWord(line []rune, start int) int {
	pos := start
	// Remove whitespace to the left
	for {
		if pos == 0 || !unicode.IsSpace(line[pos-1]) {
			break
		}
		pos--
	}
	// Remove non-whitespace to the left
	for {
		if pos == 0 || unicode.IsSpace(line[pos-1]) {
			break
		}
		pos--
	}
	return pos
}

func indexEndForwardWord(line []rune, start int) int {
	pos := start
	lineSize := len(line)
	// Remove whitespace to the right
	for {
		if pos == lineSize || !unicode.IsSpace(line[pos]) {
			break
		}
		pos++
	}
	// Remove non-whitespace to the right
	for {
		if pos == lineSize || unicode.IsSpace(line[pos]) {
			break
		}
		pos++
	}
	return pos
}