139 lines
4.7 KiB
JavaScript
139 lines
4.7 KiB
JavaScript
// rollback.js -- 统一回撤机制
|
|
// 回撤消息：后端删除后续 turn 并恢复会话所属工作区的 git 状态与文件，
|
|
// 前端刷新会话/消息/文件树，并把回撤点消息放回输入区。
|
|
|
|
import { $, showToast } from './utils.js';
|
|
import { showConfirm } from './dialog.js';
|
|
import { api } from './api.js';
|
|
import { state } from './state.js';
|
|
import { clearPendingOverlay } from './pending-overlay.js';
|
|
|
|
// 依赖注入
|
|
let renderMessagesFn = null;
|
|
let renderSessionsFn = null;
|
|
let renderAttachmentChipsFn = null;
|
|
let clearTodoOverlayFn = null;
|
|
let setDocContentFn = null;
|
|
let getDocContentFn = null;
|
|
let renderDocTreeFn = null;
|
|
let renderDocTabsFn = null;
|
|
let loadDocContentFn = null;
|
|
let setComposerCenteredFn = null;
|
|
|
|
export function setRollbackDeps(deps) {
|
|
renderMessagesFn = deps.renderMessages;
|
|
renderSessionsFn = deps.renderSessions;
|
|
renderAttachmentChipsFn = deps.renderAttachmentChips;
|
|
clearTodoOverlayFn = deps.clearTodoOverlay;
|
|
setDocContentFn = deps.setDocContent;
|
|
getDocContentFn = deps.getDocContent;
|
|
renderDocTreeFn = deps.renderDocTree;
|
|
renderDocTabsFn = deps.renderDocTabs;
|
|
loadDocContentFn = deps.loadDocContent;
|
|
setComposerCenteredFn = deps.setComposerCentered;
|
|
}
|
|
|
|
// ============================================================
|
|
// Chat 模式回撤
|
|
// ============================================================
|
|
|
|
/**
|
|
* 将聊天历史回撤到指定消息处
|
|
* @param {number} rollbackIndex - 要回退到的消息在渲染列表中的索引（包含该消息）
|
|
* @param {object} [options] - { skipConfirm: boolean } 跳过确认弹窗（重试等场景）
|
|
*/
|
|
export async function rollbackChat(rollbackIndex, options = {}) {
|
|
// 保护：Agent 运行中不允许回撤
|
|
if (state.sending) {
|
|
showToast("Agent 运行中，无法回撤。请先暂停任务。");
|
|
return false;
|
|
}
|
|
if (!state.sessionId) {
|
|
showToast("无当前会话");
|
|
return false;
|
|
}
|
|
|
|
if (!options.skipConfirm) {
|
|
const confirmed = await showConfirm(
|
|
"确认回退到此消息？\n\n将清除此消息之后的所有对话历史、TodoList、人机交互内容、工具调用记录；\n" +
|
|
"若该会话关联工作区，文件系统与 git 状态也会恢复到该消息时的状态（之后的文件改动将被丢弃）。"
|
|
);
|
|
if (!confirmed) return false;
|
|
}
|
|
|
|
try {
|
|
// 1. 调用后端回撤 API（含 git 状态与文件恢复）
|
|
const fd = new FormData();
|
|
fd.append("session_id", state.sessionId);
|
|
fd.append("rollback_index", rollbackIndex);
|
|
const data = await api("/api/chat/rollback-to-message", { method: "POST", body: fd });
|
|
|
|
// 2. 更新会话状态
|
|
state.sessionId = data.sessionId;
|
|
state.sessions.length = 0;
|
|
state.sessions.push(...(data.sessions || []));
|
|
if (data.session_workspaces) state.sessionWorkspaces = data.session_workspaces;
|
|
|
|
// 3. 清除 TodoList 和 人机交互 浮窗
|
|
if (clearTodoOverlayFn) clearTodoOverlayFn();
|
|
clearPendingOverlay();
|
|
|
|
// 4. 文件已由后端 git 恢复：刷新文件树与打开的文件预览
|
|
await refreshDocTreeAfterRollback();
|
|
|
|
// 5. 刷新视图
|
|
if (renderSessionsFn) renderSessionsFn();
|
|
if (renderMessagesFn) renderMessagesFn(data.messages || []);
|
|
|
|
// 6. 恢复回撤点用户消息到输入区
|
|
const rollbackMsg = data.rollbackMessage;
|
|
if (rollbackMsg && rollbackMsg.role === "user") {
|
|
restoreUserInput(rollbackMsg);
|
|
}
|
|
|
|
showToast("已回退到该消息（文件与 git 状态已恢复）");
|
|
return true;
|
|
} catch (e) {
|
|
showToast("回退失败: " + (e.message || "未知错误"));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 回撤后刷新文件树与已打开文件（git 状态与文件内容由后端恢复）
|
|
*/
|
|
async function refreshDocTreeAfterRollback() {
|
|
const rootPath = state._docRootPath;
|
|
try {
|
|
// 通知 doc-tree 重新拉取 git 状态并渲染
|
|
window.dispatchEvent(new CustomEvent("git-status-changed"));
|
|
if (renderDocTreeFn) renderDocTreeFn();
|
|
if (renderDocTabsFn) renderDocTabsFn();
|
|
// 重新加载当前打开的文件内容
|
|
if (rootPath && loadDocContentFn && state.activeDocFile) {
|
|
await loadDocContentFn(state.activeDocFile);
|
|
}
|
|
} catch { /* 文件树刷新失败不影响回撤主流程 */ }
|
|
}
|
|
|
|
/**
|
|
* 将回撤点的用户消息内容恢复到输入区
|
|
*/
|
|
function restoreUserInput(msg) {
|
|
const textInput = $("textInput");
|
|
if (!textInput) return;
|
|
|
|
// 恢复文本
|
|
if (typeof msg.content === "string") {
|
|
textInput.value = msg.content;
|
|
} else if (Array.isArray(msg.content)) {
|
|
const textParts = msg.content
|
|
.filter((p) => p && p.type === "text")
|
|
.map((p) => p.text || "")
|
|
.join("\n");
|
|
textInput.value = textParts;
|
|
}
|
|
|
|
textInput.focus();
|
|
}
|