From f81035dbeb2d9b2dd47f1dd4732b106c081212ff Mon Sep 17 00:00:00 2001
From: tenntenn <uedatakuya@gmail.com>
Date: Mon, 23 Nov 2020 19:56:34 +0900
Subject: [PATCH] Fix IsGeneratedFile

---
 file.go      | 4 ++--
 file_test.go | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/file.go b/file.go
index f7d0e9b..b9b2955 100644
--- a/file.go
+++ b/file.go
@@ -23,8 +23,8 @@ var genCommentRegexp = regexp.MustCompile(`^// Code generated .* DO NOT EDIT\.$`
 // IsGeneratedFile reports whether the file has been generated automatically.
 // If file is nil, IsGeneratedFile will return false.
 func IsGeneratedFile(file *ast.File) bool {
-	if file == nil || file.Doc == nil {
+	if file == nil || len(file.Comments) == 0 {
 		return false
 	}
-	return genCommentRegexp.MatchString(file.Doc.List[0].Text)
+	return genCommentRegexp.MatchString(file.Comments[0].List[0].Text)
 }
diff --git a/file_test.go b/file_test.go
index 341e7c5..e699025 100644
--- a/file_test.go
+++ b/file_test.go
@@ -20,6 +20,7 @@ func TestIsGeneratedFile(t *testing.T) {
 		"true":  {"// Code generated by test; DO NOT EDIT.", true},
 		"false": {"//Code generated by test; DO NOT EDIT.", false},
 		"empty": {"", false},
+		"blank": {"// Code generated by test; DO NOT EDIT.\n", true},
 	}
 
 	for name, tt := range cases {