From ed2b9374599cc00467b1c27121318748cc0d3f49 Mon Sep 17 00:00:00 2001 From: Stefan Majewsky Date: Sat, 28 Mar 2020 13:53:10 +0100 Subject: rename pkg/ to internal/ --- internal/prompt/misc.go | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 internal/prompt/misc.go (limited to 'internal/prompt/misc.go') diff --git a/internal/prompt/misc.go b/internal/prompt/misc.go new file mode 100644 index 0000000..124015a --- /dev/null +++ b/internal/prompt/misc.go @@ -0,0 +1,43 @@ +/******************************************************************************* +* +* Copyright 2017 Stefan Majewsky +* +* This program is free software: you can redistribute it and/or modify it under +* the terms of the GNU General Public License as published by the Free Software +* Foundation, either version 3 of the License, or (at your option) any later +* version. +* +* This program is distributed in the hope that it will be useful, but WITHOUT ANY +* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR +* A PARTICULAR PURPOSE. See the GNU General Public License for more details. +* +* You should have received a copy of the GNU General Public License along with +* this program. If not, see . +* +*******************************************************************************/ + +package prompt + +import ( + "os" + "strconv" +) + +func getTerminalField() string { + termName := os.Getenv("TERM") + if termName == "xterm-256color" { + return "" + } + if termName == "" { + termName = withColor("1;41", "not set") + } + return withType("term", termName) +} + +func getExitCodeField(arg string) string { + exitCode, err := strconv.Atoi(arg) + if err == nil && exitCode > 0 { + return withColor("1;31", "exit:"+arg) + } + return "" +} -- cgit v1.3.1