feat: open last workspace when back or refresh affine (#1413)

Co-authored-by: himself65 <himself65@outlook.com>
This commit is contained in:
Qi
2023-03-09 02:40:25 +08:00
committed by GitHub
parent 867ea9cf2b
commit 0db7868a6a
5 changed files with 65 additions and 14 deletions

View File

@@ -1,4 +1,5 @@
import { atom, useAtom, useAtomValue } from 'jotai';
import { atom, useAtom, useAtomValue, useSetAtom } from 'jotai';
import { atomWithStorage } from 'jotai/utils';
import { useCallback } from 'react';
import {
@@ -16,6 +17,11 @@ export const currentWorkspaceAtom = atom<Promise<RemWorkspace | null>>(
}
);
export const lastWorkspaceIdAtom = atomWithStorage<string | null>(
'last_workspace_id',
null
);
export function useCurrentWorkspace(): [
RemWorkspace | null,
(id: string | null) => void
@@ -23,14 +29,16 @@ export function useCurrentWorkspace(): [
const currentWorkspace = useAtomValue(currentWorkspaceAtom);
const [, setId] = useAtom(currentWorkspaceIdAtom);
const [, setPageId] = useAtom(currentPageIdAtom);
const setLast = useSetAtom(lastWorkspaceIdAtom);
return [
currentWorkspace,
useCallback(
(id: string | null) => {
setPageId(null);
setLast(id);
setId(id);
},
[setId, setPageId]
[setId, setLast, setPageId]
),
];
}