Merge pull request #40 from gostaticanalysis/fix-isgenfile

Fix IsGeneratedFile

Author: tenntenn

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)
 }

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 {