refactor: remove deprecated atoms (#2615)

apps/web/src/hooks/__tests__/index.spec.tsx

@@ -23,11 +23,9 @@ import type React from 'react';
 import { beforeAll, beforeEach, describe, expect, test, vi } from 'vitest';
 
 import { workspacesAtom } from '../../atoms';
+import { rootCurrentWorkspaceAtom } from '../../atoms/root';
 import { BlockSuiteWorkspace } from '../../shared';
-import {
-  currentWorkspaceAtom,
-  useCurrentWorkspace,
-} from '../current/use-current-workspace';
+import { useCurrentWorkspace } from '../current/use-current-workspace';
 import { useAppHelper, useWorkspaces } from '../use-workspaces';
 
 vi.mock(
@@ -169,7 +167,7 @@ describe('useWorkspacesHelper', () => {
       wrapper: ProviderWrapper,
     });
     store.set(rootCurrentWorkspaceIdAtom, workspacesHook.result.current[1].id);
-    await store.get(currentWorkspaceAtom);
+    await store.get(rootCurrentWorkspaceAtom);
     const currentWorkspaceHook = renderHook(() => useCurrentWorkspace(), {
       wrapper: ProviderWrapper,
     });

apps/web/src/hooks/__tests__/use-recent-views.spec.tsx

@@ -20,11 +20,9 @@ import { beforeAll, beforeEach, describe, expect, test, vi } from 'vitest';
 
 import { LocalAdapter } from '../../adapters/local';
 import { workspacesAtom } from '../../atoms';
+import { rootCurrentWorkspaceAtom } from '../../atoms/root';
 import { BlockSuiteWorkspace } from '../../shared';
-import {
-  currentWorkspaceAtom,
-  useCurrentWorkspace,
-} from '../current/use-current-workspace';
+import { useCurrentWorkspace } from '../current/use-current-workspace';
 import {
   useRecentlyViewed,
   useSyncRecentViewsWithRouter,
@@ -98,14 +96,14 @@ describe('useRecentlyViewed', () => {
       providers: [],
     } satisfies LocalWorkspace);
     store.set(rootCurrentWorkspaceIdAtom, blockSuiteWorkspace.id);
-    const workspace = await store.get(currentWorkspaceAtom);
+    const workspace = await store.get(rootCurrentWorkspaceAtom);
     expect(workspace?.id).toBe(blockSuiteWorkspace.id);
     const currentHook = renderHook(() => useCurrentWorkspace(), {
       wrapper: ProviderWrapper,
     });
     expect(currentHook.result.current[0]?.id).toEqual(workspaceId);
     store.set(rootCurrentWorkspaceIdAtom, blockSuiteWorkspace.id);
-    await store.get(currentWorkspaceAtom);
+    await store.get(rootCurrentWorkspaceAtom);
     const recentlyViewedHook = renderHook(() => useRecentlyViewed(), {
       wrapper: ProviderWrapper,
     });

apps/web/src/hooks/current/use-current-mode.ts

@@ -1,9 +1,10 @@
+import { rootCurrentPageIdAtom } from '@affine/workspace/atom';
 import { atom, useAtomValue } from 'jotai';
 
-import { currentPageIdAtom, workspacePreferredModeAtom } from '../../atoms';
+import { workspacePreferredModeAtom } from '../../atoms';
 
 const currentModeAtom = atom<'page' | 'edgeless'>(get => {
-  const pageId = get(currentPageIdAtom);
+  const pageId = get(rootCurrentPageIdAtom);
   const record = get(workspacePreferredModeAtom);
   if (pageId) return record[pageId] ?? 'page';
   else {

apps/web/src/hooks/current/use-current-workspace.ts

@@ -1,22 +1,20 @@
+import {
+  rootCurrentPageIdAtom,
+  rootCurrentWorkspaceIdAtom,
+} from '@affine/workspace/atom';
 import { useAtom, useAtomValue } from 'jotai';
 import { useCallback } from 'react';
 
-import { currentPageIdAtom, currentWorkspaceIdAtom } from '../../atoms';
 import { rootCurrentWorkspaceAtom } from '../../atoms/root';
 import type { AllWorkspace } from '../../shared';
 
-/**
- * @deprecated use `rootCurrentWorkspaceAtom` instead
- */
-export const currentWorkspaceAtom = rootCurrentWorkspaceAtom;
-
 export function useCurrentWorkspace(): [
   AllWorkspace,
   (id: string | null) => void
 ] {
   const currentWorkspace = useAtomValue(rootCurrentWorkspaceAtom);
-  const [, setId] = useAtom(currentWorkspaceIdAtom);
-  const [, setPageId] = useAtom(currentPageIdAtom);
+  const [, setId] = useAtom(rootCurrentWorkspaceIdAtom);
+  const [, setPageId] = useAtom(rootCurrentPageIdAtom);
   return [
     currentWorkspace,
     useCallback(