AFFiNE-Mirror/packages/common/infra/src/workspace/engine/index.ts

import { Slot } from '@blocksuite/global/utils';

import { throwIfAborted } from '../../utils/throw-if-aborted';
import type { AwarenessEngine } from './awareness';
import type { BlobEngine, BlobStatus } from './blob';
import type { SyncEngine } from './sync';
import { type SyncEngineStatus } from './sync';

export interface WorkspaceEngineStatus {
  sync: SyncEngineStatus;
  blob: BlobStatus;
}

/**
 * # WorkspaceEngine
 *
 * sync ydoc, blob, awareness together
 */
export class WorkspaceEngine {
  _status: WorkspaceEngineStatus;
  onStatusChange = new Slot<WorkspaceEngineStatus>();

  get status() {
    return this._status;
  }

  set status(status: WorkspaceEngineStatus) {
    this._status = status;
    this.onStatusChange.emit(status);
  }

  constructor(
    public blob: BlobEngine,
    public sync: SyncEngine,
    public awareness: AwarenessEngine
  ) {
    this._status = {
      sync: sync.status,
      blob: blob.status,
    };
    sync.onStatusChange.on(status => {
      this.status = {
        sync: status,
        blob: blob.status,
      };
    });
    blob.onStatusChange.on(status => {
      this.status = {
        sync: sync.status,
        blob: status,
      };
    });
  }

  start() {
    this.sync.start();
    this.awareness.connect();
    this.blob.start();
  }

  canGracefulStop() {
    return this.sync.canGracefulStop();
  }

  async waitForGracefulStop(abort?: AbortSignal) {
    await this.sync.waitForGracefulStop(abort);
    throwIfAborted(abort);
    this.forceStop();
  }

  forceStop() {
    this.sync.forceStop();
    this.awareness.disconnect();
    this.blob.stop();
  }
}

export * from './awareness';
export * from './blob';
export * from './error';
export * from './sync';