added env module

reujab · Jul 20, 2017 · 99974d0 · 99974d0
1 parent 3cd89db
commit 99974d0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/env.go b/env.go
@@ -0,0 +1,15 @@
+package main
+
+import (
+	"os"
+
+	. "github.com/reujab/bronze/types"
+)
+
+func envSegment(segment *Segment, args []string) {
+	if len(args) != 1 {
+		dief("env: expected exactly one argument")
+	}
+
+	segment.Value = os.Getenv(args[0])
+}
diff --git a/modules.go b/modules.go
@@ -22,6 +22,8 @@ func handleModule(module string, segment *Segment, args []string) {
 		cmdTimeSegment(segment)
 	case "time":
 		timeSegment(segment)
+	case "env":
+		envSegment(segment, args)
 	case "plugin":
 		pluginSegment(segment, args)
 	default: