Is there a feature rich Wayland WM? I really miss the customisability of AwesomeWM since I moved to Sway.

Nothing quite like AwesomeWM yet, but here's two I'm aware of that have some interesting features:

https://github.com/WayfireWM/wayfire/wiki

https://github.com/riverwm/river

(I'm also on Sway, FWIW)