I have something very similar that works in neovim buffers for those interested (unfortunately mostly AI written, yet to be reviewed, but is being used frequently by me)

https://github.com/mktip/bsh.nvim