[prism] Fix splitting logic for aref grouping

fixtures/small/aref_in_call_chain_actual.rb

@@ -0,0 +1,29 @@
+assert_equal(
+  1,
+  T.cast(T.must(pages[1]&.elements)[0], Name::Spacing::To::Extend::CodeQuality::AcrossManyLines).table[
+    :line_items
+  ]
+    .length
+)
+
+foo.bar[some_really_long_index][another_really_long_index]
+  .baz
+  .qux
+
+result.data[
+  :first_key,
+  :second_key
+]
+  .transform
+  .process
+
+outer_method(
+  inner.call[0]
+    .chain
+    .another
+)
+
+matrix[row][col]
+  .value
+  .to_s
+  .upcase

fixtures/small/aref_in_call_chain_expected.rb

@@ -0,0 +1,29 @@
+assert_equal(
+  1,
+  T.cast(T.must(pages[1]&.elements)[0], Name::Spacing::To::Extend::CodeQuality::AcrossManyLines).table[
+    :line_items
+  ]
+    .length
+)
+
+foo.bar[some_really_long_index][another_really_long_index]
+  .baz
+  .qux
+
+result.data[
+  :first_key,
+  :second_key
+]
+  .transform
+  .process
+
+outer_method(
+  inner.call[0]
+    .chain
+    .another
+)
+
+matrix[row][col]
+  .value
+  .to_s
+  .upcase

librubyfmt/src/format_prism.rs

@@ -3037,7 +3037,18 @@ fn split_node_into_call_chains<'src>(node: prism::Node<'src>) -> Vec<Vec<prism::
             let is_aref: bool = !node_is_dot_call;
 
             if is_aref && seen_dot_call && has_dot_after[i] {
-                split_before.push(i);
+                let mut split_idx = i;
+                while split_idx < elements.len() {
+                    if let Some(call_node) = elements[split_idx].as_call_node()
+                        && call_node.call_operator_loc().is_none()
+                    {
+                        // Still an aref, advance past it
+                        split_idx += 1;
+                        continue;
+                    }
+                    break;
+                }
+                split_before.push(split_idx);
                 seen_dot_call = false;
                 continue;
             }