feat: add TailchatWsClient and update tailchat-types

Author: moonrailgun

client/packages/tailchat-client-sdk/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tailchat-client-sdk",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "",
   "main": "lib/index.js",
   "scripts": {
@@ -16,9 +16,12 @@
     "@types/node": "^18.16.1",
     "jest": "27.5.1",
     "ts-jest": "27.1.4",
+    "tailchat-types": "*",
     "typescript": "^4.9.5"
   },
   "dependencies": {
-    "axios": "^1.3.2"
+    "axios": "^1.3.2",
+    "socket.io-client": "^4.7.1",
+    "socket.io-msgpack-parser": "^3.0.2"
   }
 }

client/packages/tailchat-client-sdk/src/openapi/client.ts renamed to client/packages/tailchat-client-sdk/src/openapi/client/base.ts

@@ -1,7 +1,7 @@
 import axios, { AxiosInstance } from 'axios';
 import crypto from 'crypto';
 
-export class TailchatClient {
+export class TailchatBaseClient {
   request: AxiosInstance;
   jwt: string | null = null;
   userId: string | null = null;
@@ -58,16 +58,20 @@ export class TailchatClient {
     }
   }
 
+  async waitingForLogin(): Promise<void> {
+    await Promise.resolve(this.loginP);
+  }
+
   async call(action: string, params = {}) {
     try {
-      await Promise.resolve(this.loginP);
+      await this.waitingForLogin();
       console.log('Calling:', action);
       const { data } = await this.request.post(
         '/api/' + action.replace(/\./g, '/'),
         params
       );
 
-      return data;
+      return data.data;
     } catch (err: any) {
       console.error('Service Call Failed:', err);
       const data: string = err?.response?.data;
@@ -84,7 +88,18 @@ export class TailchatClient {
     }
   }
 
-  async whoami() {
+  async whoami(): Promise<{
+    userAgent: string;
+    language: string;
+    user: {
+      _id: string;
+      nickname: string;
+      email: string;
+      avatar: string;
+    };
+    token: string;
+    userId: string;
+  }> {
     return this.call('user.whoami');
   }
 

client/packages/tailchat-client-sdk/src/openapi/client/http.ts

@@ -0,0 +1,3 @@
+import { TailchatBaseClient } from './base';
+
+export class TailchatHTTPClient extends TailchatBaseClient {}

client/packages/tailchat-client-sdk/src/openapi/client/index.ts

@@ -0,0 +1,8 @@
+export {
+  /**
+   * @deprecated please rename to TailchatHTTPClient
+   */
+  TailchatHTTPClient as TailchatClient,
+  TailchatHTTPClient,
+} from './http';
+export { TailchatWsClient } from './ws';

client/packages/tailchat-client-sdk/src/openapi/client/ws.ts

@@ -0,0 +1,103 @@
+import { TailchatBaseClient } from './base';
+import io, { Socket } from 'socket.io-client';
+import * as msgpackParser from 'socket.io-msgpack-parser';
+import type { ChatMessage } from 'tailchat-types';
+
+export class TailchatWsClient extends TailchatBaseClient {
+  public socket: Socket | null = null;
+
+  connect(): Promise<Socket> {
+    return new Promise<Socket>(async (resolve, reject) => {
+      await this.waitingForLogin();
+
+      const token = this.jwt;
+
+      const socket = (this.socket = io(this.url, {
+        transports: ['websocket'],
+        auth: {
+          token,
+        },
+        forceNew: true,
+        parser: msgpackParser,
+      }));
+
+      socket.once('connect', () => {
+        // 连接成功
+        this.emit('chat.converse.findAndJoinRoom')
+          .then((res) => {
+            console.log('Joined rooms', res.data);
+            resolve(socket);
+          })
+          .catch((err) => {
+            reject(err);
+          });
+      });
+      socket.once('error', () => {
+        reject();
+      });
+
+      socket.on('disconnect', (reason) => {
+        console.log(`disconnect due to ${reason}`);
+        this.socket = null;
+      });
+
+      socket.onAny((ev) => {
+        console.log('onAny', ev);
+      });
+    });
+  }
+
+  disconnect() {
+    if (!this.socket) {
+      console.warn('You should call it after connect');
+      return;
+    }
+
+    this.socket.disconnect();
+    this.socket = null;
+  }
+
+  emit(eventName: string, eventData: any = {}) {
+    if (!this.socket) {
+      console.warn('You should call it after connect');
+      throw new Error('You should call it after connect');
+    }
+
+    return this.socket.emitWithAck(eventName, eventData);
+  }
+
+  on(eventName: string, callback: (payload: any) => void) {
+    if (!this.socket) {
+      console.warn('You should call it after connect');
+      return;
+    }
+
+    this.socket.on(eventName, callback);
+  }
+
+  once(eventName: string, callback: (payload: any) => void) {
+    if (!this.socket) {
+      console.warn('You should call it after connect');
+      return;
+    }
+
+    this.socket.once(eventName, callback);
+  }
+
+  off(eventName: string, callback: (payload: any) => void) {
+    if (!this.socket) {
+      console.warn('You should call it after connect');
+      return;
+    }
+
+    this.socket.off(eventName, callback);
+  }
+
+  onMessage(callback: (messagePayload: ChatMessage) => void) {
+    this.on('notify:chat.message.add', callback);
+  }
+
+  onMessageUpdate(callback: (messagePayload: ChatMessage) => void) {
+    this.on('notify:chat.message.update', callback);
+  }
+}

client/packages/tailchat-client-sdk/test/index.ts

@@ -0,0 +1,20 @@
+import { TailchatWsClient } from '../src';
+
+const HOST = process.env.HOST;
+const APPID = process.env.APPID;
+const APPSECRET = process.env.APPSECRET;
+
+if (!HOST || !APPID || !APPSECRET) {
+  console.log('require env: HOST, APPID, APPSECRET');
+  process.exit(1);
+}
+
+const client = new TailchatWsClient(HOST, APPID, APPSECRET);
+
+client.connect().then(async () => {
+  console.log('Login Success!');
+
+  client.onMessage((message) => {
+    console.log('Receive message', message);
+  });
+});

client/packages/tailchat-client-sdk/tsconfig.json

@@ -2,6 +2,7 @@
   "compilerOptions": {
     "target": "esnext",
     "lib": ["ESNext"],
+    "skipLibCheck": true,
     "outDir": "lib",
     "declaration": true,
     "declarationMap": true,

client/shared/model/message.ts

@@ -1,31 +1,7 @@
 import { request } from '../api/request';
+import type { ChatMessageReaction, ChatMessage } from 'tailchat-types';
 
-export interface ChatMessageReaction {
-  name: string;
-  author: string;
-}
-
-export interface ChatMessage {
-  _id: string;
-
-  content: string;
-
-  author?: string;
-
-  groupId?: string;
-
-  converseId: string;
-
-  reactions?: ChatMessageReaction[];
-
-  hasRecall?: boolean;
-
-  meta?: Record<string, unknown>;
-
-  createdAt?: string;
-
-  updatedAt?: string;
-}
+export { ChatMessageReaction, ChatMessage };
 
 export interface SimpleMessagePayload {
   groupId?: string;

packages/types/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tailchat-types",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Tailchat model types",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",

packages/types/src/index.ts

@@ -1,2 +1,3 @@
 export * from './model/inbox';
 export * from './model/user';
+export * from './model/message';