Add flag to support leading comments separated by newline. (#73)

wsl.go

@@ -100,6 +100,16 @@ type Configuration struct {
 	// AllowTrailingComment will allow blocks to end with comments.
 	AllowTrailingComment bool
 
+	// AllowSeparatedLeadingComment will allow multiple comments in the
+	// beginning of a block separated with newline. Example:
+	//  func () {
+	//		// Comment one
+	//
+	//		// Comment two
+	// 		fmt.Println("x")
+	//  }
+	AllowSeparatedLeadingComment bool
+
 	// AllowCuddleDeclaration will allow multiple var/declaration statements to
 	// be cuddled. This defaults to false but setting it to true will enable the
 	// following example:
@@ -941,8 +951,9 @@ func (p *Processor) findLeadingAndTrailingWhitespaces(ident *ast.Ident, stmt, ne
 	}
 
 	var (
-		firstStatement = blockStatements[0]
-		lastStatement  = blockStatements[len(blockStatements)-1]
+		firstStatement    = blockStatements[0]
+		lastStatement     = blockStatements[len(blockStatements)-1]
+		seenCommentGroups = 0
 	)
 
 	// Get the comment related to the first statement, we do allow commends in
@@ -962,13 +973,24 @@ func (p *Processor) findLeadingAndTrailingWhitespaces(ident *ast.Ident, stmt, ne
 				break
 			}
 
+			// We store number of seen comment groups because we allow multiple
+			// groups with a newline between them.
+			seenCommentGroups++
+
 			// Support both /* multiline */ and //single line comments
 			for _, c := range commentGroup.List {
 				allowedLinesBeforeFirstStatement += len(strings.Split(c.Text, "\n"))
 			}
 		}
 	}
 
+	// If we have multiple groups, add support for newline between each group.
+	if p.config.AllowSeparatedLeadingComment {
+		if seenCommentGroups > 1 {
+			allowedLinesBeforeFirstStatement += seenCommentGroups - 1
+		}
+	}
+
 	if p.nodeStart(firstStatement) != blockStartLine+allowedLinesBeforeFirstStatement {
 		p.addError(
 			blockStartPos,

wsl_test.go

@@ -1621,6 +1621,59 @@ func TestWithConfig(t *testing.T) {
 			}`),
 			expectedErrorStrings: []string{},
 		},
+		{
+			description: "allow separated leading comment",
+			customConfig: &Configuration{
+				AllowSeparatedLeadingComment: true,
+			},
+			code: []byte(`package main
+
+			func main() {
+				// These blocks should not generate error
+				func () {
+					// Comment
+
+					// Comment
+					fmt.Println("Hello, World")
+				}
+
+				func () {
+					/*
+						Multiline
+					*/
+
+					/*
+						Multiline
+					*/
+					fmt.Println("Hello, World")
+				}
+
+				func () {
+					/*
+						Multiline
+					*/
+
+					// Comment
+					fmt.Println("Hello, World")
+				}
+
+				func () {
+					// Comment
+
+					/*
+						Multiline
+					*/
+					fmt.Println("Hello, World")
+				}
+
+				func () { // Comment
+					/*
+						Multiline
+					*/
+					fmt.Println("Hello, World")
+				}
+			}`),
+		},
 	}
 
 	for _, tc := range cases {