You're already storing the code so why not just join the invite with the email and when the user logs in with an email flagged with an invite the user's state is already known?