update

Author: admpub

cmd.go

@@ -175,6 +175,14 @@ func ParseArgs(command string) (params []string) {
 	return
 }
 
+func ParseEnvVar(v string) string {
+	return envOS.ReplaceAllStringFunc(v, getEnv)
+}
+
+func ParseWindowsEnvVar(v string) string {
+	return envWin.ReplaceAllStringFunc(v, getWinEnv)
+}
+
 func getWinEnv(s string) string {
 	s = strings.TrimPrefix(s, `{%`)
 	s = strings.TrimSuffix(s, `%}`)

cmd_test.go

@@ -0,0 +1,20 @@
+package com
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestParseEnvVar(t *testing.T) {
+	os.Setenv(`TESTENV`, `1`)
+	v := ParseEnvVar(`ab{$TESTENV}c`)
+	assert.Equal(t, `ab1c`, v)
+}
+
+func TestParseWindowsEnvVar(t *testing.T) {
+	os.Setenv(`TESTENV`, `2`)
+	v := ParseWindowsEnvVar(`ab{%TESTENV%}c`)
+	assert.Equal(t, `ab2c`, v)
+}