google_redirect

ssr/src/page/google_redirect.rs

@@ -1,10 +1,13 @@
+use candid::Principal;
+use codee::string::FromToStringCodec;
 use leptos::*;
 use leptos_router::*;
+use leptos_use::use_cookie;
 use openidconnect::CsrfToken;
 use serde::{Deserialize, Serialize};
 use server_fn::codec::{GetUrl, Json};
 
-use crate::{component::loading::Loading, utils::route::go_to_root};
+use crate::{component::loading::Loading, consts::USER_PRINCIPAL_STORE, utils::route::go_to_root};
 use yral_types::delegated_identity::DelegatedIdentityWire;
 
 pub type GoogleAuthMessage = Result<DelegatedIdentityWire, String>;
@@ -126,11 +129,25 @@ pub fn IdentitySender(identity_res: GoogleAuthMessage) -> impl IntoView {
             let win = window();
             let origin = win.origin();
             let opener = win.opener().unwrap();
+            let origin = win.origin();
+            let opener = win.opener().unwrap();
             if opener.is_null() {
                 go_to_root();
+                if let Some(principal) =
+                    use_cookie::<Principal, FromToStringCodec>(USER_PRINCIPAL_STORE)
+                        .0
+                        .get()
+                {
+                    let new_path = format!("/profile/{}/tokens", principal);
+                    use_navigate()(&new_path, Default::default());
+                } else {
+                    go_to_root();
+                }
             }
             let opener = Window::from(opener);
             let msg = serde_json::to_string(&_id).unwrap();
+            let opener = Window::from(opener);
+            let msg = serde_json::to_string(&_id).unwrap();
             _ = opener.post_message(&msg.into(), &origin);
         }
     });