// SPDX-License-Identifier: Unlicense OR MIT

package gel

import (
	"bufio"
	"bytes"
	"image"
	"io"
	"math"
	"runtime"
	"sort"
	"strings"
	"time"
	"unicode"
	"unicode/utf8"

	"github.com/p9c/p9/pkg/gel/gio/f32"
	"github.com/p9c/p9/pkg/gel/gio/io/clipboard"
	"github.com/p9c/p9/pkg/gel/gio/io/event"
	"github.com/p9c/p9/pkg/gel/gio/io/key"
	"github.com/p9c/p9/pkg/gel/gio/io/pointer"
	"github.com/p9c/p9/pkg/gel/gio/layout"
	"github.com/p9c/p9/pkg/gel/gio/op"
	"github.com/p9c/p9/pkg/gel/gio/op/clip"
	"github.com/p9c/p9/pkg/gel/gio/op/paint"
	"github.com/p9c/p9/pkg/gel/gio/text"
	"github.com/p9c/p9/pkg/gel/gio/unit"

	"github.com/p9c/p9/pkg/gel/gio/gesture"

	"golang.org/x/image/math/fixed"

	clipboard3 "github.com/p9c/p9/pkg/gel/clipboard"
)

func (w *Window) Editor() *Editor {
	e := &Editor{
		submitHook: func(string) {},
		changeHook: func(string) {},
		focusHook:  func(bool) {},
	}
	return e
}

// Editor implements an editable and scrollable text area.
type Editor struct {
	alignment text.Alignment
	// singleLine force the text to stay on a single line. singleLine also sets the scrolling direction to horizontal.
	singleLine bool
	// submit enabled translation of carriage return keys to SubmitEvents. If not enabled, carriage returns are inserted
	// as newlines in the text.
	submit bool
	// mask replaces the visual display of each rune in the contents with the given rune. Newline characters are not
	// masked. When non-zero, the unmasked contents are accessed by Len, Text, and SetText.
	mask rune

	eventKey     int
	font         text.Font
	shaper       text.Shaper
	textSize     fixed.Int26_6
	blinkStart   time.Time
	focused      bool
	editBuffer   editBuffer
	maskReader   maskReader
	lastMask     rune
	maxWidth     int
	viewSize     image.Point
	valid        bool
	lines        []text.Line
	shapes       []line
	dims         layout.Dimensions
	requestFocus bool
	caret        struct {
		on     bool
		scroll bool
		// start is the current caret position, and also the start position of
		// selected text. end is the end positon of selected text. If start.ofs
		// == end.ofs, then there's no selection. Note that it's possible (and
		// common) that the caret (start) is after the end, e.g. after
		// Shift-DownArrow.
		start combinedPos
		end   combinedPos
	}
	dragging  bool
	dragger   gesture.Drag
	scroller  gesture.Scroll
	scrollOff image.Point

	clicker gesture.Click

	// events is the list of events not yet processed.
	events []EditorEvent
	// prevEvents is the number of events from the previous frame.
	prevEvents int
	// the following are hooks for change events on the editor
	submitHook func(string)
	changeHook func(string)
	focusHook  func(bool)
}

type maskReader struct {
	// rr is the underlying reader.
	rr      io.RuneReader
	maskBuf [utf8.UTFMax]byte
	// mask is the utf-8 encoded mask rune.
	mask []byte
	// overflow contains excess mask bytes left over after the last Read call.
	overflow []byte
}

// combinedPos is a point in the editor.
type combinedPos struct {
	// editorBuffer offset. The other three fields are based off of this one.
	ofs int

	// lineCol.Y = line (offset into Editor.lines), and X = col (offset into
	// Editor.lines[Y])
	lineCol screenPos

	// Pixel coordinates
	x fixed.Int26_6
	y int

	// xoff is the offset to the current position when moving between lines.
	xoff fixed.Int26_6
}

type selectionAction int

const (
	selectionExtend selectionAction = iota
	selectionClear
)

func (m *maskReader) Reset(r io.RuneReader, mr rune) {
	m.rr = r
	n := utf8.EncodeRune(m.maskBuf[:], mr)
	m.mask = m.maskBuf[:n]
}

// Read reads from the underlying reader and replaces every rune with the mask rune.
func (m *maskReader) Read(b []byte) (n int, e error) {
	for len(b) > 0 {
		var replacement []byte
		if len(m.overflow) > 0 {
			replacement = m.overflow
		} else {
			var r rune
			r, _, e = m.rr.ReadRune()
			if e != nil {
				break
			}
			if r == '\n' {
				replacement = []byte{'\n'}
			} else {
				replacement = m.mask
			}
		}
		nn := copy(b, replacement)
		m.overflow = replacement[nn:]
		n += nn
		b = b[nn:]
	}
	return n, e
}

type EditorEvent interface {
	isEditorEvent()
}

// A ChangeEvent is generated for every user change to the text.
type ChangeEvent struct{}

// A SubmitEvent is generated when submit is set and a carriage return key is pressed.
type SubmitEvent struct {
	Text string
}

// A SelectEvent is generated when the user selects some text, or changes the
// selection (e.g. with a shift-click), including if they remove the
// selection. The selected text is not part of the event, on the theory that
// it could be a relatively expensive operation (for a large editor), most
// applications won't actually care about it, and those that do can call
// Editor.SelectedText() (which can be empty).
type SelectEvent struct{}

type line struct {
	offset         image.Point
	clip           op.CallOp
	selected       bool
	selectionYOffs int
	selectionSize  image.Point
}

const (
	blinksPerSecond  = 1
	maxBlinkDuration = 10 * time.Second
)

// Events returns available editor events.
func (e *Editor) Events() []EditorEvent {
	events := e.events
	e.events = nil
	e.prevEvents = 0
	return events
}

