feat(core): new worker workspace engine (#9257)

packages/frontend/apps/ios/src/app.tsx

@@ -12,23 +12,22 @@ import {
   DefaultServerService,
   ServersService,
   ValidatorProvider,
-  WebSocketAuthProvider,
 } from '@affine/core/modules/cloud';
 import { DocsService } from '@affine/core/modules/doc';
 import { GlobalContextService } from '@affine/core/modules/global-context';
 import { I18nProvider } from '@affine/core/modules/i18n';
 import { LifecycleService } from '@affine/core/modules/lifecycle';
-import { configureLocalStorageStateStorageImpls } from '@affine/core/modules/storage';
+import {
+  configureLocalStorageStateStorageImpls,
+  NbstoreProvider,
+} from '@affine/core/modules/storage';
 import { PopupWindowProvider } from '@affine/core/modules/url';
 import { ClientSchemeProvider } from '@affine/core/modules/url/providers/client-schema';
-import { configureIndexedDBUserspaceStorageProvider } from '@affine/core/modules/userspace';
 import { configureBrowserWorkbenchModule } from '@affine/core/modules/workbench';
 import { WorkspacesService } from '@affine/core/modules/workspace';
-import {
-  configureBrowserWorkspaceFlavours,
-  configureIndexedDBWorkspaceEngineStorageProvider,
-} from '@affine/core/modules/workspace-engine';
+import { configureBrowserWorkspaceFlavours } from '@affine/core/modules/workspace-engine';
 import { I18n } from '@affine/i18n';
+import { WorkerClient } from '@affine/nbstore/worker/client';
 import {
   defaultBlockMarkdownAdapterMatchers,
   docLinkBaseURLMiddleware,
@@ -44,16 +43,17 @@ import { Browser } from '@capacitor/browser';
 import { Haptics } from '@capacitor/haptics';
 import { Keyboard, KeyboardStyle } from '@capacitor/keyboard';
 import { Framework, FrameworkRoot, getCurrentStore } from '@toeverything/infra';
+import { OpClient } from '@toeverything/infra/op';
+import { AsyncCall } from 'async-call-rpc';
 import { useTheme } from 'next-themes';
 import { Suspense, useEffect } from 'react';
 import { RouterProvider } from 'react-router-dom';
 
 import { BlocksuiteMenuConfigProvider } from './bs-menu-config';
-import { configureFetchProvider } from './fetch';
 import { ModalConfigProvider } from './modal-config';
-import { Cookie } from './plugins/cookie';
 import { Hashcash } from './plugins/hashcash';
 import { Intelligents } from './plugins/intelligents';
+import { NbStoreNativeDBApis } from './plugins/nbstore';
 import { enableNavigationGesture$ } from './web-navigation-control';
 
 const future = {
@@ -65,9 +65,52 @@ configureCommonModules(framework);
 configureBrowserWorkbenchModule(framework);
 configureLocalStorageStateStorageImpls(framework);
 configureBrowserWorkspaceFlavours(framework);
-configureIndexedDBWorkspaceEngineStorageProvider(framework);
-configureIndexedDBUserspaceStorageProvider(framework);
 configureMobileModules(framework);
+framework.impl(NbstoreProvider, {
+  openStore(_key, options) {
+    const worker = new Worker(
+      new URL(
+        /* webpackChunkName: "nbstore-worker" */ './worker.ts',
+        import.meta.url
+      )
+    );
+    const { port1: nativeDBApiChannelServer, port2: nativeDBApiChannelClient } =
+      new MessageChannel();
+    AsyncCall<typeof NbStoreNativeDBApis>(NbStoreNativeDBApis, {
+      channel: {
+        on(listener) {
+          const f = (e: MessageEvent<any>) => {
+            listener(e.data);
+          };
+          nativeDBApiChannelServer.addEventListener('message', f);
+          return () => {
+            nativeDBApiChannelServer.removeEventListener('message', f);
+          };
+        },
+        send(data) {
+          nativeDBApiChannelServer.postMessage(data);
+        },
+      },
+      log: false,
+    });
+    nativeDBApiChannelServer.start();
+    worker.postMessage(
+      {
+        type: 'native-db-api-channel',
+        port: nativeDBApiChannelClient,
+      },
+      [nativeDBApiChannelClient]
+    );
+    const client = new WorkerClient(new OpClient(worker), options);
+    return {
+      store: client,
+      dispose: () => {
+        worker.terminate();
+        nativeDBApiChannelServer.close();
+      },
+    };
+  },
+});
 framework.impl(PopupWindowProvider, {
   open: (url: string) => {
     Browser.open({
@@ -81,18 +124,6 @@ framework.impl(ClientSchemeProvider, {
     return 'affine';
   },
 });
-configureFetchProvider(framework);
-framework.impl(WebSocketAuthProvider, {
-  getAuthToken: async url => {
-    const cookies = await Cookie.getCookies({
-      url,
-    });
-    return {
-      userId: cookies['affine_user_id'],
-      token: cookies['affine_session'],
-    };
-  },
-});
 framework.impl(ValidatorProvider, {
   async validate(_challenge, resource) {
     const res = await Hashcash.hash({ challenge: resource });