From 434df0476dcae718cb35028fe1e8f28d2f2f9e7b Mon Sep 17 00:00:00 2001
From: pagran <pagran@protonmail.com>
Date: Tue, 29 Sep 2020 22:09:39 +0300
Subject: [PATCH] fixed comments cleaning

Added cleanup of the Comment field.
In some cases, the appearance of a comment in a random place
may break the compilation (e.g. cgo and runtime package).

This is safe because the Comment field cannot contain any directives.

Part of #149.
---
 line_obfuscator.go            |  4 ++++
 testdata/scripts/debugdir.txt | 10 +++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/line_obfuscator.go b/line_obfuscator.go
index b345350..9e12801 100644
--- a/line_obfuscator.go
+++ b/line_obfuscator.go
@@ -49,12 +49,16 @@ func clearNodeComments(node ast.Node) {
 	switch n := node.(type) {
 	case *ast.Field:
 		n.Doc = clearCommentGroup(n.Doc)
+		n.Comment = nil
 	case *ast.ImportSpec:
 		n.Doc = clearCommentGroup(n.Doc)
+		n.Comment = nil
 	case *ast.ValueSpec:
 		n.Doc = clearCommentGroup(n.Doc)
+		n.Comment = nil
 	case *ast.TypeSpec:
 		n.Doc = clearCommentGroup(n.Doc)
+		n.Comment = nil
 	case *ast.GenDecl:
 		n.Doc = clearCommentGroup(n.Doc)
 	case *ast.FuncDecl:
diff --git a/testdata/scripts/debugdir.txt b/testdata/scripts/debugdir.txt
index 16b62fe..796489d 100644
--- a/testdata/scripts/debugdir.txt
+++ b/testdata/scripts/debugdir.txt
@@ -4,13 +4,21 @@ garble -debugdir ./test1 build
 exists 'test1/test/main/imported/imported.go' 'test1/main/main.go'
 ! grep ImportedFunc $WORK/test1/test/main/imported/imported.go
 ! grep ImportedFunc $WORK/test1/main/main.go
+! grep 'some comment' $WORK/test1/main/main.go
 
 -- go.mod --
 module test/main
 -- main.go --
 package main
 
-import "test/main/imported"
+import "test/main/imported" // some comment
+
+type someType int // some comment
+var someVar = 0
+
+type someStruct struct {
+    someField int // some comment
+}
 
 func main() {
 	imported.ImportedFunc()