refactor: move affine utils into @affine/workspace (#2611)

packages/workspace/src/providers/affine-download.ts

@@ -0,0 +1,61 @@
+import { DebugLogger } from '@affine/debug';
+import { assertExists, Workspace } from '@blocksuite/store';
+
+import { affineApis } from '../affine/shared';
+import type { AffineDownloadProvider } from '../type';
+
+const hashMap = new Map<string, ArrayBuffer>();
+
+const logger = new DebugLogger('affine:workspace:download-provider');
+
+export const createAffineDownloadProvider = (
+  blockSuiteWorkspace: Workspace
+): AffineDownloadProvider => {
+  assertExists(blockSuiteWorkspace.id);
+  const id = blockSuiteWorkspace.id;
+  let connected = false;
+  const callbacks = new Set<() => void>();
+  return {
+    flavour: 'affine-download',
+    background: true,
+    get connected() {
+      return connected;
+    },
+    callbacks,
+    connect: () => {
+      logger.info('connect download provider', id);
+      if (hashMap.has(id)) {
+        logger.debug('applyUpdate');
+        Workspace.Y.applyUpdate(
+          blockSuiteWorkspace.doc,
+          new Uint8Array(hashMap.get(id) as ArrayBuffer)
+        );
+        connected = true;
+        callbacks.forEach(cb => cb());
+        return;
+      }
+      affineApis
+        .downloadWorkspace(id, false)
+        .then(binary => {
+          hashMap.set(id, binary);
+          logger.debug('applyUpdate');
+          Workspace.Y.applyUpdate(
+            blockSuiteWorkspace.doc,
+            new Uint8Array(binary)
+          );
+          connected = true;
+          callbacks.forEach(cb => cb());
+        })
+        .catch(e => {
+          logger.error('downloadWorkspace', e);
+        });
+    },
+    disconnect: () => {
+      logger.info('disconnect download provider', id);
+      connected = false;
+    },
+    cleanup: () => {
+      hashMap.delete(id);
+    },
+  };
+};

packages/workspace/src/providers/index.ts

@@ -1,17 +1,4 @@
 import { config, websocketPrefixUrl } from '@affine/env';
-import { KeckProvider } from '@affine/workspace/affine/keck';
-import {
-  getLoginStorage,
-  storageChangeSlot,
-} from '@affine/workspace/affine/login';
-import type {
-  AffineWebSocketProvider,
-  LocalIndexedDBBackgroundProvider,
-  LocalIndexedDBDownloadProvider,
-  Provider,
-  SQLiteProvider,
-} from '@affine/workspace/type';
-import { CallbackSet } from '@affine/workspace/utils';
 import type { BlobManager, Disposable } from '@blocksuite/store';
 import {
   assertExists,
@@ -23,6 +10,17 @@ import {
   EarlyDisconnectError,
 } from '@toeverything/y-indexeddb';
 
+import { KeckProvider } from '../affine/keck';
+import { getLoginStorage, storageChangeSlot } from '../affine/login';
+import type {
+  AffineWebSocketProvider,
+  LocalIndexedDBBackgroundProvider,
+  LocalIndexedDBDownloadProvider,
+  Provider,
+  SQLiteProvider,
+} from '../type';
+import { CallbackSet } from '../utils';
+import { createAffineDownloadProvider } from './affine-download';
 import { createBroadCastChannelProvider } from './broad-cast-channel';
 import { localProviderLogger as logger } from './logger';
 
@@ -258,6 +256,7 @@ const createSQLiteProvider = (
 };
 
 export {
+  createAffineDownloadProvider,
   createAffineWebSocketProvider,
   createBroadCastChannelProvider,
   createIndexedDBBackgroundProvider,
@@ -278,3 +277,17 @@ export const createLocalProviders = (
     ] as any[]
   ).filter(v => Boolean(v));
 };
+
+export const createAffineProviders = (
+  blockSuiteWorkspace: BlockSuiteWorkspace
+): Provider[] => {
+  return (
+    [
+      createAffineDownloadProvider(blockSuiteWorkspace),
+      createAffineWebSocketProvider(blockSuiteWorkspace),
+      config.enableBroadCastChannelProvider &&
+        createBroadCastChannelProvider(blockSuiteWorkspace),
+      createIndexedDBDownloadProvider(blockSuiteWorkspace),
+    ] as any[]
+  ).filter(v => Boolean(v));
+};