From 4d93e901e0be715d005875868516294b69085347 Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Thu, 10 Apr 2025 21:34:56 +0800
Subject: [PATCH] =?UTF-8?q?=E7=94=9F=E6=88=90=E6=80=9D=E7=BB=B4=E5=AF=BC?=
 =?UTF-8?q?=E5=9B=BE=E6=97=B6=E5=80=99=E8=87=AA=E5=8A=A8=E7=BC=93=E5=AD=98?=
 =?UTF-8?q?=E4=B8=8A=E4=B8=80=E6=AC=A1=E7=9A=84=E7=BB=93=E6=9E=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 CHANGELOG.md              |   1 +
 web/src/views/MarkMap.vue | 241 +++++++++++++++++++-------------------
 2 files changed, 122 insertions(+), 120 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 053c0855..54d6cda0 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,7 @@
 - 功能优化：支持 GPT-4o 图片编辑功能。
 - 功能新增：对话页面支持 AI 输出语音播报（TTS）。
 - 功能优化：替换瀑布流组件，优化用户体验。
+- 功能优化：生成思维导图时候自动缓存上一次的结果
 
 ## v4.2.1
 
diff --git a/web/src/views/MarkMap.vue b/web/src/views/MarkMap.vue
index 28aeff20..a9320347 100644
--- a/web/src/views/MarkMap.vue
+++ b/web/src/views/MarkMap.vue
@@ -19,11 +19,7 @@
 
               <div class="param-line">请选择生成思维导图的AI模型</div>
               <div class="param-line">
-                <el-select
-                  v-model="modelID"
-                  placeholder="请选择模型"
-                  style="width: 100%"
-                >
+                <el-select v-model="modelID" placeholder="请选择模型" style="width: 100%">
                   <el-option
                     v-for="item in models"
                     :key="item.id"
@@ -43,9 +39,7 @@
 
               <div class="text-info">
                 <el-text type="primary"
-                  >当前可用算力：<el-text type="warning">{{
-                    loginUser.power
-                  }}</el-text></el-text
+                  >当前可用算力：<el-text type="warning">{{ loginUser.power }}</el-text></el-text
                 >
               </div>
 
@@ -115,179 +109,186 @@
 </template>
 
 <script setup>
-import { nextTick, ref } from "vue";
-import { Markmap } from "markmap-view";
-import { Transformer } from "markmap-lib";
-import { checkSession, getSystemInfo } from "@/store/cache";
-import { httpGet, httpPost } from "@/utils/http";
-import { ElMessage } from "element-plus";
-import { Download } from "@element-plus/icons-vue";
-import { Toolbar } from "markmap-toolbar";
-import { useSharedStore } from "@/store/sharedata";
+import { nextTick, onMounted, ref } from 'vue'
+import { Markmap } from 'markmap-view'
+import { Transformer } from 'markmap-lib'
+import { checkSession, getSystemInfo } from '@/store/cache'
+import { httpGet, httpPost } from '@/utils/http'
+import { ElMessage } from 'element-plus'
+import { Download } from '@element-plus/icons-vue'
+import { Toolbar } from 'markmap-toolbar'
+import { useSharedStore } from '@/store/sharedata'
 
-const leftBoxHeight = ref(window.innerHeight - 105);
+const leftBoxHeight = ref(window.innerHeight - 105)
 //const rightBoxHeight = ref(window.innerHeight - 115);
-const rightBoxHeight = ref(window.innerHeight);
+const rightBoxHeight = ref(window.innerHeight)
 
-const prompt = ref("");
-const text = ref("");
-const content = ref(text.value);
-const html = ref("");
+const prompt = ref('')
+const text = ref('')
+const content = ref(text.value)
+const html = ref('')
 
-const isLogin = ref(false);
-const loginUser = ref({ power: 0 });
-const transformer = new Transformer();
-const store = useSharedStore();
-const loading = ref(false);
+const isLogin = ref(false)
+const loginUser = ref({ power: 0 })
+const transformer = new Transformer()
+const store = useSharedStore()
+const loading = ref(false)
 
-const svgRef = ref(null);
-const markMap = ref(null);
-const models = ref([]);
-const modelID = ref(0);
+const svgRef = ref(null)
+const markMap = ref(null)
+const models = ref([])
+const modelID = ref(0)
+const cacheKey = ref('MarkMapCache')
 
