feat: show generated result

close #8
wolfogre · May 9, 2020 · bb480e0 · bb480e0
1 parent 9fc66cb
commit bb480e0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/cmd/gtag/main.go b/cmd/gtag/main.go
@@ -31,11 +31,15 @@ func main() {
 		tags = strings.Split(*Tags, ",")
 	}
 
-	_, err := gtag.Generate(context.Background(), dir, types, tags)
+	result, err := gtag.Generate(context.Background(), dir, types, tags)
 	if err != nil {
 		fmt.Println(err)
 		os.Exit(1)
 	}
+
+	for _, v := range result {
+		fmt.Printf("generated %s -> %s\n", v.Source, v.Output)
+	}
 }
 
 func printUsages() {

diff --git a/internal/gtag/gtag.go b/internal/gtag/gtag.go
@@ -17,6 +17,7 @@ import (
 
 type GenerateResult struct {
 	Content []byte
+	Source  string
 	Output  string
 }
 
@@ -115,6 +116,7 @@ func generateFile(ctx context.Context, cmd, file string, types []string, tags []
 
 	ret := &GenerateResult{
 		Content: src,
+		Source:  file,
 		Output:  strings.TrimSuffix(file, ".go") + "_tag.go",
 	}