func (e *Editor) processEvents(gtx layout.Context) {
	// Flush events from before the previous layout.
	n := copy(e.events, e.events[e.prevEvents:])
	e.events = e.events[:n]
	e.prevEvents = n
	if e.shaper == nil {
		// Can't process events without a shaper.
		return
	}
	oldStart, oldLen := min(e.caret.start.ofs, e.caret.end.ofs), e.SelectionLen()
	e.processPointer(gtx)
	e.processKey(gtx)
	if newStart, newLen := min(e.caret.start.ofs, e.caret.end.ofs),
		e.SelectionLen(); oldStart != newStart || oldLen != newLen {
		e.events = append(e.events, SelectEvent{})
		if e.SelectionLen() != 0 {
			st := e.SelectedText()
			_ = clipboard3.SetPrimary(st)
			// I.F("new primary buffer string: '%s'", st)
		}
	}
}

func (e *Editor) makeValid(positions ...*combinedPos) {
	if e.valid {
		return
	}
	e.lines, e.dims = e.layoutText(e.shaper)
	e.makeValidCaret(positions...)
	e.valid = true
}

func (e *Editor) processPointer(gtx layout.Context) {
	sbounds := e.scrollBounds()
	var smin, smax int
	var axis gesture.Axis
	if e.singleLine {
		axis = gesture.Horizontal
		smin, smax = sbounds.Min.X, sbounds.Max.X
	} else {
		axis = gesture.Vertical
		smin, smax = sbounds.Min.Y, sbounds.Max.Y
	}
	sdist := e.scroller.Scroll(gtx.Metric, gtx, gtx.Now, axis)
	var soff int
	if e.singleLine {
		e.scrollRel(sdist, 0)
		soff = e.scrollOff.X
	} else {
		e.scrollRel(0, sdist)
		soff = e.scrollOff.Y
	}
	for _, evt := range e.clickDragEvents(gtx) {
		switch evt := evt.(type) {
		case gesture.ClickEvent:
			switch {
			case evt.Type == gesture.TypePress && evt.Source == pointer.Mouse,
				evt.Type == gesture.TypeClick:
				if evt.Button == pointer.ButtonPrimary {
					prevCaretPos := e.caret.start
					e.blinkStart = gtx.Now
					e.moveCoord(image.Point{
						X: int(math.Round(float64(evt.Position.X))),
						Y: int(math.Round(float64(evt.Position.Y))),
					})
					e.requestFocus = true
					if e.scroller.State() != gesture.StateFlinging {
						e.caret.scroll = true
					}
					if evt.Modifiers == key.ModShift {
						// If they clicked closer to the end, then change the end to
						// where the caret used to be (effectively swapping start & end).
						if abs(e.caret.end.ofs-e.caret.start.ofs) < abs(e.caret.start.ofs-prevCaretPos.ofs) {
							e.caret.end = prevCaretPos
						}
					} else {
						e.ClearSelection()
					}
				}
				e.dragging = true
				// I.S(evt)
				// Process a double-click.
				// Double and triple clicks are primary button only
				if evt.Button == pointer.ButtonPrimary {
					if evt.NumClicks == 2 {
						e.moveWord(-1, selectionClear)
						e.moveWord(1, selectionExtend)
						e.dragging = false
					}
					// process a triple click - select all. This required forking github.com/p9c/p9/pkg/gel/gio/gesture
					if evt.NumClicks == 3 {
						e.dragging = false
						e.caret.end, e.caret.start = e.offsetToScreenPos2(0, e.Len())
						evt.NumClicks = 0
					}
				}
				if evt.Button == pointer.ButtonTertiary && evt.Type == gesture.TypeClick {
					e.blinkStart = gtx.Now
					e.moveCoord(image.Point{
						X: int(math.Round(float64(evt.Position.X))),
						Y: int(math.Round(float64(evt.Position.Y))),
					})
					e.ClearSelection()
					primary := clipboard3.GetPrimary()
					e.prepend(primary)
					distance := utf8.RuneCountInString(primary)
					e.MoveCaret(distance, int(selectionExtend))
				}
			}
		case pointer.Event:
			release := false
			switch {
			// on X11 process middle click as insert Primary at pointer position
			case evt.Buttons == pointer.ButtonTertiary && evt.Source == pointer.Mouse:
				e.moveCoord(image.Point{
					X: int(math.Round(float64(evt.Position.X))),
					Y: int(math.Round(float64(evt.Position.Y))),
				})
				e.prepend(clipboard3.GetPrimary())
			case evt.Type == pointer.Release && evt.Source == pointer.Mouse:
				release = true
				fallthrough
			case evt.Type == pointer.Drag && evt.Source == pointer.Mouse:
				if e.dragging {
					e.blinkStart = gtx.Now
					e.moveCoord(image.Point{
						X: int(math.Round(float64(evt.Position.X))),
						Y: int(math.Round(float64(evt.Position.Y))),
					})
					e.caret.scroll = true

					if release {
						e.dragging = false
					}
				}
			default:
				// I.S(evt)
			}
		}
	}

	if (sdist > 0 && soff >= smax) || (sdist < 0 && soff <= smin) {
		e.scroller.Stop()
	}
}

func (e *Editor) clickDragEvents(gtx layout.Context) []event.Event {
	var combinedEvents []event.Event
	for _, evt := range e.clicker.Events(gtx) {
		combinedEvents = append(combinedEvents, evt)
	}
	for _, evt := range e.dragger.Events(gtx.Metric, gtx, gesture.Both) {
		combinedEvents = append(combinedEvents, evt)
	}
	return combinedEvents
}

func (e *Editor) processKey(gtx layout.Context) {
	if e.editBuffer.Changed() {
		e.events = append(e.events, ChangeEvent{})
	}
	for _, ke := range gtx.Events(&e.eventKey) {
		e.blinkStart = gtx.Now
		switch ke := ke.(type) {
		case key.FocusEvent:
			e.focused = ke.Focus
			e.focusHook(ke.Focus)
		case key.Event:
			if !e.focused || ke.State != key.Press {
				break
			}
			if e.submit && (ke.Name == key.NameReturn || ke.Name == key.NameEnter) {
				if !ke.Modifiers.Contain(key.ModShift) {
					e.events = append(e.events, SubmitEvent{
						Text: e.Text(),
					})
					continue
				}
			}
			if e.command(gtx, ke) {
				e.caret.scroll = true
				e.scroller.Stop()
			}
		case key.EditEvent:
			e.caret.scroll = true
			e.scroller.Stop()
			e.append(ke.Text)
		// Complete a paste event, initiated by Shortcut-V in Editor.command().
		case clipboard.Event:
			e.caret.scroll = true
			e.scroller.Stop()
			e.append(ke.Text)
		}
		if e.editBuffer.Changed() {
			e.events = append(e.events, ChangeEvent{})
			e.changeHook(e.Text())
		}
	}
}

