Set tokens on context (#22)

Author: yurynix

packages/@wix-astro/src/components/login.astro

@@ -283,6 +283,8 @@
   import { authentication, verification, recovery } from '@wix/identity';
   import { getMemberTokensForDirectLogin } from './login-helpers/login-helpers.js';
   import { wixContext } from '@wix/sdk-context';
+  import type { WixClient } from '@wix/sdk';
+  import type { SiteSessionAuth } from "@wix/sdk/auth/site-session";
 
   class LoginForm extends HTMLElement {
     constructor() {
@@ -478,6 +480,9 @@
             response.sessionToken!
           );
 
+          // Update context
+          (wixContext['client'] as WixClient<undefined, ReturnType<typeof SiteSessionAuth>>).auth.setTokens(tokens.tokens);
+
           // Set cookie
           this.setCookie('wixSession', JSON.stringify(tokens), 2);