From 5d3e00dcb39a2d87919ea40bf144b4579c86c365 Mon Sep 17 00:00:00 2001 From: Stefan Majewsky Date: Sat, 19 Aug 2017 18:53:30 +0200 Subject: prettyprompt: show $TERM if not xterm-256color --- pkg/prompt/main.go | 1 + pkg/prompt/misc.go | 32 ++++++++++++++++++++++++++++++++ 2 files changed, 33 insertions(+) create mode 100644 pkg/prompt/misc.go (limited to 'pkg/prompt') 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 +* +* 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" + +func getTerminalField() string { + termName := os.Getenv("TERM") + if termName == "xterm-256color" { + return "" + } + if termName == "" { + termName = withColor("1;41", "not set") + } + return withType("term", termName) +} -- cgit v1.3.1