refactor!: next generation AFFiNE code structure (#1176)

apps/web/src/blocksuite/__tests__/index.spec.ts

@@ -0,0 +1,32 @@
+/**
+ * @vitest-environment happy-dom
+ */
+import 'fake-indexeddb/auto';
+
+import { beforeEach, describe, expect, test } from 'vitest';
+
+import { BlockSuiteWorkspace } from '../../shared';
+import { createAffineProviders, createLocalProviders } from '..';
+
+let blockSuiteWorkspace: BlockSuiteWorkspace;
+
+beforeEach(() => {
+  blockSuiteWorkspace = new BlockSuiteWorkspace({
+    room: 'test',
+  });
+});
+
+describe('blocksuite providers', () => {
+  test('should be valid provider', () => {
+    [createLocalProviders, createAffineProviders].forEach(createProviders => {
+      createProviders(blockSuiteWorkspace).forEach(provider => {
+        expect(provider).toBeTypeOf('object');
+        expect(provider).toHaveProperty('flavour');
+        expect(provider).toHaveProperty('connect');
+        expect(provider.connect).toBeTypeOf('function');
+        expect(provider).toHaveProperty('disconnect');
+        expect(provider.disconnect).toBeTypeOf('function');
+      });
+    });
+  });
+});

apps/web/src/blocksuite/index.ts

@@ -0,0 +1,26 @@
+import { BlockSuiteWorkspace, Provider } from '../shared';
+import { config } from '../shared/env';
+import { createIndexedDBProvider, createWebSocketProvider } from './providers';
+
+export const createAffineProviders = (
+  blockSuiteWorkspace: BlockSuiteWorkspace
+): Provider[] => {
+  return (
+    [
+      createWebSocketProvider(blockSuiteWorkspace),
+      config.enableIndexedDBProvider &&
+        createIndexedDBProvider(blockSuiteWorkspace),
+    ] as any[]
+  ).filter(v => Boolean(v));
+};
+
+export const createLocalProviders = (
+  blockSuiteWorkspace: BlockSuiteWorkspace
+): Provider[] => {
+  return (
+    [
+      config.enableIndexedDBProvider &&
+        createIndexedDBProvider(blockSuiteWorkspace),
+    ] as any[]
+  ).filter(v => Boolean(v));
+};

apps/web/src/blocksuite/providers/index.ts

@@ -0,0 +1,71 @@
+import { WebsocketProvider } from '@affine/datacenter';
+import { assertExists } from '@blocksuite/store';
+import { IndexeddbPersistence } from 'y-indexeddb';
+
+import {
+  AffineWebSocketProvider,
+  BlockSuiteWorkspace,
+  LocalIndexedDBProvider,
+} from '../../shared';
+import { apis } from '../../shared/apis';
+
+export const createWebSocketProvider = (
+  blockSuiteWorkspace: BlockSuiteWorkspace
+): AffineWebSocketProvider => {
+  let webSocketProvider: WebsocketProvider | null = null;
+  return {
+    flavour: 'affine-websocket',
+    cleanup: () => {
+      assertExists(webSocketProvider);
+      webSocketProvider?.destroy();
+    },
+    connect: () => {
+      const wsUrl = `${
+        window.location.protocol === 'https:' ? 'wss' : 'ws'
+      }://${window.location.host}/api/sync/`;
+      webSocketProvider = new WebsocketProvider(
+        wsUrl,
+        blockSuiteWorkspace.room as string,
+        blockSuiteWorkspace.doc,
+        {
+          params: { token: apis.auth.refresh },
+          // @ts-expect-error ignore the type
+          awareness: blockSuiteWorkspace.awarenessStore.awareness,
+        }
+      );
+      console.log('connect', webSocketProvider.roomname);
+      webSocketProvider.connect();
+    },
+    disconnect: () => {
+      assertExists(webSocketProvider);
+      console.log('disconnect', webSocketProvider.roomname);
+      webSocketProvider?.disconnect();
+    },
+  };
+};
+
+export const createIndexedDBProvider = (
+  blockSuiteWorkspace: BlockSuiteWorkspace
+): LocalIndexedDBProvider => {
+  let indexdbProvider: IndexeddbPersistence | null = null;
+  return {
+    flavour: 'local-indexeddb',
+    cleanup: () => {
+      assertExists(indexdbProvider);
+      indexdbProvider.clearData();
+    },
+    connect: () => {
+      console.info('connect indexeddb provider', blockSuiteWorkspace.room);
+      indexdbProvider = new IndexeddbPersistence(
+        blockSuiteWorkspace.room as string,
+        blockSuiteWorkspace.doc
+      );
+    },
+    disconnect: () => {
+      assertExists(indexdbProvider);
+      console.info('disconnect indexeddb provider', blockSuiteWorkspace.room);
+      indexdbProvider.destroy();
+      indexdbProvider = null;
+    },
+  };
+};