-getSystemInfo()
-  .then((res) => {
-    text.value = res.data["mark_map_text"];
-    content.value = text.value;
-    initData();
-    nextTick(() => {
-      try {
-        markMap.value = Markmap.create(svgRef.value);
-        const { el } = Toolbar.create(markMap.value);
-        document.getElementById("toolbar").append(el);
-        update();
-      } catch (e) {
-        console.error(e);
-      }
-    });
+onMounted(async () => {
+  const cache = localStorage.getItem(cacheKey.value)
+  if (cache) {
+    text.value = cache
+  } else {
+    const res = await getSystemInfo().catch((e) => {
+      ElMessage.error('获取系统配置失败：' + e.message)
+    })
+    text.value = res.data['mark_map_text']
+    content.value = text.value
+  }
+
+  initData()
+  nextTick(() => {
+    try {
+      markMap.value = Markmap.create(svgRef.value)
+      const { el } = Toolbar.create(markMap.value)
+      document.getElementById('toolbar').append(el)
+      update()
+    } catch (e) {
+      console.error(e)
+    }
   })
-  .catch((e) => {
-    ElMessage.error("获取系统配置失败：" + e.message);
-  });
+})
 
 const initData = () => {
-  httpGet("/api/model/list")
+  httpGet('/api/model/list')
     .then((res) => {
       for (let v of res.data) {
-        models.value.push(v);
+        models.value.push(v)
       }
-      modelID.value = models.value[0].id;
+      modelID.value = models.value[0].id
     })
     .catch((e) => {
-      ElMessage.error("获取模型失败：" + e.message);
-    });
+      ElMessage.error('获取模型失败：' + e.message)
+    })
 
   checkSession()
     .then((user) => {
-      loginUser.value = user;
-      isLogin.value = true;
+      loginUser.value = user
+      isLogin.value = true
     })
-    .catch(() => {});
-};
+    .catch(() => {})
+}
 
 const update = () => {
   try {
-    const { root } = transformer.transform(processContent(text.value));
-    markMap.value.setData(root);
-    markMap.value.fit();
+    const { root } = transformer.transform(processContent(text.value))
+    markMap.value.setData(root)
+    markMap.value.fit()
   } catch (e) {
-    console.error(e);
+    console.error(e)
   }
-};
+}
 
 const processContent = (text) => {
   if (!text) {
-    return text;
+    return text
   }
 
-  const arr = [];
-  const lines = text.split("\n");
+  const arr = []
+  const lines = text.split('\n')
   for (let line of lines) {
-    if (line.indexOf("```") !== -1) {
-      continue;
+    if (line.indexOf('```') !== -1) {
+      continue
     }
-    line = line.replace(/([*_~`>])|(\d+\.)\s/g, "");
-    arr.push(line);
+    line = line.replace(/([*_~`>])|(\d+\.)\s/g, '')
+    arr.push(line)
   }
-  return arr.join("\n");
-};
+  return arr.join('\n')
+}
 
 window.onresize = () => {
-  leftBoxHeight.value = window.innerHeight - 145;
-  rightBoxHeight.value = window.innerHeight - 85;
-};
+  leftBoxHeight.value = window.innerHeight - 145
+  rightBoxHeight.value = window.innerHeight - 85
+}
 
 const generate = () => {
-  text.value = content.value;
-  update();
-};
+  text.value = content.value
+  update()
+}
 
 // 使用 AI 智能生成
 const generateAI = () => {
-  html.value = "";
-  text.value = "";
-  if (prompt.value === "") {
-    return ElMessage.error("请输入你的需求");
+  html.value = ''
+  text.value = ''
+  if (prompt.value === '') {
+    return ElMessage.error('请输入你的需求')
   }
   if (!isLogin.value) {
-    store.setShowLoginDialog(true);
-    return;
+    store.setShowLoginDialog(true)
+    return
   }
-  loading.value = true;
-  httpPost("/api/markMap/gen", {
+  loading.value = true
+  httpPost('/api/markMap/gen', {
     prompt: prompt.value,
-    model_id: modelID.value
+    model_id: modelID.value,
   })
     .then((res) => {
-      text.value = res.data;
-      content.value = processContent(text.value);
-      const model = getModelById(modelID.value);
-      loginUser.value.power -= model.power;
-      nextTick(() => update());
-      loading.value = false;
+      text.value = res.data
+      content.value = processContent(text.value)
+      const model = getModelById(modelID.value)
+      loginUser.value.power -= model.power
+      nextTick(() => update())
+      loading.value = false
+      // 缓存结果
+      localStorage.setItem(cacheKey.value, text.value)
     })
     .catch((e) => {
-      ElMessage.error("生成思维导图失败：" + e.message);
-      loading.value = false;
-    });
-};
+      ElMessage.error('生成思维导图失败：' + e.message)
+      loading.value = false
+    })
+}
 
 const getModelById = (modelId) => {
   for (let m of models.value) {
     if (m.id === modelId) {
-      return m;
+      return m
     }
   }
-};
+}
 
 // download SVG to png file
 const downloadImage = () => {
-  const svgElement = document.getElementById("markmap");
+  const svgElement = document.getElementById('markmap')
   // 将 SVG 渲染到图片对象
-  const serializer = new XMLSerializer();
+  const serializer = new XMLSerializer()
   const source =
-    '<?xml version="1.0" standalone="no"?>\r\n' +
-    serializer.serializeToString(svgRef.value);
-  const image = new Image();
-  image.src = "data:image/svg+xml;charset=utf-8," + encodeURIComponent(source);
+    '<?xml version="1.0" standalone="no"?>\r\n' + serializer.serializeToString(svgRef.value)
+  const image = new Image()
+  image.src = 'data:image/svg+xml;charset=utf-8,' + encodeURIComponent(source)
 
   // 将图片对象渲染
-  const canvas = document.createElement("canvas");
-  canvas.width = svgElement.offsetWidth;
-  canvas.height = svgElement.offsetHeight;
-  let context = canvas.getContext("2d");
-  context.clearRect(0, 0, canvas.width, canvas.height);
-  context.fillStyle = "white";
-  context.fillRect(0, 0, canvas.width, canvas.height);
+  const canvas = document.createElement('canvas')
+  canvas.width = svgElement.offsetWidth
+  canvas.height = svgElement.offsetHeight
+  let context = canvas.getContext('2d')
+  context.clearRect(0, 0, canvas.width, canvas.height)
+  context.fillStyle = 'white'
+  context.fillRect(0, 0, canvas.width, canvas.height)
 
   image.onload = function () {
-    context.drawImage(image, 0, 0);
-    const a = document.createElement("a");
-    a.download = "geek-ai-xmind.png";
-    a.href = canvas.toDataURL(`image/png`);
-    a.click();
-  };
-};
+    context.drawImage(image, 0, 0)
+    const a = document.createElement('a')
+    a.download = 'geek-ai-xmind.png'
+    a.href = canvas.toDataURL(`image/png`)
+    a.click()
+  }
+}
 </script>
 
 <style lang="stylus">