1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

package editor

import "strings"

// Inline markdown formatting: wrap (or unwrap) the selection in markers, keeping
// the markup visible in the buffer (TASK-009). With no selection a marker pair is
// inserted with the cursor between. Symmetric markers (bold/italic/code) toggle
// off when the selection is already wrapped — either with the markers inside the
// selection or sitting immediately outside it.

// WrapBold wraps the selection in ** (toggles off if already bold).
func (e *Editor) WrapBold() { e.toggleWrap("**") }

// WrapItalic wraps the selection in _ (toggles off if already italic).
func (e *Editor) WrapItalic() { e.toggleWrap("_") }

// WrapCode wraps the selection in ` (toggles off if already inline code).
func (e *Editor) WrapCode() { e.toggleWrap("`") }

func (e *Editor) toggleWrap(marker string) {
	if !e.HasSelection() {
		e.insertPair(marker, marker, len([]rune(marker)))
		return
	}
	start, end, _ := e.selRange()
	n := len([]rune(marker))
	inner := e.SelectedText()
	// Toggle off: markers inside the selection.
	if len([]rune(inner)) >= 2*n && strings.HasPrefix(inner, marker) && strings.HasSuffix(inner, marker) {
		stripped := string([]rune(inner)[n : len([]rune(inner))-n])
		e.replaceRange(start, end, stripped)
		newEnd := Position{Row: start.Row, Col: start.Col + len([]rune(stripped))}
		e.setSelection(start, newEnd)
		return
	}
	// Toggle off: markers immediately outside the selection (single line).
	if start.Row == end.Row && e.markersOutside(start, end, marker) {
		line := []rune(e.Lines[start.Row])
		newLine := string(line[:start.Col-n]) + string(line[start.Col:end.Col]) + string(line[end.Col+n:])
		e.Lines[start.Row] = newLine
		e.Dirty = true
		ns := Position{Row: start.Row, Col: start.Col - n}
		ne := Position{Row: end.Row, Col: end.Col - n}
		e.setSelection(ns, ne)
		return
	}
	// Wrap: insert marker after the selection, then before it.
	e.insertAt(end, marker)
	e.insertAt(start, marker)
	ns := Position{Row: start.Row, Col: start.Col + n}
	var ne Position
	if start.Row == end.Row {
		ne = Position{Row: end.Row, Col: end.Col + n}
	} else {
		ne = end
	}
	e.setSelection(ns, ne)
}

// WrapLink turns the selection into [sel](), cursor inside the parens; with no
// selection inserts [](), cursor inside the brackets. Toggles a full
// [text](url) selection back to its label text.
func (e *Editor) WrapLink() {
	if !e.HasSelection() {
		e.insertPair("[", "]()", 1) // cursor between [ and ]
		return
	}
	start, end, _ := e.selRange()
	inner := e.SelectedText()
	// Toggle off: selection is a complete [label](url) — unwrap to the label.
	if start.Row == end.Row && strings.HasPrefix(inner, "[") && strings.HasSuffix(inner, ")") {
		if i := strings.Index(inner, "]("); i >= 0 {
			label := inner[1:i]
			e.replaceRange(start, end, label)
			e.setSelection(start, Position{Row: start.Row, Col: start.Col + len([]rune(label))})
			return
		}
	}
	e.insertAt(end, "]()")
	e.insertAt(start, "[")
	// Park the cursor inside the parens, just before the closing ).
	if start.Row == end.Row {
		e.Cursor = Position{Row: end.Row, Col: end.Col + 3} // +"[" +"]("
	} else {
		e.Cursor = Position{Row: end.Row, Col: end.Col + 2}
	}
	e.anchor = nil
	e.setGoal()
	e.followCursor()
}

// insertPair inserts left+right at the cursor and parks the cursor `offset`
// runes past the original position (i.e. between the two halves).
func (e *Editor) insertPair(left, right string, offset int) {
	pos := e.Cursor
	line := []rune(e.Lines[pos.Row])
	newLine := string(line[:pos.Col]) + left + right + string(line[pos.Col:])
	e.Lines[pos.Row] = newLine
	e.Dirty = true
	e.Cursor = Position{Row: pos.Row, Col: pos.Col + offset}
	e.anchor = nil
	e.setGoal()
	e.followCursor()
}

// insertAt inserts s at position p (single-line column splice).
func (e *Editor) insertAt(p Position, s string) {
	line := []rune(e.Lines[p.Row])
	e.Lines[p.Row] = string(line[:p.Col]) + s + string(line[p.Col:])
	e.Dirty = true
}

// markersOutside reports whether marker sits immediately before start and after
// end on the same line.
func (e *Editor) markersOutside(start, end Position, marker string) bool {
	n := len([]rune(marker))
	line := []rune(e.Lines[start.Row])
	if start.Col < n || end.Col+n > len(line) {
		return false
	}
	return string(line[start.Col-n:start.Col]) == marker && string(line[end.Col:end.Col+n]) == marker
}

// replaceRange replaces the text in [start,end) (same row) with s.
func (e *Editor) replaceRange(start, end Position, s string) {
	line := []rune(e.Lines[start.Row])
	e.Lines[start.Row] = string(line[:start.Col]) + s + string(line[end.Col:])
	e.Dirty = true
}

// setSelection anchors at a and parks the cursor at b.
func (e *Editor) setSelection(a, b Position) {
	c := a
	e.anchor = &c
	e.Cursor = b
	e.setGoal()
	e.followCursor()
}