feat: add MessageCenterHandler (#770)

Co-authored-by: Peng Xiao <pengxiao@freewheel.tv>

packages/data-center/src/datacenter.ts

@@ -6,23 +6,25 @@ import type {
   CreateWorkspaceInfoParams,
   UpdateWorkspaceMetaParams,
 } from './provider/base';
-import { LocalProvider } from './provider/local/local';
+import { LocalProvider } from './provider/local';
 import { AffineProvider } from './provider';
 import type { Message } from './types';
 import assert from 'assert';
 import { getLogger } from './logger';
-import { createBlocksuiteWorkspace } from './utils/index';
+import { createBlocksuiteWorkspace } from './utils';
 import { MessageCenter } from './message';
 import { WorkspaceUnit } from './workspace-unit';
 /**
  * @class DataCenter
  * @classdesc Data center is made for managing different providers for business
  */
+
 export class DataCenter {
   private readonly _workspaceUnitCollection = new WorkspaceUnitCollection();
   private readonly _logger = getLogger('dc');
   private _workspaceInstances: Map<string, BlocksuiteWorkspace> = new Map();
   private _messageCenter = MessageCenter.getInstance();
+
   /**
    * A mainProvider must exist as the only data trustworthy source.
    */
@@ -304,7 +306,6 @@ export class DataCenter {
 
   /**
    * remove the new member to the workspace
-   * @param {number} permissionId permission id
    */
   public async removeMember(workspaceId: string, permissionId: number) {
     const workspaceInfo = this._workspaceUnitCollection.find(workspaceId);

packages/data-center/src/message/message.ts

@@ -32,5 +32,8 @@ export class MessageCenter extends Observable<string> {
 
   public onMessage(callback: (message: Message) => void) {
     this.on('message', callback);
+    return () => {
+      this.off('message', callback);
+    };
   }
 }

packages/data-center/src/provider/affine/apis/auth.ts

@@ -105,17 +105,21 @@ export class Auth {
         type: 'Refresh',
         token: refreshToken || this._refreshToken,
       });
-      this._padding.finally(() => {
-        // clear on settled
-        this._padding = undefined;
-      });
       this._refreshToken = refreshToken || this._refreshToken;
     }
-    const res = await this._padding;
-    if (!refreshToken || refreshToken !== this._refreshToken) {
-      this.setLogin(res);
+    try {
+      const res = await this._padding;
+      if (res && (!refreshToken || refreshToken !== this._refreshToken)) {
+        this.setLogin(res);
+      }
+      return true;
+    } catch {
+      this._logger('Failed to refresh token');
+    } finally {
+      // clear on settled
+      this._padding = undefined;
     }
-    return true;
+    return false;
   }
 
   get user() {

packages/data-center/src/provider/affine/apis/request.ts

@@ -14,19 +14,15 @@ export const bareClient: KyInstance = ky.extend({
   // todo: report timeout error
   timeout: 60000,
   hooks: {
-    // afterResponse: [
-    //   async (_request, _options, response) => {
-    //     if (response.status === 200) {
-    //       const data = await response.json();
-    //       if (data.error) {
-    //         return new Response(data.error.message, {
-    //           status: data.error.code,
-    //         });
-    //       }
-    //     }
-    //     return response;
-    //   },
-    // ],
+    beforeError: [
+      error => {
+        const { response } = error;
+        if (response.status === 401) {
+          _sendMessage(MessageCenter.messageCode.noPermission);
+        }
+        return error;
+      },
+    ],
   },
 });
 
@@ -59,15 +55,5 @@ export const client: KyInstance = bareClient.extend({
         request.headers.set('Authorization', auth.token);
       },
     ],
-
-    beforeError: [
-      error => {
-        const { response } = error;
-        if (response.status === 401) {
-          _sendMessage(MessageCenter.messageCode.noPermission);
-        }
-        return error;
-      },
-    ],
   },
 });