cmd/godefs: support comment

Signed-off-by: Koichi Shiraishi <zchee.io@gmail.com>

Author: zchee

cmd/godefs/ast.go

@@ -54,7 +54,7 @@ func (f *File) ParseGo(abspath string, src []byte) {
 	// In cgo mode, we ignore ast2 and just apply edits directly
 	// the text behind ast1. In godefs mode we modify and print ast2.
 	ast1 := parse(abspath, src, parser.SkipObjectResolution|parser.ParseComments)
-	ast2 := parse(abspath, src, parser.SkipObjectResolution)
+	ast2 := parse(abspath, src, parser.SkipObjectResolution|parser.ParseComments)
 
 	f.Package = ast1.Name.Name
 	f.Name = make(map[string]*Name)
@@ -115,6 +115,7 @@ func (f *File) ParseGo(abspath string, src []byte) {
 		error_(ast1.Package, `cannot find import "C"`)
 	}
 
+	comments := []*ast.CommentGroup{}
 	// In ast2, strip the import "C" line.
 	if *godefs {
 		w := 0
@@ -141,6 +142,7 @@ func (f *File) ParseGo(abspath string, src []byte) {
 			w++
 		}
 		ast2.Decls = ast2.Decls[0:w]
+		comments = ast2.Comments[w:]
 	} else {
 		for _, decl := range ast2.Decls {
 			d, ok := decl.(*ast.GenDecl)
@@ -157,6 +159,7 @@ func (f *File) ParseGo(abspath string, src []byte) {
 			}
 		}
 	}
+	ast2.Comments = comments
 
 	// Accumulate pointers to uses of C.x.
 	if f.Ref == nil {

cmd/godefs/ast_go1.go

@@ -2,6 +2,8 @@
 // Use of this source code is governed by a BSD-style
 // license that can be found in the LICENSE file.
 
+//go:build !go1.18
+
 package main
 
 import (

cmd/godefs/ast_go118.go

@@ -2,6 +2,8 @@
 // Use of this source code is governed by a BSD-style
 // license that can be found in the LICENSE file.
 
+//go:build go1.18
+
 package main
 
 import (