aicq-chat-plugin 2.3.0 → 2.4.1

package/index.js

@@ -350,6 +350,48 @@ app.delete('/api/chat/:messageId', (req, res) => {
   res.json({ success: true });
 });
 
+// Streaming endpoints — for external systems / OpenClaw agent output
+app.post('/api/chat/stream-chunk', (req, res) => {
+  try {
+    const { targetId, friend_id, chunk_type, chunkType, data } = req.body;
+    const streamTarget = targetId || friend_id;
+    if (!streamTarget) return res.status(400).json({ error: 'targetId or friend_id is required' });
+    if (!data) return res.status(400).json({ error: 'data is required' });
+    const type = chunk_type || chunkType || 'text';
+    if (!['text', 'reasoning', 'tool_call', 'tool_result'].includes(type)) {
+      return res.status(400).json({ error: `Invalid chunk_type: ${type}` });
+    }
+    const sent = serverClient.sendWS({
+      type: 'stream_chunk',
+      to: streamTarget,
+      chunkType: type,
+      data: data,
+    });
+    if (!sent) return res.status(503).json({ error: 'Not connected to server', success: false });
+    res.json({ success: true });
+  } catch (e) {
+    res.status(500).json({ error: e.message });
+  }
+});
+
+app.post('/api/chat/stream-end', (req, res) => {
+  try {
+    const { targetId, friend_id, message_id, messageId } = req.body;
+    const streamTarget = targetId || friend_id;
+    if (!streamTarget) return res.status(400).json({ error: 'targetId or friend_id is required' });
+    const msgId = message_id || messageId || ('msg_' + Date.now() + '_' + Math.random().toString(36).substr(2, 6));
+    const sent = serverClient.sendWS({
+      type: 'stream_end',
+      to: streamTarget,
+      messageId: msgId,
+    });
+    if (!sent) return res.status(503).json({ error: 'Not connected to server', success: false });
+    res.json({ success: true, messageId: msgId });
+  } catch (e) {
+    res.status(500).json({ error: e.message });
+  }
+});
+
 // File upload
 app.post('/api/upload', upload.single('file'), async (req, res) => {
   try {
@@ -575,6 +617,33 @@ async function handleGatewayCall(method, kwargs = {}) {
       return await chat.sendMessage(currentAgentId, kwargs.targetId, kwargs.content, { isGroup: kwargs.isGroup });
     case 'aicq.chat.history':
       return { messages: db.getChatHistory(currentAgentId, kwargs.targetId, { limit: kwargs.limit || 50 }) };
+    case 'aicq.chat.streamChunk': {
+      if (!kwargs.friend_id && !kwargs.targetId) return { error: 'friend_id or targetId is required' };
+      if (!kwargs.data) return { error: 'data is required' };
+      const chunkType = kwargs.chunk_type || kwargs.chunkType || 'text';
+      if (!['text', 'reasoning', 'tool_call', 'tool_result'].includes(chunkType)) return { error: `Invalid chunk_type: ${chunkType}` };
+      const streamTarget = kwargs.friend_id || kwargs.targetId;
+      const sent = serverClient.sendWS({
+        type: 'stream_chunk',
+        to: streamTarget,
+        chunkType: chunkType,
+        data: kwargs.data,
+      });
+      if (!sent) return { error: 'Not connected to server', success: false };
+      return { success: true };
+    }
+    case 'aicq.chat.streamEnd': {
+      if (!kwargs.friend_id && !kwargs.targetId) return { error: 'friend_id or targetId is required' };
+      const endTarget = kwargs.friend_id || kwargs.targetId;
+      const msgId = kwargs.message_id || kwargs.messageId || ('msg_' + Date.now() + '_' + Math.random().toString(36).substr(2, 6));
+      const endSent = serverClient.sendWS({
+        type: 'stream_end',
+        to: endTarget,
+        messageId: msgId,
+      });
+      if (!endSent) return { error: 'Not connected to server', success: false };
+      return { success: true, messageId: msgId };
+    }
     default:
       return { error: `Unknown method: ${method}` };
   }

package/lib/chat.js

@@ -21,6 +21,8 @@ class ChatManager {
     this.server.onMessage('handshake_initiate', (data) => this._handleHandshakeRequest(data));
     this.server.onMessage('presence', (data) => this._handlePresence(data));
     this.server.onMessage('file_chunk', (data) => this._handleFileChunk(data));
+    this.server.onMessage('stream_chunk', (data) => this._handleStreamChunk(data));
+    this.server.onMessage('stream_end', (data) => this._handleStreamEnd(data));
   }
 
   setOnNewMessage(callback) {
@@ -208,6 +210,46 @@ class ChatManager {
     console.log('[Chat] File chunk from', data.from);
   }
 
+  _handleStreamChunk(data) {
+    // Incoming streaming chunk from another agent
+    const agentId = this.server.currentAgentId;
+    if (!agentId) return;
+
+    const fromId = data.from;
+    const chunkType = data.chunkType || 'text';
+    const chunkData = data.data;
+
+    // Notify callback so OpenClaw agent can process streaming input
+    if (this._onNewMessage) {
+      this._onNewMessage({
+        type: 'stream_chunk',
+        from_id: fromId,
+        chunk_type: chunkType,
+        data: chunkData,
+      });
+    }
+    console.log('[Chat] Stream chunk from', fromId, 'type:', chunkType);
+  }
+
+  _handleStreamEnd(data) {
+    // Incoming stream end signal from another agent
+    const agentId = this.server.currentAgentId;
+    if (!agentId) return;
+
+    const fromId = data.from;
+    const messageId = data.messageId || '';
+
+    // Notify callback so OpenClaw agent knows stream is complete
+    if (this._onNewMessage) {
+      this._onNewMessage({
+        type: 'stream_end',
+        from_id: fromId,
+        message_id: messageId,
+      });
+    }
+    console.log('[Chat] Stream end from', fromId, 'messageId:', messageId);
+  }
+
   // ─── Chat History ─────────────────────────────────────────────────
 
   getHistory(agentId, targetId, { limit = 50, before = null } = {}) {

package/openclaw.plugin.json

@@ -1,7 +1,7 @@
 {
   "id": "aicq-chat",
   "name": "AICQ Encrypted Chat",
-  "version": "2.1.0",
+  "version": "2.2.0",
   "description": "End-to-end encrypted chat plugin for OpenClaw agents — Node.js implementation with full UI",
   "entry": "index.js",
   "activation": {
@@ -25,6 +25,8 @@
       "aicq.chat.history",
       "aicq.chat.send",
       "aicq.chat.delete",
+      "aicq.chat.streamChunk",
+      "aicq.chat.streamEnd",
       "aicq.groups.list",
       "aicq.groups.create",
       "aicq.groups.join",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aicq-chat-plugin",
-  "version": "2.3.0",
+  "version": "2.4.1",
   "description": "AICQ End-to-end Encrypted Chat Plugin for OpenClaw — Full UI with friend management, group chat, file transfer, and AI agent communication",
   "main": "index.js",
   "bin": {

package/public/index.html

@@ -122,9 +122,28 @@ body{font-family:-apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans S
 .msg-bubble a{color:var(--info)}
 .msg-bubble table{border-collapse:collapse;margin:6px 0}
 .msg-bubble th,.msg-bubble td{border:1px solid var(--border);padding:4px 8px;font-size:13px}
+/* Toast */
+.toast{position:fixed;top:20px;left:50%;transform:translateX(-50%);background:var(--bg2);border:1px solid var(--border);color:var(--text);padding:12px 24px;border-radius:8px;font-size:14px;z-index:9999;box-shadow:0 4px 20px rgba(0,0,0,.5);max-width:90%;text-align:center;opacity:0;transition:opacity .3s;pointer-events:none}
+.toast.show{opacity:1}
+.toast.warning{border-color:var(--warning);background:rgba(245,158,11,.15);color:#fcd34d}
+/* Backup section */
+.backup-section{background:var(--bg);border:1px solid var(--border);border-radius:8px;padding:12px;margin-top:8px}
+.backup-section h4{font-size:14px;margin-bottom:8px;color:var(--text)}
+.backup-section p{font-size:12px;color:var(--text2);line-height:1.6;margin:4px 0}
+.backup-section .warning-box{font-size:12px;padding:8px;margin:8px 0}
+.backup-btns{display:flex;gap:8px;margin-top:10px}
+.backup-btns .btn{flex:1;text-align:center}
+/* Key match row */
+.key-match-row{display:flex;align-items:center;gap:10px;padding:10px 0;border-bottom:1px solid var(--border)}
+.key-match-row:last-child{border-bottom:none}
+.key-match-row label{font-size:13px;color:var(--text2);min-width:120px;white-space:nowrap;overflow:hidden;text-overflow:ellipsis}
+.key-match-row select{flex:1;padding:6px 10px;background:var(--bg);border:1px solid var(--border);border-radius:6px;color:var(--text);font-size:13px}
 </style>
 </head>
 <body>
+<!-- Toast -->
+<div class="toast" id="toast"></div>
+
 <div class="app">
   <!-- Right Panel -->
   <div class="right-panel">
@@ -143,6 +162,9 @@ body{font-family:-apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans S
       <button class="action-btn" onclick="showModal('settings')">
         <span class="icon">⚙️</span>设置
       </button>
+      <button class="action-btn" onclick="confirmLogout()">
+        <span class="icon">🚪</span>登出
+      </button>
     </div>
     <div class="list-section">
       <h4>好友</h4>
@@ -240,8 +262,25 @@ body{font-family:-apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans S
       <button class="btn btn-secondary" onclick="showKeysModal()" style="width:100%;text-align:left">🔑 显示/重新生成密钥</button>
       <button class="btn btn-secondary" onclick="createNewAgent()" style="width:100%;text-align:left">➕ 创建新 Agent</button>
       <button class="btn btn-secondary" onclick="syncData()" style="width:100%;text-align:left">🔄 同步服务器数据</button>
+      <div class="backup-section">
+        <h4>💾 数据备份</h4>
+        <p><strong>导出内容：</strong>聊天消息、好友关系、群聊信息、Agent 身份密钥（含私钥）</p>
+        <div class="warning-box">⚠️ 私钥包含在导出文件中！请妥善保管备份文件，切勿分享给他人。拥有私钥的人可以冒充你的身份发送消息。</div>
+        <p><strong>数据丢失场景：</strong></p>
+        <p>• 清除浏览器缓存数据</p>
+        <p>• 删除浏览器数据/历史记录</p>
+        <p>• 使用无痕/隐身模式（关闭窗口后数据清除）</p>
+        <p>• 更换浏览器或设备</p>
+        <p><strong>恢复方法：</strong>点击「导入备份」选择之前导出的 JSON 文件。若备份中的 Agent 私钥与当前 Agent 不匹配，系统会提示你手动对应。</p>
+        <div class="backup-btns">
+          <button class="btn btn-primary" onclick="exportBackup()">📤 导出备份</button>
+          <button class="btn btn-secondary" onclick="triggerImportBackup()">📥 导入备份</button>
+        </div>
+        <input type="file" id="backupFileInput" accept=".json" style="display:none" onchange="handleImportBackup(this)">
+      </div>
     </div>
     <div class="btn-row" style="margin-top:20px">
+      <button class="btn btn-danger" onclick="confirmLogout()">🚪 登出</button>
       <button class="btn btn-primary" onclick="hideModal('settings')">关闭</button>
     </div>
   </div>
@@ -343,6 +382,20 @@ body{font-family:-apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans S
   </div>
 </div>
 
+<!-- Backup Key Match Modal -->
+<div class="modal-overlay" id="modal-backupMatch">
+  <div class="modal" style="max-width:560px">
+    <h3>🔑 Agent 密钥匹配</h3>
+    <div class="warning-box">备份中的 Agent 私钥与当前 Agent 不匹配，请手动选择对应关系。未匹配的聊天记录将不会导入。</div>
+    <p style="font-size:13px;color:var(--text2);margin-bottom:12px">左侧为备份中的 Agent，右侧选择当前对应的 Agent：</p>
+    <div id="backupMatchList"></div>
+    <div class="btn-row">
+      <button class="btn btn-secondary" onclick="skipBackupMatch()">跳过不匹配项</button>
+      <button class="btn btn-primary" onclick="applyBackupMatch()">确认匹配并导入</button>
+    </div>
+  </div>
+</div>
+
 <script>
 // ─── State ──────────────────────────────────────────────────────────
 let currentAgentId = '';
@@ -362,19 +415,6 @@ async function api(method, path, body = null) {
   return resp.json();
 }
 
-// ─── Initialize ─────────────────────────────────────────────────────
-async function init() {
-  await loadAgents();
-  await loadFriends();
-  await loadGroups();
-  // Auto-select first agent
-  const sel = document.getElementById('agentSelect');
-  if (sel.options.length > 1) {
-    currentAgentId = sel.options[1].value;
-    sel.value = currentAgentId;
-  }
-}
-
 // ─── Agents ─────────────────────────────────────────────────────────
 async function loadAgents() {
   const data = await api('GET', '/api/agents');
@@ -391,6 +431,7 @@ async function loadAgents() {
 
 async function switchAgent(agentId) {
   if (!agentId) return;
+  saveChatToLocalStorage();
   currentAgentId = agentId;
   currentTarget = null;
   showEmptyState();
@@ -661,6 +702,7 @@ async function sendMessage() {
       mentions: extractMentions(content),
     });
     await loadMessages();
+    saveChatToLocalStorage();
   } catch (e) {
     alert('发送失败: ' + e.message);
   }
@@ -979,14 +1021,401 @@ async function handlePluginAvatarUpload(input) {
   input.value = '';
 }
 
+// ─── Toast Notification ─────────────────────────────────────────────
+let toastTimer = null;
+function showToast(message, type = '', duration = 5000) {
+  const el = document.getElementById('toast');
+  el.textContent = message;
+  el.className = 'toast show' + (type ? ' ' + type : '');
+  if (toastTimer) clearTimeout(toastTimer);
+  toastTimer = setTimeout(() => { el.className = 'toast'; }, duration);
+}
+
+// ─── localStorage Chat Cache ───────────────────────────────────────
+const CACHE_KEY = 'aicq_plugin_chat_cache';
+
+function saveChatToLocalStorage() {
+  try {
+    const cache = {
+      currentAgentId,
+      currentTarget,
+      chatMessages,
+      oldestTimestamp,
+      savedAt: Date.now()
+    };
+    localStorage.setItem(CACHE_KEY, JSON.stringify(cache));
+  } catch (e) {
+    // localStorage might be full or unavailable
+    console.warn('Failed to save chat cache:', e);
+  }
+}
+
+function loadChatFromLocalStorage() {
+  try {
+    const raw = localStorage.getItem(CACHE_KEY);
+    if (!raw) return null;
+    return JSON.parse(raw);
+  } catch (e) {
+    return null;
+  }
+}
+
+function clearChatCache() {
+  try { localStorage.removeItem(CACHE_KEY); } catch(e) {}
+}
+
+// ─── Logout ─────────────────────────────────────────────────────────
+function confirmLogout() {
+  if (confirm('确定要登出吗？登出前会自动保存聊天缓存到本地。')) {
+    saveChatToLocalStorage();
+    // Clear state
+    currentAgentId = '';
+    currentTarget = null;
+    chatMessages = [];
+    oldestTimestamp = null;
+    document.getElementById('agentSelect').value = '';
+    showEmptyState();
+    document.getElementById('friendsList').innerHTML = '';
+    document.getElementById('groupsList').innerHTML = '';
+    hideModal('settings');
+    showToast('已登出，聊天缓存已保存到本地');
+  }
+}
+
+// ─── Backup Export ──────────────────────────────────────────────────
+async function exportBackup() {
+  try {
+    showToast('正在生成备份...');
+    // Get all agents
+    const agentsData = await api('GET', '/api/agents');
+    const agents = agentsData.agents || [];
+
+    // Collect data for each agent
+    const backupAgents = [];
+    for (const agent of agents) {
+      // Get identity/keys for this agent (includes private keys)
+      let keys = {};
+      try {
+        keys = await api('GET', `/api/identity/keys?agent_id=${agent.agent_id}`);
+      } catch(e) {}
+
+      // Get friends
+      let friends = [];
+      try {
+        const fd = await api('GET', `/api/friends?agent_id=${agent.agent_id}`);
+        friends = fd.friends || [];
+      } catch(e) {}
+
+      // Get groups
+      let groups = [];
+      try {
+        const gd = await api('GET', `/api/groups?agent_id=${agent.agent_id}`);
+        groups = gd.groups || [];
+      } catch(e) {}
+
+      // Get chat messages for each friend and group
+      const chatData = {};
+      for (const f of friends) {
+        try {
+          const md = await api('GET', `/api/chat/${f.id}?agent_id=${agent.agent_id}&limit=9999`);
+          chatData['friend_' + f.id] = md.messages || [];
+        } catch(e) {}
+      }
+      for (const g of groups) {
+        try {
+          const md = await api('GET', `/api/chat/${g.id}?agent_id=${agent.agent_id}&limit=9999`);
+          chatData['group_' + g.id] = md.messages || [];
+        } catch(e) {}
+      }
+
+      backupAgents.push({
+        agent_id: agent.agent_id,
+        nickname: agent.nickname || agent.agent_id,
+        avatar: agent.avatar || null,
+        signing_public_key: keys.signing_public_key || null,
+        exchange_public_key: keys.exchange_public_key || null,
+        signing_secret_key: keys.signing_secret_key || null,
+        exchange_secret_key: keys.exchange_secret_key || null,
+        fingerprint: keys.fingerprint || null,
+        friends,
+        groups,
+        chatData
+      });
+    }
+
+    const backup = {
+      version: 'aicq-plugin-backup-v1',
+      exportedAt: new Date().toISOString(),
+      agents: backupAgents
+    };
+
+    // Download as JSON file
+    const blob = new Blob([JSON.stringify(backup, null, 2)], { type: 'application/json' });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = `aicq-backup-${new Date().toISOString().slice(0,10)}.json`;
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+    URL.revokeObjectURL(url);
+
+    showToast('备份已导出！请妥善保管此文件，内含私钥。', 'warning', 6000);
+  } catch (e) {
+    alert('导出失败: ' + e.message);
+  }
+}
+
+// ─── Backup Import ──────────────────────────────────────────────────
+function triggerImportBackup() {
+  document.getElementById('backupFileInput').click();
+}
+
+// Store pending backup for matching
+let pendingBackup = null;
+let pendingMatchMap = {};
+
+async function handleImportBackup(input) {
+  if (!input.files || !input.files[0]) return;
+  const file = input.files[0];
+  input.value = '';
+
+  try {
+    const text = await file.text();
+    const backup = JSON.parse(text);
+
+    if (!backup.version || !backup.agents || !Array.isArray(backup.agents)) {
+      alert('无效的备份文件格式');
+      return;
+    }
+
+    await processImport(backup);
+  } catch (e) {
+    alert('读取备份文件失败: ' + e.message);
+  }
+}
+
+async function processImport(backup) {
+  // Get current agents
+  const agentsData = await api('GET', '/api/agents');
+  const currentAgents = agentsData.agents || [];
+
+  // Get current agents' keys for comparison
+  const currentAgentKeys = {};
+  for (const agent of currentAgents) {
+    try {
+      const keys = await api('GET', `/api/identity/keys?agent_id=${agent.agent_id}`);
+      currentAgentKeys[agent.agent_id] = {
+        agent_id: agent.agent_id,
+        nickname: agent.nickname || agent.agent_id,
+        signing_public_key: keys.signing_public_key,
+        fingerprint: keys.fingerprint
+      };
+    } catch(e) {
+      currentAgentKeys[agent.agent_id] = {
+        agent_id: agent.agent_id,
+        nickname: agent.nickname || agent.agent_id
+      };
+    }
+  }
+
+  // Check which backup agents don't match current agents
+  const matchedAgents = [];  // { backupAgent, currentAgent }
+  const unmatchedBackupAgents = [];  // backup agents with no key match
+
+  for (const ba of backup.agents) {
+    // Try to find matching current agent by signing_public_key or fingerprint
+    let matched = null;
+    for (const [cid, ck] of Object.entries(currentAgentKeys)) {
+      if (ba.signing_public_key && ck.signing_public_key && ba.signing_public_key === ck.signing_public_key) {
+        matched = ck;
+        break;
+      }
+      if (ba.fingerprint && ck.fingerprint && ba.fingerprint === ck.fingerprint) {
+        matched = ck;
+        break;
+      }
+      // Also match by agent_id
+      if (ba.agent_id === cid) {
+        matched = ck;
+        break;
+      }
+    }
+
+    if (matched) {
+      matchedAgents.push({ backupAgent: ba, currentAgent: matched });
+    } else {
+      unmatchedBackupAgents.push(ba);
+    }
+  }
+
+  // If there are unmatched agents, show matching modal
+  if (unmatchedBackupAgents.length > 0 && Object.keys(currentAgentKeys).length > 0) {
+    pendingBackup = backup;
+    pendingMatchMap = {};
+
+    // Auto-add matched ones
+    for (const m of matchedAgents) {
+      pendingMatchMap[m.backupAgent.agent_id] = m.currentAgent.agent_id;
+    }
+
+    showBackupMatchModal(unmatchedBackupAgents, Object.values(currentAgentKeys));
+  } else {
+    // All matched or no current agents, proceed directly
+    await doImportBackup(backup, matchedAgents.reduce((map, m) => { map[m.backupAgent.agent_id] = m.currentAgent.agent_id; return map; }, {}));
+  }
+}
+
+function showBackupMatchModal(unmatchedAgents, currentAgentList) {
+  const container = document.getElementById('backupMatchList');
+  container.innerHTML = '';
+
+  for (const ba of unmatchedAgents) {
+    const row = document.createElement('div');
+    row.className = 'key-match-row';
+    row.innerHTML = `
+      <label title="${ba.agent_id}">${ba.nickname || ba.agent_id}</label>
+      <select id="match-${ba.agent_id}" data-backup-agent="${ba.agent_id}">
+        <option value="">-- 不导入 --</option>
+        ${currentAgentList.map(ca => `<option value="${ca.agent_id}">${ca.nickname || ca.agent_id}</option>`).join('')}
+      </select>
+    `;
+    container.appendChild(row);
+  }
+
+  showModal('backupMatch');
+}
+
+async function applyBackupMatch() {
+  // Collect user selections
+  const selects = document.querySelectorAll('#backupMatchList select');
+  for (const sel of selects) {
+    const backupAgentId = sel.dataset.backupAgent;
+    const chosenAgentId = sel.value;
+    if (chosenAgentId) {
+      pendingMatchMap[backupAgentId] = chosenAgentId;
+    }
+  }
+
+  hideModal('backupMatch');
+  await doImportBackup(pendingBackup, pendingMatchMap);
+  pendingBackup = null;
+  pendingMatchMap = {};
+}
+
+async function skipBackupMatch() {
+  hideModal('backupMatch');
+  // Import only the already-matched agents (without the unmatched ones)
+  await doImportBackup(pendingBackup, pendingMatchMap);
+  pendingBackup = null;
+  pendingMatchMap = {};
+}
+
+async function doImportBackup(backup, matchMap) {
+  try {
+    showToast('正在导入备份...', '', 8000);
+    let importedCount = 0;
+    let importedMessages = 0;
+
+    for (const ba of backup.agents) {
+      const targetAgentId = matchMap[ba.agent_id];
+      if (!targetAgentId) continue;
+
+      // Import friends (via API if possible, or just save to cache)
+      // The plugin uses local SQLite, so we store to localStorage cache
+      for (const friend of (ba.friends || [])) {
+        try {
+          // Try to add friend via API
+          await api('POST', '/api/friends/add-by-fingerprint', {
+            agent_id: targetAgentId,
+            fingerprint: friend.fingerprint,
+            ai_name: friend.ai_name,
+            ai_avatar: friend.ai_avatar
+          });
+        } catch(e) {
+          // Friend might already exist, that's fine
+        }
+      }
+
+      // Import chat messages to localStorage cache
+      if (ba.chatData) {
+        const chatCacheKey = `aicq_chat_${targetAgentId}`;
+        try {
+          const existing = JSON.parse(localStorage.getItem(chatCacheKey) || '{}');
+          for (const [targetKey, messages] of Object.entries(ba.chatData)) {
+            existing[targetKey] = messages;
+          }
+          localStorage.setItem(chatCacheKey, JSON.stringify(existing));
+          importedMessages += Object.keys(ba.chatData).length;
+        } catch(e) {}
+      }
+
+      importedCount++;
+    }
+
+    // Refresh UI
+    await loadAgents();
+    await loadFriends();
+    await loadGroups();
+
+    hideModal('settings');
+    showToast(`导入完成！已导入 ${importedCount} 个 Agent 的数据，${importedMessages} 个会话记录。`, '', 6000);
+  } catch (e) {
+    alert('导入失败: ' + e.message);
+  }
+}
+
 // ─── Init ───────────────────────────────────────────────────────────
+async function init() {
+  await loadAgents();
+  await loadFriends();
+  await loadGroups();
+  // Auto-select first agent
+  const sel = document.getElementById('agentSelect');
+  if (sel.options.length > 1) {
+    currentAgentId = sel.options[1].value;
+    sel.value = currentAgentId;
+  }
+
+  // Load chat cache from localStorage
+  const cached = loadChatFromLocalStorage();
+  if (cached && cached.currentAgentId) {
+    // Restore cache if agent still exists
+    const sel2 = document.getElementById('agentSelect');
+    let agentExists = false;
+    for (let i = 0; i < sel2.options.length; i++) {
+      if (sel2.options[i].value === cached.currentAgentId) {
+        agentExists = true;
+        break;
+      }
+    }
+    if (agentExists) {
+      currentAgentId = cached.currentAgentId;
+      sel2.value = currentAgentId;
+      if (cached.currentTarget) {
+        currentTarget = cached.currentTarget;
+        await loadFriends();
+        await loadGroups();
+        // Will load messages from server, which is more reliable
+        if (currentTarget) {
+          selectTarget(currentTarget.id, currentTarget.name, currentTarget.type, currentTarget.isOnline, currentTarget.silent);
+        }
+      }
+    }
+  }
+
+  // Show login reminder toast
+  showToast('提醒：聊天记录仅保存在本地浏览器中，清除缓存数据或删除浏览器内容将丢失', 'warning', 8000);
+}
+
 init();
 
-// Periodic refresh
+// Periodic refresh and auto-save
 setInterval(async () => {
   if (currentAgentId) {
     await loadFriends();
     await loadGroups();
+    saveChatToLocalStorage();
   }
 }, 30000);
 </script>