From 4b9bfb8845dc9d720c5d087f69439c9dc4a5c467 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ernesto=20Jime=CC=81nez?= <me@ernesto-jimenez.com>
Date: Sun, 25 Sep 2016 23:00:40 +0100
Subject: [PATCH] run go/format on generated code

---
 _codegen/main.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/_codegen/main.go b/_codegen/main.go
index 328009f..7cc2531 100644
--- a/_codegen/main.go
+++ b/_codegen/main.go
@@ -10,6 +10,7 @@ import (
 	"go/ast"
 	"go/build"
 	"go/doc"
+	"go/format"
 	"go/importer"
 	"go/parser"
 	"go/token"
@@ -77,13 +78,18 @@ func generateCode(importer imports.Importer, funcs []testFunc) error {
 		}
 	}
 
+	code, err := format.Source(buff.Bytes())
+	if err != nil {
+		return err
+	}
+
 	// Write file
 	output, err := outputFile()
 	if err != nil {
 		return err
 	}
 	defer output.Close()
-	_, err = io.Copy(output, buff)
+	_, err = io.Copy(output, bytes.NewReader(code))
 	return err
 }