func (e *Editor) moveLines(distance int, selAct selectionAction) {
	e.caret.start = e.movePosToLine(e.caret.start, e.caret.start.x+e.caret.start.xoff,
		e.caret.start.lineCol.Y+distance)
	e.updateSelection(selAct)
}

func (e *Editor) command(gtx layout.Context, k key.Event) bool {
	modSkip := key.ModCtrl
	if runtime.GOOS == "darwin" {
		modSkip = key.ModAlt
	}
	moveByWord := k.Modifiers.Contain(modSkip)
	selAct := selectionClear
	if k.Modifiers.Contain(key.ModShift) {
		selAct = selectionExtend
	}
	switch k.Name {
	case key.NameReturn, key.NameEnter:
		e.append("\n")
	case key.NameDeleteBackward:
		if moveByWord {
			e.deleteWord(-1)
		} else {
			e.Delete(-1)
		}
	case key.NameDeleteForward:
		if moveByWord {
			e.deleteWord(1)
		} else {
			e.Delete(1)
		}
	case key.NameUpArrow:
		e.moveLines(-1, selAct)
	case key.NameDownArrow:
		e.moveLines(+1, selAct)
	case key.NameLeftArrow:
		if moveByWord {
			e.moveWord(-1, selAct)
		} else {
			if selAct == selectionClear {
				e.ClearSelection()
			}
			e.MoveCaret(-1, -1*int(selAct))
		}
	case key.NameRightArrow:
		if moveByWord {
			e.moveWord(1, selAct)
		} else {
			if selAct == selectionClear {
				e.ClearSelection()
			}
			e.MoveCaret(1, int(selAct))
		}
	case key.NamePageUp:
		e.movePages(-1, selAct)
	case key.NamePageDown:
		e.movePages(+1, selAct)
	case key.NameHome:
		e.moveStart(selAct)
	case key.NameEnd:
		e.moveEnd(selAct)
	// Initiate a paste operation, by requesting the clipboard contents; other
	// half is in Editor.processKey() under clipboard.Event.
	case "V":
		if k.Modifiers != key.ModShortcut {
			return false
		}
		clipboard.ReadOp{Tag: &e.eventKey}.Add(gtx.Ops)
	// Copy or Cut selection -- ignored if nothing selected.
	case "C", "X":
		if k.Modifiers != key.ModShortcut {
			return false
		}
		if text := e.SelectedText(); text != "" {
			clipboard.WriteOp{Text: text}.Add(gtx.Ops)
			if k.Name == "X" {
				e.Delete(1)
			}
		}
	// Select all
	case "A":
		if k.Modifiers != key.ModShortcut {
			return false
		}
		e.caret.end, e.caret.start = e.offsetToScreenPos2(0, e.Len())
	default:
		return false
	}
	return true
}

// Focus requests the input focus for the _editor.
func (e *Editor) Focus() {
	e.requestFocus = true
}

// Focused returns whether the editor is focused or not.
func (e *Editor) Focused() bool {
	return e.focused
}

// Layout lays out the editor.
func (e *Editor) Layout(gtx layout.Context, sh text.Shaper, font text.Font, size unit.Value) layout.Dimensions {
	textSize := fixed.I(gtx.Px(size))
	if e.font != font || e.textSize != textSize {
		e.invalidate()
		e.font = font
		e.textSize = textSize
	}
	maxWidth := gtx.Constraints.Max.X
	if e.singleLine {
		maxWidth = Inf
	}
	if maxWidth != e.maxWidth {
		e.maxWidth = maxWidth
		e.invalidate()
	}
	if sh != e.shaper {
		e.shaper = sh
		e.invalidate()
	}
	if e.mask != e.lastMask {
		e.lastMask = e.mask
		e.invalidate()
	}
	e.makeValid()
	e.processEvents(gtx)
	e.makeValid()
	if viewSize := gtx.Constraints.Constrain(e.dims.Size); viewSize != e.viewSize {
		e.viewSize = viewSize
		e.invalidate()
	}
	e.makeValid()
	return e.layout(gtx)
}

func (e *Editor) layout(gtx layout.Context) layout.Dimensions {
	// Adjust scrolling for new viewport and layout.
	e.scrollRel(0, 0)
	if e.caret.scroll {
		e.caret.scroll = false
		e.scrollToCaret()
	}
	off := image.Point{
		X: -e.scrollOff.X,
		Y: -e.scrollOff.Y,
	}
	cl := textPadding(e.lines)
	cl.Max = cl.Max.Add(e.viewSize)
	startSel, endSel := sortPoints(e.caret.start.lineCol, e.caret.end.lineCol)
	it := segmentIterator{
		startSel:  startSel,
		endSel:    endSel,
		Lines:     e.lines,
		Clip:      cl,
		Alignment: e.alignment,
		Width:     e.viewSize.X,
		Offset:    off,
	}
	e.shapes = e.shapes[:0]
	for {
		lo, off, selected, yOffs, size, ok := it.Next()
		if !ok {
			break
		}
		path := e.shaper.Shape(e.font, e.textSize, lo)
		e.shapes = append(e.shapes, line{off, path, selected, yOffs, size})
	}
	key.InputOp{Tag: &e.eventKey}.Add(gtx.Ops)
	if e.requestFocus {
		key.FocusOp{Tag: &e.eventKey}.Add(gtx.Ops)
		key.SoftKeyboardOp{Show: true}.Add(gtx.Ops)
	}
	e.requestFocus = false
	// todo: this should be scaled
	pointerPadding := gtx.Px(unit.Dp(4))
	r := image.Rectangle{Max: e.viewSize}
	r.Min.X -= pointerPadding
	r.Min.Y -= pointerPadding
	r.Max.X += pointerPadding
	r.Max.X += pointerPadding
	pointer.Rect(r).Add(gtx.Ops)
	pointer.CursorNameOp{Name: pointer.CursorText}.Add(gtx.Ops)
	var scrollRange image.Rectangle
	if e.singleLine {
		scrollRange.Min.X = -e.scrollOff.X
		scrollRange.Max.X = max(0, e.dims.Size.X-(e.scrollOff.X+e.viewSize.X))
	} else {
		scrollRange.Min.Y = -e.scrollOff.Y
		scrollRange.Max.Y = max(0, e.dims.Size.Y-(e.scrollOff.Y+e.viewSize.Y))
	}
	e.scroller.Add(gtx.Ops, scrollRange)

	e.clicker.Add(gtx.Ops)
	e.dragger.Add(gtx.Ops)
	e.caret.on = false
	if e.focused {
		now := gtx.Now
		dt := now.Sub(e.blinkStart)
		blinking := dt < maxBlinkDuration
		const timePerBlink = time.Second / blinksPerSecond
		nextBlink := now.Add(timePerBlink/2 - dt%(timePerBlink/2))
		if blinking {
			redraw := op.InvalidateOp{At: nextBlink}
			redraw.Add(gtx.Ops)
		}
		e.caret.on = e.focused && (!blinking || dt%timePerBlink < timePerBlink/2)
	}

	return layout.Dimensions{Size: e.viewSize, Baseline: e.dims.Baseline}
}

