From 3047fbcab22299f4575e1adf09f6dc9f31d62ea7 Mon Sep 17 00:00:00 2001 From: Christoph Blecker Date: Mon, 25 Sep 2017 12:36:24 -0700 Subject: Update generator to use WHAT variable --- generator/app.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'generator/app.go') diff --git a/generator/app.go b/generator/app.go index 99336c0d..b44eadcd 100644 --- a/generator/app.go +++ b/generator/app.go @@ -200,14 +200,14 @@ func writeCustomContentBlock(f *os.File, content string) { func createGroupReadme(groups []Group, prefix string) error { // figure out if the user wants to generate one group var selectedGroupName *string - if envVal, ok := os.LookupEnv(strings.ToUpper(prefix)); ok { + if envVal, ok := os.LookupEnv("WHAT"); ok { selectedGroupName = &envVal } for _, group := range groups { group.Dir = group.DirName(prefix) // skip generation if the user specified only one group - if selectedGroupName != nil && *selectedGroupName != group.Dir { + if selectedGroupName != nil && strings.HasSuffix(group.Dir, *selectedGroupName) == false { fmt.Printf("Skipping %s/README.md\n", group.Dir) continue } -- cgit v1.2.3