44 lines
1.3 KiB
TypeScript
44 lines
1.3 KiB
TypeScript
import {
|
|
rootCurrentWorkspaceIdAtom,
|
|
rootWorkspacesMetadataAtom,
|
|
} from '@affine/workspace/atom';
|
|
import type { WorkspaceFlavour } from '@affine/workspace/type';
|
|
import type { WorkspaceRegistry } from '@affine/workspace/type';
|
|
import { useSetAtom } from 'jotai';
|
|
import { useCallback } from 'react';
|
|
|
|
import { WorkspaceAdapters } from '../plugins';
|
|
|
|
/**
|
|
* Transform workspace from one flavour to another
|
|
*
|
|
* The logic here is to delete the old workspace and create a new one.
|
|
*/
|
|
export function useTransformWorkspace() {
|
|
const setCurrentWorkspaceId = useSetAtom(rootCurrentWorkspaceIdAtom);
|
|
const set = useSetAtom(rootWorkspacesMetadataAtom);
|
|
return useCallback(
|
|
async <From extends WorkspaceFlavour, To extends WorkspaceFlavour>(
|
|
from: From,
|
|
to: To,
|
|
workspace: WorkspaceRegistry[From]
|
|
): Promise<string> => {
|
|
await WorkspaceAdapters[from].CRUD.delete(workspace as any);
|
|
const newId = await WorkspaceAdapters[to].CRUD.create(
|
|
workspace.blockSuiteWorkspace
|
|
);
|
|
set(workspaces => {
|
|
const idx = workspaces.findIndex(ws => ws.id === workspace.id);
|
|
workspaces.splice(idx, 1, {
|
|
id: newId,
|
|
flavour: to,
|
|
});
|
|
return [...workspaces];
|
|
});
|
|
setCurrentWorkspaceId(newId);
|
|
return newId;
|
|
},
|
|
[set, setCurrentWorkspaceId]
|
|
);
|
|
}
|