fix: Prompt商店在线导入可以导入两种recommend.json里提到的模板 (#516) (#521)

* fix(PromptStore): import prompt template using different dict keys * chore(PromptStore): clear url field after downloading
Chanzhaoyu · Mar 12, 2023 · 7bebee2 · 7bebee2
1 parent bc5e835
commit 7bebee2
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 6 deletions.
diff --git a/src/assets/recommend.json b/src/assets/recommend.json
@@ -1,8 +1,14 @@
 [
+  {
+    "key": "chatgpt-prompt-collection",
+    "desc": "Nothing1024收集整理的prompts",
+    "downloadUrl": "https://raw.githubusercontent.com/Nothing1024/chatgpt-prompt-collection/main/awesome-chatgpt-prompts-zh.json",
+    "url": "https://github.com/Nothing1024/chatgpt-prompt-collection"
+  },
   {
     "key": "awesome-chatgpt-prompts-zh",
     "desc": "ChatGPT 中文调教指南",
-    "downloadUrl": "https://raw.githubusercontent.com/Nothing1024/chatgpt-prompt-collection/main/awesome-chatgpt-prompts-zh.json",
+    "downloadUrl": "https://raw.githubusercontent.com/PlexPt/awesome-chatgpt-prompts-zh/main/prompts-zh.json",
     "url": "https://github.com/PlexPt/awesome-chatgpt-prompts-zh"
   }
 ]
diff --git a/src/components/common/PromptStore/index.vue b/src/components/common/PromptStore/index.vue
@@ -144,24 +144,41 @@ const clearPromptTemplate = () => {
 const importPromptTemplate = () => {
   try {
     const jsonData = JSON.parse(tempPromptValue.value)
+    let key = ''
+    let value = ''
+    // 可以扩展加入更多模板字典的key
+    if ('key' in jsonData[0]) {
+      key = 'key'
+      value = 'value'
+    }
+    else if ('act' in jsonData[0]) {
+      key = 'act'
+      value = 'prompt'
+    }
+    else {
+      // 不支持的字典的key防止导入 以免破坏prompt商店打开
+      message.warning('prompt key not supported.')
+      throw new Error('prompt key not supported.')
+    }
+
     for (const i of jsonData) {
       if (!('key' in i) || !('value' in i))
         throw new Error('键值不匹配')
       let safe = true
       for (const j of promptList.value) {
-        if (j.key === i.key) {
-          message.warning(`因标题重复跳过：${i.key}`)
+        if (j.key === i[key]) {
+          message.warning(`因标题重复跳过：${i[key]}`)
           safe = false
           break
         }
-        if (j.value === i.value) {
-          message.warning(`因内容重复跳过：${i.key}`)
+        if (j.value === i[value]) {
+          message.warning(`因内容重复跳过：${i[key]}`)
           safe = false
           break
         }
       }
       if (safe)
-        promptList.value.unshift({ key: i.key, value: i.value } as never)
+        promptList.value.unshift({ key: i[key], value: i[value] } as never)
     }
     message.success('导入成功')
     changeShowModal('')
@@ -194,6 +211,7 @@ const downloadPromptTemplate = async () => {
       }).then(() => {
         importPromptTemplate()
       })
+    downloadURL.value = ''
   }
   catch {
     message.error('网络导入出现问题，请检查网络状态与 JSON 文件有效性')