Merge pull request #15 from lyz-code/fix/lines-with-comments

fix: correct indentation of parent lists with comments

src/yamlfix/services.py

@@ -111,6 +111,7 @@ def _fix_top_level_lists(source_code: str) -> str:
 
     ```yaml
     ---
+    # Comment
     - item 1
     - item 2
     ```
@@ -119,6 +120,7 @@ def _fix_top_level_lists(source_code: str) -> str:
 
     ```yaml
     ---
+    # Comment
       - item 1
       - item 2
     ```
@@ -138,7 +140,7 @@ def _fix_top_level_lists(source_code: str) -> str:
     for line in source_lines:
 
         # Skip the heading and first empty lines
-        if line in ["---", ""]:
+        if re.match(r"^(---|#.*|)$", line):
             fixed_source_lines.append(line)
             continue
 

tests/unit/test_services.py

@@ -108,6 +108,21 @@ def test_fix_code_preserves_comments() -> None:
     assert result == source
 
 
+def test_fix_code_respects_parent_lists_with_comments() -> None:
+    """Do not indent lists at the first level even if there is a comment."""
+    source = dedent(
+        """\
+        ---
+        # Comment
+        - item1
+        - item2"""
+    )
+
+    result = fix_code(source)
+
+    assert result == source
+
+
 def test_fix_code_removes_extra_apostrophes() -> None:
     """Remove not needed apostrophes."""
     source = dedent(