workspace.ts
3.94 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
import { defineStore } from "pinia";
import { ref } from "vue";
import { ElMessage } from "element-plus";
import {
apiCreateDraftFromQuestion,
apiCreateDraftFromDeepSearch,
apiSaveDraftToKnowledge,
} from "@/api/drafts";
import { getFiles, createFolder } from "@/api/files";
import { useAppStore } from "@/stores/app";
export const useWorkspaceStore = defineStore("workspace", () => {
const isDocumentGenerating = ref(false);
const generatingProgress = ref(0);
const currentGeneratingTitle = ref("");
async function generateDocument(params: {
title: string;
content: string;
questionId?: string;
isDeepSearch?: boolean;
onSuccess?: (fileInfo: any, draftContent: string) => void;
}) {
if (isDocumentGenerating.value) {
ElMessage.warning("已有文档正在生成中，请稍后再试");
return;
}
try {
isDocumentGenerating.value = true;
generatingProgress.value = 10;
currentGeneratingTitle.value = params.title;
let title = params.title.replace(/\n/g, " ").substring(0, 20).trim();
if (!title) title = "未命名文档";
// 1. 检查并创建“草稿”目录
generatingProgress.value = 30;
let draftFolderId: number | undefined;
const rootFilesRes = await getFiles();
if (rootFilesRes.status === 200 && rootFilesRes.data) {
const items = Array.isArray(rootFilesRes.data)
? rootFilesRes.data
: (rootFilesRes.data.items || []);
const draftFolder = items.find(
(item: any) =>
(item.type === "folder" || item.isFolder) && item.name === "草稿",
);
if (draftFolder) {
draftFolderId = draftFolder.id;
}
}
if (!draftFolderId) {
const createFolderRes = await createFolder({ name: "草稿" });
if (createFolderRes.status === 200 && createFolderRes.data) {
draftFolderId = createFolderRes.data.id;
}
}
// 2. 调用保存API创建草稿
generatingProgress.value = 60;
let response;
if (params.isDeepSearch) {
response = await apiCreateDraftFromDeepSearch({
title,
content: params.content,
});
} else if (params.questionId) {
response = await apiCreateDraftFromQuestion(params.questionId, {
title,
});
} else {
throw new Error("缺少问题ID或内容");
}
if (response.status === 200 && response.data) {
const draftInfo = response.data;
const draftContent = draftInfo.content || "";
// 3. 将草稿保存到“草稿”目录
generatingProgress.value = 80;
const saveRes = await apiSaveDraftToKnowledge(draftInfo.id, {
parentId: draftFolderId,
fileName: title + ".md",
});
if (saveRes.status === 200 && saveRes.data) {
const fileInfo = saveRes.data;
generatingProgress.value = 100;
// 刷新文件列表
const appStore = useAppStore();
appStore.triggerRefreshFileHierarchy(draftFolderId);
window.dispatchEvent(
new CustomEvent("refresh-file-hierarchy", {
detail: { folderId: draftFolderId },
}),
);
if (params.onSuccess) {
params.onSuccess(fileInfo, draftContent);
}
ElMessage.success("已保存到草稿目录");
} else {
throw new Error("保存到目录失败");
}
} else {
throw new Error((response as any)?.data?.message || "创建草稿失败");
}
} catch (error: any) {
console.error("生成文档失败:", error);
ElMessage.error(error.message || "操作失败，请重试");
} finally {
isDocumentGenerating.value = false;
generatingProgress.value = 0;
currentGeneratingTitle.value = "";
}
}
return {
isDocumentGenerating,
generatingProgress,
currentGeneratingTitle,
generateDocument,
};
});