pulumi/pkg/util/cmdutil/console.go
Joe Duffy 811759fb77
Fix missing emojis on Windows (#966)
I was reminded of this yesterday with unprintable characters as I
debugged some things on Windows.  Inspired by Yarn, this change adds
a new flag --emoji (-e for short) that can be used to control whether
we show ASCII-only characters or not in the console.  On Mac, it
defaults to true, and on Windows and Linux, it defaults to false.

This also brings back the retro ASCII-friendly progress spinner
when --emoji is disabled.
2018-02-21 09:42:06 -08:00

65 lines
1.5 KiB
Go

// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
package cmdutil
import (
"bufio"
"fmt"
"os"
"runtime"
"strings"
"golang.org/x/crypto/ssh/terminal"
"github.com/pulumi/pulumi/pkg/diag/colors"
)
// Emoji controls whether emojis will by default be printed in the output.
var Emoji = (runtime.GOOS == "darwin")
// EmojiOr returns the emoji string e if emojis are enabled, or the string or if emojis are disabled.
func EmojiOr(e, or string) string {
if Emoji {
return e
}
return or
}
// Interactive returns true if we're in an interactive terminal session.
func Interactive() bool {
return terminal.IsTerminal(int(os.Stdin.Fd()))
}
// ReadConsole reads the console with the given prompt text.
func ReadConsole(prompt string) (string, error) {
if prompt != "" {
prompt = colors.ColorizeText(
fmt.Sprintf("%s%s:%s ", colors.BrightCyan, prompt, colors.Reset))
fmt.Print(prompt)
}
reader := bufio.NewReader(os.Stdin)
raw, err := reader.ReadString('\n')
if err != nil {
return "", err
}
return RemoveTralingNewline(raw), nil
}
// IsTruthy returns true if the given string represents a CLI input interpreted as "true".
func IsTruthy(s string) bool {
return s == "1" || strings.EqualFold(s, "true")
}
// RemoveTralingNewline removes a trailing newline from a string. On windows, we'll remove either \r\n or \n, on other
// platforms, we just remove \n.
func RemoveTralingNewline(s string) string {
s = strings.TrimSuffix(s, "\n")
if runtime.GOOS == "windows" {
s = strings.TrimSuffix(s, "\r")
}
return s
}