From a3fe21a5e6fa9386cdf3272f9b01c6454d448fa6 Mon Sep 17 00:00:00 2001
From: Zachary Yedidia <zyedidia@gmail.com>
Date: Mon, 4 Apr 2016 14:09:24 -0400
Subject: [PATCH] Implement lastX for cursor movement

---
 src/cursor.go | 9 +++++++++
 src/view.go   | 1 +
 2 files changed, 10 insertions(+)

diff --git a/src/cursor.go b/src/cursor.go
index f54ae6bf..ffbedf0a 100644
--- a/src/cursor.go
+++ b/src/cursor.go
@@ -42,6 +42,9 @@ type Cursor struct {
 	x int
 	y int
 
+	// Last cursor x position
+	lastVisualX int
+
 	// The current selection as a range of character numbers (inclusive)
 	curSelection [2]int
 	// The original selection as a range of character numbers
@@ -200,6 +203,7 @@ func (c *Cursor) Up() {
 		c.y--
 
 		runes := []rune(c.v.buf.lines[c.y])
+		c.x = c.GetCharPosInLine(c.y, c.lastVisualX)
 		if c.x > len(runes) {
 			c.x = len(runes)
 		}
@@ -212,6 +216,7 @@ func (c *Cursor) Down() {
 		c.y++
 
 		runes := []rune(c.v.buf.lines[c.y])
+		c.x = c.GetCharPosInLine(c.y, c.lastVisualX)
 		if c.x > len(runes) {
 			c.x = len(runes)
 		}
@@ -229,6 +234,7 @@ func (c *Cursor) Left() {
 		c.Up()
 		c.End()
 	}
+	c.lastVisualX = c.GetVisualX()
 }
 
 // Right moves the cursor right one cell (if possible) or to the next line if it is at the end
@@ -242,16 +248,19 @@ func (c *Cursor) Right() {
 		c.Down()
 		c.Start()
 	}
+	c.lastVisualX = c.GetVisualX()
 }
 
 // End moves the cursor to the end of the line it is on
 func (c *Cursor) End() {
 	c.x = Count(c.v.buf.lines[c.y])
+	c.lastVisualX = c.GetVisualX()
 }
 
 // Start moves the cursor to the start of the line it is on
 func (c *Cursor) Start() {
 	c.x = 0
+	c.lastVisualX = c.GetVisualX()
 }
 
 // GetCharPosInLine gets the char position of a visual x y coordinate (this is necessary because tabs are 1 char but 4 visual spaces)
diff --git a/src/view.go b/src/view.go
index 2fce237f..beef181f 100644
--- a/src/view.go
+++ b/src/view.go
@@ -330,6 +330,7 @@ func (v *View) MoveToMouseClick(x, y int) {
 	}
 	v.cursor.x = x
 	v.cursor.y = y
+	v.cursor.lastVisualX = v.cursor.GetVisualX()
 }
 
 // HandleEvent handles an event passed by the main loop
-- 
2.44.0