diff options
Diffstat (limited to 'pkg/prompt')
| -rw-r--r-- | pkg/prompt/main.go | 1 | ||||
| -rw-r--r-- | pkg/prompt/misc.go | 32 |
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) +} |