Username (windows).

This commit is contained in:
Nuno Cruces 2022-05-12 14:27:24 +01:00
parent 0abd40cb11
commit 5e39a48e5c
5 changed files with 223 additions and 32 deletions

View file

@ -158,8 +158,7 @@ func main() {
calResult(zenity.Calendar(text, opts...)) calResult(zenity.Calendar(text, opts...))
case passwordDlg: case passwordDlg:
_, pw, err := zenity.Password(opts...) pwdResult(zenity.Password(opts...))
strResult(pw, err)
case fileSelectionDlg: case fileSelectionDlg:
switch { switch {
@ -533,7 +532,9 @@ func strResult(s string, err error) {
func lstResult(l []string, err error) { func lstResult(l []string, err error) {
errResult(err) errResult(err)
os.Stdout.WriteString(strings.Join(l, zenutil.Separator)) os.Stdout.WriteString(strings.Join(l, zenutil.Separator))
os.Stdout.WriteString(zenutil.LineBreak) if len(l) > 0 {
os.Stdout.WriteString(zenutil.LineBreak)
}
} }
func calResult(d time.Time, err error) { func calResult(d time.Time, err error) {
@ -548,6 +549,16 @@ func colResult(c color.Color, err error) {
os.Stdout.WriteString(zenutil.LineBreak) os.Stdout.WriteString(zenutil.LineBreak)
} }
func pwdResult(u, p string, err error) {
errResult(err)
if username {
os.Stdout.WriteString(u)
os.Stdout.WriteString(zenutil.Separator)
}
os.Stdout.WriteString(p)
os.Stdout.WriteString(zenutil.LineBreak)
}
func ingestPath(path string) string { func ingestPath(path string) string {
if runtime.GOOS == "windows" && path != "" { if runtime.GOOS == "windows" && path != "" {
var args []string var args []string

2
pwd.go
View file

@ -4,7 +4,7 @@ package zenity
// //
// Valid options: Title, OKLabel, CancelLabel, ExtraButton, Icon, Username. // Valid options: Title, OKLabel, CancelLabel, ExtraButton, Icon, Username.
// //
// May return: ErrCanceled, ErrExtraButton, ErrUnsupported. // May return: ErrCanceled, ErrExtraButton.
func Password(options ...Option) (usr string, pwd string, err error) { func Password(options ...Option) (usr string, pwd string, err error) {
return password(applyOptions(options)) return password(applyOptions(options))
} }

View file

@ -5,7 +5,6 @@ import (
"errors" "errors"
"fmt" "fmt"
"os" "os"
"runtime"
"testing" "testing"
"time" "time"
@ -60,16 +59,8 @@ func TestPassword_username(t *testing.T) {
if skip, err := skip(err); skip { if skip, err := skip(err); skip {
t.Skip("skipping:", err) t.Skip("skipping:", err)
} }
if runtime.GOOS == "windows" { if !errors.Is(err, context.Canceled) {
if errors.Is(err, zenity.ErrUnsupported) { t.Error("was not canceled:", err)
t.Skip("was not unsupported:", err)
} else {
t.Error("was not unsupported:", err)
}
} else {
if !errors.Is(err, context.Canceled) {
t.Error("was not canceled:", err)
}
} }
} }
@ -84,7 +75,7 @@ func TestPassword_script(t *testing.T) {
}{ }{
{name: "Cancel", call: "cancel", err: zenity.ErrCanceled}, {name: "Cancel", call: "cancel", err: zenity.ErrCanceled},
{name: "Password", call: "enter pwd", pwd: "pwd"}, {name: "Password", call: "enter pwd", pwd: "pwd"},
{name: "User", call: "enter usr and pwd (if supported)", usr: "usr", pwd: "pwd", {name: "User", call: "enter usr and pwd", usr: "usr", pwd: "pwd",
opts: []zenity.Option{zenity.Username()}}, opts: []zenity.Option{zenity.Username()}},
} }
for _, tt := range tests { for _, tt := range tests {
@ -94,9 +85,6 @@ func TestPassword_script(t *testing.T) {
if skip, err := skip(err); skip { if skip, err := skip(err); skip {
t.Skip("skipping:", err) t.Skip("skipping:", err)
} }
if errors.Is(err, zenity.ErrUnsupported) {
t.Skip("was not unsupported:", err)
}
if usr != tt.usr || pwd != tt.pwd || err != tt.err { if usr != tt.usr || pwd != tt.pwd || err != tt.err {
t.Errorf("Password() = %q, %q, %v; want %q, %q, %v", usr, pwd, err, tt.usr, tt.pwd, tt.err) t.Errorf("Password() = %q, %q, %v; want %q, %q, %v", usr, pwd, err, tt.usr, tt.pwd, tt.err)
} }

View file

@ -1,12 +1,201 @@
package zenity package zenity
import "fmt" import (
"syscall"
"unsafe"
)
func password(opts options) (string, string, error) { func password(opts options) (string, string, error) {
if opts.username { if !opts.username {
return "", "", fmt.Errorf("%w: username", ErrUnsupported) opts.hideText = true
str, err := entry("Password:", opts)
return "", str, err
} }
opts.hideText = true
str, err := entry("Password:", opts) if opts.title == nil {
return "", str, err opts.title = stringPtr("")
}
if opts.okLabel == nil {
opts.okLabel = stringPtr("OK")
}
if opts.cancelLabel == nil {
opts.cancelLabel = stringPtr("Cancel")
}
dlg := &passwordDialog{}
return dlg.setup(opts)
}
type passwordDialog struct {
usr string
pwd string
err error
wnd uintptr
uTextCtl uintptr
uEditCtl uintptr
pTextCtl uintptr
pEditCtl uintptr
okBtn uintptr
cancelBtn uintptr
extraBtn uintptr
font font
}
func (dlg *passwordDialog) setup(opts options) (string, string, error) {
defer setup()()
dlg.font = getFont()
defer dlg.font.delete()
if opts.ctx != nil && opts.ctx.Err() != nil {
return "", "", opts.ctx.Err()
}
instance, _, err := getModuleHandle.Call(0)
if instance == 0 {
return "", "", err
}
cls, err := registerClass(instance, syscall.NewCallback(passwordProc))
if cls == 0 {
return "", "", err
}
defer unregisterClass.Call(cls, instance)
dlg.wnd, _, _ = createWindowEx.Call(_WS_EX_CONTROLPARENT|_WS_EX_WINDOWEDGE|_WS_EX_DLGMODALFRAME,
cls, strptr(*opts.title),
_WS_POPUPWINDOW|_WS_CLIPSIBLINGS|_WS_DLGFRAME,
_CW_USEDEFAULT, _CW_USEDEFAULT,
281, 191, 0, 0, instance, uintptr(unsafe.Pointer(dlg)))
dlg.uTextCtl, _, _ = createWindowEx.Call(0,
strptr("STATIC"), strptr("Username:"),
_WS_CHILD|_WS_VISIBLE|_WS_GROUP|_SS_WORDELLIPSIS|_SS_EDITCONTROL|_SS_NOPREFIX,
12, 10, 241, 16, dlg.wnd, 0, instance, 0)
var flags uintptr = _WS_CHILD | _WS_VISIBLE | _WS_GROUP | _WS_TABSTOP | _ES_AUTOHSCROLL
dlg.uEditCtl, _, _ = createWindowEx.Call(_WS_EX_CLIENTEDGE,
strptr("EDIT"), 0,
flags,
12, 30, 241, 24, dlg.wnd, 0, instance, 0)
dlg.pTextCtl, _, _ = createWindowEx.Call(0,
strptr("STATIC"), strptr("Password:"),
_WS_CHILD|_WS_VISIBLE|_WS_GROUP|_SS_WORDELLIPSIS|_SS_EDITCONTROL|_SS_NOPREFIX,
12, 60, 241, 16, dlg.wnd, 0, instance, 0)
dlg.pEditCtl, _, _ = createWindowEx.Call(_WS_EX_CLIENTEDGE,
strptr("EDIT"), 0,
flags|_ES_PASSWORD,
12, 80, 241, 24, dlg.wnd, 0, instance, 0)
dlg.okBtn, _, _ = createWindowEx.Call(0,
strptr("BUTTON"), strptr(*opts.okLabel),
_WS_CHILD|_WS_VISIBLE|_WS_GROUP|_WS_TABSTOP|_BS_DEFPUSHBUTTON,
12, 116, 75, 24, dlg.wnd, _IDOK, instance, 0)
dlg.cancelBtn, _, _ = createWindowEx.Call(0,
strptr("BUTTON"), strptr(*opts.cancelLabel),
_WS_CHILD|_WS_VISIBLE|_WS_GROUP|_WS_TABSTOP,
12, 116, 75, 24, dlg.wnd, _IDCANCEL, instance, 0)
if opts.extraButton != nil {
dlg.extraBtn, _, _ = createWindowEx.Call(0,
strptr("BUTTON"), strptr(*opts.extraButton),
_WS_CHILD|_WS_VISIBLE|_WS_GROUP|_WS_TABSTOP,
12, 116, 75, 24, dlg.wnd, _IDNO, instance, 0)
}
dlg.layout(getDPI(dlg.wnd))
centerWindow(dlg.wnd)
setFocus.Call(dlg.uEditCtl)
showWindow.Call(dlg.wnd, _SW_NORMAL, 0)
sendMessage.Call(dlg.uEditCtl, _EM_SETSEL, 0, intptr(-1))
if opts.ctx != nil {
wait := make(chan struct{})
defer close(wait)
go func() {
select {
case <-opts.ctx.Done():
sendMessage.Call(dlg.wnd, _WM_SYSCOMMAND, _SC_CLOSE, 0)
case <-wait:
}
}()
}
if err := messageLoop(dlg.wnd); err != nil {
return "", "", err
}
if opts.ctx != nil && opts.ctx.Err() != nil {
return "", "", opts.ctx.Err()
}
return dlg.usr, dlg.pwd, dlg.err
}
func (dlg *passwordDialog) layout(dpi dpi) {
font := dlg.font.forDPI(dpi)
sendMessage.Call(dlg.uTextCtl, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.uEditCtl, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.pTextCtl, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.pEditCtl, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.okBtn, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.cancelBtn, _WM_SETFONT, font, 1)
sendMessage.Call(dlg.extraBtn, _WM_SETFONT, font, 1)
setWindowPos.Call(dlg.wnd, 0, 0, 0, dpi.scale(281), dpi.scale(191), _SWP_NOZORDER|_SWP_NOMOVE)
setWindowPos.Call(dlg.uTextCtl, 0, dpi.scale(12), dpi.scale(10), dpi.scale(241), dpi.scale(16), _SWP_NOZORDER)
setWindowPos.Call(dlg.uEditCtl, 0, dpi.scale(12), dpi.scale(30), dpi.scale(241), dpi.scale(24), _SWP_NOZORDER)
setWindowPos.Call(dlg.pTextCtl, 0, dpi.scale(12), dpi.scale(60), dpi.scale(241), dpi.scale(16), _SWP_NOZORDER)
setWindowPos.Call(dlg.pEditCtl, 0, dpi.scale(12), dpi.scale(80), dpi.scale(241), dpi.scale(24), _SWP_NOZORDER)
if dlg.extraBtn == 0 {
setWindowPos.Call(dlg.okBtn, 0, dpi.scale(95), dpi.scale(116), dpi.scale(75), dpi.scale(24), _SWP_NOZORDER)
setWindowPos.Call(dlg.cancelBtn, 0, dpi.scale(178), dpi.scale(116), dpi.scale(75), dpi.scale(24), _SWP_NOZORDER)
} else {
setWindowPos.Call(dlg.okBtn, 0, dpi.scale(12), dpi.scale(116), dpi.scale(75), dpi.scale(24), _SWP_NOZORDER)
setWindowPos.Call(dlg.extraBtn, 0, dpi.scale(95), dpi.scale(116), dpi.scale(75), dpi.scale(24), _SWP_NOZORDER)
setWindowPos.Call(dlg.cancelBtn, 0, dpi.scale(178), dpi.scale(116), dpi.scale(75), dpi.scale(24), _SWP_NOZORDER)
}
}
func passwordProc(wnd uintptr, msg uint32, wparam uintptr, lparam *unsafe.Pointer) uintptr {
var dlg *passwordDialog
switch msg {
case _WM_NCCREATE:
saveBackRef(wnd, *lparam)
dlg = (*passwordDialog)(*lparam)
case _WM_NCDESTROY:
deleteBackRef(wnd)
default:
dlg = (*passwordDialog)(loadBackRef(wnd))
}
switch msg {
case _WM_DESTROY:
postQuitMessage.Call(0)
case _WM_CLOSE:
dlg.err = ErrCanceled
destroyWindow.Call(wnd)
case _WM_COMMAND:
switch wparam {
default:
return 1
case _IDOK, _IDYES:
dlg.usr = getWindowString(dlg.uEditCtl)
dlg.pwd = getWindowString(dlg.pEditCtl)
case _IDCANCEL:
dlg.err = ErrCanceled
case _IDNO:
dlg.err = ErrExtraButton
}
destroyWindow.Call(wnd)
case _WM_DPICHANGED:
dlg.layout(dpi(uint32(wparam) >> 16))
default:
res, _, _ := defWindowProc.Call(wnd, uintptr(msg), wparam, uintptr(unsafe.Pointer(lparam)))
return res
}
return 0
} }

View file

@ -86,13 +86,16 @@ func lstResult(opts options, out []byte, err error) ([]string, error) {
func pwdResult(sep string, opts options, out []byte, err error) (string, string, error) { func pwdResult(sep string, opts options, out []byte, err error) (string, string, error) {
str, err := strResult(opts, out, err) str, err := strResult(opts, out, err)
if opts.username { if opts.username {
if err, ok := err.(*exec.ExitError); ok && err.ExitCode() == 255 { usr, pwd, _ := cut(str, sep)
return "", "", ErrUnsupported return usr, pwd, err
}
if split := strings.SplitN(str, sep, 2); err == nil && len(split) == 2 {
return split[0], split[1], nil
}
} }
return "", str, err return "", str, err
} }
// Replace with strings.Cut after 1.18.
func cut(s, sep string) (before, after string, found bool) {
if i := strings.Index(s, sep); i >= 0 {
return s[:i], s[i+len(sep):], true
}
return s, "", false
}