AI Tiered Plans: redirect user back to editor after upgrade

[lhkowalski]

When the upgrade process started on the block editor, while editing some post or page, we should redirect the user back to the same post after the upgrade is finished.

[lhkowalski]

Going to take a look on this one.

[lhkowalski]

I did an audit on the current behavior:

• On Simple sites, we set the redirect_to on the checkout URL, and it contains the current post editing URL, but on the wp-admin format instead of the Calypso format, what can confuse some users
  • the URL set does not work since the Post ID and action parameters are lost in the redirect, so the user ends up on the list of posts of the wp-admin panel
  • unsaved changes are saved before the redirect
  • on empty, unsaved posts, the redirect_to URL is set as the "new post URL" (correct) but for the wp-admin context (not ideal)
  • non-empty, usaved posts are saved before the redirect, but the redirect_to URL is still the same "new post URL" (could be the URL for the current post)
  • redirect_to parameter is set here: https://github.com/Automattic/jetpack/blob/trunk/projects/plugins/jetpack/extensions/blocks/ai-assistant/hooks/use-ai-checkout/index.ts#L25

---

• On Jetpack sites, since there is a product interstitial page between the editor and the checkout page, the info about the post being edited is lost: we send the user to the interstitial without any extra info, then send they to the checkout URL with redirect_to set to the My Jetpack page
  • unsaved changes are saveed before the redirect

---

The things we may need to do:

• For Simple sites:
  • fix the redirect_to from Simple sites, to use the Calypso version OR to at least encode the URL so we don't lose the extra parameters and send the user to the right post instead of the list of posts
• For Jetpack sites:
  • introduce a redirect_to or referer parameter on the product interstitials of My Jetpack, that would be used as the redirect_to paramenter of the checkout URL
  • change the upgrade button on the editor to start adding this parameter when linking to the product interstitial, so it gets forwarded

I will start with Jetpack sites, then move to simple sites after it.