refactor: workspace manager (#5060)

2026-09-02 06:39:46 +08:00 · 2023-12-15 07:20:50 +00:00
parent af15aa06d4
commit fe2851d3e9
217 changed files with 3605 additions and 4244 deletions
@@ -0,0 +1,40 @@
+import { workspaceManagerAtom } from '@affine/workspace/atom';
+import type { WorkspaceMetadata } from '@affine/workspace/metadata';
+import { useAtomValue } from 'jotai';
+import { useEffect, useState } from 'react';
+
+export function useWorkspaceBlobObjectUrl(
+  meta?: WorkspaceMetadata,
+  blobKey?: string | null
+) {
+  const workspaceManager = useAtomValue(workspaceManagerAtom);
+
+  const [blob, setBlob] = useState<string | undefined>(undefined);
+
+  useEffect(() => {
+    setBlob(undefined);
+    if (!blobKey || !meta) {
+      return;
+    }
+    let canceled = false;
+    let objectUrl: string = '';
+    workspaceManager
+      .getWorkspaceBlob(meta, blobKey)
+      .then(blob => {
+        if (blob && !canceled) {
+          objectUrl = URL.createObjectURL(blob);
+          setBlob(objectUrl);
+        }
+      })
+      .catch(err => {
+        console.error('get workspace blob error: ' + err);
+      });
+
+    return () => {
+      canceled = true;
+      URL.revokeObjectURL(objectUrl);
+    };
+  }, [meta, blobKey, workspaceManager]);
+
+  return blob;
+}