mirror of
https://github.com/buckley310/nixos-config.git
synced 2024-12-21 19:24:15 +00:00
49 lines
1.2 KiB
Diff
49 lines
1.2 KiB
Diff
Subject: [PATCH] add segment-shlvl
|
|
|
|
---
|
|
main.go | 1 +
|
|
segment-shlvl.go | 23 +++++++++++++++++++++++
|
|
2 files changed, 24 insertions(+)
|
|
create mode 100644 segment-shlvl.go
|
|
|
|
diff --git a/main.go b/main.go
|
|
index 02f4a4a..2bd111e 100644
|
|
--- a/main.go
|
|
+++ b/main.go
|
|
@@ -97,6 +97,7 @@ var modules = map[string]func(*powerline) []pwl.Segment{
|
|
"root": segmentRoot,
|
|
"shell-var": segmentShellVar,
|
|
"shenv": segmentShEnv,
|
|
+ "shlvl": segmentShlvl,
|
|
"ssh": segmentSSH,
|
|
"termtitle": segmentTermTitle,
|
|
"terraform-workspace": segmentTerraformWorkspace,
|
|
diff --git a/segment-shlvl.go b/segment-shlvl.go
|
|
new file mode 100644
|
|
index 0000000..d8bba7d
|
|
--- /dev/null
|
|
+++ b/segment-shlvl.go
|
|
@@ -0,0 +1,23 @@
|
|
+package main
|
|
+
|
|
+import (
|
|
+ pwl "github.com/justjanne/powerline-go/powerline"
|
|
+ "os"
|
|
+ "strconv"
|
|
+)
|
|
+
|
|
+func segmentShlvl(p *powerline) []pwl.Segment {
|
|
+
|
|
+ level, _ := os.LookupEnv("SHLVL")
|
|
+ leveli, err := strconv.Atoi(level)
|
|
+
|
|
+ if err != nil || leveli < 1 {
|
|
+ return []pwl.Segment{}
|
|
+ }
|
|
+ return []pwl.Segment{{
|
|
+ Name: "shlvl",
|
|
+ Content: level,
|
|
+ Foreground: 231,
|
|
+ Background: 55,
|
|
+ }}
|
|
+}
|