AFFiNE-Mirror/apps/web/src/bootstrap/index.ts

import { migrateToSubdoc } from '@affine/env/blocksuite';
import { isDesktop, isServer } from '@affine/env/constant';
import { setupGlobal } from '@affine/env/global';
import type { LocalIndexedDBDownloadProvider } from '@affine/env/workspace';
import { WorkspaceFlavour, WorkspaceVersion } from '@affine/env/workspace';
import type { RootWorkspaceMetadata } from '@affine/workspace/atom';
import { rootWorkspacesMetadataAtom } from '@affine/workspace/atom';
import {
  migrateLocalBlobStorage,
  upgradeV1ToV2,
} from '@affine/workspace/migration';
import { createIndexedDBDownloadProvider } from '@affine/workspace/providers';
import { assertExists } from '@blocksuite/global/utils';
import { rootStore } from '@toeverything/plugin-infra/manager';

import { WorkspaceAdapters } from '../adapters/workspace';

setupGlobal();

if (process.env.NODE_ENV === 'development') {
  console.log('Runtime Preset', runtimeConfig);
}

if (runtimeConfig.enablePlugin && !isServer) {
  import('@affine/copilot');
}

if (!isServer) {
  import('@affine/bookmark-block');
}

if (!isDesktop && !isServer) {
  // Polyfill Electron
  const unimplemented = () => {
    throw new Error('AFFiNE Plugin Web will be supported in the future');
  };
  const affine = {
    ipcRenderer: {
      invoke: unimplemented,
      send: unimplemented,
      on: unimplemented,
      once: unimplemented,
      removeListener: unimplemented,
    },
  };

  Object.freeze(affine);

  Object.defineProperty(window, 'affine', {
    value: affine,
    writable: false,
  });
}

rootStore.sub(rootWorkspacesMetadataAtom, () => {
  const metadata = rootStore.get(rootWorkspacesMetadataAtom);
  metadata.forEach(oldMeta => {
    if (!oldMeta.version) {
      const adapter = WorkspaceAdapters[oldMeta.flavour];
      assertExists(adapter);
      const upgrade = async () => {
        const workspace = await adapter.CRUD.get(oldMeta.id);
        if (!workspace) {
          console.warn('cannot find workspace', oldMeta.id);
          return;
        }
        if (workspace.flavour !== WorkspaceFlavour.LOCAL) {
          console.warn('not supported');
          return;
        }
        const doc = workspace.blockSuiteWorkspace.doc;
        const provider = createIndexedDBDownloadProvider(workspace.id, doc, {
          awareness: workspace.blockSuiteWorkspace.awarenessStore.awareness,
        }) as LocalIndexedDBDownloadProvider;
        provider.sync();
        await provider.whenReady;
        const newDoc = migrateToSubdoc(doc);
        if (doc === newDoc) {
          console.log('doc not changed');
          rootStore.set(rootWorkspacesMetadataAtom, metadata =>
            metadata.map(newMeta =>
              newMeta.id === oldMeta.id
                ? {
                    ...newMeta,
                    version: WorkspaceVersion.SubDoc,
                  }
                : newMeta
            )
          );
          return;
        }
        const newWorkspace = upgradeV1ToV2(workspace);

        const newId = await adapter.CRUD.create(
          newWorkspace.blockSuiteWorkspace
        );

        await adapter.CRUD.delete(workspace as any);
        await migrateLocalBlobStorage(workspace.id, newId);
        rootStore.set(rootWorkspacesMetadataAtom, metadata => [
          ...metadata
            .map(newMeta => (newMeta.id === oldMeta.id ? null : newMeta))
            .filter((meta): meta is RootWorkspaceMetadata => !!meta),
          {
            id: newId,
            flavour: oldMeta.flavour,
            version: WorkspaceVersion.SubDoc,
          },
        ]);
      };

      // create a new workspace and push it to metadata
      upgrade().catch(console.error);
    }
  });
});