the ideal solution takes into account which screen you were on and leaves you on it

i used to have a setup like this on i3, using i3-msg to query the state