import (
"time"
+
+ dmp "github.com/sergi/go-diff/diffmatchpatch"
)
const (
// TextEvent holds data for a manipulation on some text that can be undone
type TextEvent struct {
- c Cursor
+ C Cursor
- eventType int
- text string
- start int
- end int
- time time.Time
+ EventType int
+ Text string
+ Start Loc
+ End Loc
+ Time time.Time
}
// ExecuteTextEvent runs a text event
func ExecuteTextEvent(t *TextEvent, buf *Buffer) {
- if t.eventType == TextEventInsert {
- buf.Insert(t.start, t.text)
- } else if t.eventType == TextEventRemove {
- t.text = buf.Remove(t.start, t.end)
+ if t.EventType == TextEventInsert {
+ buf.insert(t.Start, []byte(t.Text))
+ } else if t.EventType == TextEventRemove {
+ t.Text = buf.remove(t.Start, t.End)
}
}
// UndoTextEvent undoes a text event
func UndoTextEvent(t *TextEvent, buf *Buffer) {
- t.eventType = -t.eventType
+ t.EventType = -t.EventType
ExecuteTextEvent(t, buf)
}
// EventHandler executes text manipulations and allows undoing and redoing
type EventHandler struct {
- v *View
- undo *Stack
- redo *Stack
+ buf *Buffer
+ UndoStack *Stack
+ RedoStack *Stack
}
// NewEventHandler returns a new EventHandler
-func NewEventHandler(v *View) *EventHandler {
+func NewEventHandler(buf *Buffer) *EventHandler {
eh := new(EventHandler)
- eh.undo = new(Stack)
- eh.redo = new(Stack)
- eh.v = v
+ eh.UndoStack = new(Stack)
+ eh.RedoStack = new(Stack)
+ eh.buf = buf
return eh
}
+// ApplyDiff takes a string and runs the necessary insertion and deletion events to make
+// the buffer equal to that string
+// This means that we can transform the buffer into any string and still preserve undo/redo
+// through insert and delete events
+func (eh *EventHandler) ApplyDiff(new string) {
+ differ := dmp.New()
+ diff := differ.DiffMain(eh.buf.String(), new, false)
+ loc := eh.buf.Start()
+ for _, d := range diff {
+ if d.Type == dmp.DiffDelete {
+ eh.Remove(loc, loc.Move(Count(d.Text), eh.buf))
+ } else {
+ if d.Type == dmp.DiffInsert {
+ eh.Insert(loc, d.Text)
+ }
+ loc = loc.Move(Count(d.Text), eh.buf)
+ }
+ }
+}
+
// Insert creates an insert text event and executes it
-func (eh *EventHandler) Insert(start int, text string) {
+func (eh *EventHandler) Insert(start Loc, text string) {
e := &TextEvent{
- c: eh.v.Cursor,
- eventType: TextEventInsert,
- text: text,
- start: start,
- end: start + Count(text),
- time: time.Now(),
+ C: eh.buf.Cursor,
+ EventType: TextEventInsert,
+ Text: text,
+ Start: start,
+ Time: time.Now(),
}
eh.Execute(e)
+ e.End = start.Move(Count(text), eh.buf)
}
// Remove creates a remove text event and executes it
-func (eh *EventHandler) Remove(start, end int) {
+func (eh *EventHandler) Remove(start, end Loc) {
e := &TextEvent{
- c: eh.v.Cursor,
- eventType: TextEventRemove,
- start: start,
- end: end,
- time: time.Now(),
+ C: eh.buf.Cursor,
+ EventType: TextEventRemove,
+ Start: start,
+ End: end,
+ Time: time.Now(),
}
eh.Execute(e)
}
// Replace deletes from start to end and replaces it with the given string
-func (eh *EventHandler) Replace(start, end int, replace string) {
+func (eh *EventHandler) Replace(start, end Loc, replace string) {
eh.Remove(start, end)
eh.Insert(start, replace)
}
// Execute a textevent and add it to the undo stack
func (eh *EventHandler) Execute(t *TextEvent) {
- if eh.redo.Len() > 0 {
- eh.redo = new(Stack)
+ if eh.RedoStack.Len() > 0 {
+ eh.RedoStack = new(Stack)
}
- eh.undo.Push(t)
- ExecuteTextEvent(t, eh.v.Buf)
+ eh.UndoStack.Push(t)
+ ExecuteTextEvent(t, eh.buf)
}
// Undo the first event in the undo stack
func (eh *EventHandler) Undo() {
- t := eh.undo.Peek()
+ t := eh.UndoStack.Peek()
if t == nil {
return
}
- te := t.(*TextEvent)
- startTime := t.(*TextEvent).time.UnixNano() / int64(time.Millisecond)
+ startTime := t.Time.UnixNano() / int64(time.Millisecond)
eh.UndoOneEvent()
for {
- t = eh.undo.Peek()
+ t = eh.UndoStack.Peek()
if t == nil {
return
}
- te = t.(*TextEvent)
-
- if startTime-(te.time.UnixNano()/int64(time.Millisecond)) > undoThreshold {
+ if startTime-(t.Time.UnixNano()/int64(time.Millisecond)) > undoThreshold {
return
}
+ startTime = t.Time.UnixNano() / int64(time.Millisecond)
eh.UndoOneEvent()
}
func (eh *EventHandler) UndoOneEvent() {
// This event should be undone
// Pop it off the stack
- t := eh.undo.Pop()
+ t := eh.UndoStack.Pop()
if t == nil {
return
}
- te := t.(*TextEvent)
// Undo it
// Modifies the text event
- UndoTextEvent(te, eh.v.Buf)
+ UndoTextEvent(t, eh.buf)
// Set the cursor in the right place
- teCursor := te.c
- te.c = eh.v.Cursor
- eh.v.Cursor = teCursor
+ teCursor := t.C
+ t.C = eh.buf.Cursor
+ eh.buf.Cursor.Goto(teCursor)
// Push it to the redo stack
- eh.redo.Push(te)
+ eh.RedoStack.Push(t)
}
// Redo the first event in the redo stack
func (eh *EventHandler) Redo() {
- t := eh.redo.Peek()
+ t := eh.RedoStack.Peek()
if t == nil {
return
}
- te := t.(*TextEvent)
- startTime := t.(*TextEvent).time.UnixNano() / int64(time.Millisecond)
+ startTime := t.Time.UnixNano() / int64(time.Millisecond)
eh.RedoOneEvent()
for {
- t = eh.redo.Peek()
+ t = eh.RedoStack.Peek()
if t == nil {
return
}
- te = t.(*TextEvent)
-
- if (te.time.UnixNano()/int64(time.Millisecond))-startTime > undoThreshold {
+ if (t.Time.UnixNano()/int64(time.Millisecond))-startTime > undoThreshold {
return
}
// RedoOneEvent redoes one event
func (eh *EventHandler) RedoOneEvent() {
- t := eh.redo.Pop()
+ t := eh.RedoStack.Pop()
if t == nil {
return
}
- te := t.(*TextEvent)
// Modifies the text event
- UndoTextEvent(te, eh.v.Buf)
+ UndoTextEvent(t, eh.buf)
- teCursor := te.c
- te.c = eh.v.Cursor
- eh.v.Cursor = teCursor
+ teCursor := t.C
+ t.C = eh.buf.Cursor
+ eh.buf.Cursor.Goto(teCursor)
- eh.undo.Push(te)
+ eh.UndoStack.Push(t)
}