mirror of
https://github.com/johnkerl/miller.git
synced 2026-01-23 02:14:13 +00:00
085e831668
* Update package version * Update makefile targets * Update readme packages * Remaining old packages via rg/sd
71 lines
2 KiB
Go
71 lines
2 KiB
Go
// ================================================================
|
|
// Handling for default and customized banners/prompts for the Miller REPL.
|
|
// ================================================================
|
|
|
|
package repl
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
|
|
"golang.org/x/term"
|
|
|
|
"github.com/johnkerl/miller/v6/pkg/colorizer"
|
|
"github.com/johnkerl/miller/v6/pkg/lib"
|
|
"github.com/johnkerl/miller/v6/pkg/version"
|
|
)
|
|
|
|
const ENV_PRIMARY_PROMPT = "MLR_REPL_PS1"
|
|
const ENV_SECONDARY_PROMPT = "MLR_REPL_PS2"
|
|
const DEFAULT_PRIMARY_PROMPT = "[mlr] "
|
|
const DEFAULT_SECONDARY_PROMPT = "... "
|
|
|
|
func getInputIsTerminal() bool {
|
|
if runtime.GOOS == "windows" && os.Getenv("MSYSTEM") != "" {
|
|
// Sadly, term.IsTerminal doesn't work inside MSYS2 but does work
|
|
// outside MSYS2. So in that case we simply return true so that the mlr
|
|
// repl has a prompt.
|
|
return true
|
|
} else {
|
|
return term.IsTerminal(int(os.Stdin.Fd()))
|
|
}
|
|
}
|
|
|
|
func getPrompt1() string {
|
|
prompt1 := os.Getenv(ENV_PRIMARY_PROMPT)
|
|
if prompt1 == "" {
|
|
prompt1 = DEFAULT_PRIMARY_PROMPT
|
|
}
|
|
prompt1 = lib.UnbackslashStringLiteral(prompt1) // In case they want a "\n" in there
|
|
return colorizer.MaybeColorizeREPLPS1(prompt1, true)
|
|
}
|
|
|
|
func getPrompt2() string {
|
|
prompt2 := os.Getenv(ENV_SECONDARY_PROMPT)
|
|
if prompt2 == "" {
|
|
prompt2 = DEFAULT_SECONDARY_PROMPT
|
|
}
|
|
prompt2 = lib.UnbackslashStringLiteral(prompt2) // In case they want a "\n" in there
|
|
return colorizer.MaybeColorizeREPLPS2(prompt2, true)
|
|
}
|
|
|
|
func (repl *Repl) printStartupBanner() {
|
|
if repl.inputIsTerminal {
|
|
fmt.Printf("Miller %s REPL for %s/%s/%s\n", version.STRING, runtime.GOOS, runtime.GOARCH, runtime.Version())
|
|
fmt.Printf("Docs: %s\n", lib.DOC_URL)
|
|
fmt.Printf("Type ':h' or ':help' for online help; ':q' or ':quit' to quit.\n")
|
|
}
|
|
}
|
|
|
|
func (repl *Repl) printPrompt1() {
|
|
if repl.inputIsTerminal && repl.showPrompts {
|
|
fmt.Print(repl.prompt1)
|
|
}
|
|
}
|
|
|
|
func (repl *Repl) printPrompt2() {
|
|
if repl.inputIsTerminal && repl.showPrompts {
|
|
fmt.Print(repl.prompt2)
|
|
}
|
|
}
|