1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15

diff --git a/DOCS.md b/DOCS.md
index 5d70838b0e264580f4e2fa774fac53f099709e4c..638b13b29ee46ace88a0eb32b943fab675c2473e 100644
--- a/DOCS.md
+++ b/DOCS.md
@@ -15,6 +15,10 @@ **NOTE:** The template.toml file will be expanded, though not with the full power of the template itself.  
 The template.toml file will only expand environment variables with syntax `$USER` or `${USER}`.  
 For full documentation on the syntax, see [os.ExpandEnv](https://golang.org/pkg/os/#ExpandEnv).
 
+When using the `--defaults` flag, no prompts will be shown and only default values will be used.  
+As another alternative, any environment variable that matches a key will bypass the prompt.  
+For example, `author` would have the corresponding environment variable `TMPL_VAR_AUTHOR`.
+
 ```toml
 # Key-value pairs can be simple
 # The user will receive a basic prompt asking them to fill out the variable

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

diff --git a/registry/prompt.go b/registry/prompt.go
index 62317c2a2a984efbe5413f2e2a8e93b581c12d22..d0d34cfa2852248c909e09c5b547608325068458 100644
--- a/registry/prompt.go
+++ b/registry/prompt.go
@@ -6,6 +6,7 @@ 	"io/ioutil"
 	"os"
 	"path/filepath"
 	"sort"
+	"strings"
 	"text/template"
 
 	"github.com/AlecAivazis/survey/v2"
@@ -67,15 +68,22 @@ 		}
 		prompts[idx] = p
 	}
 
-	// Return early if we only want defaults
-	if defaults {
-		return prompts, nil
-	}
-
 	// Sort the prompts so they are consistent
 	sort.Sort(prompts)
 
 	for idx, prompt := range prompts {
+		// Check for env variable
+		if e, ok := os.LookupEnv(fmt.Sprintf("TMPL_VAR_%s", strings.ToUpper(prompt.Key))); ok {
+			prompts[idx].Value = e
+			continue
+		}
+
+		// Check if we are using defaults
+		if defaults {
+			prompts[idx].Value = prompt.Default
+			continue
+		}
+
 		var p survey.Prompt
 		switch t := prompt.Default.(type) {
 		case []string:
@@ -119,11 +127,7 @@ // ToMap converts a slice to templatePrompt into a suitable template context
 func (t templatePrompts) ToMap() map[string]interface{} {
 	m := make(map[string]interface{})
 	for _, p := range t {
-		if p.Value != nil {
-			m[p.Key] = p.Value
-			continue
-		}
-		m[p.Key] = p.Default
+		m[p.Key] = p.Value
 	}
 	return m
 }

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

diff --git a/registry/registry_test.go b/registry/registry_test.go
index 5b3e217dc851a7ced6a1abf235923a6fd5df6f67..8adc7ca075829844fd7923247675985c85ee096b 100644
--- a/registry/registry_test.go
+++ b/registry/registry_test.go
@@ -17,7 +17,7 @@ )
 
 func TestMain(m *testing.M) {
 	var err error
-	destDir, err = ioutil.TempDir(os.TempDir(), "tmpl")
+	destDir, err = ioutil.TempDir(os.TempDir(), "tmpl-dest")
 	if err != nil {
 		panic(err)
 	}
@@ -75,7 +75,7 @@ }
 
 func setupTemplate() {
 	var err error
-	tmplDir, err = ioutil.TempDir(os.TempDir(), "tmpl")
+	tmplDir, err = ioutil.TempDir(os.TempDir(), "tmpl-setup")
 	if err != nil {
 		panic(err)
 	}
@@ -122,7 +122,7 @@ }
 
 func setupRegistry() {
 	var err error
-	regDir, err = ioutil.TempDir(os.TempDir(), "tmpl")
+	regDir, err = ioutil.TempDir(os.TempDir(), "tmpl-reg")
 	if err != nil {
 		panic(err)
 	}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

diff --git a/registry/template.go b/registry/template.go
index 9f37184d2f161d27f97db5dc7d0885a8a879f813..938eff00500886ee1f3a222fd07cb432318d40b6 100644
--- a/registry/template.go
+++ b/registry/template.go
@@ -67,7 +67,7 @@ 		if err != nil {
 			return err
 		}
 
-		newDest := strings.TrimPrefix(walkPath, base+"/")
+		newDest := strings.TrimPrefix(walkPath, base+string(filepath.Separator))
 		newDest = filepath.Join(dest, newDest)
 
 		tmplDest, err := template.New("dest").Funcs(funcs).Parse(newDest)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

diff --git a/registry/template_test.go b/registry/template_test.go
index 019d3b15d71b62d03fae59c1d96ddab3d90a06e6..dd8c317c860422481d6a0c7b3ff6e995b12fc3ce 100644
--- a/registry/template_test.go
+++ b/registry/template_test.go
@@ -8,7 +8,7 @@ 	"testing"
 )
 
 var (
-	tmplContents = `{{title name}} {{if .bool}}{{.year}}{{end}}`
+	tmplContents = `{{title name}} (@{{username}}) {{if .bool}}{{.year}}{{end}}`
 	tmplTemplate = `
 name = "john olheiser"
 
@@ -20,14 +20,21 @@ default = "pkg"
 
 [bool]
 default = true
+
+[username]
+default = "username"
 `
-	tmplGold    = "John Olheiser 2020"
+	tmplGold    = "John Olheiser (@jolheiser) 2020"
 	tmplNewGold = "DO NOT OVERWRITE!"
 )
 
 func testExecute(t *testing.T) {
 	// Set environment variable
 	if err := os.Setenv("TMPL_TEST", "2020"); err != nil {
+		t.Logf("could not set environment: %v", err)
+		t.FailNow()
+	}
+	if err := os.Setenv("TMPL_VAR_USERNAME", "jolheiser"); err != nil {
 		t.Logf("could not set environment: %v", err)
 		t.FailNow()
 	}