It would be even better if the code was always valid because we only edit the AST itself, like what paredit does for Lisp modes.

No.

(The code being valid s-sexps doesn’t even guarantee it’s syntactically valid after macro expansion, but even if it does this is a bad idea.)