@aochuang/client-sdk 1.0.278

package/services/client-chat/client-chat-wechat/client-chat-socket.js

@@ -0,0 +1,215 @@
+import Socket from '../../socket'
+
+export default class Websocket {
+	get connectStatus () {
+		return this.socket.connectStatus
+	}
+	
+	get connectIpChange () {
+		return this.socket.connectIpChange
+	}
+	
+	constructor (options) {
+		options = Object.assign({
+			socket: null,
+			// 检查socket事件通知方法，有些场景不需要后端推送的命令执行，比如管理后台
+			checkSocketEventNotifyMethod: null,
+			// 执行回调事件前的方法
+			beforeFireCallback: null, 
+			onCmdNewMessage: null,
+			onCmdSendMessageResult: null,
+			onCmdMessageRecalled: null,
+			onCmdFriendInfoChanged: null,
+			onCmdDownloadVideoResult: null,
+			onCmdChatroomInfoChanged: null,
+			onCmdChatroomInfoChangedBatch: null,
+			onCmdChatroomMemberInfoChanged: null,
+			onCmdChatroomInvite: null,
+			onCmdNewGreetingMsg: null,
+			onHeartbeatCheck: null,
+			onHeartbeatSuccess: null,
+			onSendMessage: null,
+			onReceiveMessage: null,
+			onSocketOpen: null, 
+			onSocketClose: null, 
+			onSocketKickOut: null
+		}, options)
+		if (!options.socket) {
+			throw new Error('socket实例不能为空')
+		}
+		this.socket = options.socket
+		this.socket.log.onBeforeAdd(({ data }) => {
+			// 收消息需要拦截一些，因为下面有了
+			if (data && ['socket:sendMessage', 'socket:receiveMessage'].includes(data.group)) {
+				return false
+			}
+		})
+		this.socket.event.on('onCmdNewMessage', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				const cloneMsg = Object.assign({}, evt.data)
+				const newMsg = Object.assign({}, cloneMsg.chatroomMessage || cloneMsg.friendMessage)
+				if (typeof newMsg.content === 'string') {
+					newMsg.content = newMsg.content.substr(0, 100) // 只保留前100个字符，怕数据太多了
+				}
+				delete cloneMsg.chatroomMessage
+				delete cloneMsg.friendMessage
+				cloneMsg.message = newMsg
+				this.socket.logEnable && this.socket.log.addLog({
+					group: 'client-chat-socket:receiveMessage',
+					content: {
+						title: 'socket收到消息',
+						data: cloneMsg
+					}
+				})
+				options.onCmdNewMessage(evt)
+			}
+		})
+		this.socket.event.on('onCmdSendMessageResult', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdSendMessageResult(evt)
+			}
+		})
+		this.socket.event.on('onCmdFriendInfoChanged', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdFriendInfoChanged(evt)
+			}
+		})
+		this.socket.event.on('onCmdDownloadVideoResult', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdDownloadVideoResult(evt)
+			}
+		})
+		this.socket.event.on('onCmdChatroomInfoChanged', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdChatroomInfoChanged(evt)
+			}
+		})
+		this.socket.event.on('onCmdChatroomInfoChangedBatch', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdChatroomInfoChangedBatch(evt)
+			}
+		})
+		this.socket.event.on('onCmdChatroomMemberInfoChanged', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdChatroomMemberInfoChanged(evt)
+			}
+		})
+		this.socket.event.on('onCmdChatroomInvite', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdChatroomInvite(evt)
+			}
+		})
+		this.socket.event.on('onCmdNewGreetingMsg', (evt) => {
+			if (!options.checkSocketEventNotifyMethod || options.checkSocketEventNotifyMethod(evt) !== false) {
+				options.onCmdNewGreetingMsg(evt)
+			}
+		})
+		this.socket.event.on('onCmdMessageRecalled', options.onCmdMessageRecalled)
+		this.socket.event.on('onHeartbeatCheck', ({ mergeParams }) => {
+			options.onHeartbeatCheck && options.onHeartbeatCheck({
+				mergeParams
+			})
+		})
+		this.socket.event.on('onHeartbeatSuccess', (evt) => {
+			options.onHeartbeatSuccess && options.onHeartbeatSuccess(evt)
+		})
+		this.socket.event.on('onSendMessage', (evt) => {
+			const cloneData = Object.assign({}, evt.data)
+			if (cloneData && typeof cloneData.content === 'string') {
+				cloneData.content = cloneData.content.substr(0, 100)
+			}
+			if (evt.data.cmdType === 'CmdRequestWechatAccountsAliveStatus') {
+				delete cloneData.wechatAccountIds
+			}
+			this.socket.logEnable && this.socket.log.addLog({
+				group: 'client-chat-socket:sendMessage',
+				content: {
+					title: '发送消息',
+					data: cloneData
+				}
+			})
+			options.onSendMessage && options.onSendMessage(evt)
+		})
+		this.socket.event.on('onReceiveMessage', (evt) => {
+			options.onReceiveMessage && options.onReceiveMessage(evt)
+		})
+		this.socket.event.on('onOpen', (evt) => {
+			options.onSocketOpen && options.onSocketOpen(evt)
+		})
+		this.socket.event.on('onClose', (evt) => {
+			options.onSocketClose && options.onSocketClose(evt)
+		})
+		this.socket.event.on('onKickOut', (evt) => {
+			options.onSocketKickOut && options.onSocketKickOut(evt)
+		})
+		// 执行消息前回调
+		this.beforeFireCallback = options.beforeFireCallback
+	}
+	
+	connect () {
+		return this.socket.connect()
+	}
+	
+	reconnect () {
+		return this.socket.reconnect()
+	}
+	
+	close (options) {
+		return this.socket.close(options)
+	}
+
+	sendMessage (message, beforeCallback) {
+		return this.socket.sendMessage(message, (data) => {
+			let ps = Promise.resolve()
+			// 如果发送消息的时候，连接是断开或者失败的，重新连接一次，主要是防止用户发送消息的时候网络断开了，提示socket已关闭，体验不好，程序静默来处理
+			if (this.socket.connectStatus === Socket.CONNECT_STATUS_ENUM.Close || this.socket.connectStatus === Socket.CONNECT_STATUS_ENUM.Error) {
+				ps = this.socket.reconnect()
+			}
+			return ps.finally(() => {
+				if (beforeCallback) {
+					return beforeCallback(data)
+				}
+			})
+		})
+	}
+	
+	/**
+	 * 发送下载视频命令
+	*/
+	sendDownloadVideo (options) {
+		options = Object.assign({
+			wechatAccountId: null,
+			friendMessageId: null,
+			chatroomMessageId: null,
+			tencentUrl: null,
+			wechatTime: null
+		}, options, {
+			cmdType: "CmdDownloadVideo",
+		})
+        return this.socket.sendMessage(options)
+	}
+	
+	disconnect () {
+		return this.socket.disconnect()
+	}
+	
+	setUrl (url) {
+		this.socket.setUrl(url)
+	}
+
+	doHeartbeat () {
+		this.socket.doHeartbeat()
+	}
+    
+	addLog (data) {
+		this.socket.logEnable && this.socket.log.addLog(data)
+	}
+
+    exportLogData () {
+        return this.socket.exportLogData()
+    }
+    
+    clearLogData () {
+        return this.socket.clearLogData()
+    }
+}