The `sway` window manager is an attempt to clone i3 for Wayland. If you do have to switch, that part at least, shouldn't be too painful.