feat: add basic comment context formatting

sway-fmt-v2/src/formatter/mod.rs

@@ -627,8 +627,6 @@ pub struct Foo { // Here is a comment
 // This is a comment, for this one to be placed correctly we need to have Module visitor implemented
 pub struct Foo { // Here is a comment
 
-
-
     // Trying some ASCII art
     baz: u64,
     bazzz: u64,//  ________ ___  ___  _______   ___               ___       ________  ________  ________
@@ -988,12 +986,36 @@ fn main() {}
         let sway_code_to_format = r#"script;
 
 // use statements
-use std::*;"#;
+use std::*;
+
+fn main() {
+    // Array of integers with type ascription
+let array_of_integers: [u8;
+5] = [1, 2, 3, 4, 5];
+
+
+    // Array of strings
+    let array_of_strings = [ "Bob", "Jan", "Ron"];
+}
+"#;
 
         let correct_sway_code = r#"script;
 
 // use statements
 use std::*;
+
+fn main() {
+    // Array of integers with type ascription
+    let array_of_integers: [u8; 5] = [1, 2, 3, 4, 5];
+
+
+    // Array of strings
+    let array_of_strings = [
+        "Bob",
+        "Jan",
+        "Ron",
+    ];
+}
 "#;
 
         let mut formatter = Formatter::default();

sway-fmt-v2/src/utils/map/comments.rs

@@ -180,6 +180,22 @@ fn get_comments_between_spans(
     comments_with_context
 }
 
+fn format_context(context: &str, threshold: usize) -> String {
+    let mut remaining_newlines = threshold;
+    let mut formatted_context = String::new();
+    for char in context.chars() {
+        if char == '\n' {
+            if remaining_newlines > 0 {
+                formatted_context.push('\n');
+                remaining_newlines -= 1;
+            }
+        } else {
+            formatted_context.push(char);
+        }
+    }
+    formatted_context
+}
+
 /// Inserts after given span and returns the offset. While inserting comments this also inserts Context of the comments so that the alignment whitespaces/newlines are intact
 fn insert_after_span(
     from: &ByteSpan,
@@ -195,7 +211,7 @@ fn insert_after_span(
         write!(
             comment_str,
             "{}{}",
-            comment_context,
+            format_context(comment_context, 2),
             &format_comment(comment_value)
         )?;
     }