Fixing issue with #if being eaten in collection expression (#1120)

closes #1119
belav · Jan 15, 2024 · 102c6e5 · 102c6e5
1 parent 207353a
commit 102c6e5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/Src/CSharpier.Tests/FormattingTests/TestFiles/cs/CollectionExpressions.test b/Src/CSharpier.Tests/FormattingTests/TestFiles/cs/CollectionExpressions.test
@@ -79,6 +79,24 @@ CallMethod(
     ]
 );
 
+int[] someArray =
+[
+    1
+#if DEBUG
+    ,
+    2
+#endif
+];
+
+int[] someArray =
+[
+    1
+#if !DEBUG
+    ,
+    2
+#endif
+];
+
 class MyClass
 {
     private readonly List<string> _items;

diff --git a/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/CollectionExpression.cs b/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/CollectionExpression.cs
@@ -43,7 +43,7 @@ or ExpressionStatementSyntax
                     ? Doc.HardLine
                     : Doc.IfBreak(Doc.Line, Doc.Null)
                 : Doc.Null,
-            node.CloseBracketToken.LeadingTrivia.Any(o => o.IsComment())
+            node.CloseBracketToken.LeadingTrivia.Any(o => o.IsComment() || o.IsDirective)
                 ? Doc.Concat(
                     Doc.Indent(Token.PrintLeadingTrivia(node.CloseBracketToken, context)),
                     Doc.HardLine