feat: bump blocksuite (#6078)

This commit is contained in:
regischen
2024-03-13 17:04:21 +08:00
committed by GitHub
parent 573528be41
commit fddbb426a6
126 changed files with 891 additions and 918 deletions
+2 -2
View File
@@ -3,8 +3,8 @@
"private": true,
"type": "module",
"devDependencies": {
"@blocksuite/global": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/global": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"react": "18.2.0",
"react-dom": "18.2.0",
"vitest": "1.3.1"
+4 -4
View File
@@ -1,5 +1,5 @@
// This file should has not side effect
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
declare global {
interface Window {
@@ -95,12 +95,12 @@ export const Messages = {
};
export class PageNotFoundError extends TypeError {
readonly workspace: Workspace;
readonly docCollection: DocCollection;
readonly pageId: string;
constructor(workspace: Workspace, pageId: string) {
constructor(docCollection: DocCollection, pageId: string) {
super();
this.workspace = workspace;
this.docCollection = docCollection;
this.pageId = pageId;
}
}
+2 -2
View File
@@ -1,4 +1,4 @@
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { z } from 'zod';
export const literalValueSchema: z.ZodType<LiteralValue, z.ZodTypeDef> =
@@ -81,4 +81,4 @@ export const tagSchema = z.object({
});
export type Tag = z.input<typeof tagSchema>;
export type PropertiesMeta = Workspace['meta']['properties'];
export type PropertiesMeta = DocCollection['meta']['properties'];
+3 -3
View File
@@ -1,6 +1,6 @@
/// <reference types="@blocksuite/global" />
import { assertEquals } from '@blocksuite/global/utils';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { z } from 'zod';
import { isDesktop, isServer } from './constant.js';
@@ -154,9 +154,9 @@ export function setupGlobal() {
globalThis.$AFFINE_SETUP = true;
}
export function setupEditorFlags(workspace: Workspace) {
export function setupEditorFlags(docCollection: DocCollection) {
Object.entries(runtimeConfig.editorFlags).forEach(([key, value]) => {
workspace.awarenessStore.setFlag(
docCollection.awarenessStore.setFlag(
key as keyof BlockSuiteFeatureFlags,
value
);
+5 -5
View File
@@ -17,9 +17,9 @@
"@affine/debug": "workspace:*",
"@affine/env": "workspace:*",
"@affine/templates": "workspace:*",
"@blocksuite/blocks": "0.13.0-canary-202403050653-934469c",
"@blocksuite/global": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/blocks": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/global": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"foxact": "^0.2.31",
"jotai": "^2.6.5",
"jotai-effect": "^0.6.0",
@@ -33,8 +33,8 @@
"devDependencies": {
"@affine-test/fixtures": "workspace:*",
"@affine/templates": "workspace:*",
"@blocksuite/lit": "0.13.0-canary-202403050653-934469c",
"@blocksuite/presets": "0.13.0-canary-202403050653-934469c",
"@blocksuite/lit": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/presets": "0.13.0-canary-202403120738-e15d583",
"@testing-library/react": "^14.2.1",
"async-call-rpc": "^6.4.0",
"react": "^18.2.0",
@@ -35,7 +35,7 @@ export async function migratePages(
console.error(e);
}
});
schema.upgradeWorkspace(rootDoc);
schema.upgradeCollection(rootDoc);
// Hard code to upgrade page version to 2.
// Let e2e to ensure the data version is correct.
@@ -1,4 +1,4 @@
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { Array as YArray, Doc as YDoc, Map as YMap } from 'yjs';
/**
@@ -11,14 +11,14 @@ export enum MigrationPoint {
}
export function checkWorkspaceCompatibility(
workspace: Workspace,
docCollection: DocCollection,
isCloud: boolean
): MigrationPoint | null {
// check if there is any key starts with 'space:' on root doc
const spaceMetaObj = workspace.doc.share.get('space:meta') as
const spaceMetaObj = docCollection.doc.share.get('space:meta') as
| YMap<any>
| undefined;
const docKeys = Array.from(workspace.doc.share.keys());
const docKeys = Array.from(docCollection.doc.share.keys());
const haveSpaceMeta = !!spaceMetaObj && spaceMetaObj.size > 0;
const haveLegacySpace = docKeys.some(key => key.startsWith('space:'));
@@ -28,12 +28,12 @@ export function checkWorkspaceCompatibility(
}
// exit if no pages
if (!workspace.meta.docs?.length) {
if (!docCollection.meta.docs?.length) {
return null;
}
// check guid compatibility
const meta = workspace.doc.getMap('meta') as YMap<unknown>;
const meta = docCollection.doc.getMap('meta') as YMap<unknown>;
const pages = meta.get('pages') as YArray<YMap<unknown>>;
for (const page of pages) {
const pageId = page.get('id') as string | undefined;
@@ -41,23 +41,23 @@ export function checkWorkspaceCompatibility(
return MigrationPoint.GuidFix;
}
}
const spaces = workspace.doc.getMap('spaces') as YMap<YDoc>;
const spaces = docCollection.doc.getMap('spaces') as YMap<YDoc>;
for (const [pageId, _] of spaces) {
if (pageId.includes(':')) {
return MigrationPoint.GuidFix;
}
}
const hasVersion = workspace.meta.hasVersion;
const hasVersion = docCollection.meta.hasVersion;
if (!hasVersion) {
return MigrationPoint.BlockVersion;
}
// TODO: Catch compatibility error from blocksuite to show upgrade page.
// Temporarily follow the check logic of blocksuite.
if ((workspace.meta.docs?.length ?? 0) <= 1) {
if ((docCollection.meta.docs?.length ?? 0) <= 1) {
try {
workspace.meta.validateVersion(workspace);
docCollection.meta.validateVersion(docCollection);
} catch (e) {
console.info('validateVersion error', e);
return MigrationPoint.BlockVersion;
@@ -65,9 +65,9 @@ export function checkWorkspaceCompatibility(
}
// From v2, we depend on blocksuite to check and migrate data.
const blockVersions = workspace.meta.blockVersions;
const blockVersions = docCollection.meta.blockVersions;
for (const [flavour, version] of Object.entries(blockVersions ?? {})) {
const schema = workspace.schema.flavourSchemaMap.get(flavour);
const schema = docCollection.schema.flavourSchemaMap.get(flavour);
if (schema?.version !== version) {
return MigrationPoint.BlockVersion;
}
@@ -1,9 +1,9 @@
import type { WorkspaceFlavour } from '@affine/env/workspace';
import type {
CollectionInfoSnapshot,
Doc,
DocSnapshot,
JobMiddleware,
WorkspaceInfoSnapshot,
} from '@blocksuite/store';
import { Job } from '@blocksuite/store';
import { Map as YMap } from 'yjs';
@@ -49,17 +49,17 @@ export async function buildShowcaseWorkspace(
) {
const meta = await workspaceManager.createWorkspace(
flavour,
async (blockSuiteWorkspace, blobStorage) => {
blockSuiteWorkspace.meta.setName(workspaceName);
async (docCollection, blobStorage) => {
docCollection.meta.setName(workspaceName);
const { onboarding } = await import('@affine/templates');
const info = onboarding['info.json'] as WorkspaceInfoSnapshot;
const info = onboarding['info.json'] as CollectionInfoSnapshot;
const blob = onboarding['blob.json'] as { [key: string]: string };
const migrationMiddleware: JobMiddleware = ({ slots, workspace }) => {
const migrationMiddleware: JobMiddleware = ({ slots, collection }) => {
slots.afterImport.on(payload => {
if (payload.type === 'page') {
workspace.schema.upgradeDoc(
collection.schema.upgradeDoc(
info?.pageVersion ?? 0,
{},
payload.page.spaceDoc
@@ -69,11 +69,11 @@ export async function buildShowcaseWorkspace(
};
const job = new Job({
workspace: blockSuiteWorkspace,
collection: docCollection,
middlewares: [replaceIdMiddleware, migrationMiddleware],
});
job.snapshotToWorkspaceInfo(info);
job.snapshotToCollectionInfo(info);
// for now all onboarding assets are considered served via CDN
// hack assets so that every blob exists
@@ -92,8 +92,8 @@ export async function buildShowcaseWorkspace(
})
);
const newVersions = getLatestVersions(blockSuiteWorkspace.schema);
blockSuiteWorkspace.doc
const newVersions = getLatestVersions(docCollection.schema);
docCollection.doc
.getMap('meta')
.set('blockVersions', new YMap(Object.entries(newVersions)));
+1 -1
View File
@@ -20,7 +20,7 @@ export class PageManager {
if (!pageRecord) {
throw new Error('Page record not found');
}
const blockSuitePage = this.workspace.blockSuiteWorkspace.getDoc(pageId);
const blockSuitePage = this.workspace.docCollection.getDoc(pageId);
if (!blockSuitePage) {
throw new Error('Page not found');
}
@@ -18,7 +18,7 @@ export class PageRecordList {
new Observable(subscriber => {
const emit = () => {
subscriber.next(
this.workspace.blockSuiteWorkspace.meta.docMetas.map(
this.workspace.docCollection.meta.docMetas.map(
v => new PageRecord(v.id, this.workspace, this.localState)
)
);
@@ -27,7 +27,7 @@ export class PageRecordList {
emit();
const dispose =
this.workspace.blockSuiteWorkspace.meta.docMetaUpdated.on(emit).dispose;
this.workspace.docCollection.meta.docMetaUpdated.on(emit).dispose;
return () => {
dispose();
};
+3 -3
View File
@@ -16,7 +16,7 @@ export class PageRecord {
meta = LiveData.from<DocMeta>(
new Observable(subscriber => {
const emit = () => {
const meta = this.workspace.blockSuiteWorkspace.meta.docMetas.find(
const meta = this.workspace.docCollection.meta.docMetas.find(
page => page.id === this.id
);
if (meta === undefined) {
@@ -28,7 +28,7 @@ export class PageRecord {
emit();
const dispose =
this.workspace.blockSuiteWorkspace.meta.docMetaUpdated.on(emit).dispose;
this.workspace.docCollection.meta.docMetaUpdated.on(emit).dispose;
return () => {
dispose();
};
@@ -42,7 +42,7 @@ export class PageRecord {
);
setMeta(meta: Partial<DocMeta>): void {
this.workspace.blockSuiteWorkspace.setDocMeta(this.id, meta);
this.workspace.docCollection.setDocMeta(this.id, meta);
}
mode: LiveData<PageMode> = LiveData.from(
@@ -22,7 +22,7 @@ describe('Workspace System', () => {
expect(workspaceListService.workspaceList.value.length).toBe(1);
const page = workspace.blockSuiteWorkspace.createDoc({
const page = workspace.docCollection.createDoc({
id: 'page0',
});
page.load();
@@ -30,7 +30,7 @@ describe('Workspace System', () => {
title: new page.Text('test-page'),
});
expect(workspace.blockSuiteWorkspace.docs.size).toBe(1);
expect(workspace.docCollection.docs.size).toBe(1);
expect(
(page!.getBlockByFlavour('affine:page')[0] as any).title.toString()
).toBe('test-page');
@@ -18,7 +18,7 @@
* })
*/
import { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import { DocCollection } from '@blocksuite/store';
import { nanoid } from 'nanoid';
import type { Awareness } from 'y-protocols/awareness.js';
import type { Doc as YDoc } from 'yjs';
@@ -29,7 +29,7 @@ import { globalBlockSuiteSchema } from './global-schema';
import type { WorkspaceMetadata } from './metadata';
import { WorkspaceScope } from './service-scope';
export const BlockSuiteWorkspaceContext = createIdentifier<BlockSuiteWorkspace>(
export const BlockSuiteWorkspaceContext = createIdentifier<DocCollection>(
'BlockSuiteWorkspaceContext'
);
@@ -53,7 +53,7 @@ export function configureWorkspaceContext(
.addImpl(WorkspaceMetadataContext, workspaceMetadata)
.addImpl(WorkspaceIdContext, workspaceMetadata.id)
.addImpl(BlockSuiteWorkspaceContext, provider => {
return new BlockSuiteWorkspace({
return new DocCollection({
id: workspaceMetadata.id,
blobStorages: [
() => ({
@@ -1,5 +1,5 @@
import { WorkspaceFlavour } from '@affine/env/workspace';
import { Workspace } from '@blocksuite/store';
import { DocCollection } from '@blocksuite/store';
import { beforeEach, describe, expect, test, vi } from 'vitest';
import { Doc } from 'yjs';
@@ -26,14 +26,14 @@ describe('SyncEngine', () => {
const storage2 = new MemoryMemento();
let prev: any;
{
const workspace = new Workspace({
const docCollection = new DocCollection({
id: 'test',
schema: globalBlockSuiteSchema,
});
const syncEngine = new SyncEngine(
workspace.doc,
docCollection.doc,
new TestingSyncStorage(testMeta, storage),
[
new TestingSyncStorage(testMeta, storage1),
@@ -42,7 +42,7 @@ describe('SyncEngine', () => {
);
syncEngine.start();
const page = workspace.createDoc({
const page = docCollection.createDoc({
id: 'page0',
});
page.load();
@@ -69,23 +69,23 @@ describe('SyncEngine', () => {
);
await syncEngine.waitForSynced();
syncEngine.forceStop();
prev = workspace.doc.toJSON();
prev = docCollection.doc.toJSON();
}
for (const current of [storage, storage1, storage2]) {
const workspace = new Workspace({
const docCollection = new DocCollection({
id: 'test',
schema: globalBlockSuiteSchema,
});
const syncEngine = new SyncEngine(
workspace.doc,
docCollection.doc,
new TestingSyncStorage(testMeta, current),
[]
);
syncEngine.start();
await syncEngine.waitForSynced();
expect(workspace.doc.toJSON()).toEqual({
expect(docCollection.doc.toJSON()).toEqual({
...prev,
});
syncEngine.forceStop();
@@ -1,5 +1,5 @@
import { WorkspaceFlavour } from '@affine/env/workspace';
import { Workspace } from '@blocksuite/store';
import { DocCollection } from '@blocksuite/store';
import { beforeEach, describe, expect, test, vi } from 'vitest';
import { MemoryMemento } from '../../../../storage';
@@ -23,19 +23,19 @@ describe('SyncPeer', () => {
let prev: any;
{
const workspace = new Workspace({
const docCollection = new DocCollection({
id: 'test',
schema: globalBlockSuiteSchema,
});
const syncPeer = new SyncPeer(
workspace.doc,
docCollection.doc,
new TestingSyncStorage(testMeta, storage)
);
await syncPeer.waitForLoaded();
const page = workspace.createDoc({
const page = docCollection.createDoc({
id: 'page0',
});
page.load();
@@ -62,21 +62,21 @@ describe('SyncPeer', () => {
);
await syncPeer.waitForSynced();
syncPeer.stop();
prev = workspace.doc.toJSON();
prev = docCollection.doc.toJSON();
}
{
const workspace = new Workspace({
const docCollection = new DocCollection({
id: 'test',
schema: globalBlockSuiteSchema,
});
const syncPeer = new SyncPeer(
workspace.doc,
docCollection.doc,
new TestingSyncStorage(testMeta, storage)
);
await syncPeer.waitForSynced();
expect(workspace.doc.toJSON()).toEqual({
expect(docCollection.doc.toJSON()).toEqual({
...prev,
});
syncPeer.stop();
@@ -86,21 +86,21 @@ describe('SyncPeer', () => {
test('status', async () => {
const storage = new MemoryMemento();
const workspace = new Workspace({
const docCollection = new DocCollection({
id: 'test',
schema: globalBlockSuiteSchema,
});
const syncPeer = new SyncPeer(
workspace.doc,
docCollection.doc,
new TestingSyncStorage(testMeta, storage)
);
expect(syncPeer.status.step).toBe(SyncPeerStep.LoadingRootDoc);
await syncPeer.waitForSynced();
expect(syncPeer.status.step).toBe(SyncPeerStep.Synced);
const page = workspace.createDoc({
const page = docCollection.createDoc({
id: 'page0',
});
expect(syncPeer.status.step).toBe(SyncPeerStep.LoadingSubDoc);
@@ -1,6 +1,6 @@
import { DebugLogger } from '@affine/debug';
import type { WorkspaceFlavour } from '@affine/env/workspace';
import type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { differenceWith } from 'lodash-es';
import { createIdentifier } from '../../di';
@@ -34,7 +34,7 @@ export interface WorkspaceListProvider {
*/
create(
initial: (
workspace: BlockSuiteWorkspace,
docCollection: DocCollection,
blobStorage: BlobStorage
) => Promise<void>
): Promise<WorkspaceMetadata>;
@@ -124,7 +124,7 @@ export class WorkspaceListService {
async create(
flavour: WorkspaceFlavour,
initial: (
workspace: BlockSuiteWorkspace,
docCollection: DocCollection,
blobStorage: BlobStorage
) => Promise<void> = () => Promise.resolve()
) {
@@ -57,13 +57,13 @@ export class WorkspaceInformation {
*/
syncWithWorkspace(workspace: Workspace) {
this.info = {
avatar: workspace.blockSuiteWorkspace.meta.avatar ?? this.info.avatar,
name: workspace.blockSuiteWorkspace.meta.name ?? this.info.name,
avatar: workspace.docCollection.meta.avatar ?? this.info.avatar,
name: workspace.docCollection.meta.name ?? this.info.name,
};
workspace.blockSuiteWorkspace.meta.commonFieldsUpdated.on(() => {
workspace.docCollection.meta.commonFieldsUpdated.on(() => {
this.info = {
avatar: workspace.blockSuiteWorkspace.meta.avatar ?? this.info.avatar,
name: workspace.blockSuiteWorkspace.meta.name ?? this.info.name,
avatar: workspace.docCollection.meta.avatar ?? this.info.avatar,
name: workspace.docCollection.meta.name ?? this.info.name,
};
});
}
@@ -2,7 +2,7 @@ import { DebugLogger } from '@affine/debug';
import { setupEditorFlags } from '@affine/env/global';
import { WorkspaceFlavour } from '@affine/env/workspace';
import { assertEquals } from '@blocksuite/global/utils';
import type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { applyUpdate, encodeStateAsUpdate } from 'yjs';
import { fixWorkspaceVersion } from '../blocksuite';
@@ -105,7 +105,7 @@ export class WorkspaceManager {
createWorkspace(
flavour: WorkspaceFlavour,
initial?: (
workspace: BlockSuiteWorkspace,
docCollection: DocCollection,
blobStorage: BlobStorage
) => Promise<void>
): Promise<WorkspaceMetadata> {
@@ -131,9 +131,9 @@ export class WorkspaceManager {
const newId = await this.list.create(
WorkspaceFlavour.AFFINE_CLOUD,
async (ws, bs) => {
applyUpdate(ws.doc, encodeStateAsUpdate(local.blockSuiteWorkspace.doc));
applyUpdate(ws.doc, encodeStateAsUpdate(local.docCollection.doc));
for (const subdoc of local.blockSuiteWorkspace.doc.getSubdocs()) {
for (const subdoc of local.docCollection.doc.getSubdocs()) {
for (const newSubdoc of ws.doc.getSubdocs()) {
if (newSubdoc.guid === subdoc.guid) {
applyUpdate(newSubdoc, encodeStateAsUpdate(subdoc));
@@ -191,9 +191,9 @@ export class WorkspaceManager {
const workspace = provider.get(Workspace);
// apply compatibility fix
fixWorkspaceVersion(workspace.blockSuiteWorkspace.doc);
fixWorkspaceVersion(workspace.docCollection.doc);
setupEditorFlags(workspace.blockSuiteWorkspace);
setupEditorFlags(workspace.docCollection);
return workspace;
}
@@ -1,5 +1,5 @@
import { WorkspaceFlavour } from '@affine/env/workspace';
import { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import { DocCollection } from '@blocksuite/store';
import { differenceBy } from 'lodash-es';
import { nanoid } from 'nanoid';
import { applyUpdate, encodeStateAsUpdate } from 'yjs';
@@ -43,7 +43,7 @@ export class TestingLocalWorkspaceListProvider
}
async create(
initial: (
workspace: BlockSuiteWorkspace,
docCollection: DocCollection,
blobStorage: BlobStorage
) => Promise<void>
): Promise<WorkspaceMetadata> {
@@ -53,18 +53,18 @@ export class TestingLocalWorkspaceListProvider
const blobStorage = new TestingBlobStorage(meta, this.state);
const syncStorage = new TestingSyncStorage(meta, this.state);
const workspace = new BlockSuiteWorkspace({
const docCollection = new DocCollection({
id: id,
idGenerator: () => nanoid(),
schema: globalBlockSuiteSchema,
});
// apply initial state
await initial(workspace, blobStorage);
await initial(docCollection, blobStorage);
// save workspace to storage
await syncStorage.push(id, encodeStateAsUpdate(workspace.doc));
for (const subdocs of workspace.doc.getSubdocs()) {
await syncStorage.push(id, encodeStateAsUpdate(docCollection.doc));
for (const subdocs of docCollection.doc.getSubdocs()) {
await syncStorage.push(subdocs.guid, encodeStateAsUpdate(subdocs));
}
@@ -117,7 +117,7 @@ export class TestingLocalWorkspaceListProvider
return;
}
const bs = new BlockSuiteWorkspace({
const bs = new DocCollection({
id,
schema: globalBlockSuiteSchema,
});
+14 -17
View File
@@ -1,7 +1,7 @@
import { Unreachable } from '@affine/env/constant';
import { WorkspaceFlavour } from '@affine/env/workspace';
import { Slot } from '@blocksuite/global/utils';
import type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { applyUpdate, Doc as YDoc, encodeStateAsUpdate } from 'yjs';
import { checkWorkspaceCompatibility, MigrationPoint } from '../blocksuite';
@@ -38,18 +38,18 @@ export class WorkspaceUpgradeController {
}
constructor(
private readonly blockSuiteWorkspace: BlockSuiteWorkspace,
private readonly docCollection: DocCollection,
private readonly sync: SyncEngine,
private readonly workspaceMetadata: WorkspaceMetadata
) {
blockSuiteWorkspace.doc.on('update', () => {
docCollection.doc.on('update', () => {
this.checkIfNeedUpgrade();
});
}
checkIfNeedUpgrade() {
const needUpgrade = !!checkWorkspaceCompatibility(
this.blockSuiteWorkspace,
this.docCollection,
this.workspaceMetadata.flavour === WorkspaceFlavour.AFFINE_CLOUD
);
this.status = {
@@ -72,7 +72,7 @@ export class WorkspaceUpgradeController {
await this.sync.waitForSynced();
const step = checkWorkspaceCompatibility(
this.blockSuiteWorkspace,
this.docCollection,
this.workspaceMetadata.flavour === WorkspaceFlavour.AFFINE_CLOUD
);
@@ -82,9 +82,9 @@ export class WorkspaceUpgradeController {
// Clone a new doc to prevent change events.
const clonedDoc = new YDoc({
guid: this.blockSuiteWorkspace.doc.guid,
guid: this.docCollection.doc.guid,
});
applyDoc(clonedDoc, this.blockSuiteWorkspace.doc);
applyDoc(clonedDoc, this.docCollection.doc);
if (step === MigrationPoint.SubDoc) {
const newWorkspace = await workspaceManager.createWorkspace(
@@ -92,14 +92,11 @@ export class WorkspaceUpgradeController {
async (workspace, blobStorage) => {
await upgradeV1ToV2(clonedDoc, workspace.doc);
migrateGuidCompatibility(clonedDoc);
await forceUpgradePages(
workspace.doc,
this.blockSuiteWorkspace.schema
);
const blobList = await this.blockSuiteWorkspace.blob.list();
await forceUpgradePages(workspace.doc, this.docCollection.schema);
const blobList = await this.docCollection.blob.list();
for (const blobKey of blobList) {
const blob = await this.blockSuiteWorkspace.blob.get(blobKey);
const blob = await this.docCollection.blob.get(blobKey);
if (blob) {
await blobStorage.set(blobKey, blob);
}
@@ -110,13 +107,13 @@ export class WorkspaceUpgradeController {
return newWorkspace;
} else if (step === MigrationPoint.GuidFix) {
migrateGuidCompatibility(clonedDoc);
await forceUpgradePages(clonedDoc, this.blockSuiteWorkspace.schema);
applyDoc(this.blockSuiteWorkspace.doc, clonedDoc);
await forceUpgradePages(clonedDoc, this.docCollection.schema);
applyDoc(this.docCollection.doc, clonedDoc);
await this.sync.waitForSynced();
return null;
} else if (step === MigrationPoint.BlockVersion) {
await forceUpgradePages(clonedDoc, this.blockSuiteWorkspace.schema);
applyDoc(this.blockSuiteWorkspace.doc, clonedDoc);
await forceUpgradePages(clonedDoc, this.docCollection.schema);
applyDoc(this.docCollection.doc, clonedDoc);
await this.sync.waitForSynced();
return null;
} else {
@@ -1,6 +1,6 @@
import { DebugLogger } from '@affine/debug';
import { Slot } from '@blocksuite/global/utils';
import type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { ServiceProvider } from '../di';
import { CleanupService } from '../lifecycle';
@@ -10,7 +10,7 @@ import { type WorkspaceMetadata } from './metadata';
import type { WorkspaceUpgradeController } from './upgrade';
import { type WorkspaceUpgradeStatus } from './upgrade';
export type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
export type { DocCollection } from '@blocksuite/store';
const logger = new DebugLogger('affine:workspace');
@@ -67,7 +67,7 @@ export class Workspace {
constructor(
public meta: WorkspaceMetadata,
public engine: WorkspaceEngine,
public blockSuiteWorkspace: BlockSuiteWorkspace,
public docCollection: DocCollection,
public upgrade: WorkspaceUpgradeController,
public services: ServiceProvider
) {
+3 -3
View File
@@ -32,14 +32,14 @@
}
},
"dependencies": {
"@blocksuite/global": "0.13.0-canary-202403050653-934469c",
"@blocksuite/global": "0.13.0-canary-202403120738-e15d583",
"idb": "^8.0.0",
"nanoid": "^5.0.6",
"y-provider": "workspace:*"
},
"devDependencies": {
"@blocksuite/blocks": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/blocks": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"fake-indexeddb": "^5.0.2",
"vite": "^5.1.4",
"vite-plugin-dts": "3.7.3",
@@ -8,7 +8,7 @@ import { setTimeout } from 'node:timers/promises';
import { AffineSchemas } from '@blocksuite/blocks/schemas';
import { assertExists } from '@blocksuite/global/utils';
import type { Doc } from '@blocksuite/store';
import { Schema, Workspace } from '@blocksuite/store';
import { DocCollection, Schema } from '@blocksuite/store';
import { openDB } from 'idb';
import { nanoid } from 'nanoid';
import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest';
@@ -69,7 +69,7 @@ async function getUpdates(id: string): Promise<Uint8Array[]> {
}
let id: string;
let workspace: Workspace;
let docCollection: DocCollection;
const rootDBName = DEFAULT_DB_NAME;
const schema = new Schema();
@@ -78,7 +78,7 @@ schema.register(AffineSchemas);
beforeEach(() => {
id = nanoid();
workspace = new Workspace({
docCollection = new DocCollection({
id,
schema,
@@ -93,7 +93,7 @@ afterEach(() => {
describe('indexeddb provider', () => {
test('connect', async () => {
const provider = createIndexedDBProvider(workspace.doc);
const provider = createIndexedDBProvider(docCollection.doc);
provider.connect();
// todo: has a better way to know when data is synced
@@ -110,11 +110,11 @@ describe('indexeddb provider', () => {
updates: [
{
timestamp: expect.any(Number),
update: encodeStateAsUpdate(workspace.doc),
update: encodeStateAsUpdate(docCollection.doc),
},
],
});
const page = workspace.createDoc({ id: 'page0' });
const page = docCollection.createDoc({ id: 'page0' });
page.load();
const pageBlockId = page.addBlock(
'affine:page' as keyof BlockSuite.BlockModels,
@@ -139,13 +139,13 @@ describe('indexeddb provider', () => {
const data = (await store.get(id)) as WorkspacePersist | undefined;
assertExists(data);
expect(data.id).toBe(id);
const testWorkspace = new Workspace({
const testWorkspace = new DocCollection({
id: 'test',
schema,
});
// data should only contain updates for the root doc
data.updates.forEach(({ update }) => {
Workspace.Y.applyUpdate(testWorkspace.doc, update);
DocCollection.Y.applyUpdate(testWorkspace.doc, update);
});
const subPage = testWorkspace.doc.spaces.get('page0');
{
@@ -157,23 +157,23 @@ describe('indexeddb provider', () => {
assertExists(data);
testWorkspace.getDoc('page0')?.load();
data.updates.forEach(({ update }) => {
Workspace.Y.applyUpdate(subPage, update);
DocCollection.Y.applyUpdate(subPage, update);
});
}
expect(workspace.doc.toJSON()).toEqual(testWorkspace.doc.toJSON());
expect(docCollection.doc.toJSON()).toEqual(testWorkspace.doc.toJSON());
}
});
test('connect and disconnect', async () => {
const provider = createIndexedDBProvider(workspace.doc, rootDBName);
const provider = createIndexedDBProvider(docCollection.doc, rootDBName);
provider.connect();
expect(provider.connected).toBe(true);
await setTimeout(200);
const snapshot = encodeStateAsUpdate(workspace.doc);
const snapshot = encodeStateAsUpdate(docCollection.doc);
provider.disconnect();
expect(provider.connected).toBe(false);
{
const page = workspace.createDoc({ id: 'page0' });
const page = docCollection.createDoc({ id: 'page0' });
page.load();
const pageBlockId = page.addBlock(
'affine:page' as keyof BlockSuite.BlockModels
@@ -190,7 +190,7 @@ describe('indexeddb provider', () => {
);
}
{
const updates = await getUpdates(workspace.id);
const updates = await getUpdates(docCollection.id);
expect(updates.length).toBe(1);
expect(updates[0]).toEqual(snapshot);
}
@@ -199,7 +199,7 @@ describe('indexeddb provider', () => {
expect(provider.connected).toBe(true);
await setTimeout(200);
{
const updates = await getUpdates(workspace.id);
const updates = await getUpdates(docCollection.id);
expect(updates).not.toEqual([]);
}
expect(provider.connected).toBe(true);
@@ -208,7 +208,7 @@ describe('indexeddb provider', () => {
});
test('cleanup', async () => {
const provider = createIndexedDBProvider(workspace.doc);
const provider = createIndexedDBProvider(docCollection.doc);
provider.connect();
await setTimeout(200);
const db = await openDB(rootDBName, dbVersion);
@@ -218,7 +218,7 @@ describe('indexeddb provider', () => {
.transaction('workspace', 'readonly')
.objectStore('workspace');
const keys = await store.getAllKeys();
expect(keys).contain(workspace.id);
expect(keys).contain(docCollection.id);
}
await provider.cleanup();
@@ -229,16 +229,16 @@ describe('indexeddb provider', () => {
.transaction('workspace', 'readonly')
.objectStore('workspace');
const keys = await store.getAllKeys();
expect(keys).not.contain(workspace.id);
expect(keys).not.contain(docCollection.id);
}
});
test('merge', async () => {
setMergeCount(5);
const provider = createIndexedDBProvider(workspace.doc, rootDBName);
const provider = createIndexedDBProvider(docCollection.doc, rootDBName);
provider.connect();
{
const page = workspace.createDoc({ id: 'page0' });
const page = docCollection.createDoc({ id: 'page0' });
page.load();
const pageBlockId = page.addBlock(
'affine:page' as keyof BlockSuite.BlockModels
@@ -264,7 +264,7 @@ describe('indexeddb provider', () => {
});
test("data won't be lost", async () => {
const doc = new Workspace.Y.Doc();
const doc = new DocCollection.Y.Doc();
const map = doc.getMap('map');
for (let i = 0; i < 100; i++) {
map.set(`${i}`, i);
@@ -275,7 +275,7 @@ describe('indexeddb provider', () => {
provider.disconnect();
}
{
const newDoc = new Workspace.Y.Doc();
const newDoc = new DocCollection.Y.Doc();
const provider = createIndexedDBProvider(newDoc, rootDBName);
provider.connect();
provider.disconnect();
@@ -412,14 +412,14 @@ describe('subDoc', () => {
});
test('blocksuite', async () => {
const page0 = workspace.createDoc({
const page0 = docCollection.createDoc({
id: 'page0',
});
page0.load();
const { paragraphBlockId: paragraphBlockIdPage1 } = initEmptyPage(page0);
const provider = createIndexedDBProvider(workspace.doc, rootDBName);
const provider = createIndexedDBProvider(docCollection.doc, rootDBName);
provider.connect();
const page1 = workspace.createDoc({
const page1 = docCollection.createDoc({
id: 'page1',
});
page1.load();
@@ -427,22 +427,22 @@ describe('subDoc', () => {
await setTimeout(200);
provider.disconnect();
{
const newWorkspace = new Workspace({
const docCollection = new DocCollection({
id,
schema,
});
const provider = createIndexedDBProvider(newWorkspace.doc, rootDBName);
const provider = createIndexedDBProvider(docCollection.doc, rootDBName);
provider.connect();
await setTimeout(200);
const page0 = newWorkspace.getDoc('page0') as Doc;
const page0 = docCollection.getDoc('page0') as Doc;
page0.load();
await setTimeout(200);
{
const block = page0.getBlockById(paragraphBlockIdPage1);
assertExists(block);
}
const page1 = newWorkspace.getDoc('page1') as Doc;
const page1 = docCollection.getDoc('page1') as Doc;
page1.load();
await setTimeout(200);
{
@@ -455,30 +455,30 @@ describe('subDoc', () => {
describe('utils', () => {
test('download binary', async () => {
const page = workspace.createDoc({ id: 'page0' });
const page = docCollection.createDoc({ id: 'page0' });
page.load();
initEmptyPage(page);
const provider = createIndexedDBProvider(workspace.doc, rootDBName);
const provider = createIndexedDBProvider(docCollection.doc, rootDBName);
provider.connect();
await setTimeout(200);
provider.disconnect();
const update = (await downloadBinary(
workspace.id,
docCollection.id,
rootDBName
)) as Uint8Array;
expect(update).toBeInstanceOf(Uint8Array);
const newWorkspace = new Workspace({
const newDocCollection = new DocCollection({
id,
schema,
});
applyUpdate(newWorkspace.doc, update);
applyUpdate(newDocCollection.doc, update);
await setTimeout();
expect(workspace.doc.toJSON()['meta']).toEqual(
newWorkspace.doc.toJSON()['meta']
expect(docCollection.doc.toJSON()['meta']).toEqual(
newDocCollection.doc.toJSON()['meta']
);
expect(Object.keys(workspace.doc.toJSON()['spaces'])).toEqual(
Object.keys(newWorkspace.doc.toJSON()['spaces'])
expect(Object.keys(docCollection.doc.toJSON()['spaces'])).toEqual(
Object.keys(newDocCollection.doc.toJSON()['spaces'])
);
});
+1 -1
View File
@@ -24,7 +24,7 @@
"build": "vite build"
},
"devDependencies": {
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"vite": "^5.1.4",
"vite-plugin-dts": "3.7.3",
"vitest": "1.3.1",
+5 -5
View File
@@ -72,12 +72,12 @@
"uuid": "^9.0.1"
},
"devDependencies": {
"@blocksuite/blocks": "0.13.0-canary-202403050653-934469c",
"@blocksuite/global": "0.13.0-canary-202403050653-934469c",
"@blocksuite/blocks": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/global": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/icons": "2.1.45",
"@blocksuite/lit": "0.13.0-canary-202403050653-934469c",
"@blocksuite/presets": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/lit": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/presets": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"@storybook/addon-actions": "^7.6.17",
"@storybook/addon-essentials": "^7.6.17",
"@storybook/addon-interactions": "^7.6.17",
+7 -7
View File
@@ -25,14 +25,14 @@
"@affine/i18n": "workspace:*",
"@affine/templates": "workspace:*",
"@affine/workspace-impl": "workspace:*",
"@blocksuite/block-std": "0.13.0-canary-202403050653-934469c",
"@blocksuite/blocks": "0.13.0-canary-202403050653-934469c",
"@blocksuite/global": "0.13.0-canary-202403050653-934469c",
"@blocksuite/block-std": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/blocks": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/global": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/icons": "2.1.45",
"@blocksuite/inline": "0.13.0-canary-202403050653-934469c",
"@blocksuite/lit": "0.13.0-canary-202403050653-934469c",
"@blocksuite/presets": "0.13.0-canary-202403050653-934469c",
"@blocksuite/store": "0.13.0-canary-202403050653-934469c",
"@blocksuite/inline": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/lit": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/presets": "0.13.0-canary-202403120738-e15d583",
"@blocksuite/store": "0.13.0-canary-202403120738-e15d583",
"@dnd-kit/core": "^6.1.0",
"@dnd-kit/modifiers": "^7.0.0",
"@dnd-kit/sortable": "^8.0.0",
@@ -1,7 +1,7 @@
import { WorkspaceSubPath } from '@affine/core/shared';
import type { useAFFiNEI18N } from '@affine/i18n/hooks';
import { ArrowRightBigIcon } from '@blocksuite/icons';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { registerAffineCommand } from '@toeverything/infra/command';
import type { createStore } from 'jotai';
@@ -11,13 +11,13 @@ import type { useNavigateHelper } from '../hooks/use-navigate-helper';
export function registerAffineNavigationCommands({
t,
store,
workspace,
docCollection,
navigationHelper,
}: {
t: ReturnType<typeof useAFFiNEI18N>;
store: ReturnType<typeof createStore>;
navigationHelper: ReturnType<typeof useNavigateHelper>;
workspace: Workspace;
docCollection: DocCollection;
}) {
const unsubs: Array<() => void> = [];
unsubs.push(
@@ -27,7 +27,7 @@ export function registerAffineNavigationCommands({
icon: <ArrowRightBigIcon />,
label: t['com.affine.cmdk.affine.navigation.goto-all-pages'](),
run() {
navigationHelper.jumpToSubPath(workspace.id, WorkspaceSubPath.ALL);
navigationHelper.jumpToSubPath(docCollection.id, WorkspaceSubPath.ALL);
},
})
);
@@ -39,7 +39,7 @@ export function registerAffineNavigationCommands({
icon: <ArrowRightBigIcon />,
label: 'Go to Collection List',
run() {
navigationHelper.jumpToCollections(workspace.id);
navigationHelper.jumpToCollections(docCollection.id);
},
})
);
@@ -51,7 +51,7 @@ export function registerAffineNavigationCommands({
icon: <ArrowRightBigIcon />,
label: 'Go to Tag List',
run() {
navigationHelper.jumpToTags(workspace.id);
navigationHelper.jumpToTags(docCollection.id);
},
})
);
@@ -91,7 +91,10 @@ export function registerAffineNavigationCommands({
icon: <ArrowRightBigIcon />,
label: t['com.affine.cmdk.affine.navigation.goto-trash'](),
run() {
navigationHelper.jumpToSubPath(workspace.id, WorkspaceSubPath.TRASH);
navigationHelper.jumpToSubPath(
docCollection.id,
WorkspaceSubPath.TRASH
);
},
})
);
@@ -14,7 +14,7 @@ const SyncAwarenessInnerLoggedIn = () => {
useEffect(() => {
if (user && currentWorkspace) {
currentWorkspace.blockSuiteWorkspace.awarenessStore.awareness.setLocalStateField(
currentWorkspace.docCollection.awarenessStore.awareness.setLocalStateField(
'user',
{
name: user.name,
@@ -23,7 +23,7 @@ const SyncAwarenessInnerLoggedIn = () => {
);
return () => {
currentWorkspace.blockSuiteWorkspace.awarenessStore.awareness.setLocalStateField(
currentWorkspace.docCollection.awarenessStore.awareness.setLocalStateField(
'user',
null
);
@@ -1,5 +1,5 @@
import { useDocMetaHelper } from '@affine/core/hooks/use-block-suite-page-meta';
import { useBlockSuiteWorkspacePage } from '@affine/core/hooks/use-block-suite-workspace-page';
import { useDocCollectionPage } from '@affine/core/hooks/use-block-suite-workspace-page';
import { timestampToLocalDate } from '@affine/core/utils';
import { DebugLogger } from '@affine/debug';
import {
@@ -10,7 +10,7 @@ import {
} from '@affine/graphql';
import { AffineCloudBlobStorage } from '@affine/workspace-impl';
import { assertEquals } from '@blocksuite/global/utils';
import { Workspace } from '@blocksuite/store';
import { DocCollection } from '@blocksuite/store';
import { globalBlockSuiteSchema } from '@toeverything/infra';
import { revertUpdate } from '@toeverything/y-indexeddb';
import { useEffect, useMemo } from 'react';
@@ -98,14 +98,14 @@ const snapshotFetcher = async (
// so that we do not need to worry about providers etc
// todo: fix references to the page (the referenced page will shown as deleted)
// if we simply clone the current workspace, it maybe time consuming right?
const workspaceMap = new Map<string, Workspace>();
const docCollectionMap = new Map<string, DocCollection>();
// assume the workspace is a cloud workspace since the history feature is only enabled for cloud workspace
const getOrCreateShellWorkspace = (workspaceId: string) => {
let workspace = workspaceMap.get(workspaceId);
if (!workspace) {
let docCollection = docCollectionMap.get(workspaceId);
if (!docCollection) {
const blobStorage = new AffineCloudBlobStorage(workspaceId);
workspace = new Workspace({
docCollection = new DocCollection({
id: workspaceId,
blobStorages: [
() => ({
@@ -114,10 +114,10 @@ const getOrCreateShellWorkspace = (workspaceId: string) => {
],
schema: globalBlockSuiteSchema,
});
workspaceMap.set(workspaceId, workspace);
workspace.doc.emit('sync', []);
docCollectionMap.set(workspaceId, docCollection);
docCollection.doc.emit('sync', []);
}
return workspace;
return docCollection;
};
// workspace id + page id + timestamp -> snapshot (update binary)
@@ -139,17 +139,17 @@ export const usePageHistory = (
// workspace id + page id + timestamp + snapshot -> Page (to be used for rendering in blocksuite editor)
export const useSnapshotPage = (
workspace: Workspace,
docCollection: DocCollection,
pageDocId: string,
ts?: string
) => {
const snapshot = usePageHistory(workspace.id, pageDocId, ts);
const snapshot = usePageHistory(docCollection.id, pageDocId, ts);
const page = useMemo(() => {
if (!ts) {
return;
}
const pageId = pageDocId + '-' + ts;
const historyShellWorkspace = getOrCreateShellWorkspace(workspace.id);
const historyShellWorkspace = getOrCreateShellWorkspace(docCollection.id);
let page = historyShellWorkspace.getDoc(pageId);
if (!page && snapshot) {
page = historyShellWorkspace.createDoc({
@@ -163,15 +163,15 @@ export const useSnapshotPage = (
}); // must load before applyUpdate
}
return page ?? undefined;
}, [pageDocId, snapshot, ts, workspace]);
}, [pageDocId, snapshot, ts, docCollection]);
useEffect(() => {
const historyShellWorkspace = getOrCreateShellWorkspace(workspace.id);
const historyShellWorkspace = getOrCreateShellWorkspace(docCollection.id);
// apply the rootdoc's update to the current workspace
// this makes sure the page reference links are not deleted ones in the preview
const update = encodeStateAsUpdate(workspace.doc);
const update = encodeStateAsUpdate(docCollection.doc);
applyUpdate(historyShellWorkspace.doc, update);
}, [workspace]);
}, [docCollection]);
return page;
};
@@ -187,13 +187,16 @@ export const historyListGroupByDay = (histories: DocHistory[]) => {
return [...map.entries()];
};
export const useRestorePage = (workspace: Workspace, pageId: string) => {
const page = useBlockSuiteWorkspacePage(workspace, pageId);
export const useRestorePage = (
docCollection: DocCollection,
pageId: string
) => {
const page = useDocCollectionPage(docCollection, pageId);
const mutateQueryResource = useMutateQueryResource();
const { trigger: recover, isMutating } = useMutation({
mutation: recoverDocMutation,
});
const { getDocMeta, setDocTitle } = useDocMetaHelper(workspace);
const { getDocMeta, setDocTitle } = useDocMetaHelper(docCollection);
const onRestore = useMemo(() => {
return async (version: string, update: Uint8Array) => {
@@ -216,12 +219,12 @@ export const useRestorePage = (workspace: Workspace, pageId: string) => {
await recover({
docId: pageDocId,
timestamp: version,
workspaceId: workspace.id,
workspaceId: docCollection.id,
});
await mutateQueryResource(listHistoryQuery, vars => {
return (
vars.pageDocId === pageDocId && vars.workspaceId === workspace.id
vars.pageDocId === pageDocId && vars.workspaceId === docCollection.id
);
});
@@ -234,7 +237,7 @@ export const useRestorePage = (workspace: Workspace, pageId: string) => {
pageId,
recover,
setDocTitle,
workspace.id,
docCollection.id,
]);
return {
@@ -5,13 +5,13 @@ import { ConfirmModal, Modal } from '@affine/component/ui/modal';
import { openSettingModalAtom } from '@affine/core/atoms';
import { useIsWorkspaceOwner } from '@affine/core/hooks/affine/use-is-workspace-owner';
import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
import { useBlockSuiteWorkspacePageTitle } from '@affine/core/hooks/use-block-suite-workspace-page-title';
import { useDocCollectionPageTitle } from '@affine/core/hooks/use-block-suite-workspace-page-title';
import { useWorkspaceQuota } from '@affine/core/hooks/use-workspace-quota';
import { Trans } from '@affine/i18n';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { CloseIcon, ToggleCollapseIcon } from '@blocksuite/icons';
import type { Doc as BlockSuiteDoc } from '@blocksuite/store';
import { type Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import { type DocCollection } from '@blocksuite/store';
import * as Collapsible from '@radix-ui/react-collapsible';
import type { DialogContentProps } from '@radix-ui/react-dialog';
import { Doc, type PageMode, Workspace } from '@toeverything/infra';
@@ -49,7 +49,7 @@ import * as styles from './styles.css';
export interface PageHistoryModalProps {
open: boolean;
onOpenChange: (open: boolean) => void;
workspace: BlockSuiteWorkspace;
docCollection: DocCollection;
pageId: string;
}
@@ -444,26 +444,26 @@ const EmptyHistoryPrompt = () => {
};
const PageHistoryManager = ({
workspace,
docCollection,
pageId,
onClose,
}: {
workspace: BlockSuiteWorkspace;
docCollection: DocCollection;
pageId: string;
onClose: () => void;
}) => {
const workspaceId = workspace.id;
const workspaceId = docCollection.id;
const [activeVersion, setActiveVersion] = useState<string>();
const pageDocId = useMemo(() => {
return workspace.getDoc(pageId)?.spaceDoc.guid ?? pageId;
}, [pageId, workspace]);
return docCollection.getDoc(pageId)?.spaceDoc.guid ?? pageId;
}, [pageId, docCollection]);
const snapshotPage = useSnapshotPage(workspace, pageDocId, activeVersion);
const snapshotPage = useSnapshotPage(docCollection, pageDocId, activeVersion);
const t = useAFFiNEI18N();
const { onRestore, isMutating } = useRestorePage(workspace, pageId);
const { onRestore, isMutating } = useRestorePage(docCollection, pageId);
const handleRestore = useMemo(
() => async () => {
@@ -481,7 +481,7 @@ const PageHistoryManager = ({
const page = useService(Doc);
const [mode, setMode] = useState<PageMode>(page.mode.value);
const title = useBlockSuiteWorkspacePageTitle(workspace, pageId);
const title = useDocCollectionPageTitle(docCollection, pageId);
const [showRestoreConfirmModal, setShowRestoreConfirmModal] = useState(false);
@@ -558,7 +558,7 @@ export const PageHistoryModal = ({
onOpenChange,
open,
pageId,
workspace,
docCollection: workspace,
}: PageHistoryModalProps) => {
const onClose = useCallback(() => {
onOpenChange(false);
@@ -570,7 +570,7 @@ export const PageHistoryModal = ({
<PageHistoryManager
onClose={onClose}
pageId={pageId}
workspace={workspace}
docCollection={workspace}
/>
</Suspense>
</ModalContainer>
@@ -595,7 +595,7 @@ export const GlobalPageHistoryModal = () => {
open={open}
onOpenChange={handleOpenChange}
pageId={pageId}
workspace={workspace.blockSuiteWorkspace}
docCollection={workspace.docCollection}
/>
);
};
@@ -124,7 +124,7 @@ export class PagePropertiesMetaManager {
// returns page schema properties -> related page
getPropertyStatistics() {
const mapping = new Map<string, Set<string>>();
for (const page of this.adapter.workspace.blockSuiteWorkspace.docs.values()) {
for (const page of this.adapter.workspace.docCollection.docs.values()) {
const properties = this.adapter.getPageProperties(page.id);
for (const id of Object.keys(properties.custom)) {
if (!mapping.has(id)) mapping.set(id, new Set());
@@ -169,7 +169,7 @@ export class PagePropertiesManager {
}
get page() {
return this.adapter.workspace.blockSuiteWorkspace.getDoc(this.pageId);
return this.adapter.workspace.docCollection.getDoc(this.pageId);
}
get intrinsicMeta() {
@@ -177,8 +177,8 @@ export const NumberValue = ({ property }: PropertyRowValueProps) => {
export const TagsValue = () => {
const workspace = useService(Workspace);
const page = useService(Doc);
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const pageMetas = useAllBlockSuiteDocMeta(blockSuiteWorkspace);
const docCollection = workspace.docCollection;
const pageMetas = useAllBlockSuiteDocMeta(docCollection);
const legacyProperties = useService(WorkspaceLegacyProperties);
const options = useLiveData(legacyProperties.tagOptions$);
@@ -405,7 +405,7 @@ export const PageBacklinksPopup = ({
key={pageId}
wrapper={MenuItem}
pageId={pageId}
workspace={manager.workspace.blockSuiteWorkspace}
docCollection={manager.workspace.docCollection}
/>
))}
</div>
@@ -600,7 +600,7 @@ export const PagePropertiesTableHeader = ({
const t = useAFFiNEI18N();
const backlinks = useBlockSuitePageBacklinks(
manager.workspace.blockSuiteWorkspace,
manager.workspace.docCollection,
manager.pageId
);
@@ -1015,7 +1015,7 @@ export const PagePropertiesAddProperty = () => {
const PagePropertiesTableInner = () => {
const manager = useContext(managerContext);
const [expanded, setExpanded] = useState(false);
use(manager.workspace.blockSuiteWorkspace.doc.whenSynced);
use(manager.workspace.docCollection.doc.whenSynced);
return (
<div className={styles.root}>
<Collapsible.Root
@@ -3,7 +3,7 @@ import { useJournalHelper } from '@affine/core/hooks/use-journal';
import { WorkbenchLink } from '@affine/core/modules/workbench';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { LinkedPageIcon, TodayIcon } from '@blocksuite/icons';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { PropsWithChildren } from 'react';
import * as styles from './styles.css';
@@ -44,15 +44,15 @@ export function pageReferenceRenderer({
export function AffinePageReference({
pageId,
workspace,
docCollection,
wrapper: Wrapper,
}: {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
wrapper?: React.ComponentType<PropsWithChildren>;
}) {
const pageMetaHelper = useDocMetaHelper(workspace);
const journalHelper = useJournalHelper(workspace);
const pageMetaHelper = useDocMetaHelper(docCollection);
const journalHelper = useJournalHelper(docCollection);
const t = useAFFiNEI18N();
const el = pageReferenceRenderer({
pageId,
@@ -46,17 +46,13 @@ export const ProfilePanel = ({ isOwner, workspace }: ProfilePanelProps) => {
const avatarUrl = useWorkspaceBlobObjectUrl(workspace?.meta, avatarBlob);
useEffect(() => {
if (workspace?.blockSuiteWorkspace) {
setAvatarBlob(workspace.blockSuiteWorkspace.meta.avatar ?? null);
setName(
workspace.blockSuiteWorkspace.meta.name ?? UNTITLED_WORKSPACE_NAME
);
const dispose = workspace.blockSuiteWorkspace.meta.commonFieldsUpdated.on(
if (workspace?.docCollection) {
setAvatarBlob(workspace.docCollection.meta.avatar ?? null);
setName(workspace.docCollection.meta.name ?? UNTITLED_WORKSPACE_NAME);
const dispose = workspace.docCollection.meta.commonFieldsUpdated.on(
() => {
setAvatarBlob(workspace.blockSuiteWorkspace.meta.avatar ?? null);
setName(
workspace.blockSuiteWorkspace.meta.name ?? UNTITLED_WORKSPACE_NAME
);
setAvatarBlob(workspace.docCollection.meta.avatar ?? null);
setName(workspace.docCollection.meta.name ?? UNTITLED_WORKSPACE_NAME);
}
);
return () => {
@@ -75,14 +71,14 @@ export const ProfilePanel = ({ isOwner, workspace }: ProfilePanelProps) => {
return;
}
if (!file) {
workspace.blockSuiteWorkspace.meta.setAvatar('');
workspace.docCollection.meta.setAvatar('');
return;
}
try {
const reducedFile = await validateAndReduceImage(file);
const blobs = workspace.blockSuiteWorkspace.blob;
const blobs = workspace.docCollection.blob;
const blobId = await blobs.set(reducedFile);
workspace.blockSuiteWorkspace.meta.setAvatar(blobId);
workspace.docCollection.meta.setAvatar(blobId);
} catch (error) {
console.error(error);
throw error;
@@ -96,7 +92,7 @@ export const ProfilePanel = ({ isOwner, workspace }: ProfilePanelProps) => {
if (!workspace) {
return;
}
workspace.blockSuiteWorkspace.meta.setName(name);
workspace.docCollection.meta.setName(name);
},
[workspace]
);
@@ -110,8 +110,8 @@ const BlockSuiteEditorImpl = forwardRef<AffineEditorContainer, EditorProps>(
};
}, []);
const pageMetaHelper = useDocMetaHelper(page.workspace);
const journalHelper = useJournalHelper(page.workspace);
const pageMetaHelper = useDocMetaHelper(page.collection);
const journalHelper = useJournalHelper(page.collection);
const t = useAFFiNEI18N();
const customRenderers = useMemo(() => {
@@ -6,7 +6,7 @@ import * as styles from './styles.css';
export const BlocksuiteEditorJournalDocTitle = ({ page }: { page: Doc }) => {
const { localizedJournalDate, isTodayJournal, journalDate } =
useJournalInfoHelper(page.workspace, page.id);
useJournalInfoHelper(page.collection, page.id);
const t = useAFFiNEI18N();
// TODO: i18n
@@ -64,7 +64,7 @@ export const BlocksuiteDocEditor = forwardRef<
const docRef = useRef<PageEditor | null>(null);
const [docPage, setDocPage] =
useState<HTMLElementTagNameMap['affine-page-root']>();
const { isJournal } = useJournalInfoHelper(page.workspace, page.id);
const { isJournal } = useJournalInfoHelper(page.collection, page.id);
const onDocRef = useCallback(
(el: PageEditor) => {
@@ -15,17 +15,17 @@ export interface FavoriteButtonProps {
export const useFavorite = (pageId: string) => {
const t = useAFFiNEI18N();
const workspace = useService(Workspace);
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const currentPage = blockSuiteWorkspace.getDoc(pageId);
const docCollection = workspace.docCollection;
const currentPage = docCollection.getDoc(pageId);
assertExists(currentPage);
const pageMeta = useBlockSuiteDocMeta(blockSuiteWorkspace).find(
const pageMeta = useBlockSuiteDocMeta(docCollection).find(
meta => meta.id === pageId
);
const favorite = pageMeta?.favorite ?? false;
const { toggleFavorite: _toggleFavorite } =
useBlockSuiteMetaHelper(blockSuiteWorkspace);
useBlockSuiteMetaHelper(docCollection);
const toggleFavorite = useCallback(() => {
_toggleFavorite(pageId);
@@ -3,24 +3,24 @@ import {
useJournalInfoHelper,
useJournalRouteHelper,
} from '@affine/core/hooks/use-journal';
import type { BlockSuiteWorkspace } from '@affine/core/shared';
import type { DocCollection } from '@affine/core/shared';
import type { Doc } from '@blocksuite/store';
import dayjs from 'dayjs';
import { useEffect, useRef, useState } from 'react';
export interface JournalWeekDatePickerProps {
workspace: BlockSuiteWorkspace;
docCollection: DocCollection;
page: Doc;
}
const weekStyle = { maxWidth: 800, width: '100%' };
export const JournalWeekDatePicker = ({
workspace,
docCollection,
page,
}: JournalWeekDatePickerProps) => {
const handleRef = useRef<WeekDatePickerHandle>(null);
const { journalDate } = useJournalInfoHelper(workspace, page.id);
const { openJournal } = useJournalRouteHelper(workspace);
const { journalDate } = useJournalInfoHelper(docCollection, page.id);
const { openJournal } = useJournalRouteHelper(docCollection);
const [date, setDate] = useState(
(journalDate ?? dayjs()).format('YYYY-MM-DD')
);
@@ -1,16 +1,18 @@
import { Button } from '@affine/component';
import { useJournalRouteHelper } from '@affine/core/hooks/use-journal';
import type { BlockSuiteWorkspace } from '@affine/core/shared';
import type { DocCollection } from '@affine/core/shared';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { useCallback } from 'react';
export interface JournalTodayButtonProps {
workspace: BlockSuiteWorkspace;
docCollection: DocCollection;
}
export const JournalTodayButton = ({ workspace }: JournalTodayButtonProps) => {
export const JournalTodayButton = ({
docCollection,
}: JournalTodayButtonProps) => {
const t = useAFFiNEI18N();
const journalHelper = useJournalRouteHelper(workspace);
const journalHelper = useJournalRouteHelper(docCollection);
const onToday = useCallback(() => {
journalHelper.openToday();
@@ -47,11 +47,11 @@ export const PageHeaderMenuButton = ({
const t = useAFFiNEI18N();
const workspace = useService(Workspace);
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const currentPage = blockSuiteWorkspace.getDoc(pageId);
const docCollection = workspace.docCollection;
const currentPage = docCollection.getDoc(pageId);
assertExists(currentPage);
const pageMeta = useBlockSuiteDocMeta(blockSuiteWorkspace).find(
const pageMeta = useBlockSuiteDocMeta(docCollection).find(
meta => meta.id === pageId
);
const page = useService(Doc);
@@ -59,9 +59,9 @@ export const PageHeaderMenuButton = ({
const { favorite, toggleFavorite } = useFavorite(pageId);
const { duplicate } = useBlockSuiteMetaHelper(blockSuiteWorkspace);
const { importFile } = usePageHelper(blockSuiteWorkspace);
const { setTrashModal } = useTrashModalHelper(blockSuiteWorkspace);
const { duplicate } = useBlockSuiteMetaHelper(docCollection);
const { importFile } = usePageHelper(docCollection);
const { setTrashModal } = useTrashModalHelper(docCollection);
const [historyModalOpen, setHistoryModalOpen] = useState(false);
const setOpenHistoryTipsModal = useSetAtom(openHistoryTipsModalAtom);
@@ -227,7 +227,7 @@ export const PageHeaderMenuButton = ({
</Menu>
{workspace.flavour === WorkspaceFlavour.AFFINE_CLOUD ? (
<PageHistoryModal
workspace={workspace.blockSuiteWorkspace}
docCollection={workspace.docCollection}
open={historyModalOpen}
pageId={pageId}
onOpenChange={setHistoryModalOpen}
@@ -3,14 +3,14 @@ import {
useBlockSuiteDocMeta,
useDocMetaHelper,
} from '@affine/core/hooks/use-block-suite-page-meta';
import type { BlockSuiteWorkspace } from '@affine/core/shared';
import type { DocCollection } from '@affine/core/shared';
import type { HTMLAttributes } from 'react';
import { useCallback } from 'react';
import * as styles from './style.css';
export interface BlockSuiteHeaderTitleProps {
blockSuiteWorkspace: BlockSuiteWorkspace;
docCollection: DocCollection;
pageId: string;
/** if set, title cannot be edited */
isPublic?: boolean;
@@ -21,18 +21,13 @@ const inputAttrs = {
'data-testid': 'title-content',
} as HTMLAttributes<HTMLInputElement>;
export const BlocksuiteHeaderTitle = (props: BlockSuiteHeaderTitleProps) => {
const {
blockSuiteWorkspace: workspace,
pageId,
isPublic,
inputHandleRef,
} = props;
const currentPage = workspace.getDoc(pageId);
const pageMeta = useBlockSuiteDocMeta(workspace).find(
const { docCollection, pageId, isPublic, inputHandleRef } = props;
const currentPage = docCollection.getDoc(pageId);
const pageMeta = useBlockSuiteDocMeta(docCollection).find(
meta => meta.id === currentPage?.id
);
const title = pageMeta?.title;
const { setDocTitle } = useDocMetaHelper(workspace);
const { setDocTitle } = useDocMetaHelper(docCollection);
const onChange = useCallback(
(v: string) => {
@@ -10,14 +10,14 @@ import {
import type { CSSProperties } from 'react';
import { useCallback, useEffect } from 'react';
import type { BlockSuiteWorkspace } from '../../../shared';
import type { DocCollection } from '../../../shared';
import { toast } from '../../../utils';
import { StyledEditorModeSwitch, StyledKeyboardItem } from './style';
import { EdgelessSwitchItem, PageSwitchItem } from './switch-items';
export type EditorModeSwitchProps = {
// todo(himself65): combine these two properties
blockSuiteWorkspace: BlockSuiteWorkspace;
docCollection: DocCollection;
pageId: string;
style?: CSSProperties;
isPublic?: boolean;
@@ -36,13 +36,13 @@ const TooltipContent = () => {
};
export const EditorModeSwitch = ({
style,
blockSuiteWorkspace,
docCollection,
pageId,
isPublic,
publicMode,
}: EditorModeSwitchProps) => {
const t = useAFFiNEI18N();
const pageMeta = useBlockSuiteDocMeta(blockSuiteWorkspace).find(
const pageMeta = useBlockSuiteDocMeta(docCollection).find(
meta => meta.id === pageId
);
const trash = pageMeta?.trash ?? false;
@@ -1,7 +1,7 @@
import { toast } from '@affine/component';
import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
import { useDocMetaHelper } from '@affine/core/hooks/use-block-suite-page-meta';
import { useBlockSuiteWorkspaceHelper } from '@affine/core/hooks/use-block-suite-workspace-helper';
import { useDocCollectionHelper } from '@affine/core/hooks/use-block-suite-workspace-helper';
import { WorkspaceSubPath } from '@affine/core/shared';
import { useService } from '@toeverything/infra';
import { PageRecordList } from '@toeverything/infra';
@@ -9,12 +9,12 @@ import { initEmptyPage } from '@toeverything/infra';
import { useCallback, useMemo } from 'react';
import { useNavigateHelper } from '../../../hooks/use-navigate-helper';
import type { BlockSuiteWorkspace } from '../../../shared';
import type { DocCollection } from '../../../shared';
export const usePageHelper = (blockSuiteWorkspace: BlockSuiteWorkspace) => {
export const usePageHelper = (docCollection: DocCollection) => {
const { openPage, jumpToSubPath } = useNavigateHelper();
const { createDoc } = useBlockSuiteWorkspaceHelper(blockSuiteWorkspace);
const { setDocMeta } = useDocMetaHelper(blockSuiteWorkspace);
const { createDoc } = useDocCollectionHelper(docCollection);
const { setDocMeta } = useDocMetaHelper(docCollection);
const pageRecordList = useService(PageRecordList);
const isPreferredEdgeless = useCallback(
@@ -28,10 +28,10 @@ export const usePageHelper = (blockSuiteWorkspace: BlockSuiteWorkspace) => {
const page = createDoc();
initEmptyPage(page);
pageRecordList.record(page.id).value?.setMode(mode || 'page');
openPage(blockSuiteWorkspace.id, page.id);
openPage(docCollection.id, page.id);
return page;
},
[blockSuiteWorkspace.id, createDoc, openPage, pageRecordList]
[docCollection.id, createDoc, openPage, pageRecordList]
);
const createEdgelessAndOpen = useCallback(() => {
@@ -50,7 +50,7 @@ export const usePageHelper = (blockSuiteWorkspace: BlockSuiteWorkspace) => {
}.`
);
if (options.isWorkspaceFile) {
jumpToSubPath(blockSuiteWorkspace.id, WorkspaceSubPath.ALL);
jumpToSubPath(docCollection.id, WorkspaceSubPath.ALL);
return;
}
@@ -58,16 +58,16 @@ export const usePageHelper = (blockSuiteWorkspace: BlockSuiteWorkspace) => {
return;
}
const pageId = pageIds[0];
openPage(blockSuiteWorkspace.id, pageId);
openPage(docCollection.id, pageId);
};
showImportModal({ workspace: blockSuiteWorkspace, onSuccess });
}, [blockSuiteWorkspace, openPage, jumpToSubPath]);
showImportModal({ collection: docCollection, onSuccess });
}, [docCollection, openPage, jumpToSubPath]);
const createLinkedPageAndOpen = useAsyncCallback(
async (pageId: string) => {
const page = createPageAndOpen();
page.load();
const parentPage = blockSuiteWorkspace.getDoc(pageId);
const parentPage = docCollection.getDoc(pageId);
if (parentPage) {
parentPage.load();
const text = parentPage.Text.fromDelta([
@@ -86,7 +86,7 @@ export const usePageHelper = (blockSuiteWorkspace: BlockSuiteWorkspace) => {
setDocMeta(page.id, {});
}
},
[blockSuiteWorkspace, createPageAndOpen, setDocMeta]
[docCollection, createPageAndOpen, setDocMeta]
);
return useMemo(() => {
@@ -13,7 +13,7 @@ import {
PlusIcon,
ViewBarIcon,
} from '@blocksuite/icons';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import clsx from 'clsx';
import { useErrorBoundary } from 'foxact/use-error-boundary';
import { useAtom } from 'jotai';
@@ -45,7 +45,7 @@ import {
import { hasAnimationPlayedAtom, previewBlockIdAtom } from './index.jotai';
export type ImagePreviewModalProps = {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
};
@@ -92,7 +92,7 @@ const ImagePreviewModalImpl = (
const nextImageHandler = useCallback(
(blockId: string | null) => {
assertExists(blockId);
const workspace = props.workspace;
const workspace = props.docCollection;
if (!hasPlayedAnimation) {
setHasPlayedAnimation(true);
}
@@ -109,13 +109,13 @@ const ImagePreviewModalImpl = (
setBlockId(nextBlock.id);
}
},
[props.pageId, props.workspace, setBlockId, hasPlayedAnimation]
[props.pageId, props.docCollection, setBlockId, hasPlayedAnimation]
);
const previousImageHandler = useCallback(
(blockId: string | null) => {
assertExists(blockId);
const workspace = props.workspace;
const workspace = props.docCollection;
const page = workspace.getDoc(props.pageId);
assertExists(page);
const block = page.getBlockById(blockId);
@@ -130,12 +130,12 @@ const ImagePreviewModalImpl = (
}
resetZoom();
},
[props.pageId, props.workspace, setBlockId, resetZoom]
[props.pageId, props.docCollection, setBlockId, resetZoom]
);
const deleteHandler = useCallback(
(blockId: string) => {
const { pageId, workspace, onClose } = props;
const { pageId, docCollection: workspace, onClose } = props;
const page = workspace.getDoc(pageId);
assertExists(page);
@@ -185,7 +185,7 @@ const ImagePreviewModalImpl = (
const downloadHandler = useCallback(
async (blockId: string | null) => {
const workspace = props.workspace;
const workspace = props.docCollection;
const page = workspace.getDoc(props.pageId);
assertExists(page);
if (typeof blockId === 'string') {
@@ -237,31 +237,31 @@ const ImagePreviewModalImpl = (
a.remove();
}
},
[props.pageId, props.workspace]
[props.pageId, props.docCollection]
);
const [caption, setCaption] = useState(() => {
const page = props.workspace.getDoc(props.pageId);
const page = props.docCollection.getDoc(props.pageId);
assertExists(page);
const block = page.getBlockById(props.blockId) as ImageBlockModel;
assertExists(block);
return block?.caption;
});
useEffect(() => {
const page = props.workspace.getDoc(props.pageId);
const page = props.docCollection.getDoc(props.pageId);
assertExists(page);
const block = page.getBlockById(props.blockId) as ImageBlockModel;
assertExists(block);
setCaption(block?.caption);
}, [props.blockId, props.pageId, props.workspace]);
}, [props.blockId, props.pageId, props.docCollection]);
const { data, error } = useSWR(
['workspace', 'image', props.pageId, props.blockId],
{
fetcher: ([_, __, pageId, blockId]) => {
const page = props.workspace.getDoc(pageId);
const page = props.docCollection.getDoc(pageId);
assertExists(page);
const block = page.getBlockById(blockId) as ImageBlockModel;
assertExists(block);
return props.workspace.blob.get(block?.sourceId as string);
return props.docCollection.blob.get(block?.sourceId as string);
},
suspense: true,
}
@@ -508,7 +508,7 @@ export const ImagePreviewModal = (
return;
}
const workspace = props.workspace;
const workspace = props.docCollection;
const page = workspace.getDoc(props.pageId);
assertExists(page);
@@ -541,7 +541,7 @@ export const ImagePreviewModal = (
event.preventDefault();
event.stopPropagation();
},
[blockId, setBlockId, props.workspace, props.pageId, isOpen, setIsOpen]
[blockId, setBlockId, props.docCollection, props.pageId, isOpen, setIsOpen]
);
useEffect(() => {
@@ -1,9 +1,9 @@
import './page-detail-editor.css';
import { useBlockSuiteWorkspacePage } from '@affine/core/hooks/use-block-suite-workspace-page';
import { useDocCollectionPage } from '@affine/core/hooks/use-block-suite-workspace-page';
import { assertExists, DisposableGroup } from '@blocksuite/global/utils';
import type { AffineEditorContainer } from '@blocksuite/presets';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { Doc as BlockSuiteDoc } from '@blocksuite/store';
import {
Doc,
@@ -34,7 +34,7 @@ export type OnLoadEditor = (
export interface PageDetailEditorProps {
isPublic?: boolean;
publishMode?: PageMode;
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
onLoad?: OnLoadEditor;
}
@@ -78,7 +78,7 @@ const PageDetailEditorMain = memo(function PageDetailEditorMain({
const disposableGroup = new DisposableGroup();
disposableGroup.add(
page.slots.blockUpdated.once(() => {
page.workspace.setDocMeta(page.id, {
page.collection.setDocMeta(page.id, {
updatedDate: Date.now(),
});
})
@@ -121,8 +121,8 @@ const PageDetailEditorMain = memo(function PageDetailEditorMain({
});
export const PageDetailEditor = (props: PageDetailEditorProps) => {
const { workspace, pageId } = props;
const page = useBlockSuiteWorkspacePage(workspace, pageId);
const { docCollection, pageId } = props;
const page = useDocCollectionPage(docCollection, pageId);
if (!page) {
return null;
}
@@ -6,21 +6,21 @@ import 'fake-indexeddb/auto';
import { AffineSchemas } from '@blocksuite/blocks/schemas';
import { assertExists } from '@blocksuite/global/utils';
import type { Doc } from '@blocksuite/store';
import { Schema, Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import { DocCollection, Schema } from '@blocksuite/store';
import { renderHook } from '@testing-library/react';
import { useAtomValue } from 'jotai';
import { describe, expect, test, vi } from 'vitest';
import { beforeEach } from 'vitest';
import { useBlockSuitePagePreview } from '../use-block-suite-page-preview';
let blockSuiteWorkspace: BlockSuiteWorkspace;
let docCollection: DocCollection;
const schema = new Schema();
schema.register(AffineSchemas);
beforeEach(async () => {
vi.useFakeTimers({ toFake: ['requestIdleCallback'] });
blockSuiteWorkspace = new BlockSuiteWorkspace({ id: 'test', schema });
docCollection = new DocCollection({ id: 'test', schema });
const initPage = async (page: Doc) => {
page.load();
expect(page).not.toBeNull();
@@ -31,12 +31,12 @@ beforeEach(async () => {
const frameId = page.addBlock('affine:note', {}, pageBlockId);
page.addBlock('affine:paragraph', {}, frameId);
};
await initPage(blockSuiteWorkspace.createDoc({ id: 'page0' }));
await initPage(docCollection.createDoc({ id: 'page0' }));
});
describe('useBlockSuitePagePreview', () => {
test('basic', async () => {
const page = blockSuiteWorkspace.getDoc('page0') as Doc;
const page = docCollection.getDoc('page0') as Doc;
const id = page.addBlock(
'affine:paragraph',
{
@@ -125,7 +125,7 @@ export const VirtualizedCollectionList = ({
items={collectionMetas}
itemRenderer={collectionItemRenderer}
rowAsLink
blockSuiteWorkspace={currentWorkspace.blockSuiteWorkspace}
docCollection={currentWorkspace.docCollection}
operationsRenderer={collectionOperationRenderer}
headerRenderer={collectionHeaderRenderer}
/>
@@ -18,7 +18,7 @@ export const PageListNewPageButton = ({
}>) => {
const currentWorkspace = useService(Workspace);
const { importFile, createEdgeless, createPage } = usePageHelper(
currentWorkspace.blockSuiteWorkspace
currentWorkspace.docCollection
);
return (
<div className={className} data-testid={testId}>
@@ -28,11 +28,9 @@ import {
const usePageOperationsRenderer = () => {
const currentWorkspace = useService(Workspace);
const { setTrashModal } = useTrashModalHelper(
currentWorkspace.blockSuiteWorkspace
);
const { setTrashModal } = useTrashModalHelper(currentWorkspace.docCollection);
const { toggleFavorite, duplicate } = useBlockSuiteMetaHelper(
currentWorkspace.blockSuiteWorkspace
currentWorkspace.docCollection
);
const t = useAFFiNEI18N();
@@ -97,11 +95,9 @@ export const VirtualizedPageList = ({
const [showFloatingToolbar, setShowFloatingToolbar] = useState(false);
const [selectedPageIds, setSelectedPageIds] = useState<string[]>([]);
const currentWorkspace = useService(Workspace);
const pageMetas = useBlockSuiteDocMeta(currentWorkspace.blockSuiteWorkspace);
const pageMetas = useBlockSuiteDocMeta(currentWorkspace.docCollection);
const pageOperations = usePageOperationsRenderer();
const { isPreferredEdgeless } = usePageHelper(
currentWorkspace.blockSuiteWorkspace
);
const { isPreferredEdgeless } = usePageHelper(currentWorkspace.docCollection);
const filteredPageMetas = useFilteredPageMetas(currentWorkspace, pageMetas, {
filters,
@@ -155,9 +151,7 @@ export const VirtualizedPageList = ({
return <PageListHeader />;
}, [collection, config, currentWorkspace.id, tag]);
const { setTrashModal } = useTrashModalHelper(
currentWorkspace.blockSuiteWorkspace
);
const { setTrashModal } = useTrashModalHelper(currentWorkspace.docCollection);
const handleMultiDelete = useCallback(() => {
const pageNameMapping = Object.fromEntries(
@@ -190,7 +184,7 @@ export const VirtualizedPageList = ({
items={pageMetasToRender}
rowAsLink
isPreferredEdgeless={isPreferredEdgeless}
blockSuiteWorkspace={currentWorkspace.blockSuiteWorkspace}
docCollection={currentWorkspace.docCollection}
operationsRenderer={pageOperationRenderer}
itemRenderer={pageItemRenderer}
headerRenderer={pageHeaderRenderer}
@@ -1,31 +1,34 @@
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { useAtomValue } from 'jotai';
import { Suspense } from 'react';
import { useBlockSuitePagePreview } from './use-block-suite-page-preview';
import { useBlockSuiteWorkspacePage } from './use-block-suite-workspace-page';
import { useDocCollectionPage } from './use-block-suite-workspace-page';
interface PagePreviewInnerProps {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
}
const PagePreviewInner = ({ workspace, pageId }: PagePreviewInnerProps) => {
const page = useBlockSuiteWorkspacePage(workspace, pageId);
const PagePreviewInner = ({
docCollection: workspace,
pageId,
}: PagePreviewInnerProps) => {
const page = useDocCollectionPage(workspace, pageId);
const previewAtom = useBlockSuitePagePreview(page);
const preview = useAtomValue(previewAtom);
return preview ? preview : null;
};
interface PagePreviewProps {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
}
export const PagePreview = ({ workspace, pageId }: PagePreviewProps) => {
export const PagePreview = ({ docCollection, pageId }: PagePreviewProps) => {
return (
<Suspense>
<PagePreviewInner workspace={workspace} pageId={pageId} />
<PagePreviewInner docCollection={docCollection} pageId={pageId} />
</Suspense>
);
};
@@ -9,7 +9,7 @@ import {
ToggleCollapseIcon,
ViewLayersIcon,
} from '@blocksuite/icons';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import * as Collapsible from '@radix-ui/react-collapsible';
import { useService } from '@toeverything/infra';
import { useLiveData } from '@toeverything/infra';
@@ -196,7 +196,7 @@ export const ItemGroup = <T extends ListItem>({
// todo: optimize how to render page meta list item
const requiredPropNames = [
'blockSuiteWorkspace',
'docCollection',
'rowAsLink',
'isPreferredEdgeless',
'operationsRenderer',
@@ -266,9 +266,9 @@ export const TagListItemRenderer = (item: ListItem) => {
function tagIdToTagOption(
tagId: string,
blockSuiteWorkspace: Workspace
docCollection: DocCollection
): Tag | undefined {
return blockSuiteWorkspace.meta.properties.tags?.options.find(
return docCollection.meta.properties.tags?.options.find(
opt => opt.id === tagId
);
}
@@ -286,15 +286,15 @@ const PageTitle = ({ id }: { id: string }) => {
const UnifiedPageIcon = ({
id,
workspace,
docCollection,
isPreferredEdgeless,
}: {
id: string;
workspace: Workspace;
docCollection: DocCollection;
isPreferredEdgeless?: (id: string) => boolean;
}) => {
const isEdgeless = isPreferredEdgeless ? isPreferredEdgeless(id) : false;
const { isJournal } = useJournalInfoHelper(workspace, id);
const { isJournal } = useJournalInfoHelper(docCollection, id);
if (isJournal) {
return <TodayIcon />;
}
@@ -325,7 +325,7 @@ function pageMetaToListItemProp(
pageId: item.id,
title: <PageTitle id={item.id} />,
preview: (
<PagePreview workspace={props.blockSuiteWorkspace} pageId={item.id} />
<PagePreview docCollection={props.docCollection} pageId={item.id} />
),
createDate: new Date(item.createDate),
updatedDate: item.updatedDate ? new Date(item.updatedDate) : undefined,
@@ -334,13 +334,13 @@ function pageMetaToListItemProp(
icon: (
<UnifiedPageIcon
id={item.id}
workspace={props.blockSuiteWorkspace}
docCollection={props.docCollection}
isPreferredEdgeless={props.isPreferredEdgeless}
/>
),
tags:
item.tags
?.map(id => tagIdToTagOption(id, props.blockSuiteWorkspace))
?.map(id => tagIdToTagOption(id, props.docCollection))
.filter((v): v is Tag => v != null) ?? [],
operations: props.operationsRenderer?.(item),
selectable: props.selectable,
@@ -98,7 +98,7 @@ export const VirtualizedTagList = ({
items={tagMetas}
itemRenderer={tagItemRenderer}
rowAsLink
blockSuiteWorkspace={currentWorkspace.blockSuiteWorkspace}
docCollection={currentWorkspace.docCollection}
operationsRenderer={tagOperationRenderer}
headerRenderer={tagHeaderRenderer}
/>
@@ -1,5 +1,5 @@
import type { Collection, Tag } from '@affine/env/filter';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import type { PropsWithChildren, ReactNode } from 'react';
import type { To } from 'react-router-dom';
@@ -86,7 +86,7 @@ export type DateKey = 'createDate' | 'updatedDate';
export interface ListProps<T> {
// required data:
items: T[];
blockSuiteWorkspace: Workspace;
docCollection: DocCollection;
className?: string;
hideHeader?: boolean; // whether or not to hide the header. default is false (showing header)
groupBy?: ItemGroupByType | false;
@@ -1,29 +1,29 @@
import { DebugLogger } from '@affine/debug';
import { DisposableGroup } from '@blocksuite/global/utils';
import type { Doc, Workspace } from '@blocksuite/store';
import type { Doc, DocCollection } from '@blocksuite/store';
import { useEffect, useState } from 'react';
const logger = new DebugLogger('useBlockSuiteWorkspacePage');
export function useBlockSuiteWorkspacePage(
blockSuiteWorkspace: Workspace,
export function useDocCollectionPage(
docCollection: DocCollection,
pageId: string | null
): Doc | null {
const [page, setPage] = useState(
pageId ? blockSuiteWorkspace.getDoc(pageId) : null
pageId ? docCollection.getDoc(pageId) : null
);
useEffect(() => {
const group = new DisposableGroup();
group.add(
blockSuiteWorkspace.slots.docAdded.on(id => {
docCollection.slots.docAdded.on(id => {
if (pageId === id) {
setPage(blockSuiteWorkspace.getDoc(id));
setPage(docCollection.getDoc(id));
}
})
);
group.add(
blockSuiteWorkspace.slots.docRemoved.on(id => {
docCollection.slots.docRemoved.on(id => {
if (pageId === id) {
setPage(null);
}
@@ -32,7 +32,7 @@ export function useBlockSuiteWorkspacePage(
return () => {
group.dispose();
};
}, [blockSuiteWorkspace, pageId]);
}, [docCollection, pageId]);
useEffect(() => {
if (page && !page.loaded) {
@@ -46,9 +46,9 @@ export function useBlockSuiteWorkspacePage(
useEffect(() => {
if (page?.id !== pageId) {
setPage(pageId ? blockSuiteWorkspace.getDoc(pageId) : null);
setPage(pageId ? docCollection.getDoc(pageId) : null);
}
}, [blockSuiteWorkspace, page?.id, pageId]);
}, [docCollection, page?.id, pageId]);
return page;
}
@@ -6,7 +6,7 @@ import {
} from '@affine/component';
import type { Collection } from '@affine/env/filter';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import type { DialogContentProps } from '@radix-ui/react-dialog';
import { type ReactNode, useCallback, useMemo, useState } from 'react';
@@ -199,7 +199,7 @@ export const EditCollection = ({
export type AllPageListConfig = {
allPages: DocMeta[];
workspace: Workspace;
docCollection: DocCollection;
isEdgeless: (id: string) => boolean;
/**
* Return `undefined` if the page is not public
@@ -87,7 +87,9 @@ export const PagesMode = ({
<Menu
items={
<VariableSelect
propertiesMeta={allPageListConfig.workspace.meta.properties}
propertiesMeta={
allPageListConfig.docCollection.meta.properties
}
selected={filters}
onSelect={createFilter}
/>
@@ -114,7 +116,7 @@ export const PagesMode = ({
{showFilter ? (
<div style={{ padding: '12px 16px 16px' }}>
<FilterList
propertiesMeta={allPageListConfig.workspace.meta.properties}
propertiesMeta={allPageListConfig.docCollection.meta.properties}
value={filters}
onChange={updateFilters}
/>
@@ -125,7 +127,7 @@ export const PagesMode = ({
className={styles.pageList}
items={searchedList}
groupBy={false}
blockSuiteWorkspace={allPageListConfig.workspace}
docCollection={allPageListConfig.docCollection}
selectable
onSelectedIdsChange={ids => {
updateCollection({
@@ -123,7 +123,7 @@ export const RulesMode = ({
}}
>
<FilterList
propertiesMeta={allPageListConfig.workspace.meta.properties}
propertiesMeta={allPageListConfig.docCollection.meta.properties}
value={collection.filterList}
onChange={useCallback(
filterList => updateCollection({ ...collection, filterList }),
@@ -265,7 +265,7 @@ export const RulesMode = ({
className={styles.resultPages}
items={rulesPages}
groupBy={false}
blockSuiteWorkspace={allPageListConfig.workspace}
docCollection={allPageListConfig.docCollection}
isPreferredEdgeless={allPageListConfig.isEdgeless}
operationsRenderer={operationsRenderer}
></List>
@@ -285,7 +285,7 @@ export const RulesMode = ({
className={styles.resultPages}
items={allowListPages}
groupBy={false}
blockSuiteWorkspace={allPageListConfig.workspace}
docCollection={allPageListConfig.docCollection}
isPreferredEdgeless={allPageListConfig.isEdgeless}
operationsRenderer={operationsRenderer}
></List>
@@ -76,7 +76,9 @@ export const SelectPage = ({
<Menu
items={
<VariableSelect
propertiesMeta={allPageListConfig.workspace.meta.properties}
propertiesMeta={
allPageListConfig.docCollection.meta.properties
}
selected={filters}
onSelect={createFilter}
/>
@@ -103,7 +105,7 @@ export const SelectPage = ({
{showFilter ? (
<div style={{ padding: '12px 16px 16px' }}>
<FilterList
propertiesMeta={allPageListConfig.workspace.meta.properties}
propertiesMeta={allPageListConfig.docCollection.meta.properties}
value={filters}
onChange={updateFilters}
/>
@@ -113,7 +115,7 @@ export const SelectPage = ({
<VirtualizedList
className={styles.pageList}
items={searchedList}
blockSuiteWorkspace={allPageListConfig.workspace}
docCollection={allPageListConfig.docCollection}
selectable
groupBy={false}
onSelectedIdsChange={onChange}
@@ -2,7 +2,7 @@ import {
useBlockSuiteDocMeta,
useDocMetaHelper,
} from '@affine/core/hooks/use-block-suite-page-meta';
import { useGetBlockSuiteWorkspacePageTitle } from '@affine/core/hooks/use-block-suite-workspace-page-title';
import { useGetDocCollectionPageTitle } from '@affine/core/hooks/use-block-suite-workspace-page-title';
import { useJournalHelper } from '@affine/core/hooks/use-journal';
import { CollectionService } from '@affine/core/modules/collection';
import { WorkspaceSubPath } from '@affine/core/shared';
@@ -79,7 +79,7 @@ function getAllCommand(context: CommandContext) {
const useWorkspacePages = () => {
const workspace = useService(Workspace);
const pages = useBlockSuiteDocMeta(workspace.blockSuiteWorkspace);
const pages = useBlockSuiteDocMeta(workspace.docCollection);
return pages;
};
@@ -100,7 +100,7 @@ export const pageToCommand = (
category: CommandCategory,
page: DocMeta,
navigationHelper: ReturnType<typeof useNavigateHelper>,
getPageTitle: ReturnType<typeof useGetBlockSuiteWorkspacePageTitle>,
getPageTitle: ReturnType<typeof useGetDocCollectionPageTitle>,
isPageJournal: (pageId: string) => boolean,
t: ReturnType<typeof useAFFiNEI18N>,
workspace: Workspace,
@@ -151,16 +151,14 @@ export const usePageCommands = () => {
const recentPages = useRecentPages();
const pages = useWorkspacePages();
const workspace = useService(Workspace);
const pageHelper = usePageHelper(workspace.blockSuiteWorkspace);
const pageMetaHelper = useDocMetaHelper(workspace.blockSuiteWorkspace);
const pageHelper = usePageHelper(workspace.docCollection);
const pageMetaHelper = useDocMetaHelper(workspace.docCollection);
const query = useAtomValue(cmdkQueryAtom);
const navigationHelper = useNavigateHelper();
const journalHelper = useJournalHelper(workspace.blockSuiteWorkspace);
const journalHelper = useJournalHelper(workspace.docCollection);
const t = useAFFiNEI18N();
const getPageTitle = useGetBlockSuiteWorkspacePageTitle(
workspace.blockSuiteWorkspace
);
const { isPageJournal } = useJournalHelper(workspace.blockSuiteWorkspace);
const getPageTitle = useGetDocCollectionPageTitle(workspace.docCollection);
const { isPageJournal } = useJournalHelper(workspace.docCollection);
const [searchTime, setSearchTime] = useState<number>(0);
@@ -197,7 +195,7 @@ export const usePageCommands = () => {
} else {
// queried pages that has matched contents
// TODO: we shall have a debounce for global search here
const searchResults = workspace.blockSuiteWorkspace.search({
const searchResults = workspace.docCollection.search({
query,
}) as unknown as Map<string, SearchResultsValue>;
const resultValues = Array.from(searchResults.values());
@@ -242,7 +240,7 @@ export const usePageCommands = () => {
if (!appendRes) return;
const { page, blockId } = appendRes;
navigationHelper.jumpToPageBlock(
page.workspace.id,
page.collection.id,
page.id,
blockId
);
@@ -22,15 +22,15 @@ export const TrashPageFooter = ({ pageId }: { pageId: string }) => {
useService(CurrentWorkspaceService).currentWorkspace
);
assertExists(workspace);
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const pageMeta = useBlockSuiteDocMeta(blockSuiteWorkspace).find(
const docCollection = workspace.docCollection;
const pageMeta = useBlockSuiteDocMeta(docCollection).find(
meta => meta.id === pageId
);
assertExists(pageMeta);
const t = useAFFiNEI18N();
const { appSettings } = useAppSettingHelper();
const { jumpToSubPath } = useNavigateHelper();
const { restoreFromTrash } = useBlockSuiteMetaHelper(blockSuiteWorkspace);
const { restoreFromTrash } = useBlockSuiteMetaHelper(docCollection);
const [open, setOpen] = useState(false);
const hintText = t['com.affine.cmdk.affine.editor.trash-footer-hint']();
@@ -45,9 +45,9 @@ export const TrashPageFooter = ({ pageId }: { pageId: string }) => {
const onConfirmDelete = useCallback(() => {
jumpToSubPath(workspace.id, WorkspaceSubPath.ALL);
blockSuiteWorkspace.removeDoc(pageId);
docCollection.removeDoc(pageId);
toast(t['com.affine.toastMessage.permanentlyDeleted']());
}, [blockSuiteWorkspace, jumpToSubPath, pageId, workspace.id, t]);
}, [docCollection, jumpToSubPath, pageId, workspace.id, t]);
const onDelete = useCallback(() => {
setOpen(true);
@@ -10,7 +10,7 @@ import { CollectionService } from '@affine/core/modules/collection';
import type { Collection, DeleteCollectionInfo } from '@affine/env/filter';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { MoreHorizontalIcon, ViewLayersIcon } from '@blocksuite/icons';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import { useDroppable } from '@dnd-kit/core';
import * as Collapsible from '@radix-ui/react-collapsible';
import { useService } from '@toeverything/infra';
@@ -30,12 +30,12 @@ import * as styles from './styles.css';
const CollectionRenderer = ({
collection,
pages,
workspace,
docCollection,
info,
}: {
collection: Collection;
pages: DocMeta[];
workspace: Workspace;
docCollection: DocCollection;
info: DeleteCollectionInfo;
}) => {
const [collapsed, setCollapsed] = useState(true);
@@ -159,7 +159,7 @@ const CollectionRenderer = ({
allPageMeta={allPagesMeta}
page={page}
key={page.id}
workspace={workspace}
docCollection={docCollection}
/>
);
})}
@@ -169,7 +169,7 @@ const CollectionRenderer = ({
);
};
export const CollectionsList = ({
workspace,
docCollection: workspace,
info,
onCreate,
}: CollectionsListProps) => {
@@ -205,7 +205,7 @@ export const CollectionsList = ({
key={view.id}
collection={view}
pages={metas}
workspace={workspace}
docCollection={workspace}
/>
);
})}
@@ -1,7 +1,7 @@
import { useBlockSuitePageReferences } from '@affine/core/hooks/use-block-suite-page-references';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { EdgelessIcon, PageIcon } from '@blocksuite/icons';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import { useDraggable } from '@dnd-kit/core';
import * as Collapsible from '@radix-ui/react-collapsible';
import { PageRecordList, useLiveData, useService } from '@toeverything/infra';
@@ -18,7 +18,7 @@ import * as styles from './styles.css';
export const Page = ({
page,
workspace,
docCollection,
allPageMeta,
inAllowList,
removeFromAllowList,
@@ -26,7 +26,7 @@ export const Page = ({
page: DocMeta;
inAllowList: boolean;
removeFromAllowList: (id: string) => void;
workspace: Workspace;
docCollection: DocCollection;
allPageMeta: Record<string, DocMeta>;
}) => {
const [collapsed, setCollapsed] = React.useState(true);
@@ -46,10 +46,10 @@ export const Page = ({
const { jumpToPage } = useNavigateHelper();
const clickPage = useCallback(() => {
jumpToPage(workspace.id, page.id);
}, [jumpToPage, page.id, workspace.id]);
jumpToPage(docCollection.id, page.id);
}, [jumpToPage, page.id, docCollection.id]);
const references = useBlockSuitePageReferences(workspace, pageId);
const references = useBlockSuitePageReferences(docCollection, pageId);
const referencesToRender = references.filter(
id => allPageMeta[id] && !allPageMeta[id]?.trash
);
@@ -85,7 +85,7 @@ export const Page = ({
onCollapsedChange={setCollapsed}
postfix={
<PostfixItem
workspace={workspace}
docCollection={docCollection}
pageId={pageId}
pageTitle={pageTitle}
removeFromAllowList={removeFromAllowList}
@@ -103,7 +103,7 @@ export const Page = ({
return (
<ReferencePage
key={id}
workspace={workspace}
docCollection={docCollection}
pageId={id}
metaMapping={allPageMeta}
parentIds={new Set([pageId])}
@@ -3,7 +3,7 @@ import { IconButton } from '@affine/component/ui/button';
import { Menu } from '@affine/component/ui/menu';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { MoreHorizontalIcon } from '@blocksuite/icons';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { useCallback } from 'react';
import { useBlockSuiteMetaHelper } from '../../../../hooks/affine/use-block-suite-meta-helper';
@@ -13,7 +13,7 @@ import { OperationItems } from './operation-item';
export type OperationMenuButtonProps = {
pageId: string;
workspace: Workspace;
docCollection: DocCollection;
pageTitle: string;
setRenameModalOpen: () => void;
inFavorites?: boolean;
@@ -24,7 +24,7 @@ export type OperationMenuButtonProps = {
export const OperationMenuButton = ({ ...props }: OperationMenuButtonProps) => {
const {
workspace,
docCollection,
pageId,
pageTitle,
setRenameModalOpen,
@@ -34,9 +34,9 @@ export const OperationMenuButton = ({ ...props }: OperationMenuButtonProps) => {
isReferencePage,
} = props;
const t = useAFFiNEI18N();
const { createLinkedPage } = usePageHelper(workspace);
const { setTrashModal } = useTrashModalHelper(workspace);
const { removeFromFavorite } = useBlockSuiteMetaHelper(workspace);
const { createLinkedPage } = usePageHelper(docCollection);
const { setTrashModal } = useTrashModalHelper(docCollection);
const { removeFromFavorite } = useBlockSuiteMetaHelper(docCollection);
const handleRename = useCallback(() => {
setRenameModalOpen?.();
@@ -2,7 +2,7 @@ import { toast } from '@affine/component';
import { RenameModal } from '@affine/component/rename-modal';
import { useDocMetaHelper } from '@affine/core/hooks/use-block-suite-page-meta';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { useCallback, useState } from 'react';
import { AddFavouriteButton } from '../favorite/add-favourite-button';
@@ -10,7 +10,7 @@ import * as styles from '../favorite/styles.css';
import { OperationMenuButton } from './operation-menu-button';
type PostfixItemProps = {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
pageTitle: string;
inFavorites?: boolean;
@@ -20,10 +20,10 @@ type PostfixItemProps = {
};
export const PostfixItem = ({ ...props }: PostfixItemProps) => {
const { workspace, pageId, pageTitle } = props;
const { docCollection, pageId, pageTitle } = props;
const t = useAFFiNEI18N();
const [open, setOpen] = useState(false);
const { setDocTitle } = useDocMetaHelper(workspace);
const { setDocTitle } = useDocMetaHelper(docCollection);
const handleRename = useCallback(
(newName: string) => {
@@ -1,7 +1,7 @@
import { useBlockSuitePageReferences } from '@affine/core/hooks/use-block-suite-page-references';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { EdgelessIcon, PageIcon } from '@blocksuite/icons';
import { type DocMeta, type Workspace } from '@blocksuite/store';
import { type DocCollection, type DocMeta } from '@blocksuite/store';
import * as Collapsible from '@radix-ui/react-collapsible';
import { PageRecordList, useLiveData, useService } from '@toeverything/infra';
import { useMemo, useState } from 'react';
@@ -11,14 +11,14 @@ import { MenuLinkItem } from '../../../app-sidebar';
import * as styles from '../favorite/styles.css';
import { PostfixItem } from './postfix-item';
export interface ReferencePageProps {
workspace: Workspace;
docCollection: DocCollection;
pageId: string;
metaMapping: Record<string, DocMeta>;
parentIds: Set<string>;
}
export const ReferencePage = ({
workspace,
docCollection,
pageId,
metaMapping,
parentIds,
@@ -33,7 +33,7 @@ export const ReferencePage = ({
return pageMode === 'edgeless' ? <EdgelessIcon /> : <PageIcon />;
}, [pageMode]);
const references = useBlockSuitePageReferences(workspace, pageId);
const references = useBlockSuitePageReferences(docCollection, pageId);
const referencesToShow = useMemo(() => {
return [
...new Set(
@@ -59,13 +59,13 @@ export const ReferencePage = ({
data-type="reference-page"
data-testid={`reference-page-${pageId}`}
active={active}
to={`/workspace/${workspace.id}/${pageId}`}
to={`/workspace/${docCollection.id}/${pageId}`}
icon={icon}
collapsed={collapsible ? collapsed : undefined}
onCollapsedChange={setCollapsed}
postfix={
<PostfixItem
workspace={workspace}
docCollection={docCollection}
pageId={pageId}
pageTitle={pageTitle}
isReferencePage={true}
@@ -83,7 +83,7 @@ export const ReferencePage = ({
return (
<ReferencePage
key={ref}
workspace={workspace}
docCollection={docCollection}
pageId={ref}
metaMapping={metaMapping}
parentIds={new Set([...parentIds, pageId])}
@@ -2,21 +2,21 @@ import { IconButton } from '@affine/component/ui/button';
import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
import { useDocMetaHelper } from '@affine/core/hooks/use-block-suite-page-meta';
import { PlusIcon } from '@blocksuite/icons';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { usePageHelper } from '../../../blocksuite/block-suite-page-list/utils';
type AddFavouriteButtonProps = {
workspace: Workspace;
docCollection: DocCollection;
pageId?: string;
};
export const AddFavouriteButton = ({
workspace,
docCollection,
pageId,
}: AddFavouriteButtonProps) => {
const { createPage, createLinkedPage } = usePageHelper(workspace);
const { setDocMeta } = useDocMetaHelper(workspace);
const { createPage, createLinkedPage } = usePageHelper(docCollection);
const { setDocMeta } = useDocMetaHelper(docCollection);
const handleAddFavorite = useAsyncCallback(
async e => {
if (pageId) {
@@ -11,7 +11,9 @@ import * as styles from './styles.css';
const emptyPageIdSet = new Set<string>();
export const FavoriteList = ({ workspace }: FavoriteListProps) => {
export const FavoriteList = ({
docCollection: workspace,
}: FavoriteListProps) => {
const metas = useBlockSuiteDocMeta(workspace);
const dropItemId = getDropItemId('favorites');
@@ -51,7 +53,7 @@ export const FavoriteList = ({ workspace }: FavoriteListProps) => {
pageId={pageMeta.id}
// memo?
parentIds={emptyPageIdSet}
workspace={workspace}
docCollection={workspace}
/>
);
})}
@@ -20,7 +20,7 @@ import {
import * as styles from './styles.css';
export const FavouritePage = ({
workspace,
docCollection: workspace,
pageId,
metaMapping,
parentIds,
@@ -86,7 +86,7 @@ export const FavouritePage = ({
{...listeners}
postfix={
<PostfixItem
workspace={workspace}
docCollection={workspace}
pageId={pageId}
pageTitle={pageTitle}
inFavorites={true}
@@ -102,7 +102,7 @@ export const FavouritePage = ({
return (
<ReferencePage
key={id}
workspace={workspace}
docCollection={workspace}
pageId={id}
metaMapping={metaMapping}
parentIds={new Set([pageId])}
@@ -1,12 +1,12 @@
import type { DeleteCollectionInfo } from '@affine/env/filter';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
export type FavoriteListProps = {
workspace: Workspace;
docCollection: DocCollection;
};
export type CollectionsListProps = {
workspace: Workspace;
docCollection: DocCollection;
info: DeleteCollectionInfo;
onCreate?: () => void;
};
@@ -1,17 +1,13 @@
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { ImportIcon } from '@blocksuite/icons';
import type { BlockSuiteWorkspace } from '../../shared';
import type { DocCollection } from '../../shared';
import { MenuItem } from '../app-sidebar';
import { usePageHelper } from '../blocksuite/block-suite-page-list/utils';
const ImportPage = ({
blocksuiteWorkspace,
}: {
blocksuiteWorkspace: BlockSuiteWorkspace;
}) => {
const ImportPage = ({ docCollection }: { docCollection: DocCollection }) => {
const t = useAFFiNEI18N();
const { importFile } = usePageHelper(blocksuiteWorkspace);
const { importFile } = usePageHelper(docCollection);
return (
<MenuItem icon={<ImportIcon />} onClick={importFile}>
{t['Import']()}
@@ -100,7 +100,7 @@ export const RootAppSidebar = ({
}: RootAppSidebarProps): ReactElement => {
const currentWorkspaceId = currentWorkspace.id;
const { appSettings } = useAppSettingHelper();
const blockSuiteWorkspace = currentWorkspace.blockSuiteWorkspace;
const docCollection = currentWorkspace.docCollection;
const t = useAFFiNEI18N();
const [openUserWorkspaceList, setOpenUserWorkspaceList] = useAtom(
openWorkspaceListModalAtom
@@ -114,7 +114,7 @@ export const RootAppSidebar = ({
}, [createPage, openPage]);
const { trashModal, setTrashModal, handleOnConfirm } =
useTrashModalHelper(blockSuiteWorkspace);
useTrashModalHelper(docCollection);
const deletePageTitles = trashModal.pageTitles;
const trashConfirmOpen = trashModal.open;
const onTrashConfirmOpenChange = useCallback(
@@ -164,12 +164,12 @@ export const RootAppSidebar = ({
.then(name => {
const id = nanoid();
collection.addCollection(createEmptyCollection(id, { name }));
navigateHelper.jumpToCollection(blockSuiteWorkspace.id, id);
navigateHelper.jumpToCollection(docCollection.id, id);
})
.catch(err => {
console.error(err);
});
}, [blockSuiteWorkspace.id, collection, navigateHelper, open]);
}, [docCollection.id, collection, navigateHelper, open]);
const allPageActive = currentPath === '/all';
@@ -227,7 +227,7 @@ export const RootAppSidebar = ({
</span>
</RouteMenuLinkItem>
<AppSidebarJournalButton
workspace={currentWorkspace.blockSuiteWorkspace}
docCollection={currentWorkspace.docCollection}
/>
{runtimeConfig.enableNewSettingModal ? (
<MenuItem
@@ -244,14 +244,14 @@ export const RootAppSidebar = ({
<SidebarScrollableContainer>
<CategoryDivider label={t['com.affine.rootAppSidebar.favorites']()}>
<AddFavouriteButton workspace={blockSuiteWorkspace} />
<AddFavouriteButton docCollection={docCollection} />
</CategoryDivider>
<FavoriteList workspace={blockSuiteWorkspace} />
<FavoriteList docCollection={docCollection} />
<CategoryDivider label={t['com.affine.rootAppSidebar.collections']()}>
<AddCollectionButton node={node} onClick={handleCreateCollection} />
</CategoryDivider>
<CollectionsList
workspace={blockSuiteWorkspace}
docCollection={docCollection}
info={userInfo}
onCreate={handleCreateCollection}
/>
@@ -268,7 +268,7 @@ export const RootAppSidebar = ({
{t['com.affine.workspaceSubPath.trash']()}
</span>
</RouteMenuLinkItem>
<ImportPage blocksuiteWorkspace={blockSuiteWorkspace} />
<ImportPage docCollection={docCollection} />
</SidebarScrollableContainer>
<SidebarContainer>
{environment.isDesktop ? <UpdaterButton /> : <AppDownloadButton />}
@@ -2,7 +2,7 @@ import {
useJournalInfoHelper,
useJournalRouteHelper,
} from '@affine/core/hooks/use-journal';
import type { BlockSuiteWorkspace } from '@affine/core/shared';
import type { DocCollection } from '@affine/core/shared';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import { TodayIcon, TomorrowIcon, YesterdayIcon } from '@blocksuite/icons';
import { Doc, useServiceOptional } from '@toeverything/infra';
@@ -11,17 +11,17 @@ import { useParams } from 'react-router-dom';
import { MenuItem } from '../app-sidebar';
interface AppSidebarJournalButtonProps {
workspace: BlockSuiteWorkspace;
docCollection: DocCollection;
}
export const AppSidebarJournalButton = ({
workspace,
docCollection,
}: AppSidebarJournalButtonProps) => {
const t = useAFFiNEI18N();
const currentPage = useServiceOptional(Doc);
const { openToday } = useJournalRouteHelper(workspace);
const { openToday } = useJournalRouteHelper(docCollection);
const { journalDate, isJournal } = useJournalInfoHelper(
workspace,
docCollection,
currentPage?.id
);
const params = useParams();
@@ -11,14 +11,14 @@ import type { PropsWithChildren } from 'react';
import { beforeEach, describe, expect, test, vi } from 'vitest';
import { useBlockSuiteDocMeta } from '../use-block-suite-page-meta';
import { useBlockSuiteWorkspaceHelper } from '../use-block-suite-workspace-helper';
import { useDocCollectionHelper } from '../use-block-suite-workspace-helper';
const configureTestingWorkspace = async () => {
const { workspace } = await configureTestingEnvironment();
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const docCollection = workspace.docCollection;
initEmptyPage(blockSuiteWorkspace.createDoc({ id: 'page1' }));
initEmptyPage(blockSuiteWorkspace.createDoc({ id: 'page2' }));
initEmptyPage(docCollection.createDoc({ id: 'page1' }));
initEmptyPage(docCollection.createDoc({ id: 'page2' }));
return workspace;
};
@@ -36,31 +36,25 @@ const getWrapper = (workspace: Workspace) =>
);
};
describe('useBlockSuiteWorkspaceHelper', () => {
describe('useDocCollectionHelper', () => {
test('should create page', async () => {
const workspace = await configureTestingWorkspace();
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const docCollection = workspace.docCollection;
const Wrapper = getWrapper(workspace);
expect(blockSuiteWorkspace.meta.docMetas.length).toBe(3);
const helperHook = renderHook(
() => useBlockSuiteWorkspaceHelper(blockSuiteWorkspace),
{
wrapper: Wrapper,
}
);
const pageMetaHook = renderHook(
() => useBlockSuiteDocMeta(blockSuiteWorkspace),
{
wrapper: Wrapper,
}
);
expect(docCollection.meta.docMetas.length).toBe(3);
const helperHook = renderHook(() => useDocCollectionHelper(docCollection), {
wrapper: Wrapper,
});
const pageMetaHook = renderHook(() => useBlockSuiteDocMeta(docCollection), {
wrapper: Wrapper,
});
await new Promise(resolve => setTimeout(resolve));
expect(pageMetaHook.result.current.length).toBe(3);
expect(blockSuiteWorkspace.meta.docMetas.length).toBe(3);
expect(docCollection.meta.docMetas.length).toBe(3);
const page = helperHook.result.current.createDoc('page4');
expect(page.id).toBe('page4');
expect(blockSuiteWorkspace.meta.docMetas.length).toBe(4);
expect(docCollection.meta.docMetas.length).toBe(4);
pageMetaHook.rerender();
expect(pageMetaHook.result.current.length).toBe(4);
});
@@ -13,16 +13,13 @@ import { describe, expect, test, vi } from 'vitest';
import { beforeEach } from 'vitest';
import { configureTestingEnvironment } from '../../testing';
import { useBlockSuiteWorkspacePageTitle } from '../use-block-suite-workspace-page-title';
import { useDocCollectionPageTitle } from '../use-block-suite-workspace-page-title';
const store = createStore();
const Component = () => {
const workspace = useService(Workspace);
const title = useBlockSuiteWorkspacePageTitle(
workspace.blockSuiteWorkspace,
'page0'
);
const title = useDocCollectionPageTitle(workspace.docCollection, 'page0');
return <div>title: {title}</div>;
};
@@ -30,7 +27,7 @@ beforeEach(async () => {
vi.useFakeTimers({ toFake: ['requestIdleCallback'] });
});
describe('useBlockSuiteWorkspacePageTitle', () => {
describe('useDocCollectionPageTitle', () => {
test('basic', async () => {
const { workspace, page } = await configureTestingEnvironment();
const { findByText, rerender } = render(
@@ -43,7 +40,7 @@ describe('useBlockSuiteWorkspacePageTitle', () => {
</ServiceProviderContext.Provider>
);
expect(await findByText('title: Untitled')).toBeDefined();
workspace.blockSuiteWorkspace.setDocMeta(page.id, { title: '1' });
workspace.docCollection.setDocMeta(page.id, { title: '1' });
rerender(
<ServiceProviderContext.Provider value={page.services}>
<Provider store={store}>
@@ -17,7 +17,7 @@ import { usePublicPages } from './use-is-shared-page';
export const useAllPageListConfig = () => {
const currentWorkspace = useService(Workspace);
const { getPublicMode } = usePublicPages(currentWorkspace);
const workspace = currentWorkspace.blockSuiteWorkspace;
const workspace = currentWorkspace.docCollection;
const pageMetas = useBlockSuiteDocMeta(workspace);
const { isPreferredEdgeless } = usePageHelper(workspace);
const pageMap = useMemo(
@@ -25,7 +25,7 @@ export const useAllPageListConfig = () => {
[pageMetas]
);
const { toggleFavorite } = useBlockSuiteMetaHelper(
currentWorkspace.blockSuiteWorkspace
currentWorkspace.docCollection
);
const t = useAFFiNEI18N();
const onToggleFavoritePage = useCallback(
@@ -45,7 +45,7 @@ export const useAllPageListConfig = () => {
allPages: pageMetas,
isEdgeless: isPreferredEdgeless,
getPublicMode,
workspace: currentWorkspace.blockSuiteWorkspace,
docCollection: currentWorkspace.docCollection,
getPage: id => pageMap[id],
favoriteRender: page => {
return (
@@ -61,7 +61,7 @@ export const useAllPageListConfig = () => {
pageMetas,
isPreferredEdgeless,
getPublicMode,
currentWorkspace.blockSuiteWorkspace,
currentWorkspace.docCollection,
pageMap,
onToggleFavoritePage,
]);
@@ -1,22 +1,20 @@
import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
import { useDocMetaHelper } from '@affine/core/hooks/use-block-suite-page-meta';
import { useBlockSuiteWorkspaceHelper } from '@affine/core/hooks/use-block-suite-workspace-helper';
import { useDocCollectionHelper } from '@affine/core/hooks/use-block-suite-workspace-helper';
import { CollectionService } from '@affine/core/modules/collection';
import { PageRecordList, useService } from '@toeverything/infra';
import { useCallback } from 'react';
import { applyUpdate, encodeStateAsUpdate } from 'yjs';
import type { BlockSuiteWorkspace } from '../../shared';
import type { DocCollection } from '../../shared';
import { useNavigateHelper } from '../use-navigate-helper';
import { useReferenceLinkHelper } from './use-reference-link-helper';
export function useBlockSuiteMetaHelper(
blockSuiteWorkspace: BlockSuiteWorkspace
) {
export function useBlockSuiteMetaHelper(docCollection: DocCollection) {
const { setDocMeta, getDocMeta, setDocReadonly, setDocTitle } =
useDocMetaHelper(blockSuiteWorkspace);
const { addReferenceLink } = useReferenceLinkHelper(blockSuiteWorkspace);
const { createDoc } = useBlockSuiteWorkspaceHelper(blockSuiteWorkspace);
useDocMetaHelper(docCollection);
const { addReferenceLink } = useReferenceLinkHelper(docCollection);
const { createDoc } = useDocCollectionHelper(docCollection);
const { openPage } = useNavigateHelper();
const collectionService = useService(CollectionService);
const pageRecordList = useService(PageRecordList);
@@ -82,9 +80,9 @@ export function useBlockSuiteMetaHelper(
const permanentlyDeletePage = useCallback(
(pageId: string) => {
blockSuiteWorkspace.removeDoc(pageId);
docCollection.removeDoc(pageId);
},
[blockSuiteWorkspace]
[docCollection]
);
/**
@@ -116,7 +114,7 @@ export function useBlockSuiteMetaHelper(
const currentPageMode = pageRecordList.record(pageId).value?.mode.value;
const currentPageMeta = getDocMeta(pageId);
const newPage = createDoc();
const currentPage = blockSuiteWorkspace.getDoc(pageId);
const currentPage = docCollection.getDoc(pageId);
newPage.load();
if (!currentPageMeta || !currentPage) {
@@ -142,10 +140,10 @@ export function useBlockSuiteMetaHelper(
.record(newPage.id)
.value?.setMode(currentPageMode || 'page');
setDocTitle(newPage.id, newPageTitle);
openPageAfterDuplication && openPage(blockSuiteWorkspace.id, newPage.id);
openPageAfterDuplication && openPage(docCollection.id, newPage.id);
},
[
blockSuiteWorkspace,
docCollection,
createDoc,
getDocMeta,
openPage,
@@ -1,13 +1,11 @@
import { useCallback } from 'react';
import type { BlockSuiteWorkspace } from '../../shared';
import type { DocCollection } from '../../shared';
export function useReferenceLinkHelper(
blockSuiteWorkspace: BlockSuiteWorkspace
) {
export function useReferenceLinkHelper(docCollection: DocCollection) {
const addReferenceLink = useCallback(
(pageId: string, referenceId: string) => {
const page = blockSuiteWorkspace?.getDoc(pageId);
const page = docCollection?.getDoc(pageId);
if (!page) {
return;
}
@@ -26,7 +24,7 @@ export function useReferenceLinkHelper(
frame && page.addBlock('affine:paragraph', { text }, frame.id);
},
[blockSuiteWorkspace]
[docCollection]
);
return {
@@ -24,9 +24,9 @@ export function useRegisterBlocksuiteEditorCommands() {
const mode = useLiveData(page.mode);
const t = useAFFiNEI18N();
const workspace = useService(Workspace);
const blockSuiteWorkspace = workspace.blockSuiteWorkspace;
const { getDocMeta } = useDocMetaHelper(blockSuiteWorkspace);
const currentPage = blockSuiteWorkspace.getDoc(pageId);
const docCollection = workspace.docCollection;
const { getDocMeta } = useDocMetaHelper(docCollection);
const currentPage = docCollection.getDoc(pageId);
assertExists(currentPage);
const pageMeta = getDocMeta(pageId);
assertExists(pageMeta);
@@ -43,9 +43,9 @@ export function useRegisterBlocksuiteEditorCommands() {
}, [pageId, setPageHistoryModalState]);
const { toggleFavorite, restoreFromTrash, duplicate } =
useBlockSuiteMetaHelper(blockSuiteWorkspace);
useBlockSuiteMetaHelper(docCollection);
const exportHandler = useExportPage(currentPage);
const { setTrashModal } = useTrashModalHelper(blockSuiteWorkspace);
const { setTrashModal } = useTrashModalHelper(docCollection);
const onClickDelete = useCallback(() => {
setTrashModal({
open: true,
@@ -70,7 +70,7 @@ export function getDragItemId(
export const useSidebarDrag = () => {
const t = useAFFiNEI18N();
const currentWorkspace = useService(Workspace);
const workspace = currentWorkspace.blockSuiteWorkspace;
const workspace = currentWorkspace.docCollection;
const { setTrashModal } = useTrashModalHelper(workspace);
const { addToFavorite, removeFromFavorite } =
useBlockSuiteMetaHelper(workspace);
@@ -1,17 +1,17 @@
import { toast } from '@affine/component';
import { useAFFiNEI18N } from '@affine/i18n/hooks';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import { useAtom } from 'jotai';
import { useCallback } from 'react';
import { trashModalAtom } from '../../atoms/trash-modal';
import { useBlockSuiteMetaHelper } from './use-block-suite-meta-helper';
export function useTrashModalHelper(blocksuiteWorkspace: Workspace) {
export function useTrashModalHelper(docCollection: DocCollection) {
const t = useAFFiNEI18N();
const [trashModal, setTrashModal] = useAtom(trashModalAtom);
const { pageIds } = trashModal;
const { removeToTrash } = useBlockSuiteMetaHelper(blocksuiteWorkspace);
const { removeToTrash } = useBlockSuiteMetaHelper(docCollection);
const handleOnConfirm = useCallback(() => {
pageIds.forEach(pageId => {
@@ -20,8 +20,7 @@ const useReactiveAdapter = (adapter: WorkspacePropertiesAdapter) => {
}
const disposables: (() => void)[] = [];
disposables.push(
adapter.workspace.blockSuiteWorkspace.meta.docMetaUpdated.on(observe)
.dispose
adapter.workspace.docCollection.meta.docMetaUpdated.on(observe).dispose
);
adapter.properties.observeDeep(observe);
disposables.push(() => adapter.properties.unobserveDeep(observe));
@@ -1,25 +1,25 @@
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import type { Atom } from 'jotai';
import { atom, useAtomValue } from 'jotai';
const weakMap = new WeakMap<Workspace, Atom<DocMeta[]>>();
const weakMap = new WeakMap<DocCollection, Atom<DocMeta[]>>();
// this hook is extracted from './use-block-suite-page-meta.ts' to avoid circular dependency
export function useAllBlockSuiteDocMeta(
blockSuiteWorkspace: Workspace
docCollection: DocCollection
): DocMeta[] {
if (!weakMap.has(blockSuiteWorkspace)) {
const baseAtom = atom<DocMeta[]>(blockSuiteWorkspace.meta.docMetas);
weakMap.set(blockSuiteWorkspace, baseAtom);
if (!weakMap.has(docCollection)) {
const baseAtom = atom<DocMeta[]>(docCollection.meta.docMetas);
weakMap.set(docCollection, baseAtom);
baseAtom.onMount = set => {
set(blockSuiteWorkspace.meta.docMetas);
const dispose = blockSuiteWorkspace.meta.docMetaUpdated.on(() => {
set(blockSuiteWorkspace.meta.docMetas);
set(docCollection.meta.docMetas);
const dispose = docCollection.meta.docMetaUpdated.on(() => {
set(docCollection.meta.docMetas);
});
return () => {
dispose.dispose();
};
};
}
return useAtomValue(weakMap.get(blockSuiteWorkspace) as Atom<DocMeta[]>);
return useAtomValue(weakMap.get(docCollection) as Atom<DocMeta[]>);
}
@@ -1,11 +1,11 @@
import type { Doc, Workspace } from '@blocksuite/store';
import type { Doc, DocCollection } from '@blocksuite/store';
import { type Atom, atom, useAtomValue } from 'jotai';
import { useBlockSuiteWorkspacePage } from './use-block-suite-workspace-page';
import { useDocCollectionPage } from './use-block-suite-workspace-page';
const weakMap = new WeakMap<Doc, Atom<string[]>>();
function getPageBacklinks(page: Doc): string[] {
return page.workspace.indexer.backlink
return page.collection.indexer.backlink
.getBacklink(page.id)
.map(linkNode => linkNode.pageId)
.filter(id => id !== page.id);
@@ -23,7 +23,7 @@ const getPageBacklinksAtom = (page: Doc | null) => {
page.slots.ready.on(() => {
set(getPageBacklinks(page));
}),
page.workspace.indexer.backlink.slots.indexUpdated.on(() => {
page.collection.indexer.backlink.slots.indexUpdated.on(() => {
set(getPageBacklinks(page));
}),
];
@@ -38,9 +38,9 @@ const getPageBacklinksAtom = (page: Doc | null) => {
};
export function useBlockSuitePageBacklinks(
blockSuiteWorkspace: Workspace,
pageId: string
docCollection: DocCollection,
docId: string
): string[] {
const page = useBlockSuiteWorkspacePage(blockSuiteWorkspace, pageId);
return useAtomValue(getPageBacklinksAtom(page));
const doc = useDocCollectionPage(docCollection, docId);
return useAtomValue(getPageBacklinksAtom(doc));
}
@@ -1,6 +1,6 @@
import type { RootBlockModel } from '@blocksuite/blocks';
import { assertExists } from '@blocksuite/global/utils';
import type { DocMeta, Workspace } from '@blocksuite/store';
import type { DocCollection, DocMeta } from '@blocksuite/store';
import { useMemo } from 'react';
import { useAllBlockSuiteDocMeta } from './use-all-block-suite-page-meta';
@@ -11,9 +11,9 @@ import { useJournalHelper } from './use-journal';
* If you want to get all pageMetas, use `useAllBlockSuitePageMeta` instead
* @returns
*/
export function useBlockSuiteDocMeta(blocksuiteWorkspace: Workspace) {
const pageMetas = useAllBlockSuiteDocMeta(blocksuiteWorkspace);
const { isPageJournal } = useJournalHelper(blocksuiteWorkspace);
export function useBlockSuiteDocMeta(docCollection: DocCollection) {
const pageMetas = useAllBlockSuiteDocMeta(docCollection);
const { isPageJournal } = useJournalHelper(docCollection);
return useMemo(
() =>
pageMetas.filter(
@@ -23,11 +23,11 @@ export function useBlockSuiteDocMeta(blocksuiteWorkspace: Workspace) {
);
}
export function useDocMetaHelper(blockSuiteWorkspace: Workspace) {
export function useDocMetaHelper(docCollection: DocCollection) {
return useMemo(
() => ({
setDocTitle: (docId: string, newTitle: string) => {
const page = blockSuiteWorkspace.getDoc(docId);
const page = docCollection.getDoc(docId);
assertExists(page);
const pageBlock = page
.getBlockByFlavour('affine:page')
@@ -37,20 +37,20 @@ export function useDocMetaHelper(blockSuiteWorkspace: Workspace) {
pageBlock.title.delete(0, pageBlock.title.length);
pageBlock.title.insert(newTitle, 0);
});
blockSuiteWorkspace.meta.setDocMeta(docId, { title: newTitle });
docCollection.meta.setDocMeta(docId, { title: newTitle });
},
setDocReadonly: (docId: string, readonly: boolean) => {
const page = blockSuiteWorkspace.getDoc(docId);
const page = docCollection.getDoc(docId);
assertExists(page);
page.awarenessStore.setReadonly(page, readonly);
},
setDocMeta: (docId: string, docMeta: Partial<DocMeta>) => {
blockSuiteWorkspace.meta.setDocMeta(docId, docMeta);
docCollection.meta.setDocMeta(docId, docMeta);
},
getDocMeta: (docId: string) => {
return blockSuiteWorkspace.meta.getDocMeta(docId);
return docCollection.meta.getDocMeta(docId);
},
}),
[blockSuiteWorkspace]
[docCollection]
);
}
@@ -1,12 +1,12 @@
import type { Doc, Workspace } from '@blocksuite/store';
import type { Doc, DocCollection } from '@blocksuite/store';
import { type Atom, atom, useAtomValue } from 'jotai';
import { useBlockSuiteWorkspacePage } from './use-block-suite-workspace-page';
import { useDocCollectionPage } from './use-block-suite-workspace-page';
const weakMap = new WeakMap<Doc, Atom<string[]>>();
function getPageReferences(page: Doc): string[] {
return Object.values(
page.workspace.indexer.backlink.linkIndexMap[page.id] ?? {}
page.collection.indexer.backlink.linkIndexMap[page.id] ?? {}
).flatMap(linkNodes => linkNodes.map(linkNode => linkNode.pageId));
}
@@ -22,7 +22,7 @@ const getPageReferencesAtom = (page: Doc | null) => {
page.slots.ready.on(() => {
set(getPageReferences(page));
}),
page.workspace.indexer.backlink.slots.indexUpdated.on(() => {
page.collection.indexer.backlink.slots.indexUpdated.on(() => {
set(getPageReferences(page));
}),
];
@@ -37,9 +37,9 @@ const getPageReferencesAtom = (page: Doc | null) => {
};
export function useBlockSuitePageReferences(
blockSuiteWorkspace: Workspace,
docCollection: DocCollection,
pageId: string
): string[] {
const page = useBlockSuiteWorkspacePage(blockSuiteWorkspace, pageId);
const page = useDocCollectionPage(docCollection, pageId);
return useAtomValue(getPageReferencesAtom(page));
}
@@ -1,13 +1,13 @@
import type { Doc, Workspace } from '@blocksuite/store';
import type { Doc, DocCollection } from '@blocksuite/store';
import { useMemo } from 'react';
export function useBlockSuiteWorkspaceHelper(blockSuiteWorkspace: Workspace) {
export function useDocCollectionHelper(docCollection: DocCollection) {
return useMemo(
() => ({
createDoc: (pageId?: string): Doc => {
return blockSuiteWorkspace.createDoc({ id: pageId });
return docCollection.createDoc({ id: pageId });
},
}),
[blockSuiteWorkspace]
[docCollection]
);
}
@@ -1,14 +1,14 @@
import { assertExists } from '@blocksuite/global/utils';
import type { Workspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { Atom } from 'jotai';
import { atom, useAtomValue } from 'jotai';
import { useCallback } from 'react';
import { useJournalHelper, useJournalInfoHelper } from './use-journal';
const weakMap = new WeakMap<Workspace, Map<string, Atom<string>>>();
const weakMap = new WeakMap<DocCollection, Map<string, Atom<string>>>();
function getAtom(w: Workspace, pageId: string): Atom<string> {
function getAtom(w: DocCollection, pageId: string): Atom<string> {
if (!weakMap.has(w)) {
weakMap.set(w, new Map());
}
@@ -32,33 +32,27 @@ function getAtom(w: Workspace, pageId: string): Atom<string> {
}
}
export function useBlockSuiteWorkspacePageTitle(
blockSuiteWorkspace: Workspace,
export function useDocCollectionPageTitle(
docCollection: DocCollection,
pageId: string
) {
const titleAtom = getAtom(blockSuiteWorkspace, pageId);
const titleAtom = getAtom(docCollection, pageId);
assertExists(titleAtom);
const title = useAtomValue(titleAtom);
const { localizedJournalDate } = useJournalInfoHelper(
blockSuiteWorkspace,
pageId
);
const { localizedJournalDate } = useJournalInfoHelper(docCollection, pageId);
return localizedJournalDate || title;
}
// This hook is NOT reactive to the page title change
export function useGetBlockSuiteWorkspacePageTitle(
blockSuiteWorkspace: Workspace
) {
const { getLocalizedJournalDateString } =
useJournalHelper(blockSuiteWorkspace);
export function useGetDocCollectionPageTitle(docCollection: DocCollection) {
const { getLocalizedJournalDateString } = useJournalHelper(docCollection);
return useCallback(
(pageId: string) => {
return (
getLocalizedJournalDateString(pageId) ||
blockSuiteWorkspace.getDoc(pageId)?.meta?.title
docCollection.getDoc(pageId)?.meta?.title
);
},
[blockSuiteWorkspace, getLocalizedJournalDateString]
[docCollection, getLocalizedJournalDateString]
);
}
@@ -1,26 +1,26 @@
import { DisposableGroup } from '@blocksuite/global/utils';
import type { Doc, Workspace } from '@blocksuite/store';
import type { Doc, DocCollection } from '@blocksuite/store';
import { useEffect, useState } from 'react';
export function useBlockSuiteWorkspacePage(
blockSuiteWorkspace: Workspace,
export function useDocCollectionPage(
docCollection: DocCollection,
pageId: string | null
): Doc | null {
const [page, setPage] = useState(
pageId ? blockSuiteWorkspace.getDoc(pageId) : null
pageId ? docCollection.getDoc(pageId) : null
);
useEffect(() => {
const group = new DisposableGroup();
group.add(
blockSuiteWorkspace.slots.docAdded.on(id => {
docCollection.slots.docAdded.on(id => {
if (pageId === id) {
setPage(blockSuiteWorkspace.getDoc(id));
setPage(docCollection.getDoc(id));
}
})
);
group.add(
blockSuiteWorkspace.slots.docRemoved.on(id => {
docCollection.slots.docRemoved.on(id => {
if (pageId === id) {
setPage(null);
}
@@ -29,7 +29,7 @@ export function useBlockSuiteWorkspacePage(
return () => {
group.dispose();
};
}, [blockSuiteWorkspace, pageId]);
}, [docCollection, pageId]);
useEffect(() => {
if (page && !page.loaded) {
+13 -13
View File
@@ -2,10 +2,10 @@ import { initEmptyPage } from '@toeverything/infra';
import dayjs from 'dayjs';
import { useCallback, useMemo } from 'react';
import type { BlockSuiteWorkspace } from '../shared';
import type { DocCollection } from '../shared';
import { timestampToLocalDate } from '../utils';
import { useCurrentWorkspacePropertiesAdapter } from './use-affine-adapter';
import { useBlockSuiteWorkspaceHelper } from './use-block-suite-workspace-helper';
import { useDocCollectionHelper } from './use-block-suite-workspace-helper';
import { useNavigateHelper } from './use-navigate-helper';
type MaybeDate = Date | string | number;
@@ -22,8 +22,8 @@ function toDayjs(j?: string | false) {
return day;
}
export const useJournalHelper = (workspace: BlockSuiteWorkspace) => {
const bsWorkspaceHelper = useBlockSuiteWorkspaceHelper(workspace);
export const useJournalHelper = (docCollection: DocCollection) => {
const bsWorkspaceHelper = useDocCollectionHelper(docCollection);
const adapter = useCurrentWorkspacePropertiesAdapter();
/**
@@ -35,7 +35,7 @@ export const useJournalHelper = (workspace: BlockSuiteWorkspace) => {
const title = day.format(JOURNAL_DATE_FORMAT);
const page = bsWorkspaceHelper.createDoc();
// set created date to match the journal date
page.workspace.setDocMeta(page.id, {
page.collection.setDocMeta(page.id, {
createDate: dayjs()
.set('year', day.year())
.set('month', day.month())
@@ -63,7 +63,7 @@ export const useJournalHelper = (workspace: BlockSuiteWorkspace) => {
const getJournalsByDate = useCallback(
(maybeDate: MaybeDate) => {
const day = dayjs(maybeDate);
return Array.from(workspace.docs.values()).filter(page => {
return Array.from(docCollection.docs.values()).filter(page => {
const pageId = page.id;
if (!isPageJournal(pageId)) return false;
if (page.meta?.trash) return false;
@@ -72,7 +72,7 @@ export const useJournalHelper = (workspace: BlockSuiteWorkspace) => {
return day.isSame(journalDate, 'day');
});
},
[adapter, isPageJournal, workspace.docs]
[adapter, isPageJournal, docCollection.docs]
);
/**
@@ -150,18 +150,18 @@ export const useJournalHelper = (workspace: BlockSuiteWorkspace) => {
};
// split useJournalRouteHelper since it requires a <Route /> context, which may not work in lit
export const useJournalRouteHelper = (workspace: BlockSuiteWorkspace) => {
export const useJournalRouteHelper = (docCollection: DocCollection) => {
const navigateHelper = useNavigateHelper();
const { getJournalByDate } = useJournalHelper(workspace);
const { getJournalByDate } = useJournalHelper(docCollection);
/**
* open journal by date, create one if not exist
*/
const openJournal = useCallback(
(maybeDate: MaybeDate) => {
const page = getJournalByDate(maybeDate);
navigateHelper.openPage(workspace.id, page.id);
navigateHelper.openPage(docCollection.id, page.id);
},
[getJournalByDate, navigateHelper, workspace.id]
[getJournalByDate, navigateHelper, docCollection.id]
);
/**
@@ -182,7 +182,7 @@ export const useJournalRouteHelper = (workspace: BlockSuiteWorkspace) => {
};
export const useJournalInfoHelper = (
workspace: BlockSuiteWorkspace,
docCollection: DocCollection,
pageId?: string | null
) => {
const {
@@ -190,7 +190,7 @@ export const useJournalInfoHelper = (
getJournalDateString,
getLocalizedJournalDateString,
isPageTodayJournal,
} = useJournalHelper(workspace);
} = useJournalHelper(docCollection);
return useMemo(
() => ({
@@ -24,7 +24,7 @@ export function useRegisterWorkspaceCommands() {
const theme = useTheme();
const currentWorkspace = useService(Workspace);
const languageHelper = useLanguageHelper();
const pageHelper = usePageHelper(currentWorkspace.blockSuiteWorkspace);
const pageHelper = usePageHelper(currentWorkspace.docCollection);
const navigationHelper = useNavigateHelper();
const [editor] = useActiveBlocksuiteEditor();
@@ -45,14 +45,14 @@ export function useRegisterWorkspaceCommands() {
const unsub = registerAffineNavigationCommands({
store,
t,
workspace: currentWorkspace.blockSuiteWorkspace,
docCollection: currentWorkspace.docCollection,
navigationHelper,
});
return () => {
unsub();
};
}, [store, t, currentWorkspace.blockSuiteWorkspace, navigationHelper]);
}, [store, t, currentWorkspace.docCollection, navigationHelper]);
// register AffineSettingsCommands
useEffect(() => {
@@ -94,7 +94,7 @@ export const WorkspaceLayout = function WorkspaceLayout({
export const WorkspaceLayoutInner = ({ children }: PropsWithChildren) => {
const currentWorkspace = useService(Workspace);
const { openPage } = useNavigateHelper();
const pageHelper = usePageHelper(currentWorkspace.blockSuiteWorkspace);
const pageHelper = usePageHelper(currentWorkspace.docCollection);
useRegisterWorkspaceCommands();
@@ -104,7 +104,7 @@ export const WorkspaceLayoutInner = ({ children }: PropsWithChildren) => {
// 0.8.0 ~ 0.8.1
// 0.8.0-beta.0 ~ 0.8.0-beta.3
// 0.8.0-canary.17 ~ 0.9.0-canary.3
const meta = currentWorkspace.blockSuiteWorkspace.doc.getMap('meta');
const meta = currentWorkspace.docCollection.doc.getMap('meta');
const blockVersions = meta.get('blockVersions');
if (
!(blockVersions instanceof YMap) &&
@@ -117,7 +117,7 @@ export const WorkspaceLayoutInner = ({ children }: PropsWithChildren) => {
new YMap(Object.entries(blockVersions as Record<string, number>))
);
}
}, [currentWorkspace.blockSuiteWorkspace.doc]);
}, [currentWorkspace.docCollection.doc]);
const handleCreatePage = useCallback(() => {
return pageHelper.createPage();
@@ -17,11 +17,11 @@ export class CollectionService {
constructor(private readonly workspace: Workspace) {}
private get doc() {
return this.workspace.blockSuiteWorkspace.doc;
return this.workspace.docCollection.doc;
}
private get setting() {
return this.workspace.blockSuiteWorkspace.doc.getMap(SETTING_KEY);
return this.workspace.docCollection.doc.getMap(SETTING_KEY);
}
private get collectionsYArray(): YArray<Collection> | undefined {
@@ -96,7 +96,7 @@ export class CollectionService {
return;
}
const set = new Set(ids);
this.workspace.blockSuiteWorkspace.doc.transact(() => {
this.workspace.docCollection.doc.transact(() => {
const indexList: number[] = [];
const list: Collection[] = [];
collectionsYArray.forEach((collection, i) => {
@@ -61,7 +61,7 @@ const PageItem = ({
const mode = useLiveData(pageRecord.mode);
const workspace = useService(Workspace);
const { isJournal } = useJournalInfoHelper(
workspace.blockSuiteWorkspace,
workspace.docCollection,
pageRecord.id
);
@@ -100,10 +100,10 @@ const EditorJournalPanel = () => {
const doc = useService(Doc);
const workspace = useService(Workspace);
const { journalDate, isJournal } = useJournalInfoHelper(
workspace.blockSuiteWorkspace,
workspace.docCollection,
doc.id
);
const { openJournal } = useJournalRouteHelper(workspace.blockSuiteWorkspace);
const { openJournal } = useJournalRouteHelper(workspace.docCollection);
const [date, setDate] = useState(dayjs().format('YYYY-MM-DD'));
useEffect(() => {
@@ -314,7 +314,7 @@ const ConflictList = ({
const navigateHelper = useNavigateHelper();
const workspace = useService(Workspace);
const currentDoc = useService(Doc);
const { setTrashModal } = useTrashModalHelper(workspace.blockSuiteWorkspace);
const { setTrashModal } = useTrashModalHelper(workspace.docCollection);
const handleOpenTrashModal = useCallback(
(pageRecord: PageRecord) => {
@@ -361,7 +361,7 @@ const JournalConflictBlock = ({ date }: JournalBlockProps) => {
const t = useAFFiNEI18N();
const workspace = useService(Workspace);
const pageRecordList = useService(PageRecordList);
const journalHelper = useJournalHelper(workspace.blockSuiteWorkspace);
const journalHelper = useJournalHelper(workspace.docCollection);
const docs = journalHelper.getJournalsByDate(date.format('YYYY-MM-DD'));
const pageRecords = useLiveData(pageRecordList.records).filter(v => {
return docs.some(doc => doc.id === v.id);
@@ -28,10 +28,7 @@ export const MultiTabSidebarHeaderSwitcher = ({
FeatureType.Copilot
);
const { isJournal } = useJournalInfoHelper(
workspace.blockSuiteWorkspace,
doc.id
);
const { isJournal } = useJournalInfoHelper(workspace.docCollection, doc.id);
const exts = useMemo(
() =>
@@ -30,7 +30,7 @@ export class WorkspacePropertiesAdapter {
constructor(public readonly workspace: Workspace) {
// check if properties exists, if not, create one
const rootDoc = workspace.blockSuiteWorkspace.doc;
const rootDoc = workspace.docCollection.doc;
this.properties = rootDoc.getMap(AFFINE_PROPERTIES_ID);
this.proxy = createYProxy(this.properties);
@@ -58,8 +58,8 @@ export class WorkspacePropertiesAdapter {
source: 'system',
type: PagePropertyType.Tags,
options:
this.workspace.blockSuiteWorkspace.meta.properties.tags
?.options ?? [], // better use a one time migration
this.workspace.docCollection.meta.properties.tags?.options ??
[], // better use a one time migration
},
},
},
@@ -89,8 +89,8 @@ export class WorkspacePropertiesAdapter {
}
// leak some yjs abstraction to modify multiple properties at once
transact = this.workspace.blockSuiteWorkspace.doc.transact.bind(
this.workspace.blockSuiteWorkspace.doc
transact = this.workspace.docCollection.doc.transact.bind(
this.workspace.docCollection.doc
);
get schema() {
@@ -14,19 +14,18 @@ export class WorkspaceLegacyProperties {
}
get properties() {
return this.workspace.blockSuiteWorkspace.meta.properties;
return this.workspace.docCollection.meta.properties;
}
get tagOptions() {
return this.properties.tags?.options ?? [];
}
updateProperties = (properties: DocsPropertiesMeta) => {
this.workspace.blockSuiteWorkspace.meta.setProperties(properties);
this.workspace.docCollection.meta.setProperties(properties);
};
subscribe(cb: () => void) {
const disposable =
this.workspace.blockSuiteWorkspace.meta.docMetaUpdated.on(cb);
const disposable = this.workspace.docCollection.meta.docMetaUpdated.on(cb);
return disposable.dispose;
}
@@ -58,10 +57,10 @@ export class WorkspaceLegacyProperties {
};
removeTagOption = (id: string) => {
this.workspace.blockSuiteWorkspace.doc.transact(() => {
this.workspace.docCollection.doc.transact(() => {
this.updateTagOptions(this.tagOptions.filter(o => o.id !== id));
// need to remove tag from all pages
this.workspace.blockSuiteWorkspace.docs.forEach(doc => {
this.workspace.docCollection.docs.forEach(doc => {
const tags = doc.meta?.tags ?? [];
if (tags.includes(id)) {
this.updatePageTags(
@@ -74,7 +73,7 @@ export class WorkspaceLegacyProperties {
};
updatePageTags = (pageId: string, tags: string[]) => {
this.workspace.blockSuiteWorkspace.setDocMeta(pageId, {
this.workspace.docCollection.setDocMeta(pageId, {
tags,
});
};
@@ -162,7 +162,7 @@ export const Component = () => {
.then(() => {
const { page } = workspace.services.get(PageManager).open(pageId);
workspace.blockSuiteWorkspace.awarenessStore.setReadonly(
workspace.docCollection.awarenessStore.setReadonly(
page.blockSuiteDoc,
true
);
@@ -200,14 +200,14 @@ export const Component = () => {
<ShareHeader
pageId={page.id}
publishMode={publishMode}
blockSuiteWorkspace={page.blockSuiteDoc.workspace}
docCollection={page.blockSuiteDoc.collection}
/>
<Scrollable.Root>
<Scrollable.Viewport className={styles.editorContainer}>
<PageDetailEditor
isPublic
publishMode={publishMode}
workspace={page.blockSuiteDoc.workspace}
docCollection={page.blockSuiteDoc.collection}
pageId={page.id}
onLoad={() => noop}
/>
@@ -1,6 +1,6 @@
import { EditorModeSwitch } from '@affine/core/components/blocksuite/block-suite-mode-switch';
import ShareHeaderRightItem from '@affine/core/components/cloud/share-header-right-item';
import type { Workspace as BlockSuiteWorkspace } from '@blocksuite/store';
import type { DocCollection } from '@blocksuite/store';
import type { PageMode } from '@toeverything/infra';
import { BlocksuiteHeaderTitle } from '../../components/blocksuite/block-suite-header/title/index';
@@ -9,28 +9,28 @@ import * as styles from './share-header.css';
export function ShareHeader({
pageId,
publishMode,
blockSuiteWorkspace,
docCollection,
}: {
pageId: string;
publishMode: PageMode;
blockSuiteWorkspace: BlockSuiteWorkspace;
docCollection: DocCollection;
}) {
return (
<div className={styles.header}>
<EditorModeSwitch
isPublic
blockSuiteWorkspace={blockSuiteWorkspace}
docCollection={docCollection}
pageId={pageId}
publicMode={publishMode}
/>
<BlocksuiteHeaderTitle
blockSuiteWorkspace={blockSuiteWorkspace}
docCollection={docCollection}
pageId={pageId}
isPublic={true}
/>
<div className={styles.spacer} />
<ShareHeaderRightItem
workspaceId={blockSuiteWorkspace.id}
workspaceId={docCollection.id}
pageId={pageId}
publishMode={publishMode}
/>

Some files were not shown because too many files have changed in this diff Show More