feat(server): use doc service (#9967)

close CLOUD-94

packages/backend/server/src/core/doc-renderer/__tests__/controller.spec.ts

@@ -0,0 +1,85 @@
+import { randomUUID } from 'node:crypto';
+
+import { User, Workspace } from '@prisma/client';
+import ava, { TestFn } from 'ava';
+import request from 'supertest';
+import { Doc as YDoc } from 'yjs';
+
+import { createTestingApp, type TestingApp } from '../../../__tests__/utils';
+import { AppModule } from '../../../app.module';
+import { Config } from '../../../base';
+import { ConfigModule } from '../../../base/config';
+import { Models } from '../../../models';
+import { PgWorkspaceDocStorageAdapter } from '../../doc';
+import { PermissionService } from '../../permission';
+
+const test = ava as TestFn<{
+  models: Models;
+  app: TestingApp;
+  config: Config;
+  adapter: PgWorkspaceDocStorageAdapter;
+  permission: PermissionService;
+}>;
+
+test.before(async t => {
+  const { app } = await createTestingApp({
+    imports: [
+      ConfigModule.forRoot({
+        flavor: {
+          doc: false,
+        },
+        docService: {
+          endpoint: '',
+        },
+      }),
+      AppModule,
+    ],
+  });
+
+  t.context.models = app.get(Models);
+  t.context.config = app.get(Config);
+  t.context.adapter = app.get(PgWorkspaceDocStorageAdapter);
+  t.context.permission = app.get(PermissionService);
+  t.context.app = app;
+});
+
+let user: User;
+let workspace: Workspace;
+
+test.beforeEach(async t => {
+  t.context.config.docService.endpoint = t.context.app.getHttpServerUrl();
+  await t.context.app.initTestingDB();
+  user = await t.context.models.user.create({
+    email: 'test@affine.pro',
+  });
+  workspace = await t.context.models.workspace.create(user.id);
+});
+
+test.after.always(async t => {
+  await t.context.app.close();
+});
+
+test('should render page success', async t => {
+  const docId = randomUUID();
+  const { app, adapter, permission } = t.context;
+
+  const doc = new YDoc();
+  const text = doc.getText('content');
+  const updates: Buffer[] = [];
+
+  doc.on('update', update => {
+    updates.push(Buffer.from(update));
+  });
+
+  text.insert(0, 'hello');
+  text.insert(5, 'world');
+  text.insert(5, ' ');
+
+  await adapter.pushDocUpdates(workspace.id, docId, updates, user.id);
+  await permission.publishPage(workspace.id, docId);
+
+  await request(app.getHttpServer())
+    .get(`/workspace/${workspace.id}/${docId}`)
+    .expect(200);
+  t.pass();
+});

packages/backend/server/src/core/doc-renderer/__tests__/service.spec.ts

@@ -0,0 +1,85 @@
+import { randomUUID } from 'node:crypto';
+
+import { User, Workspace } from '@prisma/client';
+import ava, { TestFn } from 'ava';
+import { Doc as YDoc } from 'yjs';
+
+import { createTestingApp, type TestingApp } from '../../../__tests__/utils';
+import { AppModule } from '../../../app.module';
+import { Config } from '../../../base';
+import { ConfigModule } from '../../../base/config';
+import { Models } from '../../../models';
+import { PgWorkspaceDocStorageAdapter } from '../../doc';
+import { DocContentService } from '..';
+
+const test = ava as TestFn<{
+  models: Models;
+  app: TestingApp;
+  docContentService: DocContentService;
+  config: Config;
+  adapter: PgWorkspaceDocStorageAdapter;
+}>;
+
+test.before(async t => {
+  const { app } = await createTestingApp({
+    imports: [
+      ConfigModule.forRoot({
+        flavor: {
+          doc: false,
+        },
+        docService: {
+          endpoint: '',
+        },
+      }),
+      AppModule,
+    ],
+  });
+
+  t.context.models = app.get(Models);
+  t.context.docContentService = app.get(DocContentService);
+  t.context.config = app.get(Config);
+  t.context.adapter = app.get(PgWorkspaceDocStorageAdapter);
+  t.context.app = app;
+});
+
+let user: User;
+let workspace: Workspace;
+
+test.beforeEach(async t => {
+  t.context.config.docService.endpoint = t.context.app.getHttpServerUrl();
+  await t.context.app.initTestingDB();
+  user = await t.context.models.user.create({
+    email: 'test@affine.pro',
+  });
+  workspace = await t.context.models.workspace.create(user.id);
+});
+
+test.after.always(async t => {
+  await t.context.app.close();
+});
+
+test('should get doc content from doc service rpc', async t => {
+  const docId = randomUUID();
+  const { docContentService } = t.context;
+
+  const doc = new YDoc();
+  const text = doc.getText('content');
+  const updates: Buffer[] = [];
+
+  doc.on('update', update => {
+    updates.push(Buffer.from(update));
+  });
+
+  text.insert(0, 'hello');
+  text.insert(5, 'world');
+  text.insert(5, ' ');
+
+  await t.context.adapter.pushDocUpdates(workspace.id, docId, updates, user.id);
+
+  const docContent = await docContentService.getPageContent(
+    workspace.id,
+    docId
+  );
+  // TODO(@fengmk2): should create a test ydoc with blocks
+  t.is(docContent, null);
+});

packages/backend/server/src/core/doc-renderer/service.ts

@@ -2,7 +2,7 @@ import { Injectable } from '@nestjs/common';
 import { applyUpdate, Doc } from 'yjs';
 
 import { Cache, OnEvent } from '../../base';
-import { PgWorkspaceDocStorageAdapter } from '../doc';
+import { DocReader } from '../doc';
 import {
   type PageDocContent,
   parsePageDoc,
@@ -14,7 +14,7 @@ import {
 export class DocContentService {
   constructor(
     private readonly cache: Cache,
-    private readonly workspace: PgWorkspaceDocStorageAdapter
+    private readonly docReader: DocReader
   ) {}
 
   async getPageContent(
@@ -28,7 +28,7 @@ export class DocContentService {
       return cachedResult;
     }
 
-    const docRecord = await this.workspace.getDoc(workspaceId, guid);
+    const docRecord = await this.docReader.getDoc(workspaceId, guid);
     if (!docRecord) {
       return null;
     }
@@ -61,7 +61,7 @@ export class DocContentService {
       return cachedResult;
     }
 
-    const docRecord = await this.workspace.getDoc(workspaceId, workspaceId);
+    const docRecord = await this.docReader.getDoc(workspaceId, workspaceId);
     if (!docRecord) {
       return null;
     }