// PaintSelection paints the contrasting background for selected text.
func (e *Editor) PaintSelection(gtx layout.Context) {
	cl := textPadding(e.lines)
	cl.Max = cl.Max.Add(e.viewSize)
	clip.Rect(cl).Add(gtx.Ops)
	for _, shape := range e.shapes {
		if !shape.selected {
			continue
		}
		stack := op.Save(gtx.Ops)
		offset := shape.offset
		offset.Y += shape.selectionYOffs
		op.Offset(layout.FPt(offset)).Add(gtx.Ops)
		clip.Rect(image.Rectangle{Max: shape.selectionSize}).Add(gtx.Ops)
		paint.PaintOp{}.Add(gtx.Ops)
		stack.Load()
	}
}

func (e *Editor) PaintText(gtx layout.Context) {
	cl := textPadding(e.lines)
	cl.Max = cl.Max.Add(e.viewSize)
	clip.Rect(cl).Add(gtx.Ops)
	for _, shape := range e.shapes {
		stack := op.Save(gtx.Ops)
		op.Offset(layout.FPt(shape.offset)).Add(gtx.Ops)
		shape.clip.Add(gtx.Ops)
		paint.PaintOp{}.Add(gtx.Ops)
		stack.Load()
	}
}

func (e *Editor) PaintCaret(gtx layout.Context) {
	if !e.caret.on {
		return
	}
	e.makeValid()
	// todo: this should also be scaled and like blink/sec, configured in theme
	carWidth := fixed.I(gtx.Px(unit.Dp(1)))
	carX := e.caret.start.x
	carY := e.caret.start.y

	defer op.Save(gtx.Ops).Load()
	carX -= carWidth / 2
	carAsc, carDesc := -e.lines[e.caret.start.lineCol.Y].Bounds.Min.Y,
		e.lines[e.caret.start.lineCol.Y].Bounds.Max.Y
	carRect := image.Rectangle{
		Min: image.Point{X: carX.Ceil(), Y: carY - carAsc.Ceil()},
		Max: image.Point{X: carX.Ceil() + carWidth.Ceil(), Y: carY + carDesc.Ceil()},
	}
	carRect = carRect.Add(image.Point{
		X: -e.scrollOff.X,
		Y: -e.scrollOff.Y,
	})
	cl := textPadding(e.lines)
	// Account for caret width to each side.
	whalf := (carWidth / 2).Ceil()
	if cl.Max.X < whalf {
		cl.Max.X = whalf
	}
	if cl.Min.X > -whalf {
		cl.Min.X = -whalf
	}
	cl.Max = cl.Max.Add(e.viewSize)
	carRect = cl.Intersect(carRect)
	if !carRect.Empty() {
		st := op.Save(gtx.Ops)
		clip.Rect(carRect).Add(gtx.Ops)
		paint.PaintOp{}.Add(gtx.Ops)
		st.Load()
	}
}

// Len is the length of the editor contents.
func (e *Editor) Len() int {
	return e.editBuffer.len()
}

// Text returns the contents of the editor.
func (e *Editor) Text() string {
	return e.editBuffer.String()
}

// SetText replaces the contents of the editor, clearing any selection first.
func (e *Editor) SetText(s string) *Editor {
	e.editBuffer = editBuffer{}
	e.caret.start = combinedPos{}
	e.caret.end = combinedPos{}
	e.prepend(s)
	return e
}

func (e *Editor) scrollBounds() image.Rectangle {
	var b image.Rectangle
	if e.singleLine {
		if len(e.lines) > 0 {
			b.Min.X = align(e.alignment, e.lines[0].Width, e.viewSize.X).Floor()
			if b.Min.X > 0 {
				b.Min.X = 0
			}
		}
		b.Max.X = e.dims.Size.X + b.Min.X - e.viewSize.X
	} else {
		b.Max.Y = e.dims.Size.Y - e.viewSize.Y
	}
	return b
}

func (e *Editor) scrollRel(dx, dy int) {
	e.scrollAbs(e.scrollOff.X+dx, e.scrollOff.Y+dy)
}

func (e *Editor) scrollAbs(x, y int) {
	e.scrollOff.X = x
	e.scrollOff.Y = y
	b := e.scrollBounds()
	if e.scrollOff.X > b.Max.X {
		e.scrollOff.X = b.Max.X
	}
	if e.scrollOff.X < b.Min.X {
		e.scrollOff.X = b.Min.X
	}
	if e.scrollOff.Y > b.Max.Y {
		e.scrollOff.Y = b.Max.Y
	}
	if e.scrollOff.Y < b.Min.Y {
		e.scrollOff.Y = b.Min.Y
	}
}

