A more convenient window creation method

[archiif]

Right now, when you call vterm-extra-edit-command-in-new-buffer, the new buffer it creates will potentially occupy up an existing window which is fine, but the problem is that once you do a C-c C-c that window will be closed along with the editing buffer.
Is it possible to have a nicer window behavior? Maybe something like org-edit-src-code?

Edit: My dirty hack to fix this for now is by advising vterm-extra-edit-done with winner-undo:

(advice-add #'vterm-extra-edit-done :after #'winner-undo))

[Sbozzolo]

This is certainly a good point, but I am not sure how to implement it at the moment. I would like to see an example that is not as complicated as org-mode or helm.

(advice-add #'vterm-extra-edit-done :after #'winner-undo))

This would work in most cases, but what if you change the window configuration? E.g., you open another buffer to check documentaiton.