Set tables workspace selection cookie upon user signup/signin

Author: cezudas

packages/server/api/src/app/authentication/basic/authentication-service.ts

@@ -58,6 +58,7 @@ export const authenticationService = {
       projectId: projectContext.project.id,
       projectRole: projectContext.projectRole,
       tablesRefreshToken: projectContext.tablesRefreshToken,
+      tablesWorkspaceId: projectContext.project.tablesWorkspaceId,
     };
   },
 };

packages/server/api/src/app/authentication/context/authentication-cookies.ts

@@ -15,7 +15,7 @@ export function setAuthCookiesAndReply(
   const date = jwtDecode<{ exp: number }>(response.tablesRefreshToken);
   const cookieExpiryDate = new Date(date.exp * 1000);
 
-  return reply
+  let replyWithCookies = reply
     .setCookie('jwt_token', response.tablesRefreshToken, {
       domain: getOpenOpsSubDomain(),
       path: '/',
@@ -29,8 +29,23 @@ export function setAuthCookiesAndReply(
       httpOnly: false,
       expires: cookieExpiryDate,
       sameSite: 'lax',
-    })
-    .send(response);
+    });
+
+  if (response.tablesWorkspaceId !== undefined) {
+    replyWithCookies = replyWithCookies.setCookie(
+      'baserow_group_id',
+      String(response.tablesWorkspaceId),
+      {
+        domain: getOpenOpsSubDomain(),
+        path: '/',
+        signed: true,
+        httpOnly: false,
+        expires: cookieExpiryDate,
+      },
+    );
+  }
+
+  return replyWithCookies.send(response);
 }
 
 export function removeAuthCookiesAndReply(reply: FastifyReply): FastifyReply {
@@ -42,6 +57,10 @@ export function removeAuthCookiesAndReply(reply: FastifyReply): FastifyReply {
     .clearCookie('token', {
       path: '/',
     })
+    .clearCookie('baserow_group_id', {
+      domain: getOpenOpsSubDomain(),
+      path: '/',
+    })
     .send('Cookies removed');
 }
 

packages/shared/src/lib/authentication/dto/authentication-response.ts

@@ -8,4 +8,5 @@ export type AuthenticationResponse = UserWithoutPassword & {
   projectId: string;
   projectRole: ProjectMemberRole | null;
   tablesRefreshToken: string;
+  tablesWorkspaceId?: number;
 };