func (e *Editor) moveCoord(pos image.Point) {
	var (
		prevDesc fixed.Int26_6
		carLine  int
		y        int
	)
	for _, l := range e.lines {
		y += (prevDesc + l.Ascent).Ceil()
		prevDesc = l.Descent
		if y+prevDesc.Ceil() >= pos.Y+e.scrollOff.Y {
			break
		}
		carLine++
	}
	x := fixed.I(pos.X + e.scrollOff.X)
	e.caret.start = e.movePosToLine(e.caret.start, x, carLine)
	e.caret.start.xoff = 0
}

func (e *Editor) layoutText(s text.Shaper) ([]text.Line, layout.Dimensions) {
	e.editBuffer.Reset()
	var r io.Reader = &e.editBuffer
	if e.mask != 0 {
		e.maskReader.Reset(&e.editBuffer, e.mask)
		r = &e.maskReader
	}
	var lines []text.Line
	if s != nil {
		lines, _ = s.Layout(e.font, e.textSize, e.maxWidth, r)
	} else {
		lines, _ = nullLayout(r)
	}
	dims := linesDimens(lines)
	for i := 0; i < len(lines)-1; i++ {
		// To avoid l flickering while editing, assume a soft newline takes up all available space.
		if lay := lines[i].Layout; len(lay.Text) > 0 {
			r := lay.Text[len(lay.Text)-1]
			if r != '\n' {
				dims.Size.X = e.maxWidth
				break
			}
		}
	}
	return lines, dims
}

// CaretPos returns the line & column numbers of the caret.
func (e *Editor) CaretPos() (line, col int) {
	e.makeValid()
	return e.caret.start.lineCol.Y, e.caret.start.lineCol.X
}

// CaretCoords returns the coordinates of the caret, relative to the
// editor itself.
func (e *Editor) CaretCoords() f32.Point {
	e.makeValid()
	return f32.Pt(float32(e.caret.start.x)/64, float32(e.caret.start.y))
}

// offsetToScreenPos2 is a utility function to shortcut the common case of
// wanting the positions of exactly two offsets.
func (e *Editor) offsetToScreenPos2(o1, o2 int) (combinedPos, combinedPos) {
	cp1, iter := e.offsetToScreenPos(o1)
	return cp1, iter(o2)
}

// offsetToScreenPos takes an offset into the editor text (e.g.
// e.caret.end.ofs) and returns a combinedPos that corresponds to its current
// screen position, as well as an iterator that lets you get the combinedPos
// of a later offset. The offsets given to offsetToScreenPos and to the
// returned iterator must be sorted, lowest first, and they must be valid (0
// <= offset <= e.Len()).
//
// This function is written this way to take advantage of previous work done
// for offsets after the first. Otherwise you have to start from the top each
// time.
func (e *Editor) offsetToScreenPos(offset int) (combinedPos, func(int) combinedPos) {
	var col, line, idx int
	var x fixed.Int26_6

	l := e.lines[line]
	y := l.Ascent.Ceil()
	prevDesc := l.Descent

	iter := func(offset int) combinedPos {
	LOOP:
		for {
			for ; col < len(l.Layout.Advances); col++ {
				if idx >= offset {
					break LOOP
				}

				x += l.Layout.Advances[col]
				_, s := e.editBuffer.runeAt(idx)
				idx += s
			}
			if lastLine := line == len(e.lines)-1; lastLine || idx > offset {
				break LOOP
			}

			line++
			x = 0
			col = 0
			l = e.lines[line]
			y += (prevDesc + l.Ascent).Ceil()
			prevDesc = l.Descent
		}
		return combinedPos{
			lineCol: screenPos{Y: line, X: col},
			x:       x + align(e.alignment, e.lines[line].Width, e.viewSize.X),
			y:       y,
			ofs:     offset,
		}
	}
	return iter(offset), iter
}

func (e *Editor) invalidate() {
	e.valid = false
}

// Delete runes from the caret position. The sign of runes specifies the
// direction to delete: positive is forward, negative is backward.
//
// If there is a selection, it is deleted and counts as a single rune.
func (e *Editor) Delete(runes int) {
	if runes == 0 {
		return
	}

	if l := e.caret.end.ofs - e.caret.start.ofs; l != 0 {
		e.caret.start.ofs = e.editBuffer.deleteRunes(e.caret.start.ofs, l)
		runes -= sign(runes)
	}

	e.caret.start.ofs = e.editBuffer.deleteRunes(e.caret.start.ofs, runes)
	e.caret.start.xoff = 0
	e.ClearSelection()
	e.invalidate()
}

// Insert inserts text at the caret, moving the caret forward. If there is a
// selection, Insert overwrites it.
func (e *Editor) Insert(s string) {
	e.append(s)
	e.caret.scroll = true
}

func (e *Editor) append(s string) {
	e.prepend(s)
	e.caret.start.ofs += len(s)
	e.caret.end.ofs = e.caret.start.ofs
}

// prepend inserts s after the cursor; the caret does not change. If there is
// a selection, prepend overwrites it.
// xxx|yyy + prepend zzz => xxx|zzzyyy
func (e *Editor) prepend(s string) {
	if e.singleLine {
		s = strings.ReplaceAll(s, "\n", " ")
	}
	e.caret.start.ofs = e.editBuffer.deleteRunes(e.caret.start.ofs,
		e.caret.end.ofs-e.caret.start.ofs) // Delete any selection first.
	e.editBuffer.prepend(e.caret.start.ofs, s)
	e.caret.start.xoff = 0
	e.invalidate()
}

func (e *Editor) movePages(pages int, selAct selectionAction) {
	e.makeValid()
	y := e.caret.start.y + pages*e.viewSize.Y
	var (
		prevDesc fixed.Int26_6
		carLine2 int
	)
	y2 := e.lines[0].Ascent.Ceil()
	for i := 1; i < len(e.lines); i++ {
		if y2 >= y {
			break
		}
		l := e.lines[i]
		h := (prevDesc + l.Ascent).Ceil()
		prevDesc = l.Descent
		if y2+h-y >= y-y2 {
			break
		}
		y2 += h
		carLine2++
	}
	e.caret.start = e.movePosToLine(e.caret.start, e.caret.start.x+e.caret.start.xoff, carLine2)
	e.updateSelection(selAct)
}

