prettyprompt: show $TERM if not xterm-256color

2 files changed, 33 insertions, 0 deletions

diff --git a/pkg/prompt/main.go b/pkg/prompt/main.go
index 4146b73..728f2f2 100644
--- a/pkg/prompt/main.go
+++ b/pkg/prompt/main.go
@@ -37,6 +37,7 @@ func Exec() int {
 	cwd := CurrentDirectory()
 	fields = appendUnlessEmpty(fields, getDirectoryField(cwd))
 	fields = appendUnlessEmpty(fields, getDeletedMessageField(cwd))
+	fields = appendUnlessEmpty(fields, getTerminalField())
 
 	line := strings.Join(fields, " ")
 	lineWidth := getPrintableLength(line)
diff --git a/pkg/prompt/misc.go b/pkg/prompt/misc.go
new file mode 100644
index 0000000..eaa1848
--- /dev/null
+++ b/pkg/prompt/misc.go
@@ -0,0 +1,32 @@
+/*******************************************************************************
+*
+* Copyright 2017 Stefan Majewsky <majewsky@gmx.net>
+*
+* 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 <http://www.gnu.org/licenses/>.
+*
+*******************************************************************************/
+
+package prompt
+
+import "os"
+
+func getTerminalField() string {
+	termName := os.Getenv("TERM")
+	if termName == "xterm-256color" {
+		return ""
+	}
+	if termName == "" {
+		termName = withColor("1;41", "not set")
+	}
+	return withType("term", termName)
+}