Hi, looking for help to perform ir manipulation th...

# compiler

[Justin Tullgren]

Hi, looking for help to perform ir manipulation that results in “around” advice, in a function body.

fun() {
  statement()
  statement()
}

to

fun() {
  begin()
  statement()
  statement()
  end()
}

[Justin Tullgren]

I can’t tell if I need to modify the IrBody or the IrFunction call…

[Javier]

you need to modify the body based on the code you shared

[Justin Tullgren]

so thats visitBody or do it at the visitSimpleFunction call? thats where I am struggling

[Justin Tullgren]

IE return a new body in visitBody with my added statements or return the IrFunction with a modified body and don’t visit them

[Javier]

I don’t know, I use a Tree with all the file children so I can filter to find any function by name for example. I haven’t used all visitor APIs

[Justin Tullgren]

ok i’ll keep messing around

[Javier]

if you already have the function, you can get the body and modify the statements by adding those two elements in the positions you want as statements is a mutable list

[Justin Tullgren]

ya thats what I am doing now. okay thanks!

[Javier]

Here I am replacing a statement, you could do the same but instead just put yours

https://github.com/JavierSegoviaCordoba/mokoki/blob/refactor-compiler-plugin/mokoki-compiler/main/kotlin/com/javiersc/mokoki/compiler/IrMokokiExtension.kt