func (e *Editor) movePosToLine(pos combinedPos, x fixed.Int26_6, line int) combinedPos {
	e.makeValid(&pos)
	if line < 0 {
		line = 0
	}
	if line >= len(e.lines) {
		line = len(e.lines) - 1
	}

	prevDesc := e.lines[line].Descent
	for pos.lineCol.Y < line {
		pos = e.movePosToEnd(pos)
		l := e.lines[pos.lineCol.Y]
		_, s := e.editBuffer.runeAt(pos.ofs)
		pos.ofs += s
		pos.y += (prevDesc + l.Ascent).Ceil()
		pos.lineCol.X = 0
		prevDesc = l.Descent
		pos.lineCol.Y++
	}
	for pos.lineCol.Y > line {
		pos = e.movePosToStart(pos)
		l := e.lines[pos.lineCol.Y]
		_, s := e.editBuffer.runeBefore(pos.ofs)
		pos.ofs -= s
		pos.y -= (prevDesc + l.Ascent).Ceil()
		prevDesc = l.Descent
		pos.lineCol.Y--
		l = e.lines[pos.lineCol.Y]
		pos.lineCol.X = len(l.Layout.Advances) - 1
	}

	pos = e.movePosToStart(pos)
	l := e.lines[line]
	pos.x = align(e.alignment, l.Width, e.viewSize.X)
	// Only move past the end of the last line
	end := 0
	if line < len(e.lines)-1 {
		end = 1
	}
	// Move to rune closest to x.
	for i := 0; i < len(l.Layout.Advances)-end; i++ {
		adv := l.Layout.Advances[i]
		if pos.x >= x {
			break
		}
		if pos.x+adv-x >= x-pos.x {
			break
		}
		pos.x += adv
		_, s := e.editBuffer.runeAt(pos.ofs)
		pos.ofs += s
		pos.lineCol.X++
	}
	pos.xoff = x - pos.x
	return pos
}

// MoveCaret moves the caret (aka selection start) and the selection end
// relative to their current positions. Positive distances moves forward,
// negative distances moves backward. Distances are in runes.
func (e *Editor) MoveCaret(startDelta, endDelta int) {
	e.makeValid()
	keepSame := e.caret.start.ofs == e.caret.end.ofs && startDelta == endDelta
	e.caret.start = e.movePos(e.caret.start, startDelta)
	e.caret.start.xoff = 0
	// If they were in the same place, and we're moving them the same distance,
	// just assign the new position, instead of recalculating it.
	if keepSame {
		e.caret.end = e.caret.start
	} else {
		e.caret.end = e.movePos(e.caret.end, endDelta)
		e.caret.end.xoff = 0
	}
}

func (e *Editor) movePos(pos combinedPos, distance int) combinedPos {
	for ; distance < 0 && pos.ofs > 0; distance++ {
		if pos.lineCol.X == 0 {
			// Move to end of previous line.
			pos = e.movePosToLine(pos, fixed.I(e.maxWidth), pos.lineCol.Y-1)
			continue
		}
		l := e.lines[pos.lineCol.Y].Layout
		_, s := e.editBuffer.runeBefore(pos.ofs)
		pos.ofs -= s
		pos.lineCol.X--
		pos.x -= l.Advances[pos.lineCol.X]
	}
	for ; distance > 0 && pos.ofs < e.editBuffer.len(); distance-- {
		l := e.lines[pos.lineCol.Y].Layout
		// Only move past the end of the last line
		end := 0
		if pos.lineCol.Y < len(e.lines)-1 {
			end = 1
		}
		if pos.lineCol.X >= len(l.Advances)-end {
			// Move to start of next line.
			pos = e.movePosToLine(pos, 0, pos.lineCol.Y+1)
			continue
		}
		pos.x += l.Advances[pos.lineCol.X]
		_, s := e.editBuffer.runeAt(pos.ofs)
		pos.ofs += s
		pos.lineCol.X++
	}
	return pos
}

func (e *Editor) moveStart(selAct selectionAction) {
	e.caret.start = e.movePosToStart(e.caret.start)
	e.updateSelection(selAct)
}

func (e *Editor) movePosToStart(pos combinedPos) combinedPos {
	e.makeValid(&pos)
	layout := e.lines[pos.lineCol.Y].Layout
	for i := pos.lineCol.X - 1; i >= 0; i-- {
		_, s := e.editBuffer.runeBefore(pos.ofs)
		pos.ofs -= s
		pos.x -= layout.Advances[i]
	}
	pos.lineCol.X = 0
	pos.xoff = -pos.x
	return pos
}

func (e *Editor) moveEnd(selAct selectionAction) {
	e.caret.start = e.movePosToEnd(e.caret.start)
	e.updateSelection(selAct)
}

func (e *Editor) movePosToEnd(pos combinedPos) combinedPos {
	e.makeValid(&pos)
	l := e.lines[pos.lineCol.Y]
	// Only move past the end of the last line
	end := 0
	if pos.lineCol.Y < len(e.lines)-1 {
		end = 1
	}
	layout := l.Layout
	for i := pos.lineCol.X; i < len(layout.Advances)-end; i++ {
		adv := layout.Advances[i]
		_, s := e.editBuffer.runeAt(pos.ofs)
		pos.ofs += s
		pos.x += adv
		pos.lineCol.X++
	}
	a := align(e.alignment, l.Width, e.viewSize.X)
	pos.xoff = l.Width + a - pos.x
	return pos
}

