Since the formatter supports this now, seems like a good feature for the VSCode extension to support!

Started hacking away at this and then saw the README said to open an issue first, so this is me formally asking: is it cool if I send a PR for this?