fixes

Author: reflog

main.go

@@ -87,7 +87,7 @@ func generateInterface(folder, outputFile, pkgName, structName, ifName, outputTe
 
 	funcs := make([]string, 0)
 	for _, file := range appPkg.Files {
-		// fmt.Printf("parsing %s\n", fset.File(file.Pos()).Name())
+		log.Printf("parsing %s\n", fset.File(file.Pos()).Name())
 		if fset.File(file.Pos()).Name() == outputFile {
 			continue
 		}
@@ -120,14 +120,15 @@ func generateInterface(folder, outputFile, pkgName, structName, ifName, outputTe
 		log.Panic(err)
 	}
 	os.Remove(outputFile)
-	formatted, err := imports.Process(outputFile, out.Bytes(), &imports.Options{})
+	formatted, err := imports.Process(outputFile, out.Bytes(), &imports.Options{Comments: true})
 	if err != nil {
 		log.Panic(err)
 	}
 	err = ioutil.WriteFile(outputFile, formatted, 0644)
 	if err != nil {
 		log.Panic(err)
 	}
+	log.Printf("Written %s successfully", outputFile)
 }
 
 func main() {
@@ -139,6 +140,15 @@ func main() {
 		Use:   "struct2interface",
 		Short: "Extract an interface from a Golang struct",
 		Run: func(cmd *cobra.Command, args []string) {
+
+			if outputTemplateFile != "" {
+				d, err := ioutil.ReadFile(outputTemplateFile)
+				if err != nil {
+					log.Panic(err)
+				}
+				outputTemplate = string(d)
+			}
+
 			generateInterface(folder, outputFile, pkgName, structName, ifName, outputTemplate)
 		},
 	}
@@ -160,12 +170,4 @@ func main() {
 		os.Exit(1)
 	}
 
-	if outputTemplateFile != "" {
-		d, err := ioutil.ReadFile(outputTemplateFile)
-		if err != nil {
-			log.Panic(err)
-		}
-		outputTemplate = string(d)
-	}
-
 }