Fix #11.

ncruces · May 22, 2021 · 6033f51 · 6033f51
1 parent 6e3393b
commit 6033f51
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/util_unix.go b/util_unix.go
@@ -3,6 +3,7 @@
 package zenity
 
 import (
+	"bytes"
 	"os/exec"
 	"strconv"
 	"strings"
@@ -55,8 +56,9 @@ func appendIcon(args []string, opts options) []string {
 }
 
 func strResult(opts options, out []byte, err error) (string, error) {
+	out = bytes.TrimSuffix(out, []byte{'\n'})
 	if err, ok := err.(*exec.ExitError); ok && err.ExitCode() == 1 {
-		if opts.extraButton != nil && *opts.extraButton+"\n" == string(out) {
+		if opts.extraButton != nil && *opts.extraButton == string(out) {
 			return "", ErrExtraButton
 		}
 		return "", ErrCanceled