zenity/msg_unix.go

// +build !windows,!darwin

package zenity

import (
	"os/exec"

	"github.com/ncruces/zenity/internal/zen"
)

// Display question dialog.
//
// Returns true on OK, false on Cancel, or ErrExtraButton.
//
// Valid options: Title, Icon, OKLabel, CancelLabel, ExtraButton, NoWrap,
// Ellipsize, DefaultCancel.
func Question(text string, options ...Option) (bool, error) {
	return message("--question", text, options)
}

// Display info dialog.
//
// Returns true on OK, false on dismiss, or ErrExtraButton.
//
// Valid options: Title, Icon, OKLabel, ExtraButton, NoWrap, Ellipsize.
func Info(text string, options ...Option) (bool, error) {
	return message("--info", text, options)
}

// Display warning dialog.
//
// Returns true on OK, false on dismiss, or ErrExtraButton.
//
// Valid options: Title, Icon, OKLabel, ExtraButton, NoWrap, Ellipsize.
func Warning(text string, options ...Option) (bool, error) {
	return message("--warning", text, options)
}

// Display error dialog.
//
// Returns true on OK, false on dismiss, or ErrExtraButton.
//
// Valid options: Title, Icon, OKLabel, ExtraButton, NoWrap, Ellipsize.
func Error(text string, options ...Option) (bool, error) {
	return message("--error", text, options)
}

func message(arg, text string, options []Option) (bool, error) {
	opts := optsParse(options)

	args := []string{arg}
	if text != "" {
		args = append(args, "--text", text, "--no-markup")
	}
	if opts.title != "" {
		args = append(args, "--title", opts.title)
	}
	if opts.ok != "" {
		args = append(args, "--ok-label", opts.ok)
	}
	if opts.cancel != "" {
		args = append(args, "--cancel-label", opts.cancel)
	}
	if opts.extra != "" {
		args = append(args, "--extra-button", opts.extra)
	}
	if opts.nowrap {
		args = append(args, "--no-wrap")
	}
	if opts.ellipsize {
		args = append(args, "--ellipsize")
	}
	if opts.defcancel {
		args = append(args, "--default-cancel")
	}
	switch opts.icon {
	case ErrorIcon:
		args = append(args, "--icon-name=dialog-error")
	case InfoIcon:
		args = append(args, "--icon-name=dialog-information")
	case QuestionIcon:
		args = append(args, "--icon-name=dialog-question")
	case WarningIcon:
		args = append(args, "--icon-name=dialog-warning")
	}

	out, err := zen.Run(args)
	if err, ok := err.(*exec.ExitError); ok && err.ExitCode() != 255 {
		if len(out) > 0 && string(out[:len(out)-1]) == opts.extra {
			return false, ErrExtraButton
		}
		return false, nil
	}
	if err != nil {
		return false, err
	}
	return true, err
}