Adding formatting for spread element (#1179)

belav · web-flow · commit 0983917c12db · 2024-02-18T09:54:27.000-06:00
* Adding formatting for spread element closes #1167 * add spread walker
diff --git a/Src/CSharpier.Tests/FormattingTests/TestFiles/cs/SpreadElements.test b/Src/CSharpier.Tests/FormattingTests/TestFiles/cs/SpreadElements.test
@@ -0,0 +1,21 @@
+int[] someArray = [.. someOtherArray];
+int[] someOtherArray = [.. value1, .. value2, .. value3];
+
+int[] someOtherArray =
+[
+    .. value1________________________________,
+    .. value2________________________________,
+    .. value3________________________________
+];
+
+List<KeyValuePair<string, string>> list =
+[
+    .. attribute
+        .Targets.Select(target =>
+            KeyValuePair.Create(
+                target,
+                context.EntityDefinitions.TryGetValue(target, out var type) ? type.ClassName : null
+            )
+        )
+        .Where(kvp => kvp.Value != null)
+];
diff --git a/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/SpreadElement.cs b/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/SpreadElement.cs
@@ -0,0 +1,13 @@
+namespace CSharpier.SyntaxPrinter.SyntaxNodePrinters;
+
+internal static class SpreadElement
+{
+    public static Doc Print(SpreadElementSyntax node, FormattingContext context)
+    {
+        return Doc.Group(
+            Token.Print(node.OperatorToken, context),
+            " ",
+            Node.Print(node.Expression, context)
+        );
+    }
+}
diff --git a/Src/SyntaxFinder/Program.cs b/Src/SyntaxFinder/Program.cs
@@ -1,4 +1,5 @@
-﻿using Microsoft.CodeAnalysis.CSharp;
+﻿using System.Reflection;
+using Microsoft.CodeAnalysis.CSharp;
 using SyntaxFinder;
 
 var files = Directory.EnumerateFiles(
@@ -47,4 +48,4 @@
     }
 );
 
-// ObjectInitializerWalker.WriteResult();
+typeToRun.GetMethod("WriteResult")?.Invoke(null, null);
diff --git a/Src/SyntaxFinder/SpreadWalker.cs b/Src/SyntaxFinder/SpreadWalker.cs
@@ -0,0 +1,44 @@
+namespace SyntaxFinder;
+
+using System.Collections.Concurrent;
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+
+public class SpreadWalker : CSharpSyntaxWalker
+{
+    private static int total;
+    private static int matching;
+    private readonly string file;
+
+    public SpreadWalker(string file)
+    {
+        this.file = file;
+    }
+
+    public override void VisitSpreadElement(SpreadElementSyntax node)
+    {
+        if (node.Kind() is SyntaxKind.SpreadElement)
+        {
+            this.VisitNode(node);
+        }
+
+        base.VisitSpreadElement(node);
+    }
+
+    private void VisitNode(SpreadElementSyntax node)
+    {
+        Console.WriteLine(node.Parent!.SyntaxTree.GetText().ToString(node.Parent!.FullSpan));
+        Interlocked.Increment(ref total);
+
+        if (node.Expression.GetLeadingTrivia().Any(o => o.Kind() is SyntaxKind.WhitespaceTrivia))
+        {
+            Interlocked.Increment(ref matching);
+        }
+    }
+
+    public static void WriteResult()
+    {
+        ResultWriter.WriteMatching(total, matching);
+    }
+}
