Pasting multi cursor clipboard with newlines for non matching cursors

Author: Neko-Box-Coder

internal/action/actions.go

@@ -1583,23 +1583,31 @@ func (h *BufPane) MoveLinesDown() bool {
 // Paste whatever is in the system clipboard into the buffer
 // Delete and paste if the user has a selection
 func (h *BufPane) Paste() bool {
-	clip, err := clipboard.ReadMulti(clipboard.ClipboardReg, h.Cursor.Num, h.Buf.NumCursors())
-	if err != nil {
-		InfoBar.Error(err)
-	} else {
-		h.paste(clip)
-	}
-	h.Relocate()
-	return true
+	return h.pasteReg(clipboard.ClipboardReg)
 }
 
 // PastePrimary pastes from the primary clipboard (only use on linux)
 func (h *BufPane) PastePrimary() bool {
-	clip, err := clipboard.ReadMulti(clipboard.PrimaryReg, h.Cursor.Num, h.Buf.NumCursors())
+	return h.pasteReg(clipboard.PrimaryReg)
+}
+
+func (h *BufPane) pasteReg(reg clipboard.Register) bool {
+	clips, err := clipboard.ReadMulti(reg)
 	if err != nil {
 		InfoBar.Error(err)
 	} else {
-		h.paste(clip)
+		if h.Buf.NumCursors() != len(clips) {
+			// If cursor clipboard length doesn't match number of cursors, paste each cursor
+			// clipboard to a newline
+			for i, clip := range clips {
+				h.paste(clip)
+				if i != len(clips)-1 {
+					h.InsertNewline()
+				}
+			}
+		} else {
+			h.paste(clips[h.Cursor.Num])
+		}
 	}
 	h.Relocate()
 	return true

internal/clipboard/clipboard.go

@@ -1,6 +1,7 @@
 package clipboard
 
 import (
+	"bytes"
 	"errors"
 
 	"github.com/zyedidia/clipper"
@@ -78,16 +79,27 @@ func Write(text string, r Register) error {
 	return write(text, r, CurrentMethod)
 }
 
-// ReadMulti reads text from a clipboard register for a certain multi-cursor
-func ReadMulti(r Register, num, ncursors int) (string, error) {
+// ReadMulti reads text array from a clipboard register, which can be a multi cursor clipboard
+func ReadMulti(r Register) ([]string, error) {
 	clip, err := Read(r)
-	if err != nil {
-		return "", err
+	multivalid := false
+	if err == nil {
+		multivalid = ValidMulti(r, &clip)
+	} else {
+		multivalid = ValidMulti(r, nil)
 	}
-	if ValidMulti(r, clip, ncursors) {
-		return multi.getText(r, num), nil
+
+	if !multivalid {
+		returnarray := make([]string, 1, 1)
+		if err == nil {
+			returnarray[0] = clip
+		} else {
+			returnarray[0] = ""
+		}
+		return returnarray, err
 	}
-	return clip, nil
+
+	return multi.getAllText(r), nil
 }
 
 // WriteMulti writes text to a clipboard register for a certain multi-cursor
@@ -97,13 +109,27 @@ func WriteMulti(text string, r Register, num int, ncursors int) error {
 
 // ValidMulti checks if the internal multi-clipboard is valid and up-to-date
 // with the system clipboard
-func ValidMulti(r Register, clip string, ncursors int) bool {
-	return multi.isValid(r, clip, ncursors)
+func ValidMulti(r Register, clip *string) bool {
+	return multi.isValid(r, clip)
 }
 
 func writeMulti(text string, r Register, num int, ncursors int, m Method) error {
+	// Write to multi cursor clipboard
 	multi.writeText(text, r, num, ncursors)
-	return write(multi.getAllText(r), r, m)
+	multitext := multi.getAllText(r)
+	if multitext == nil {
+		return write("", r, m)
+	}
+
+	// Write to normal cliipboard
+	buf := &bytes.Buffer{}
+	for i, s := range multitext {
+		buf.WriteString(s)
+		if i != len(multitext)-1 {
+			buf.WriteString("\n")
+		}
+	}
+	return write(buf.String(), r, m)
 }
 
 func read(r Register, m Method) (string, error) {

internal/clipboard/multi.go

@@ -9,17 +9,9 @@ type multiClipboard map[Register][]string
 
 var multi multiClipboard
 
-func (c multiClipboard) getAllText(r Register) string {
-	content := c[r]
-	if content == nil {
-		return ""
-	}
+func (c multiClipboard) getAllText(r Register) []string {
+	return c[r]
 
-	buf := &bytes.Buffer{}
-	for _, s := range content {
-		buf.WriteString(s)
-	}
-	return buf.String()
 }
 
 func (c multiClipboard) getText(r Register, num int) string {
@@ -35,13 +27,30 @@ func (c multiClipboard) getText(r Register, num int) string {
 // text stored in the system clipboard (provided as an argument), and therefore
 // if it is safe to use the multi-clipboard for pasting instead of the system
 // clipboard.
-func (c multiClipboard) isValid(r Register, clipboard string, ncursors int) bool {
+func (c multiClipboard) isValid(r Register, clipboard *string) bool {
 	content := c[r]
-	if content == nil || len(content) != ncursors {
+	if content == nil {
 		return false
 	}
 
-	return clipboard == c.getAllText(r)
+	multiclips := c.getAllText(r)
+	if multiclips == nil || len(multiclips) == 0 {
+		return false
+	}
+
+	if clipboard == nil {
+		return true
+	}
+
+	buf := &bytes.Buffer{}
+	for i, s := range multiclips {
+		buf.WriteString(s)
+		if i != len(multiclips)-1 {
+			buf.WriteString("\n")
+		}
+	}
+
+	return buf.String() == *clipboard
 }
 
 func (c multiClipboard) writeText(text string, r Register, num int, ncursors int) {