I think think this should be completely possible with replace ranges. I don’t think adding lines creates any issues and it would basically just require that LSP added a few newlines to inlay diagnostics.
And regarding folding I did get it to work with replace ranges. It’s not perfect but it’s something and I’m sure it could be improved