I'm working on something similar (not as far along), a vscode extension that has ideas like functions, parameters, etc as first class ideas (so press f to select a function) and also does things like adding a newline and a space where appropriate during insert actions.
Here's a short video demo if anyone is interested: https://youtu.be/owuuFxDBQh4