yunzhu-im-mobile/composables/useWebSocket.js

/**
 * 消息 WebSocket：登录后连接，收到新消息时更新 chatStore（消息列表、本地未读、会话预览），不整表刷新
 */
import { getToken } from '../utils/api'
import { chatStore } from '../store/chat'

const WS_BASE = 'wss://api.hnyunzhu.com/api/v1/ws/messages'
const HEARTBEAT_INTERVAL = 30000
let socketTask = null
let fetchContactsRef = null
let heartbeatTimer = null

function normalizeWsMessage(raw) {
	const m = raw.message || raw
	const type = m.type ?? 'MESSAGE'
	// 通知类消息：应用发给用户，当前用户始终为接收方
	const isMe = type === 'NOTIFICATION' ? false : (m.is_me ?? m.isMe)
	return {
		id: String(m.id),
		type,
		senderId: m.sender_id ?? m.senderId,
		receiverId: m.receiver_id ?? m.receiverId,
		content: m.content ?? m.text ?? '',
		contentType: m.content_type ?? m.contentType ?? 'TEXT',
		title: m.title,
		createdAt: m.created_at ?? m.createdAt,
		isMe,
		actionUrl: m.action_url ?? m.actionUrl,
		actionText: m.action_text ?? m.actionText
	}
}

/**
 * 根据推送消息计算会话 ID：私信用 sender_id/receiver_id，通知用负的 app_id
 */
function getContactIdFromMessage(msg, currentUserId) {
	const type = msg.type ?? 'MESSAGE'
	const appId = msg.app_id ?? msg.appId
	if (type === 'NOTIFICATION' && appId != null && appId !== '') {
		return -Math.abs(Number(appId))
	}
	const senderId = msg.sender_id ?? msg.senderId
	const receiverId = msg.receiver_id ?? msg.receiverId
	// 当前用户是接收方则会话对方是 sender_id，否则是 receiver_id
	if (String(receiverId) === String(currentUserId)) return senderId
	return receiverId
}

export function useWebSocket(fetchContacts) {
	fetchContactsRef = fetchContacts

	function connect() {
		const token = getToken()
		if (!token || socketTask) return
		const url = `${WS_BASE}?token=${encodeURIComponent(token)}`
		socketTask = uni.connectSocket({
			url,
			success: () => {}
		})
		uni.onSocketOpen(() => {
			console.log('[WS] messages connected')
			heartbeatTimer = setInterval(() => {
				try {
					if (socketTask) uni.sendSocketMessage({ data: 'ping' })
				} catch (e) {}
			}, HEARTBEAT_INTERVAL)
		})
		uni.onSocketMessage((res) => {
			try {
				// 心跳：服务端回复 pong
				if (res.data === 'pong') return
				const data = typeof res.data === 'string' ? JSON.parse(res.data) : res.data
				// 文档格式：{ type: 'NEW_MESSAGE', data: { id, sender_id, receiver_id, ... } }
				const msg = data.type === 'NEW_MESSAGE' ? data.data : (data.message ?? data)
				if (!msg) return
				const normalized = normalizeWsMessage({ message: msg })
				// 需要当前用户 id 判断会话方：若后端推送里带 current_user_id 用那个，否则用 receiver_id 判断
				const currentUserId = data.current_user_id ?? normalized.receiverId
				const contactId = getContactIdFromMessage(msg, currentUserId)
				if (contactId != null) {
					const cid = String(contactId)
					const list = chatStore.messages[cid] || []
					const hasId = normalized.id && list.some((m) => String(m.id) === String(normalized.id))
					if (hasId) return
					chatStore.appendMessage(cid, normalized)
					// 前台 & 后台消息通知：若当前不在该会话，则给出提示
					const isActive = String(chatStore.activeContactId || '') === String(contactId)
					// 别人发给我且不在当前会话：本地未读 +1，并更新底部消息 Tab 角标
					if (!normalized.isMe && !isActive) {
						chatStore.incrementUnread(cid)
						chatStore.updateTabBarUnreadBadge()
					}
					if (!isActive) {
						const contact = (chatStore.contacts || []).find(
							(c) => String(c.user_id || c.id) === String(contactId)
						)
						const title = (contact && contact.title) || '新消息'
						let body = ''
						if (normalized.contentType === 'TEXT') {
							body = normalized.content ? String(normalized.content).slice(0, 50) : ''
						} else if (normalized.contentType === 'IMAGE') {
							body = '[图片]'
						} else if (normalized.contentType === 'VIDEO') {
							body = '[视频]'
						} else if (normalized.contentType === 'USER_NOTIFICATION') {
							body = normalized.title
								? String(normalized.title).slice(0, 50)
								: normalized.content
									? String(normalized.content).slice(0, 50)
									: '[通知]'
						} else {
							body = normalized.title || '[文件]'
						}
						// #ifdef APP-PLUS
						try {
							plus.push.createMessage(body || '您有一条新消息', { contactId }, { title })
						} catch (e) {
							// 兜底为 Toast
							uni.showToast({ title: `${title}: ${body || '新消息'}`, icon: 'none' })
						}
						// #endif
						// #ifndef APP-PLUS
						uni.showToast({ title: `${title}: ${body || '新消息'}`, icon: 'none' })
						// #endif
					}
					// 只更新该会话在列表中的最后一条预览与时间，不整表刷新
					let preview = ''
					if (normalized.contentType === 'TEXT') {
						preview = normalized.content ? String(normalized.content).slice(0, 50) : ''
					} else if (normalized.contentType === 'IMAGE') preview = '[图片]'
					else if (normalized.contentType === 'VIDEO') preview = '[视频]'
					else if (normalized.contentType === 'USER_NOTIFICATION') {
						preview = normalized.title
							? String(normalized.title).slice(0, 50)
							: normalized.content
								? String(normalized.content).slice(0, 50)
								: '[通知]'
					} else preview = normalized.title || '[文件]'
					chatStore.updateContactPreview(cid, { lastMessage: preview, time: normalized.createdAt })
				}
			} catch (e) {
				console.warn('[WS] parse message error', e)
			}
		})
		uni.onSocketError((err) => {
			console.warn('[WS] error', err)
		})
		uni.onSocketClose(() => {
			if (heartbeatTimer) {
				clearInterval(heartbeatTimer)
				heartbeatTimer = null
			}
			socketTask = null
		})
	}

	function disconnect() {
		if (heartbeatTimer) {
			clearInterval(heartbeatTimer)
			heartbeatTimer = null
		}
		if (socketTask) {
			uni.closeSocket()
			socketTask = null
		}
	}

	return { connect, disconnect }
}