// moveWord moves the caret to the next word in the specified direction.
// Positive is forward, negative is backward.
// Absolute values greater than one will skip that many words.
func (e *Editor) moveWord(distance int, selAct selectionAction) {
	e.makeValid()
	// split the distance information into constituent parts to be
	// used independently.
	words, direction := distance, 1
	if distance < 0 {
		words, direction = distance*-1, -1
	}
	// atEnd if caret is at either side of the buffer.
	atEnd := func() bool {
		return e.caret.start.ofs == 0 || e.caret.start.ofs == e.editBuffer.len()
	}
	// next returns the appropriate rune given the direction.
	next := func() (r rune) {
		if direction < 0 {
			r, _ = e.editBuffer.runeBefore(e.caret.start.ofs)
		} else {
			r, _ = e.editBuffer.runeAt(e.caret.start.ofs)
		}
		return r
	}
	for ii := 0; ii < words; ii++ {
		for r := next(); unicode.IsSpace(r) && !atEnd(); r = next() {
			e.MoveCaret(direction, 0)
		}
		e.MoveCaret(direction, 0)
		for r := next(); !unicode.IsSpace(r) && !atEnd(); r = next() {
			e.MoveCaret(direction, 0)
		}
	}
	e.updateSelection(selAct)
}

// deleteWord the next word(s) in the specified direction. Unlike moveWord, deleteWord treats whitespace as a word
// itself.
//
// Positive is forward, negative is backward.
//
// Absolute values greater than one will delete that many words.
func (e *Editor) deleteWord(distance int) {
	if distance == 0 {
		return
	}

	e.makeValid()

	if e.caret.start.ofs != e.caret.end.ofs {
		e.Delete(1)
		distance -= sign(distance)
	}
	if distance == 0 {
		return
	}

	// split the distance information into constituent parts to be
	// used independently.
	words, direction := distance, 1
	if distance < 0 {
		words, direction = distance*-1, -1
	}
	// atEnd if offset is at or beyond either side of the buffer.
	atEnd := func(offset int) bool {
		idx := e.caret.start.ofs + offset*direction
		return idx <= 0 || idx >= e.editBuffer.len()
	}
	// next returns the appropriate rune given the direction and offset.
	next := func(offset int) (r rune) {
		idx := e.caret.start.ofs + offset*direction
		if idx < 0 {
			idx = 0
		} else if idx > e.editBuffer.len() {
			idx = e.editBuffer.len()
		}
		if direction < 0 {
			r, _ = e.editBuffer.runeBefore(idx)
		} else {
			r, _ = e.editBuffer.runeAt(idx)
		}
		return r
	}
	var runes = 1
	for ii := 0; ii < words; ii++ {
		if r := next(runes); unicode.IsSpace(r) {
			for r := next(runes); unicode.IsSpace(r) && !atEnd(runes); r = next(runes) {
				runes += 1
			}
		} else {
			for r := next(runes); !unicode.IsSpace(r) && !atEnd(runes); r = next(runes) {
				runes += 1
			}
		}
	}
	e.Delete(runes * direction)
}

func (e *Editor) scrollToCaret() {
	e.makeValid()
	l := e.lines[e.caret.start.lineCol.Y]
	if e.singleLine {
		var dist int
		if d := e.caret.start.x.Floor() - e.scrollOff.X; d < 0 {
			dist = d
		} else if d = e.caret.start.x.Ceil() - (e.scrollOff.X + e.viewSize.X); d > 0 {
			dist = d
		}
		e.scrollRel(dist, 0)
	} else {
		miny := e.caret.start.y - l.Ascent.Ceil()
		maxy := e.caret.start.y + l.Descent.Ceil()
		var dist int
		if d := miny - e.scrollOff.Y; d < 0 {
			dist = d
		} else if d = maxy - (e.scrollOff.Y + e.viewSize.Y); d > 0 {
			dist = d
		}
		e.scrollRel(0, dist)
	}
}

// NumLines returns the number of lines in the editor.
func (e *Editor) NumLines() int {
	e.makeValid()
	return len(e.lines)
}

// SelectionLen returns the length of the selection, in bytes; it is
// equivalent to len(e.SelectedText()).
func (e *Editor) SelectionLen() int {
	return abs(e.caret.start.ofs - e.caret.end.ofs)
}

// Selection returns the start and end of the selection, as offsets into the
// editor text. start can be > end.
func (e *Editor) Selection() (start, end int) {
	return e.caret.start.ofs, e.caret.end.ofs
}

// SetCaret moves the caret to start, and sets the selection end to end. start
// and end are in bytes, and represent offsets into the editor text. start and
// end must be at a rune boundary.
func (e *Editor) SetCaret(start, end int) {
	e.makeValid()
	// Constrain start and end to [0, e.Len()].
	l := e.Len()
	start = max(min(start, l), 0)
	end = max(min(end, l), 0)
	e.caret.start.ofs, e.caret.end.ofs = start, end
	e.makeValidCaret()
	e.caret.scroll = true
	e.scroller.Stop()
}

func (e *Editor) makeValidCaret(positions ...*combinedPos) {
	// Jump through some hoops to order the offsets given to offsetToScreenPos,
	// but still be able to update them correctly with the results thereof.
	positions = append(positions, &e.caret.start, &e.caret.end)
	sort.Slice(positions, func(i, j int) bool {
		return positions[i].ofs < positions[j].ofs
	})
	var iter func(offset int) combinedPos
	*positions[0], iter = e.offsetToScreenPos(positions[0].ofs)
	for _, cp := range positions[1:] {
		*cp = iter(cp.ofs)
	}
}

// SelectedText returns the currently selected text (if any) from the editor.
func (e *Editor) SelectedText() string {
	l := e.SelectionLen()
	if l == 0 {
		return ""
	}
	buf := make([]byte, l)
	e.editBuffer.Seek(int64(min(e.caret.start.ofs, e.caret.end.ofs)), io.SeekStart)
	_, err := e.editBuffer.Read(buf)
	if err != nil {
		// The only error that rr.Read can return is EOF, which just means no
		// selection, but we've already made sure that shouldn't happen.
		panic("impossible error because end is before e.rr.Len()")
	}
	return string(buf)
}

func (e *Editor) updateSelection(selAct selectionAction) {
	if selAct == selectionClear {
		e.ClearSelection()
	}
}

// ClearSelection clears the selection, by setting the selection end equal to
// the selection start.
func (e *Editor) ClearSelection() {
	e.caret.end = e.caret.start
}

func max(a, b int) int {
	if a > b {
		return a
	}
	return b
}

func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}

func abs(n int) int {
	if n < 0 {
		return -n
	}
	return n
}

