emacs has a bunch of packages for treesitter based ast motions—might be worth seeing if vim has any analogues.