refactor: rename BlocksuiteWorkspace and WorkspaceInfo

This commit is contained in:
alt0
2023-01-09 21:10:04 +08:00
parent 72a39f67c0
commit 0a4a9fcdb1
7 changed files with 60 additions and 53 deletions

View File

@@ -1,6 +1,6 @@
import { Workspaces } from './workspaces'; import { Workspaces } from './workspaces';
import type { WorkspacesChangeEvent } from './workspaces'; import type { WorkspacesChangeEvent } from './workspaces';
import { Workspace } from '@blocksuite/store'; import { Workspace as BlocksuiteWorkspace } from '@blocksuite/store';
import { BaseProvider } from './provider/base'; import { BaseProvider } from './provider/base';
import { LocalProvider } from './provider/local/local'; import { LocalProvider } from './provider/local/local';
import { AffineProvider } from './provider'; import { AffineProvider } from './provider';
@@ -17,7 +17,7 @@ import { applyUpdate, encodeStateAsUpdate } from 'yjs';
export class DataCenter { export class DataCenter {
private readonly _workspaces = new Workspaces(); private readonly _workspaces = new Workspaces();
private readonly _logger = getLogger('dc'); private readonly _logger = getLogger('dc');
private _workspaceInstances: Map<string, Workspace> = new Map(); private _workspaceInstances: Map<string, BlocksuiteWorkspace> = new Map();
/** /**
* A mainProvider must exist as the only data trustworthy source. * A mainProvider must exist as the only data trustworthy source.
*/ */
@@ -114,7 +114,7 @@ export class DataCenter {
assert(workspaceInfo, 'Workspace not found'); assert(workspaceInfo, 'Workspace not found');
return ( return (
this._workspaceInstances.get(workspaceId) || this._workspaceInstances.get(workspaceId) ||
new Workspace({ new BlocksuiteWorkspace({
room: workspaceId, room: workspaceId,
}).register(BlockSchema) }).register(BlockSchema)
); );
@@ -143,7 +143,7 @@ export class DataCenter {
/** /**
* load workspace instance by id * load workspace instance by id
* @param {string} workspaceId workspace id * @param {string} workspaceId workspace id
* @returns {Promise<Workspace>} * @returns {Promise<BlocksuiteWorkspace>}
*/ */
public async loadWorkspace(workspaceId: string) { public async loadWorkspace(workspaceId: string) {
const workspaceInfo = this._workspaces.find(workspaceId); const workspaceInfo = this._workspaces.find(workspaceId);
@@ -185,11 +185,11 @@ export class DataCenter {
/** /**
* change workspaces meta * change workspaces meta
* @param {WorkspaceMeta} workspaceMeta workspace meta * @param {WorkspaceMeta} workspaceMeta workspace meta
* @param {Workspace} workspace workspace instance * @param {BlocksuiteWorkspace} workspace workspace instance
*/ */
public async resetWorkspaceMeta( public async resetWorkspaceMeta(
{ name, avatar }: WorkspaceMeta, { name, avatar }: WorkspaceMeta,
workspace: Workspace workspace: BlocksuiteWorkspace
) { ) {
assert(workspace?.room, 'No workspace to set meta'); assert(workspace?.room, 'No workspace to set meta');
const update: Partial<WorkspaceMeta> = {}; const update: Partial<WorkspaceMeta> = {};
@@ -273,7 +273,7 @@ export class DataCenter {
} }
private async _transWorkspaceProvider( private async _transWorkspaceProvider(
workspace: Workspace, workspace: BlocksuiteWorkspace,
providerId: string providerId: string
) { ) {
assert(workspace.room, 'No workspace id'); assert(workspace.room, 'No workspace id');
@@ -306,7 +306,7 @@ export class DataCenter {
* Enable workspace cloud * Enable workspace cloud
* @param {string} id ID of workspace. * @param {string} id ID of workspace.
*/ */
public async enableWorkspaceCloud(workspace: Workspace) { public async enableWorkspaceCloud(workspace: BlocksuiteWorkspace) {
assert(workspace?.room, 'No workspace to enable cloud'); assert(workspace?.room, 'No workspace to enable cloud');
return await this._transWorkspaceProvider(workspace, 'affine'); return await this._transWorkspaceProvider(workspace, 'affine');
} }
@@ -344,7 +344,10 @@ export class DataCenter {
* @param id * @param id
* @returns {Promise<string | null>} blob url * @returns {Promise<string | null>} blob url
*/ */
async getBlob(workspace: Workspace, id: string): Promise<string | null> { async getBlob(
workspace: BlocksuiteWorkspace,
id: string
): Promise<string | null> {
const blob = await workspace.blobs; const blob = await workspace.blobs;
return (await blob?.get(id)) || ''; return (await blob?.get(id)) || '';
} }
@@ -354,7 +357,7 @@ export class DataCenter {
* @param id * @param id
* @returns {Promise<string | null>} blob url * @returns {Promise<string | null>} blob url
*/ */
async setBlob(workspace: Workspace, blob: Blob): Promise<string> { async setBlob(workspace: BlocksuiteWorkspace, blob: Blob): Promise<string> {
const blobStorage = await workspace.blobs; const blobStorage = await workspace.blobs;
return (await blobStorage?.set(blob)) || ''; return (await blobStorage?.set(blob)) || '';
} }

View File

@@ -26,5 +26,5 @@ const _initializeDataCenter = () => {
export const getDataCenter = _initializeDataCenter(); export const getDataCenter = _initializeDataCenter();
export type { AccessTokenMessage } from './provider/affine/apis'; export type { AccessTokenMessage } from './provider/affine/apis';
export type { Workspace } from './types'; export type { WorkspaceInfo } from './types';
export { getLogger } from './logger'; export { getLogger } from './logger';

View File

@@ -1,7 +1,7 @@
import { BaseProvider } from '../base.js'; import { BaseProvider } from '../base.js';
import type { ProviderConstructorParams } from '../base'; import type { ProviderConstructorParams } from '../base';
import type { User, Workspace as WS, WorkspaceMeta } from '../../types'; import type { User, WorkspaceInfo, WorkspaceMeta } from '../../types';
import { Workspace } from '@blocksuite/store'; import { Workspace as BlocksuiteWorkspace } from '@blocksuite/store';
import { BlockSchema } from '@blocksuite/blocks/models'; import { BlockSchema } from '@blocksuite/blocks/models';
import { applyUpdate } from 'yjs'; import { applyUpdate } from 'yjs';
import { storage } from './storage.js'; import { storage } from './storage.js';
@@ -19,7 +19,7 @@ export interface AffineProviderConstructorParams
export class AffineProvider extends BaseProvider { export class AffineProvider extends BaseProvider {
public id = 'affine'; public id = 'affine';
private _workspacesCache: Map<string, Workspace> = new Map(); private _workspacesCache: Map<string, BlocksuiteWorkspace> = new Map();
private _onTokenRefresh?: Callback = undefined; private _onTokenRefresh?: Callback = undefined;
private _wsMap: Map<string, WebsocketProvider> = new Map(); private _wsMap: Map<string, WebsocketProvider> = new Map();
private _apis: Apis; private _apis: Apis;
@@ -58,7 +58,7 @@ export class AffineProvider extends BaseProvider {
} }
} }
override async warpWorkspace(workspace: Workspace) { override async warpWorkspace(workspace: BlocksuiteWorkspace) {
const { doc, room } = workspace; const { doc, room } = workspace;
assert(room); assert(room);
this.linkLocal(workspace); this.linkLocal(workspace);
@@ -97,7 +97,7 @@ export class AffineProvider extends BaseProvider {
return []; return [];
} }
const workspacesList = await this._apis.getWorkspaces(); const workspacesList = await this._apis.getWorkspaces();
const workspaces: WS[] = workspacesList.map(w => { const workspaces: WorkspaceInfo[] = workspacesList.map(w => {
return { return {
...w, ...w,
memberCount: 0, memberCount: 0,
@@ -108,12 +108,12 @@ export class AffineProvider extends BaseProvider {
const workspaceInstances = workspaces.map(({ id }) => { const workspaceInstances = workspaces.map(({ id }) => {
const workspace = const workspace =
this._workspacesCache.get(id) || this._workspacesCache.get(id) ||
new Workspace({ new BlocksuiteWorkspace({
room: id, room: id,
}).register(BlockSchema); }).register(BlockSchema);
this._workspacesCache.set(id, workspace); this._workspacesCache.set(id, workspace);
if (workspace) { if (workspace) {
return new Promise<Workspace>(resolve => { return new Promise<BlocksuiteWorkspace>(resolve => {
this._apis.downloadWorkspace(id).then(data => { this._apis.downloadWorkspace(id).then(data => {
applyUpdate(workspace.doc, new Uint8Array(data)); applyUpdate(workspace.doc, new Uint8Array(data));
resolve(workspace); resolve(workspace);
@@ -235,7 +235,7 @@ export class AffineProvider extends BaseProvider {
return await this._apis.removeMember({ permissionId }); return await this._apis.removeMember({ permissionId });
} }
public override async linkLocal(workspace: Workspace) { public override async linkLocal(workspace: BlocksuiteWorkspace) {
return workspace; return workspace;
// assert(workspace.room); // assert(workspace.room);
// let idb = this._idbMap.get(workspace.room); // let idb = this._idbMap.get(workspace.room);
@@ -249,19 +249,19 @@ export class AffineProvider extends BaseProvider {
public override async createWorkspace( public override async createWorkspace(
meta: WorkspaceMeta meta: WorkspaceMeta
): Promise<Workspace | undefined> { ): Promise<BlocksuiteWorkspace | undefined> {
assert(meta.name, 'Workspace name is required'); assert(meta.name, 'Workspace name is required');
const { id } = await this._apis.createWorkspace( const { id } = await this._apis.createWorkspace(
meta as Required<WorkspaceMeta> meta as Required<WorkspaceMeta>
); );
this._logger('Creating affine workspace'); this._logger('Creating affine workspace');
const nw = new Workspace({ const nw = new BlocksuiteWorkspace({
room: id, room: id,
}).register(BlockSchema); }).register(BlockSchema);
nw.meta.setName(meta.name); nw.meta.setName(meta.name);
this.linkLocal(nw); this.linkLocal(nw);
const workspaceInfo: WS = { const workspaceInfo: WorkspaceInfo = {
name: meta.name, name: meta.name,
id, id,
isPublish: false, isPublish: false,

View File

@@ -1,5 +1,5 @@
import { BlobStorage, Workspace } from '@blocksuite/store'; import { Workspace as BlocksuiteWorkspace } from '@blocksuite/store';
import { Logger, User, Workspace as WS, WorkspaceMeta } from '../types'; import { Logger, User, WorkspaceInfo, WorkspaceMeta } from '../types';
import type { WorkspacesScope } from '../workspaces'; import type { WorkspacesScope } from '../workspaces';
const defaultLogger = () => { const defaultLogger = () => {
@@ -47,14 +47,16 @@ export class BaseProvider {
* @param workspace * @param workspace
* @returns * @returns
*/ */
public async warpWorkspace(workspace: Workspace): Promise<Workspace> { public async warpWorkspace(
workspace: BlocksuiteWorkspace
): Promise<BlocksuiteWorkspace> {
return workspace; return workspace;
} }
/** /**
* load workspaces * load workspaces
**/ **/
public async loadWorkspaces(): Promise<WS[]> { public async loadWorkspaces(): Promise<WorkspaceInfo[]> {
throw new Error(`provider: ${this.id} loadWorkSpace Not implemented`); throw new Error(`provider: ${this.id} loadWorkSpace Not implemented`);
} }
@@ -154,7 +156,7 @@ export class BaseProvider {
*/ */
public async createWorkspace( public async createWorkspace(
meta: WorkspaceMeta meta: WorkspaceMeta
): Promise<Workspace | undefined> { ): Promise<BlocksuiteWorkspace | undefined> {
meta; meta;
return; return;
} }
@@ -175,7 +177,9 @@ export class BaseProvider {
* @param workspace * @param workspace
* @returns * @returns
*/ */
public async linkLocal(workspace: Workspace): Promise<Workspace> { public async linkLocal(
workspace: BlocksuiteWorkspace
): Promise<BlocksuiteWorkspace> {
return workspace; return workspace;
} }
} }

View File

@@ -1,8 +1,8 @@
import { BaseProvider } from '../base.js'; import { BaseProvider } from '../base.js';
import type { ProviderConstructorParams } from '../base'; import type { ProviderConstructorParams } from '../base';
import { varStorage as storage } from 'lib0/storage'; import { varStorage as storage } from 'lib0/storage';
import { Workspace as WS, WorkspaceMeta } from '../../types'; import { WorkspaceInfo, WorkspaceMeta } from '../../types';
import { Workspace, uuidv4 } from '@blocksuite/store'; import { Workspace as BlocksuiteWorkspace, uuidv4 } from '@blocksuite/store';
import { IndexedDBProvider } from './indexeddb.js'; import { IndexedDBProvider } from './indexeddb.js';
import assert from 'assert'; import assert from 'assert';
import { getDefaultHeadImgBlob } from '../../utils/index.js'; import { getDefaultHeadImgBlob } from '../../utils/index.js';
@@ -18,11 +18,11 @@ export class LocalProvider extends BaseProvider {
this.loadWorkspaces(); this.loadWorkspaces();
} }
private _storeWorkspaces(workspaces: WS[]) { private _storeWorkspaces(workspaces: WorkspaceInfo[]) {
storage.setItem(WORKSPACE_KEY, JSON.stringify(workspaces)); storage.setItem(WORKSPACE_KEY, JSON.stringify(workspaces));
} }
public override async linkLocal(workspace: Workspace) { public override async linkLocal(workspace: BlocksuiteWorkspace) {
assert(workspace.room); assert(workspace.room);
let idb = this._idbMap.get(workspace.room); let idb = this._idbMap.get(workspace.room);
idb?.destroy(); idb?.destroy();
@@ -33,19 +33,19 @@ export class LocalProvider extends BaseProvider {
} }
public override async warpWorkspace( public override async warpWorkspace(
workspace: Workspace workspace: BlocksuiteWorkspace
): Promise<Workspace> { ): Promise<BlocksuiteWorkspace> {
assert(workspace.room); assert(workspace.room);
await this.linkLocal(workspace); await this.linkLocal(workspace);
return workspace; return workspace;
} }
override loadWorkspaces(): Promise<WS[]> { override loadWorkspaces(): Promise<WorkspaceInfo[]> {
const workspaceStr = storage.getItem(WORKSPACE_KEY); const workspaceStr = storage.getItem(WORKSPACE_KEY);
let workspaces: WS[] = []; let workspaces: WorkspaceInfo[] = [];
if (workspaceStr) { if (workspaceStr) {
try { try {
workspaces = JSON.parse(workspaceStr) as WS[]; workspaces = JSON.parse(workspaceStr) as WorkspaceInfo[];
workspaces.forEach(workspace => { workspaces.forEach(workspace => {
this._workspaces.add(workspace); this._workspaces.add(workspace);
}); });
@@ -77,11 +77,11 @@ export class LocalProvider extends BaseProvider {
public override async createWorkspace( public override async createWorkspace(
meta: WorkspaceMeta meta: WorkspaceMeta
): Promise<Workspace | undefined> { ): Promise<BlocksuiteWorkspace | undefined> {
assert(meta.name, 'Workspace name is required'); assert(meta.name, 'Workspace name is required');
this._logger('Creating affine workspace'); this._logger('Creating affine workspace');
const workspaceInfo: WS = { const workspaceInfo: WorkspaceInfo = {
name: meta.name, name: meta.name,
id: uuidv4(), id: uuidv4(),
isPublish: false, isPublish: false,
@@ -92,7 +92,7 @@ export class LocalProvider extends BaseProvider {
provider: 'local', provider: 'local',
}; };
const workspace = new Workspace({ room: workspaceInfo.id }); const workspace = new BlocksuiteWorkspace({ room: workspaceInfo.id });
this.linkLocal(workspace); this.linkLocal(workspace);
workspace.meta.setName(meta.name); workspace.meta.setName(meta.name);
if (!meta.avatar) { if (!meta.avatar) {

View File

@@ -1,6 +1,6 @@
import { getLogger } from '../logger'; import { getLogger } from '../logger';
export type Workspace = { export type WorkspaceInfo = {
name: string; name: string;
id: string; id: string;
isPublish?: boolean; isPublish?: boolean;
@@ -18,6 +18,6 @@ export type User = {
avatar: string; avatar: string;
}; };
export type WorkspaceMeta = Pick<Workspace, 'name' | 'avatar'>; export type WorkspaceMeta = Pick<WorkspaceInfo, 'name' | 'avatar'>;
export type Logger = ReturnType<typeof getLogger>; export type Logger = ReturnType<typeof getLogger>;

View File

@@ -1,25 +1,25 @@
import { Observable } from 'lib0/observable'; import { Observable } from 'lib0/observable';
import type { Workspace, WorkspaceMeta } from '../types'; import type { WorkspaceInfo, WorkspaceMeta } from '../types';
export interface WorkspacesScope { export interface WorkspacesScope {
get: (workspaceId: string) => Workspace | undefined; get: (workspaceId: string) => WorkspaceInfo | undefined;
list: () => Workspace[]; list: () => WorkspaceInfo[];
add: (workspace: Workspace) => void; add: (workspace: WorkspaceInfo) => void;
remove: (workspaceId: string) => boolean; remove: (workspaceId: string) => boolean;
clear: () => void; clear: () => void;
update: (workspaceId: string, workspaceMeta: Partial<WorkspaceMeta>) => void; update: (workspaceId: string, workspaceMeta: Partial<WorkspaceMeta>) => void;
} }
export interface WorkspacesChangeEvent { export interface WorkspacesChangeEvent {
added?: Workspace; added?: WorkspaceInfo;
deleted?: Workspace; deleted?: WorkspaceInfo;
updated?: Workspace; updated?: WorkspaceInfo;
} }
export class Workspaces extends Observable<'change'> { export class Workspaces extends Observable<'change'> {
private _workspacesMap = new Map<string, Workspace>(); private _workspacesMap = new Map<string, WorkspaceInfo>();
get workspaces(): Workspace[] { get workspaces(): WorkspaceInfo[] {
return Array.from(this._workspacesMap.values()); return Array.from(this._workspacesMap.values());
} }
@@ -37,7 +37,7 @@ export class Workspaces extends Observable<'change'> {
return this._workspacesMap.get(workspaceId); return this._workspacesMap.get(workspaceId);
}; };
const add = (workspace: Workspace) => { const add = (workspace: WorkspaceInfo) => {
if (this._workspacesMap.has(workspace.id)) { if (this._workspacesMap.has(workspace.id)) {
throw new Error(`Duplicate workspace id.`); throw new Error(`Duplicate workspace id.`);
} }
@@ -105,7 +105,7 @@ export class Workspaces extends Observable<'change'> {
// TODO: need to optimize // TODO: need to optimize
const list = () => { const list = () => {
const workspaces: Workspace[] = []; const workspaces: WorkspaceInfo[] = [];
scopedWorkspaceIds.forEach(id => { scopedWorkspaceIds.forEach(id => {
const workspace = this._workspacesMap.get(id); const workspace = this._workspacesMap.get(id);
if (workspace) { if (workspace) {