From bdd8d4582f185f32a68ebb8394e1cc1f63262098 Mon Sep 17 00:00:00 2001
From: Noboru Saito <noborusai@gmail.com>
Date: Mon, 29 Nov 2021 12:40:50 +0900
Subject: [PATCH] Prepare a command to use PRIMARY in wayland

Wayland defaults to the clipboard,
so users will need to add an argument if they want to use PRIMARY.
PRIMARY is provided as a separate command, so
that the order of the arguments doesn't matter.
---
 clipboard_unix.go | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/clipboard_unix.go b/clipboard_unix.go
index d9f6a56..01e7de0 100644
--- a/clipboard_unix.go
+++ b/clipboard_unix.go
@@ -30,18 +30,30 @@ var (
 	pasteCmdArgs []string
 	copyCmdArgs  []string
 
+	pastePrimaryCmdArgs []string
+	copyPrimaryCmdArgs  []string
+
 	xselPasteArgs = []string{xsel, "--output", "--clipboard"}
 	xselCopyArgs  = []string{xsel, "--input", "--clipboard"}
 
+	xselPrimaryPasteArgs = []string{xsel, "--output", "--primary"}
+	xselPrimaryCopyArgs  = []string{xsel, "--input", "--primary"}
+
 	xclipPasteArgs = []string{xclip, "-out", "-selection", "clipboard"}
 	xclipCopyArgs  = []string{xclip, "-in", "-selection", "clipboard"}
 
+	xclipPrimaryPasteArgs = []string{xclip, "-out", "-selection", "primary"}
+	xclipPrimaryCopyArgs  = []string{xclip, "-in", "-selection", "primary"}
+
 	powershellExePasteArgs = []string{powershellExe, "Get-Clipboard"}
 	clipExeCopyArgs        = []string{clipExe}
 
 	wlpasteArgs = []string{wlpaste, "--no-newline"}
 	wlcopyArgs  = []string{wlcopy}
 
+	wlPrimarypasteArgs = []string{wlpaste, "--no-newline", "--primary"}
+	wlPrimarycopyArgs  = []string{wlcopy, "--primary"}
+
 	termuxPasteArgs = []string{termuxClipboardGet}
 	termuxCopyArgs  = []string{termuxClipboardSet}
 
@@ -52,6 +64,8 @@ func init() {
 	if os.Getenv("WAYLAND_DISPLAY") != "" {
 		pasteCmdArgs = wlpasteArgs
 		copyCmdArgs = wlcopyArgs
+		pastePrimaryCmdArgs = wlPrimarypasteArgs
+		copyPrimaryCmdArgs = wlPrimarycopyArgs
 
 		if _, err := exec.LookPath(wlcopy); err == nil {
 			if _, err := exec.LookPath(wlpaste); err == nil {
@@ -62,6 +76,8 @@ func init() {
 
 	pasteCmdArgs = xclipPasteArgs
 	copyCmdArgs = xclipCopyArgs
+	pastePrimaryCmdArgs = xclipPrimaryPasteArgs
+	copyPrimaryCmdArgs = xclipPrimaryCopyArgs
 
 	if _, err := exec.LookPath(xclip); err == nil {
 		return
@@ -69,6 +85,8 @@ func init() {
 
 	pasteCmdArgs = xselPasteArgs
 	copyCmdArgs = xselCopyArgs
+	pastePrimaryCmdArgs = xselPrimaryPasteArgs
+	copyPrimaryCmdArgs = xselPrimaryCopyArgs
 
 	if _, err := exec.LookPath(xsel); err == nil {
 		return
@@ -76,6 +94,8 @@ func init() {
 
 	pasteCmdArgs = termuxPasteArgs
 	copyCmdArgs = termuxCopyArgs
+	pastePrimaryCmdArgs = termuxPasteArgs
+	copyPrimaryCmdArgs = termuxCopyArgs
 
 	if _, err := exec.LookPath(termuxClipboardSet); err == nil {
 		if _, err := exec.LookPath(termuxClipboardGet); err == nil {
@@ -85,6 +105,8 @@ func init() {
 
 	pasteCmdArgs = powershellExePasteArgs
 	copyCmdArgs = clipExeCopyArgs
+	pastePrimaryCmdArgs = powershellExePasteArgs
+	copyPrimaryCmdArgs = clipExeCopyArgs
 	trimDos = true
 
 	if _, err := exec.LookPath(clipExe); err == nil {
@@ -98,14 +120,14 @@ func init() {
 
 func getPasteCommand() *exec.Cmd {
 	if Primary {
-		pasteCmdArgs = pasteCmdArgs[:1]
+		return exec.Command(pastePrimaryCmdArgs[0], pastePrimaryCmdArgs[1:]...)
 	}
 	return exec.Command(pasteCmdArgs[0], pasteCmdArgs[1:]...)
 }
 
 func getCopyCommand() *exec.Cmd {
 	if Primary {
-		copyCmdArgs = copyCmdArgs[:1]
+		return exec.Command(copyPrimaryCmdArgs[0], copyPrimaryCmdArgs[1:]...)
 	}
 	return exec.Command(copyCmdArgs[0], copyCmdArgs[1:]...)
 }