remember last workspace visited by user

apps/api/src/auth/token.ts

@@ -110,6 +110,7 @@ export async function sessionFromCookies(
       email: true,
       name: true,
       picture: true,
+      lastWorkspaceVisited: true,
       createdAt: true,
       updatedAt: true,
       workspaces: true,
@@ -130,6 +131,7 @@ export async function sessionFromCookies(
       email: user.email,
       name: user.name,
       picture: user.picture,
+      lastWorkspaceVisited: user.lastWorkspaceVisited,
       createdAt: user.createdAt,
       updatedAt: user.updatedAt,
     },
@@ -319,9 +321,7 @@ export const isAuthorizedForDataSource = async (
       return result !== null
     }
     case 'databrickssql': {
-      const result = await prisma().databricksSQLDataSource.findFirst(
-        query,
-      )
+      const result = await prisma().databricksSQLDataSource.findFirst(query)
       return result !== null
     }
   }

apps/api/src/websocket/workspace/index.ts

@@ -37,6 +37,12 @@ export const joinWorkspace =
 
       if (!socket.rooms.has(workspaceId)) {
         await socket.join(workspaceId)
+        await prisma().user.update({
+          where: { id: session.user.id },
+          data: {
+            lastWorkspaceVisited: workspaceId,
+          },
+        })
       }
 
       await emitInitialData(io, socket, workspaceId)

apps/api/src/workspace/index.ts

@@ -50,6 +50,7 @@ export class WorkspaceCreator implements IWorkspaceCreator {
                 email: true,
                 name: true,
                 picture: true,
+                lastWorkspaceVisited: true,
                 createdAt: true,
                 updatedAt: true,
               },

apps/api/src/yjs/v2/executor/executor.ts

@@ -63,6 +63,7 @@ export function unknownUser(): ApiUser {
     name: 'unknown',
     email: 'unknown',
     picture: null,
+    lastWorkspaceVisited: null,
     createdAt: new Date(),
     updatedAt: new Date(),
   }

apps/web/src/hooks/useSessionRedirect.ts

@@ -38,10 +38,19 @@ export const useSessionRedirect = (shouldRedirect = true) => {
       } else {
         redirectToLogin(router)
       }
-    } else if (workspaces.data.length === 0) {
-      signOut()
     } else {
-      router.replace(`/workspaces/${workspaces.data[0].id}/documents`)
+      const user = session.data
+      const workspace =
+        workspaces.data.find(
+          (workspace) => workspace.id === user.lastWorkspaceVisited
+        ) ??
+        workspaces.data.find((workspace) => workspace.ownerId === user.id) ??
+        workspaces.data[0]
+      if (workspace) {
+        router.replace(`/workspaces/${workspaces.data[0].id}/documents`)
+      } else {
+        signOut()
+      }
     }
   }, [properties, workspaces, session, router, shouldRedirect, signOut])
 }

apps/web/src/pages/workspaces/index.tsx

@@ -1,4 +1,4 @@
-import { useSignout } from '@/hooks/useAuth'
+import { useSession, useSignout } from '@/hooks/useAuth'
 import { useWorkspaces } from '@/hooks/useWorkspaces'
 import { useRouter } from 'next/router'
 import { useEffect } from 'react'
@@ -7,13 +7,19 @@ export default function WorkspacesPage() {
   const router = useRouter()
   const [{ isLoading, data }] = useWorkspaces()
   const signOut = useSignout()
+  const session = useSession()
 
   useEffect(() => {
-    if (isLoading) {
+    if (isLoading || session.isLoading) {
       return
     }
 
-    const workspace = data[0]
+    const workspace =
+      data.find(
+        (workspace) => workspace.id === session.data?.lastWorkspaceVisited
+      ) ??
+      data.find((workspace) => workspace.ownerId === session.data?.id) ??
+      data[0]
     if (!workspace) {
       signOut()
     } else {

packages/database/prisma/migrations/20241129201145_add_last_visited_workspace_to_user/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "User" ADD COLUMN     "lastWorkspaceVisited" TEXT;

packages/database/prisma/schema.prisma

@@ -398,10 +398,11 @@ model User {
   createdAt      DateTime @default(now())
   updatedAt      DateTime @default(now()) @updatedAt
 
-  workspacesInvitees UserWorkspace[] @relation("UserWorkspaceInviter")
-  workspaces         UserWorkspace[] @relation("UserWorkspaceUser")
-  ownedWorkspaces    Workspace[]
-  confirmedAt        DateTime?
+  workspacesInvitees   UserWorkspace[] @relation("UserWorkspaceInviter")
+  workspaces           UserWorkspace[] @relation("UserWorkspaceUser")
+  ownedWorkspaces      Workspace[]
+  confirmedAt          DateTime?
+  lastWorkspaceVisited String?
 
   comments           Comment[]
   Favorite           Favorite[]

packages/database/src/users.ts

@@ -26,6 +26,7 @@ export async function createUser(
       picture: true,
       createdAt: true,
       updatedAt: true,
+      lastWorkspaceVisited: true,
     },
   })
 }
@@ -40,6 +41,7 @@ export async function getUserById(id: string): Promise<ApiUser | null> {
       email: true,
       name: true,
       picture: true,
+      lastWorkspaceVisited: true,
       createdAt: true,
       updatedAt: true,
     },
@@ -58,6 +60,7 @@ export async function getUserByEmail(email: string): Promise<ApiUser | null> {
       email: true,
       name: true,
       picture: true,
+      lastWorkspaceVisited: true,
       createdAt: true,
       updatedAt: true,
     },
@@ -78,6 +81,7 @@ export async function listWorkspaceUsers(
           email: true,
           name: true,
           picture: true,
+          lastWorkspaceVisited: true,
           createdAt: true,
           updatedAt: true,
         },
@@ -111,6 +115,7 @@ export async function addUserToWorkspace(
           email: true,
           name: true,
           picture: true,
+          lastWorkspaceVisited: true,
           createdAt: true,
           updatedAt: true,
         },
@@ -143,6 +148,7 @@ export async function deleteUserFromWorkspace(
           email: true,
           name: true,
           picture: true,
+          lastWorkspaceVisited: true,
           createdAt: true,
           updatedAt: true,
         },
@@ -170,6 +176,7 @@ export async function confirmUser(userId: string): Promise<ApiUser | null> {
         email: true,
         name: true,
         picture: true,
+        lastWorkspaceVisited: true,
         createdAt: true,
         updatedAt: true,
       },