The sway window manager has this feature built in and it's wonderful. I use it all the time.