tmp

pages/ai/index.js

@@ -1,66 +1,237 @@
-// pages/ai/index.js
+const AIChatSocket = require('../../utils/chatmsg.js')
+
 Page({
-
-  /**
-   * 页面的初始数据
-   */
   data: {
-
+    messages: [],
+    inputValue: '',
+    isTyping: false,
+    scrollToMessage: '',
+    quickQuestions: [
+      '今日待处理订单有哪些？',
+      '最近客户反馈统计',
+      '如何修改服务价格？',
+      '陪诊员排班情况'
+    ]
   },
 
-  /**
-   * 生命周期函数--监听页面加载
-   */
-  onLoad(options) {
+  messageIdCounter: 0,
+  socket: null,
 
+  onLoad() {
+    this.loadChatHistory()
+    this.initSocket()
   },
 
-  /**
-   * 生命周期函数--监听页面初次渲染完成
-   */
-  onReady() {
-
-  },
-
-  /**
-   * 生命周期函数--监听页面显示
-   */
-  onShow() {
-
-  },
-
-  /**
-   * 生命周期函数--监听页面隐藏
-   */
-  onHide() {
-
-  },
-
-  /**
-   * 生命周期函数--监听页面卸载
-   */
   onUnload() {
-
+    if (this.socket) {
+      this.socket.close()
+      this.socket = null
+    }
   },
 
-  /**
-   * 页面相关事件处理函数--监听用户下拉动作
-   */
-  onPullDownRefresh() {
+  initSocket() {
+    this.socket = new AIChatSocket()
 
+    this.socket.onOpen(() => {
+      console.log('[AIChat] Socket opened')
+    })
+
+    this.socket.onMessage((data) => {
+      if (data && data.type === 'ai') {
+        this.handleAIReply(data)
+      } else if (data && data.type === 'system') {
+        console.log('[AIChat] System:', data.content)
+      }
+    })
+
+    this.socket.onError((err) => {
+      console.error('[AIChat] WebSocket error', err)
+      this.handleAIError('网络连接失败，请检查网络设置')
+    })
+
+    this.socket.onClose((res) => {
+      console.log('[AIChat] Socket closed', res)
+    })
+
+    this.socket.connect().catch((err) => {
+      console.error('[AIChat] Connect failed', err)
+    })
   },
 
-  /**
-   * 页面上拉触底事件的处理函数
-   */
-  onReachBottom() {
+  handleAIReply(data) {
+    let messages = null
+    if (this.data.messages.length > 0) {
+      let lastMessage = this.data.messages[this.data.messages.length - 1]
+      if (lastMessage.type === 'ai' && lastMessage._serverId === data.id) {
+        lastMessage.content += data.content || ''
+        messages = this.data.messages
+      }
+    }
 
+    if (!messages) {
+      const aiResponse = {
+        id: `msg_${++this.messageIdCounter}`,
+        _serverId: data.id,
+        type: 'ai',
+        contentType: 'text',
+        content: data.content || '',
+        time: this.formatTime(new Date())
+      }
+
+      if (data.sessionId) {
+        wx.setStorageSync('admin_ai_session_id', data.sessionId)
+      }
+
+      messages = [...this.data.messages, aiResponse]
+    }
+
+    this.setData({
+      messages: messages,
+      isTyping: false
+    })
+    this.saveChatHistory(messages)
+    this.scrollToBottom()
   },
 
-  /**
-   * 用户点击右上角分享
-   */
-  onShareAppMessage() {
+  loadChatHistory() {
+    try {
+      const history = wx.getStorageSync('admin_ai_chat_history')
+      if (history && history.length > 0) {
+        this.setData({ messages: history })
+        this.scrollToBottom()
+      }
+    } catch (e) {
+      console.log('加载聊天记录失败', e)
+    }
+  },
 
+  saveChatHistory(messages) {
+    try {
+      wx.setStorageSync('admin_ai_chat_history', messages)
+    } catch (e) {
+      console.log('保存聊天记录失败', e)
+    }
+  },
+
+  onInputChange(e) {
+    const value = e.detail.value
+    this.setData({
+      inputValue: value,
+      canSend: !!value.trim()
+    })
+  },
+
+  onQuickQuestionTap(e) {
+    const question = e.currentTarget.dataset.question
+    this.setData({ inputValue: question, canSend: true })
+    this.sendMessage()
+  },
+
+  sendMessage() {
+    const content = this.data.inputValue.trim()
+    if (!content) return
+
+    const userMessage = {
+      id: `msg_${++this.messageIdCounter}`,
+      type: 'user',
+      contentType: 'text',
+      content: content,
+      time: this.formatTime(new Date())
+    }
+
+    const messages = [...this.data.messages, userMessage]
+    this.setData({
+      messages: messages,
+      inputValue: '',
+      canSend: false,
+      isTyping: true
+    })
+
+    this.saveChatHistory(messages)
+    this.scrollToBottom()
+
+    this.sendToAI(content)
+  },
+
+  sendToAI(content, type = 'chat') {
+    const app = getApp()
+    const user = app.globalData.user
+
+    if (!this.socket) {
+      this.handleAIError('网络连接失败，请检查网络设置')
+      return
+    }
+
+    if (!this.socket.isConnected) {
+      this.socket.connect().then(() => {
+        this._doSend(type, content, user, app)
+      }).catch((err) => {
+        console.error('[AIChat] Reconnect failed', err)
+        this.handleAIError('网络连接失败，请检查网络设置')
+      })
+    } else {
+      this._doSend(type, content, user, app)
+    }
+  },
+
+  _doSend(type, content, user, app) {
+    this.socket.send({
+      type: type,
+      content: content,
+      userId: user ? user._id : '',
+      token: user ? user.security.token : '',
+      appId: app.globalData.appId || '',
+      agent: 'escort-admin'
+    })
+  },
+
+  handleAIError(errorMessage) {
+    const errorResponse = {
+      id: `msg_${++this.messageIdCounter}`,
+      type: 'ai',
+      contentType: 'text',
+      content: errorMessage,
+      time: this.formatTime(new Date())
+    }
+
+    const messages = [...this.data.messages, errorResponse]
+    this.setData({
+      messages: messages,
+      isTyping: false
+    })
+    this.saveChatHistory(messages)
+    this.scrollToBottom()
+  },
+
+  scrollToBottom() {
+    setTimeout(() => {
+      const lastMessage = this.data.messages[this.data.messages.length - 1]
+      if (lastMessage) {
+        this.setData({
+          scrollToMessage: `msg-${lastMessage.id}`
+        })
+      }
+    }, 100)
+  },
+
+  formatTime(date) {
+    const hours = String(date.getHours()).padStart(2, '0')
+    const minutes = String(date.getMinutes()).padStart(2, '0')
+    return `${hours}:${minutes}`
+  },
+
+  clearChat() {
+    wx.showModal({
+      title: '确认清空',
+      content: '确定要清空所有聊天记录吗？',
+      success: (res) => {
+        if (res.confirm) {
+          this.sendToAI('clear', 'clear')
+          this.setData({ messages: [] })
+          wx.removeStorageSync('admin_ai_chat_history')
+          wx.removeStorageSync('admin_ai_session_id')
+        }
+      }
+    })
   }
-})
+})