Handle multiline comments better

closes #580
belav · Feb 5, 2022 · a4f1b40 · a4f1b40
1 parent 1518657
commit a4f1b40
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/Src/CSharpier/DocPrinter/DocPrinter.cs b/Src/CSharpier/DocPrinter/DocPrinter.cs
@@ -131,7 +131,19 @@ private void ProcessNextCommand()
                     this.Output.Append(this.EndOfLine);
                 }
 
-                this.Output.Append(indent.Value).Append(leadingComment.Comment);
+                // TODO do trailing comments need to do the same thing?
+                var stringReader = new StringReader(leadingComment.Comment);
+                var line = stringReader.ReadLine();
+                while (line != null)
+                {
+                    this.Output.Append(indent.Value).Append(line.Trim());
+                    line = stringReader.ReadLine();
+                    if (line != null)
+                    {
+                        this.Output.Append(this.EndOfLine);
+                    }
+                }
+
                 this.CurrentWidth = indent.Length;
                 this.NewLineNextStringValue = false;
                 this.SkipNextNewLine = false;

diff --git a/Src/CSharpier/DocTypes/LeadingComment.cs b/Src/CSharpier/DocTypes/LeadingComment.cs
@@ -2,6 +2,6 @@ namespace CSharpier.DocTypes;
 
 internal class LeadingComment : Doc
 {
-    public CommentType Type { get; set; }
-    public string Comment { get; set; } = string.Empty;
+    public CommentType Type { get; init; }
+    public string Comment { get; init; } = string.Empty;
 }