Playground

I am looking for a way to preserve code formatting after ast-grep replacements. For instance, if I have the following code:

expect_equal(names(x), c('a', 'b'))

expect_equal(
  names(x),
  c('a', 'b')
)

and the following rule:

rule:
  pattern: expect_equal(names($VAR), $VALUES)
    
fix: expect_named($VAR, $VALUES)

then the output is this:

expect_named(x, c('a', 'b'))

expect_named(x, c('a', 'b'))

while I would like it to be this:

expect_named(x, c('a', 'b'))

expect_named(
  x,
  c('a', 'b')
)

Is there a way to do this that wouldn't require me to write the patterns and fixes with this indentation?