refactor(infra): directory structure (#4615)

packages/common/infra/src/__tests__/blocksuite-atom.ts

@@ -0,0 +1,55 @@
+/**
+ * @vitest-environment happy-dom
+ */
+import { Schema, Workspace } from '@blocksuite/store';
+import { waitFor } from '@testing-library/react';
+import { getDefaultStore } from 'jotai/vanilla';
+import { expect, test, vi } from 'vitest';
+
+import {
+  getBlockSuiteWorkspaceAtom,
+  INTERNAL_BLOCKSUITE_HASH_MAP,
+} from '../__internal__/workspace.js';
+
+test('blocksuite atom', async () => {
+  const sync = vi.fn();
+  let connected = false;
+  const connect = vi.fn(() => (connected = true));
+  const workspace = new Workspace({
+    schema: new Schema(),
+    id: '1',
+    providerCreators: [
+      () => ({
+        flavour: 'fake',
+        active: true,
+        sync,
+        get whenReady(): Promise<void> {
+          return Promise.resolve();
+        },
+      }),
+      () => ({
+        flavour: 'fake-2',
+        passive: true,
+        get connected() {
+          return connected;
+        },
+        connect,
+        disconnect: vi.fn(),
+      }),
+    ],
+  });
+  INTERNAL_BLOCKSUITE_HASH_MAP.set('1', workspace);
+
+  {
+    const [atom, effectAtom] = getBlockSuiteWorkspaceAtom('1');
+    const store = getDefaultStore();
+    const result = await store.get(atom);
+    expect(result).toBe(workspace);
+    expect(sync).toBeCalledTimes(1);
+    expect(connect).not.toHaveBeenCalled();
+
+    store.sub(effectAtom, vi.fn());
+    await waitFor(() => expect(connect).toBeCalledTimes(1));
+    expect(connected).toBe(true);
+  }
+});

packages/common/infra/src/__tests__/migration.spec.ts

@@ -0,0 +1,83 @@
+import { readFileSync } from 'fs';
+import { dirname, resolve } from 'path';
+import { fileURLToPath } from 'url';
+import { describe, expect, test } from 'vitest';
+import type { Array as YArray, Map as YMap } from 'yjs';
+import { applyUpdate, Doc } from 'yjs';
+
+import { migrateToSubdoc } from '../blocksuite/index.js';
+
+const fixturePath = resolve(
+  dirname(fileURLToPath(import.meta.url)),
+  'workspace.ydoc'
+);
+const yDocBuffer = readFileSync(fixturePath);
+const doc = new Doc();
+applyUpdate(doc, new Uint8Array(yDocBuffer));
+const migratedDoc = migrateToSubdoc(doc);
+
+describe('migration', () => {
+  test('migration to subdoc', async () => {
+    const { default: json } = await import('@affine-test/fixtures/output.json');
+    const length = Object.keys(json).length;
+    const binary = new Uint8Array(length);
+    for (let i = 0; i < length; i++) {
+      binary[i] = (json as any)[i];
+    }
+    const doc = new Doc();
+    applyUpdate(doc, binary);
+    {
+      // invoke data
+      doc.getMap('space:hello-world');
+      doc.getMap('space:meta');
+    }
+    const blocks = doc.getMap('space:hello-world').toJSON();
+    const newDoc = migrateToSubdoc(doc);
+    const subDoc = newDoc.getMap('spaces').values().next().value as Doc;
+    const data = (subDoc.toJSON() as any).blocks;
+    Object.keys(data).forEach(id => {
+      if (id === 'xyWNqindHH') {
+        return;
+      }
+      if (
+        blocks[id]['sys:flavour'] === 'affine:surface' &&
+        !blocks[id]['prop:elements']
+      ) {
+        blocks[id]['prop:elements'] = data[id]['prop:elements'];
+      }
+      expect(data[id]).toEqual(blocks[id]);
+    });
+  });
+
+  test('test fixture should be set correctly', () => {
+    const meta = doc.getMap('space:meta');
+    const versions = meta.get('versions') as YMap<unknown>;
+    expect(versions.get('affine:code')).toBeTypeOf('number');
+  });
+
+  test('metadata should be migrated correctly', () => {
+    const originalMeta = doc.getMap('space:meta');
+    const originalVersions = originalMeta.get('versions') as YMap<unknown>;
+
+    const meta = migratedDoc.getMap('meta');
+    const blockVersions = meta.get('blockVersions') as YMap<unknown>;
+
+    expect(meta.get('workspaceVersion')).toBe(1);
+    expect(blockVersions.get('affine:code')).toBe(
+      originalVersions.get('affine:code')
+    );
+    expect((meta.get('pages') as YArray<unknown>).length).toBe(
+      (originalMeta.get('pages') as YArray<unknown>).length
+    );
+
+    expect(blockVersions.get('affine:embed')).toBeUndefined();
+    expect(blockVersions.get('affine:image')).toBe(
+      originalVersions.get('affine:embed')
+    );
+
+    expect(blockVersions.get('affine:frame')).toBeUndefined();
+    expect(blockVersions.get('affine:note')).toBe(
+      originalVersions.get('affine:frame')
+    );
+  });
+});