]> git.lizzy.rs Git - micro.git/blobdiff - cmd/micro/cursor.go
Start refactor
[micro.git] / cmd / micro / cursor.go
index 4d2bd1f62b8b3fd37891d08e6e7365df0ddea02d..9e9f1d8192447f73979f52d870364a010eac196e 100644 (file)
 package main
 
 import (
-       "strings"
-)
-
-// FromCharPos converts from a character position to an x, y position
-func FromCharPos(loc int, buf *Buffer) (int, int) {
-       return FromCharPosStart(0, 0, 0, loc, buf)
-}
+       "unicode/utf8"
 
-// FromCharPosStart converts from a character position to an x, y position, starting at the specified character location
-func FromCharPosStart(startLoc, startX, startY, loc int, buf *Buffer) (int, int) {
-       charNum := startLoc
-       x, y := startX, startY
+       runewidth "github.com/mattn/go-runewidth"
+       "github.com/zyedidia/clipboard"
+)
 
-       lineLen := Count(buf.lines[y]) + 1
-       for charNum+lineLen <= loc {
-               charNum += lineLen
-               y++
-               lineLen = Count(buf.lines[y]) + 1
+// InBounds returns whether the given location is a valid character position in the given buffer
+func InBounds(pos Loc, buf *Buffer) bool {
+       if pos.Y < 0 || pos.Y >= len(buf.lines) || pos.X < 0 || pos.X > utf8.RuneCount(buf.LineBytes(pos.Y)) {
+               return false
        }
-       x = loc - charNum
 
-       return x, y
-}
-
-// ToCharPos converts from an x, y position to a character position
-func ToCharPos(x, y int, buf *Buffer) int {
-       loc := 0
-       for i := 0; i < y; i++ {
-               // + 1 for the newline
-               loc += Count(buf.lines[i]) + 1
-       }
-       loc += x
-       return loc
+       return true
 }
 
-// The Cursor struct stores the location of the cursor in the view
-// The complicated part about the cursor is storing its location.
-// The cursor must be displayed at an x, y location, but since the buffer
-// uses a rope to store text, to insert text we must have an index. It
-// is also simpler to use character indicies for other tasks such as
-// selection.
+// The Cursor struct stores the location of the cursor in the buffer
+// as well as the selection
 type Cursor struct {
-       v *View
-
-       // The cursor display location
-       x int
-       y int
+       buf *Buffer
+       Loc
 
        // Last cursor x position
-       lastVisualX int
+       LastVisualX int
 
        // The current selection as a range of character numbers (inclusive)
-       curSelection [2]int
+       CurSelection [2]Loc
        // The original selection as a range of character numbers
        // This is used for line and word selection where it is necessary
        // to know what the original selection was
-       origSelection [2]int
+       OrigSelection [2]Loc
+
+       // Which cursor index is this (for multiple cursors)
+       Num int
+}
+
+// Goto puts the cursor at the given cursor's location and gives
+// the current cursor its selection too
+func (c *Cursor) Goto(b Cursor) {
+       c.X, c.Y, c.LastVisualX = b.X, b.Y, b.LastVisualX
+       c.OrigSelection, c.CurSelection = b.OrigSelection, b.CurSelection
+}
+
+// GotoLoc puts the cursor at the given cursor's location and gives
+// the current cursor its selection too
+func (c *Cursor) GotoLoc(l Loc) {
+       c.X, c.Y = l.X, l.Y
+       c.LastVisualX = c.GetVisualX()
+}
+
+// GetVisualX returns the x value of the cursor in visual spaces
+func (c *Cursor) GetVisualX() int {
+       if c.X <= 0 {
+               c.X = 0
+               return 0
+       }
+
+       bytes := c.buf.LineBytes(c.Y)
+       tabsize := int(c.buf.Settings["tabsize"].(float64))
+       if c.X > utf8.RuneCount(bytes) {
+               c.X = utf8.RuneCount(bytes) - 1
+       }
+
+       return StringWidth(bytes, c.X, tabsize)
+}
+
+// GetCharPosInLine gets the char position of a visual x y
+// coordinate (this is necessary because tabs are 1 char but
+// 4 visual spaces)
+func (c *Cursor) GetCharPosInLine(b []byte, visualPos int) int {
+       tabsize := int(c.buf.Settings["tabsize"].(float64))
+
+       // Scan rune by rune until we exceed the visual width that we are
+       // looking for. Then we can return the character position we have found
+       i := 0     // char pos
+       width := 0 // string visual width
+       for len(b) > 0 {
+               r, size := utf8.DecodeRune(b)
+               b = b[size:]
+
+               switch r {
+               case '\t':
+                       ts := tabsize - (width % tabsize)
+                       width += ts
+               default:
+                       width += runewidth.RuneWidth(r)
+               }
+
+               i++
+
+               if width >= visualPos {
+                       break
+               }
+       }
+
+       return i
 }
 
-// SetLoc sets the location of the cursor in terms of character number
-// and not x, y location
-// It's just a simple wrapper of FromCharPos
-func (c *Cursor) SetLoc(loc int) {
-       c.x, c.y = FromCharPos(loc, c.v.buf)
-       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()
 }
 
-// Loc gets the cursor location in terms of character number instead
-// of x, y location
-// It's just a simple wrapper of ToCharPos
-func (c *Cursor) Loc() int {
-       return ToCharPos(c.x, c.y, c.v.buf)
+// End moves the cursor to the end of the line it is on
+func (c *Cursor) End() {
+       c.X = utf8.RuneCount(c.buf.LineBytes(c.Y))
+       c.LastVisualX = c.GetVisualX()
+}
+
+// CopySelection copies the user's selection to either "primary"
+// or "clipboard"
+func (c *Cursor) CopySelection(target string) {
+       if c.HasSelection() {
+               if target != "primary" || c.buf.Settings["useprimary"].(bool) {
+                       clipboard.WriteAll(string(c.GetSelection()), target)
+               }
+       }
 }
 
 // ResetSelection resets the user's selection
 func (c *Cursor) ResetSelection() {
-       c.curSelection[0] = 0
-       c.curSelection[1] = 0
+       c.CurSelection[0] = c.buf.Start()
+       c.CurSelection[1] = c.buf.Start()
+}
+
+// SetSelectionStart sets the start of the selection
+func (c *Cursor) SetSelectionStart(pos Loc) {
+       c.CurSelection[0] = pos
+}
+
+// SetSelectionEnd sets the end of the selection
+func (c *Cursor) SetSelectionEnd(pos Loc) {
+       c.CurSelection[1] = pos
 }
 
 // HasSelection returns whether or not the user has selected anything
 func (c *Cursor) HasSelection() bool {
-       return c.curSelection[0] != c.curSelection[1]
+       return c.CurSelection[0] != c.CurSelection[1]
 }
 
 // DeleteSelection deletes the currently selected text
 func (c *Cursor) DeleteSelection() {
-       if c.curSelection[0] > c.curSelection[1] {
-               c.v.eh.Remove(c.curSelection[1], c.curSelection[0])
-               c.SetLoc(c.curSelection[1])
-       } else if c.GetSelection() == "" {
+       if c.CurSelection[0].GreaterThan(c.CurSelection[1]) {
+               c.buf.Remove(c.CurSelection[1], c.CurSelection[0])
+               c.Loc = c.CurSelection[1]
+       } else if !c.HasSelection() {
                return
        } else {
-               c.v.eh.Remove(c.curSelection[0], c.curSelection[1])
-               c.SetLoc(c.curSelection[0])
+               c.buf.Remove(c.CurSelection[0], c.CurSelection[1])
+               c.Loc = c.CurSelection[0]
        }
 }
 
 // GetSelection returns the cursor's selection
-func (c *Cursor) GetSelection() string {
-       if c.curSelection[0] > c.curSelection[1] {
-               return string([]rune(c.v.buf.text)[c.curSelection[1]:c.curSelection[0]])
+func (c *Cursor) GetSelection() []byte {
+       if InBounds(c.CurSelection[0], c.buf) && InBounds(c.CurSelection[1], c.buf) {
+               if c.CurSelection[0].GreaterThan(c.CurSelection[1]) {
+                       return c.buf.Substr(c.CurSelection[1], c.CurSelection[0])
+               }
+               return c.buf.Substr(c.CurSelection[0], c.CurSelection[1])
        }
-       return string([]rune(c.v.buf.text)[c.curSelection[0]:c.curSelection[1]])
+       return []byte{}
 }
 
 // SelectLine selects the current line
 func (c *Cursor) SelectLine() {
        c.Start()
-       c.curSelection[0] = c.Loc()
+       c.SetSelectionStart(c.Loc)
        c.End()
-       if len(c.v.buf.lines)-1 > c.y {
-               c.curSelection[1] = c.Loc() + 1
+       if len(c.buf.lines)-1 > c.Y {
+               c.SetSelectionEnd(c.Loc.Move(1, c.buf))
        } else {
-               c.curSelection[1] = c.Loc()
+               c.SetSelectionEnd(c.Loc)
        }
 
-       c.origSelection = c.curSelection
+       c.OrigSelection = c.CurSelection
 }
 
 // AddLineToSelection adds the current line to the selection
 func (c *Cursor) AddLineToSelection() {
-       loc := c.Loc()
-
-       if loc < c.origSelection[0] {
+       if c.Loc.LessThan(c.OrigSelection[0]) {
                c.Start()
-               c.curSelection[0] = c.Loc()
-               c.curSelection[1] = c.origSelection[1]
+               c.SetSelectionStart(c.Loc)
+               c.SetSelectionEnd(c.OrigSelection[1])
        }
-       if loc > c.origSelection[1] {
+       if c.Loc.GreaterThan(c.OrigSelection[1]) {
                c.End()
-               c.curSelection[1] = c.Loc()
-               c.curSelection[0] = c.origSelection[0]
+               c.SetSelectionEnd(c.Loc.Move(1, c.buf))
+               c.SetSelectionStart(c.OrigSelection[0])
        }
 
-       if loc < c.origSelection[1] && loc > c.origSelection[0] {
-               c.curSelection = c.origSelection
+       if c.Loc.LessThan(c.OrigSelection[1]) && c.Loc.GreaterThan(c.OrigSelection[0]) {
+               c.CurSelection = c.OrigSelection
        }
 }
 
-// SelectWord selects the word the cursor is currently on
-func (c *Cursor) SelectWord() {
-       if len(c.v.buf.lines[c.y]) == 0 {
-               return
+// UpN moves the cursor up N lines (if possible)
+func (c *Cursor) UpN(amount int) {
+       proposedY := c.Y - amount
+       if proposedY < 0 {
+               proposedY = 0
+       } else if proposedY >= len(c.buf.lines) {
+               proposedY = len(c.buf.lines) - 1
        }
 
-       if !IsWordChar(string(c.RuneUnder(c.x))) {
-               loc := c.Loc()
-               c.curSelection[0] = loc
-               c.curSelection[1] = loc + 1
-               c.origSelection = c.curSelection
-               return
-       }
+       bytes := c.buf.LineBytes(proposedY)
+       c.X = c.GetCharPosInLine(bytes, c.LastVisualX)
 
-       forward, backward := c.x, c.x
-
-       for backward > 0 && IsWordChar(string(c.RuneUnder(backward-1))) {
-               backward--
-       }
-
-       c.curSelection[0] = ToCharPos(backward, c.y, c.v.buf)
-       c.origSelection[0] = c.curSelection[0]
-
-       for forward < Count(c.v.buf.lines[c.y])-1 && IsWordChar(string(c.RuneUnder(forward+1))) {
-               forward++
+       if c.X > utf8.RuneCount(bytes) || (amount < 0 && proposedY == c.Y) {
+               c.X = utf8.RuneCount(bytes)
        }
 
-       c.curSelection[1] = ToCharPos(forward, c.y, c.v.buf) + 1
-       c.origSelection[1] = c.curSelection[1]
+       c.Y = proposedY
 }
 
-// AddWordToSelection adds the word the cursor is currently on to the selection
-func (c *Cursor) AddWordToSelection() {
-       loc := c.Loc()
-
-       if loc > c.origSelection[0] && loc < c.origSelection[1] {
-               c.curSelection = c.origSelection
-               return
-       }
-
-       if loc < c.origSelection[0] {
-               backward := c.x
-
-               for backward > 0 && IsWordChar(string(c.RuneUnder(backward-1))) {
-                       backward--
-               }
-
-               c.curSelection[0] = ToCharPos(backward, c.y, c.v.buf)
-               c.curSelection[1] = c.origSelection[1]
-       }
-
-       if loc > c.origSelection[1] {
-               forward := c.x
-
-               for forward < Count(c.v.buf.lines[c.y])-1 && IsWordChar(string(c.RuneUnder(forward+1))) {
-                       forward++
-               }
-
-               c.curSelection[1] = ToCharPos(forward, c.y, c.v.buf) + 1
-               c.curSelection[0] = c.origSelection[0]
-       }
-}
-
-func (c *Cursor) SelectTo(loc int) {
-       if loc > c.origSelection[0] {
-               c.curSelection[0] = c.origSelection[0]
-               c.curSelection[1] = loc + 1
-       } else {
-               c.curSelection[0] = loc
-               c.curSelection[1] = c.origSelection[0] + 1
-       }
-}
-
-func (c *Cursor) WordRight() {
-       for IsWordChar(string(c.RuneUnder(c.x))) {
-               c.Right()
-       }
-       c.Right()
-}
-
-func (c *Cursor) WordLeft() {
-       for IsWordChar(string(c.RuneUnder(c.x))) {
-               c.Left()
-       }
-       c.Left()
-}
-
-// RuneUnder returns the rune under the given x position
-func (c *Cursor) RuneUnder(x int) rune {
-       line := []rune(c.v.buf.lines[c.y])
-       if len(line) == 0 {
-               return '\n'
-       }
-       if x >= len(line) {
-               x = len(line) - 1
-       } else if x < 0 {
-               x = 0
-       }
-       return line[x]
+// DownN moves the cursor down N lines (if possible)
+func (c *Cursor) DownN(amount int) {
+       c.UpN(-amount)
 }
 
 // Up moves the cursor up one line (if possible)
 func (c *Cursor) Up() {
-       if c.y > 0 {
-               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)
-               }
-       }
+       c.UpN(1)
 }
 
 // Down moves the cursor down one line (if possible)
 func (c *Cursor) Down() {
-       if c.y < len(c.v.buf.lines)-1 {
-               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)
-               }
-       }
+       c.DownN(1)
 }
 
-// Left moves the cursor left one cell (if possible) or to the last line if it is at the beginning
+// Left moves the cursor left one cell (if possible) or to
+// the previous line if it is at the beginning
 func (c *Cursor) Left() {
-       if c.Loc() == 0 {
+       if c.Loc == c.buf.Start() {
                return
        }
-       if c.x > 0 {
-               c.x--
+       if c.X > 0 {
+               c.X--
        } else {
                c.Up()
                c.End()
        }
-       c.lastVisualX = c.GetVisualX()
+       c.LastVisualX = c.GetVisualX()
 }
 
-// Right moves the cursor right one cell (if possible) or to the next line if it is at the end
+// Right moves the cursor right one cell (if possible) or
+// to the next line if it is at the end
 func (c *Cursor) Right() {
-       if c.Loc() == c.v.buf.Len() {
+       if c.Loc == c.buf.End() {
                return
        }
-       if c.x < Count(c.v.buf.lines[c.y]) {
-               c.x++
+       if c.X < utf8.RuneCount(c.buf.LineBytes(c.Y)) {
+               c.X++
        } else {
                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()
+       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)
-func (c *Cursor) GetCharPosInLine(lineNum, visualPos int) int {
-       // Get the tab size
-       tabSize := settings.TabSize
-       // This is the visual line -- every \t replaced with the correct number of spaces
-       visualLine := strings.Replace(c.v.buf.lines[lineNum], "\t", "\t"+Spaces(tabSize-1), -1)
-       if visualPos > Count(visualLine) {
-               visualPos = Count(visualLine)
-       }
-       numTabs := NumOccurences(visualLine[:visualPos], '\t')
-       if visualPos >= (tabSize-1)*numTabs {
-               return visualPos - (tabSize-1)*numTabs
-       }
-       return visualPos / tabSize
-}
-
-// GetVisualX returns the x value of the cursor in visual spaces
-func (c *Cursor) GetVisualX() int {
-       runes := []rune(c.v.buf.lines[c.y])
-       tabSize := settings.TabSize
-       return c.x + NumOccurences(string(runes[:c.x]), '\t')*(tabSize-1)
-}
-
-// Relocate makes sure that the cursor is inside the bounds of the buffer
-// If it isn't, it moves it to be within the buffer's lines
+// Relocate makes sure that the cursor is inside the bounds
+// of the buffer If it isn't, it moves it to be within the
+// buffer's lines
 func (c *Cursor) Relocate() {
-       if c.y < 0 {
-               c.y = 0
-       } else if c.y >= len(c.v.buf.lines) {
-               c.y = len(c.v.buf.lines) - 1
-       }
-
-       if c.x < 0 {
-               c.x = 0
-       } else if c.x > Count(c.v.buf.lines[c.y]) {
-               c.x = Count(c.v.buf.lines[c.y])
+       if c.Y < 0 {
+               c.Y = 0
+       } else if c.Y >= len(c.buf.lines) {
+               c.Y = len(c.buf.lines) - 1
        }
-}
 
-// Display draws the cursor to the screen at the correct position
-func (c *Cursor) Display() {
-       // Don't draw the cursor if it is out of the viewport or if it has a selection
-       if (c.y-c.v.topline < 0 || c.y-c.v.topline > c.v.height-1) || c.HasSelection() {
-               screen.HideCursor()
-       } else {
-               screen.ShowCursor(c.GetVisualX()+c.v.lineNumOffset-c.v.leftCol, c.y-c.v.topline)
+       if c.X < 0 {
+               c.X = 0
+       } else if c.X > utf8.RuneCount(c.buf.LineBytes(c.Y)) {
+               c.X = utf8.RuneCount(c.buf.LineBytes(c.Y))
        }
 }