initrd/ask-secret.go

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	"os"
)

func askSecret(prompt string) []byte {
	stdinTTY.EchoOff()

	var (
		in  io.Reader = stdin
		out io.Writer = stdout
	)

	if stdin == nil {
		in = os.Stdin
		out = os.Stdout
	}

	out.Write([]byte(prompt + ": "))

	if stdin != nil {
		stdout.HideInput()
	}

	s, err := bufio.NewReader(in).ReadBytes('\n')

	if stdin != nil {
		stdout.ShowInput()
	}

	stdinTTY.Restore()

	if err != nil {
		fatalf("failed to read from stdin: %v", err)
	}

	fmt.Println()

	s = bytes.TrimRight(s, "\r\n")
	return s
}