func sign(n int) int {
	switch {
	case n < 0:
		return -1
	case n > 0:
		return 1
	default:
		return 0
	}
}

// sortPoints returns a and b sorted such that a2 <= b2.
func sortPoints(a, b screenPos) (a2, b2 screenPos) {
	if b.Less(a) {
		return b, a
	}
	return a, b
}

func nullLayout(r io.Reader) ([]text.Line, error) {
	rr := bufio.NewReader(r)
	var rerr error
	var n int
	var buf bytes.Buffer
	for {
		r, s, e := rr.ReadRune()
		n += s
		buf.WriteRune(r)
		if e != nil {
			rerr = e
			break
		}
	}
	return []text.Line{
		{
			Layout: text.Layout{
				Text:     buf.String(),
				Advances: make([]fixed.Int26_6, n),
			},
		},
	}, rerr
}

func (s ChangeEvent) isEditorEvent() {}
func (s SubmitEvent) isEditorEvent() {}
func (s SelectEvent) isEditorEvent() {}

// func (e *Editor) moveToLine(x fixed.Int26_6, line int) {
// 	e.makeValid()
// 	if line < 0 {
// 		line = 0
// 	}
// 	if line >= len(e.lines) {
// 		line = len(e.lines) - 1
// 	}
//
// 	prevDesc := e.lines[line].Descent
// 	for e.caret.Line < line {
// 		e.moveEnd()
// 		l := e.lines[e.caret.Line]
// 		_, s := e.editBuffer.runeAt(e.editBuffer.caret)
// 		e.editBuffer.caret += s
// 		e.caret.y += (prevDesc + l.Ascent).Ceil()
// 		e.caret.Col = 0
// 		prevDesc = l.Descent
// 		e.caret.Line++
// 	}
// 	for e.caret.Line > line {
// 		e.moveStart()
// 		l := e.lines[e.caret.Line]
// 		_, s := e.editBuffer.runeBefore(e.editBuffer.caret)
// 		e.editBuffer.caret -= s
// 		e.caret.y -= (prevDesc + l.Ascent).Ceil()
// 		prevDesc = l.Descent
// 		e.caret.Line--
// 		l = e.lines[e.caret.Line]
// 		e.caret.Col = len(l.Layout.Advances) - 1
// 	}
//
// 	e.moveStart()
// 	l := e.lines[line]
// 	e.caret.x = align(e.alignment, l.Width, e.viewSize.X)
// 	// Only move past the end of the last line
// 	end := 0
// 	if line < len(e.lines)-1 {
// 		end = 1
// 	}
// 	// Move to rune closest to x.
// 	for i := 0; i < len(l.Layout.Advances)-end; i++ {
// 		adv := l.Layout.Advances[i]
// 		if e.caret.x >= x {
// 			break
// 		}
// 		if e.caret.x+adv-x >= x-e.caret.x {
// 			break
// 		}
// 		e.caret.x += adv
// 		_, s := e.editBuffer.runeAt(e.editBuffer.caret)
// 		e.editBuffer.caret += s
// 		e.caret.Col++
// 	}
// 	e.caret.xoff = x - e.caret.x
// }
//
// // Move the caret: positive distance moves forward, negative distance moves
// // backward.
// func (e *Editor) Move(distance int) {
// 	e.makeValid()
// 	for ; distance < 0 && e.editBuffer.caret > 0; distance++ {
// 		if e.caret.Col == 0 {
// 			// Move to end of previous line.
// 			e.moveToLine(fixed.I(e.maxWidth), e.caret.Line-1)
// 			continue
// 		}
// 		l := e.lines[e.caret.Line].Layout
// 		_, s := e.editBuffer.runeBefore(e.editBuffer.caret)
// 		e.editBuffer.caret -= s
// 		e.caret.Col--
// 		e.caret.x -= l.Advances[e.caret.Col]
// 	}
// 	for ; distance > 0 && e.editBuffer.caret < e.editBuffer.len(); distance-- {
// 		l := e.lines[e.caret.Line].Layout
// 		// Only move past the end of the last line
// 		end := 0
// 		if e.caret.Line < len(e.lines)-1 {
// 			end = 1
// 		}
// 		if e.caret.Col >= len(l.Advances)-end {
// 			// Move to start of next line.
// 			e.moveToLine(0, e.caret.Line+1)
// 			continue
// 		}
// 		e.caret.x += l.Advances[e.caret.Col]
// 		_, s := e.editBuffer.runeAt(e.editBuffer.caret)
// 		e.editBuffer.caret += s
// 		e.caret.Col++
// 	}
// 	e.caret.xoff = 0
// }

func (e *Editor) layoutCaret() (line, col int, x fixed.Int26_6, y int) {
	var idx int
	var prevDesc fixed.Int26_6
loop:
	for {
		x = 0
		col = 0
		l := e.lines[line]
		y += (prevDesc + l.Ascent).Ceil()
		prevDesc = l.Descent
		for _, adv := range l.Layout.Advances {
			if idx == e.editBuffer.caret {
				break loop
			}
			x += adv
			_, s := e.editBuffer.runeAt(idx)
			idx += s
			col++
		}
		if line == len(e.lines)-1 || idx > e.editBuffer.caret {
			break
		}
		line++
	}
	x += align(e.alignment, e.lines[line].Width, e.viewSize.X)
	return
}

func (e *Editor) SingleLine() *Editor {
	e.singleLine = true
	return e
}

func (e *Editor) Submit(submit bool) *Editor {
	e.submit = submit
	return e
}

func (e *Editor) Mask(mask rune) *Editor {
	e.mask = mask
	return e
}

func (e *Editor) SetSubmit(submitFn func(txt string)) *Editor {
	e.submitHook = submitFn
	return e
}

func (e *Editor) SetChange(changeFn func(txt string)) *Editor {
	e.changeHook = changeFn
	return e
}

func (e *Editor) SetFocus(focusFn func(is bool)) *Editor {
	e.focusHook = focusFn
	return e
}

func (e *Editor) Alignment(alignment text.Alignment) *Editor {
	e.alignment = alignment
	return e
}