mirror of
https://github.com/int128/kubelogin.git
synced 2026-05-18 13:56:34 +00:00
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
// Package reader provides the reader of standard input.
|
|
package reader
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/google/wire"
|
|
"github.com/int128/kubelogin/pkg/adaptors/stdio"
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
//go:generate mockgen -destination mock_reader/mock_reader.go github.com/int128/kubelogin/pkg/adaptors/reader Interface
|
|
|
|
// Set provides an implementation and interface for Reader.
|
|
var Set = wire.NewSet(
|
|
wire.Struct(new(Reader), "*"),
|
|
wire.Bind(new(Interface), new(*Reader)),
|
|
)
|
|
|
|
type Interface interface {
|
|
ReadString(prompt string) (string, error)
|
|
ReadPassword(prompt string) (string, error)
|
|
}
|
|
|
|
type Reader struct {
|
|
Stdin stdio.Stdin
|
|
}
|
|
|
|
// ReadString reads a string from the stdin.
|
|
func (x *Reader) ReadString(prompt string) (string, error) {
|
|
if _, err := fmt.Fprint(os.Stderr, prompt); err != nil {
|
|
return "", xerrors.Errorf("write error: %w", err)
|
|
}
|
|
r := bufio.NewReader(x.Stdin)
|
|
s, err := r.ReadString('\n')
|
|
if err != nil {
|
|
return "", xerrors.Errorf("read error: %w", err)
|
|
}
|
|
s = strings.TrimRight(s, "\r\n")
|
|
return s, nil
|
|
}
|
|
|
|
// ReadPassword reads a password from the stdin without echo back.
|
|
func (*Reader) ReadPassword(prompt string) (string, error) {
|
|
if _, err := fmt.Fprint(os.Stderr, prompt); err != nil {
|
|
return "", xerrors.Errorf("write error: %w", err)
|
|
}
|
|
b, err := terminal.ReadPassword(int(syscall.Stdin))
|
|
if err != nil {
|
|
return "", xerrors.Errorf("read error: %w", err)
|
|
}
|
|
if _, err := fmt.Fprintln(os.Stderr); err != nil {
|
|
return "", xerrors.Errorf("write error: %w", err)
|
|
}
|
|
return string(b), nil
|
|
}
|