feat: some more edge cases with anonymous nodes

max397574 · Jan 8, 2023 · e65cad1 · e65cad1
1 parent ac5effd
commit e65cad1
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -74,6 +74,14 @@ fn main() {
             output.push_str(&"  ".repeat(indent_level));
         }
 
+        if cursor.node().kind() == "identifier"
+            && check_parent("anonymous_node", cursor.node())
+            && !check_parent("list", cursor.node().parent().unwrap())
+        {
+            output.push('\n');
+            output.push_str(&"  ".repeat(indent_level));
+        }
+
         if check_parent("parameters", cursor.node()) {
             output.push(' ')
         }
@@ -91,6 +99,12 @@ fn main() {
         if cursor.node().kind() == "anonymous_node" && check_parent("list", cursor.node()) {
             output.push_str(cursor.node().utf8_text(source_code.as_bytes()).unwrap());
         }
+        if cursor.node().kind() == "identifier"
+            && check_parent("anonymous_node", cursor.node())
+            && !check_parent("list", cursor.node().parent().unwrap())
+        {
+            output.push_str(cursor.node().utf8_text(source_code.as_bytes()).unwrap());
+        }
         if cursor.node().kind() == ":" {
             output.push(' ');
         }