Persist query parameters on ai and custom roadmap page

src/components/AuthenticationFlow/GitHubButton.tsx

@@ -101,7 +101,7 @@ export function GitHubButton(props: GitHubButtonProps) {
     // For non authentication pages, we want to redirect back to the page
     // the user was on before they clicked the social login button
     if (!['/login', '/signup'].includes(window.location.pathname)) {
-      const pagePath = ['/respond-invite', '/befriend'].includes(
+      const pagePath = ['/respond-invite', '/befriend', '/r', '/ai'].includes(
         window.location.pathname,
       )
         ? window.location.pathname + window.location.search

src/components/AuthenticationFlow/GoogleButton.tsx

@@ -97,7 +97,7 @@ export function GoogleButton(props: GoogleButtonProps) {
         // For non authentication pages, we want to redirect back to the page
         // the user was on before they clicked the social login button
         if (!['/login', '/signup'].includes(window.location.pathname)) {
-          const pagePath = ['/respond-invite', '/befriend'].includes(
+          const pagePath = ['/respond-invite', '/befriend', '/r', '/ai'].includes(
             window.location.pathname,
           )
             ? window.location.pathname + window.location.search

src/components/AuthenticationFlow/LinkedInButton.tsx

@@ -97,7 +97,7 @@ export function LinkedInButton(props: LinkedInButtonProps) {
         // For non authentication pages, we want to redirect back to the page
         // the user was on before they clicked the social login button
         if (!['/login', '/signup'].includes(window.location.pathname)) {
-          const pagePath = ['/respond-invite', '/befriend'].includes(
+          const pagePath = ['/respond-invite', '/befriend', '/r', '/ai'].includes(
             window.location.pathname,
           )
             ? window.location.pathname + window.location.search