change code gen to create writer after gen to avoid errors when writing file in inspected package

Author: francoispqt

gojay/main.go

@@ -82,19 +82,18 @@ func getOutput() (stringWriter, error) {
 	return os.Stdout, nil
 }
 
-func parseArgs() (p string, t []string, o stringWriter, err error) {
+func parseArgs() (p string, t []string, err error) {
 	flag.Parse()
 	p, err = getPath()
 	if err != nil {
-		return p, t, o, err
+		return p, t, err
 	}
 	t = getTypes()
-	o, err = getOutput()
-	return p, t, o, err
+	return p, t, err
 }
 
 func main() {
-	p, t, o, err := parseArgs()
+	p, t, err := parseArgs()
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -111,6 +110,12 @@ func main() {
 		log.Fatal(err)
 		return
 	}
+	var o stringWriter
+	o, err = getOutput()
+	if err != nil {
+		log.Fatal(err)
+		return
+	}
 	// write content to output
 	o.WriteString(g.b.String())
 }