fix: fix aligning comments

g-plane · Aug 25, 2024 · 2b34c7b · 2b34c7b
1 parent b3fde89
commit 2b34c7b
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/malva/src/doc_gen/comment.rs b/malva/src/doc_gen/comment.rs
@@ -80,11 +80,25 @@ pub(crate) fn format_comment<'s>(comment: &Comment<'s>, ctx: &Ctx<'_, 's>) -> Do
 }
 
 pub(super) fn reflow<'s>(comment: &Comment<'s>, ctx: &Ctx<'_, 's>) -> Vec<Doc<'s>> {
-    let col = ctx
-        .line_bounds
-        .get_line_col(comment.span.start)
-        .1
-        .saturating_sub(1);
+    let col = comment
+        .content
+        .lines()
+        .skip(1)
+        .filter(|line| !line.trim().is_empty())
+        .map(|line| {
+            line.as_bytes()
+                .iter()
+                .take_while(|byte| byte.is_ascii_whitespace())
+                .count()
+        })
+        .min()
+        .unwrap_or_default()
+        .min(
+            ctx.line_bounds
+                .get_line_col(comment.span.start)
+                .1
+                .saturating_sub(1),
+        );
     let mut docs = Vec::with_capacity(2);
     let mut lines = comment.content.split('\n').enumerate().peekable();
     while let Some((i, line)) = lines.next() {