refactor: prompts sync

src-tauri/Cargo.toml

@@ -25,7 +25,7 @@ thiserror = "1.0.38"
 walkdir = "2.3.2"
 regex = "1.7.0"
 tokio = { version = "1.23.0", features = ["macros"] }
-# reqwest = "0.11.13"
+reqwest = "0.11.13"
 
 [dependencies.tauri-plugin-log]
 git = "https://github.com/tauri-apps/tauri-plugin-log"

src-tauri/src/app/cmd.rs

@@ -1,7 +1,8 @@
 use crate::{
-    conf::ChatConfJson,
+    conf::{ChatConfJson, GITHUB_PROMPTS_CSV_URL},
     utils::{self, exists},
 };
+use log::info;
 use std::{collections::HashMap, fs, path::PathBuf};
 use tauri::{api, command, AppHandle, Manager};
 
@@ -134,8 +135,13 @@ pub fn cmd_list() -> Vec<ModelRecord> {
 }
 
 #[command]
-pub fn sync_prompts(app: AppHandle, data: String, time: u64) {
-    let data = parse_prompt(data)
+pub async fn sync_prompts(app: AppHandle, time: u64) -> Option<Vec<ModelRecord>> {
+    let res = utils::get_data(GITHUB_PROMPTS_CSV_URL, Some(&app))
+        .await
+        .unwrap();
+
+    if let Some(v) = res {
+        let data = parse_prompt(v)
             .iter()
             .map(move |i| ModelRecord {
                 cmd: if i.cmd.is_some() {
@@ -150,6 +156,8 @@ pub fn sync_prompts(app: AppHandle, data: String, time: u64) {
             })
             .collect::<Vec<ModelRecord>>();
 
+        let data2 = data.clone();
+
         let model = chat_root().join("chat.model.json");
         let model_cmd = chat_root().join("chat.model.cmd.json");
         let chatgpt_prompts = chat_root().join("cache_model").join("chatgpt_prompts.json");
@@ -205,4 +213,52 @@ pub fn sync_prompts(app: AppHandle, data: String, time: u64) {
             "ChatGPT Prompts data has been synchronized!",
         );
         window_reload(app, "core");
+
+        return Some(data2);
+    }
+
+    None
+}
+
+#[command]
+pub async fn sync_user_prompts(url: String, data_type: String) -> Option<Vec<ModelRecord>> {
+    let res = utils::get_data(&url, None).await.unwrap();
+
+    info!("chatgpt_http_url: {}", url);
+
+    if let Some(v) = res {
+        let data;
+        if data_type == "csv" {
+            info!("chatgpt_http_csv_parser");
+            data = parse_prompt(v);
+        } else if data_type == "json" {
+            info!("chatgpt_http_json_parser");
+            data = serde_json::from_str(&v).unwrap_or_else(|err| {
+                info!("chatgpt_http_json_parser_error: {}", err);
+                vec![]
+            });
+        } else {
+            info!("chatgpt_http_unknown_type");
+            data = vec![];
+        }
+
+        let data = data
+            .iter()
+            .map(move |i| ModelRecord {
+                cmd: if i.cmd.is_some() {
+                    i.cmd.clone().unwrap()
+                } else {
+                    utils::gen_cmd(i.act.clone())
+                },
+                act: i.act.clone(),
+                prompt: i.prompt.clone(),
+                tags: vec!["user-sync".to_string()],
+                enable: true,
+            })
+            .collect::<Vec<ModelRecord>>();
+
+        return Some(data);
+    }
+
+    None
 }

src-tauri/src/app/setup.rs

@@ -17,7 +17,8 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
         info!("global_shortcut");
         let handle = app.app_handle();
         let mut shortcut = app.global_shortcut_manager();
-        shortcut.register(&v, move|| {
+        shortcut
+            .register(&v, move || {
                 if let Some(w) = handle.get_window("core") {
                     if w.is_visible().unwrap() {
                         w.hide().unwrap();
@@ -26,7 +27,8 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
                         w.set_focus().unwrap();
                     }
                 }
-        }).unwrap_or_else(|err| {
+            })
+            .unwrap_or_else(|err| {
                 info!("global_shortcut_register_error: {}", err);
             });
     } else {

src-tauri/src/assets/core.js

@@ -88,21 +88,7 @@ async function init() {
   });
 
   window.__sync_prompts = async function() {
-    const res = await fetch('https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv');
-    if (res.ok) {
-      const data = await res.text();
-      await invoke('sync_prompts', { data, time: Date.now() });
-    } else {
-      invoke('messageDialog', {
-        __tauriModule: 'Dialog',
-        message: {
-          cmd: 'messageDialog',
-          message: 'ChatGPT Prompts data sync failed, please try again!'.toString(),
-          title: 'Sync Prompts'.toString(),
-          type: 'error'
-        }
-      })
-    }
+    await invoke('sync_prompts', { time: Date.now() });
   }
 }
 

src-tauri/src/conf.rs

@@ -14,6 +14,8 @@ use tauri::TitleBarStyle;
 pub const ISSUES_URL: &str = "https://github.com/lencx/ChatGPT/issues";
 pub const UPDATE_LOG_URL: &str = "https://github.com/lencx/ChatGPT/blob/main/UPDATE_LOG.md";
 pub const AWESOME_URL: &str = "https://github.com/lencx/ChatGPT/blob/main/AWESOME.md";
+pub const GITHUB_PROMPTS_CSV_URL: &str =
+    "https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv";
 pub const DEFAULT_CHAT_CONF: &str = r#"{
     "stay_on_top": false,
     "theme": "Light",

src-tauri/src/main.rs

@@ -63,6 +63,7 @@ async fn main() {
             cmd::get_chat_model_cmd,
             cmd::parse_prompt,
             cmd::sync_prompts,
+            cmd::sync_user_prompts,
             cmd::window_reload,
             cmd::cmd_list,
             fs_extra::metadata,

src-tauri/src/utils.rs

@@ -109,3 +109,22 @@ pub fn gen_cmd(name: String) -> String {
     let re = Regex::new(r"[^a-zA-Z0-9]").unwrap();
     re.replace_all(&name, "_").to_lowercase()
 }
+
+pub async fn get_data(
+    url: &str,
+    app: Option<&tauri::AppHandle>,
+) -> Result<Option<String>, reqwest::Error> {
+    let res = reqwest::get(url).await?;
+    let is_ok = res.status() == 200;
+    let body = res.text().await?;
+
+    if is_ok {
+        Ok(Some(body))
+    } else {
+        info!("chatgpt_http_error: {}", body);
+        if let Some(v) = app {
+            tauri::api::dialog::message(v.get_window("core").as_ref(), "ChatGPT HTTP", body);
+        }
+        Ok(None)
+    }
+}

src-tauri/tauri.conf.json

@@ -15,13 +15,6 @@
       "globalShortcut": {
         "all": true
       },
-      "http": {
-        "all": true,
-        "scope": [
-          "https://**",
-          "http://**"
-        ]
-      },
       "fs": {
         "all": true,
         "scope": [

src/view/model/SyncCustom/index.tsx

@@ -1,6 +1,6 @@
 import { useState, useRef, useEffect } from 'react';
 import { Table, Modal, Button, message } from 'antd';
-import { invoke, http, path, fs } from '@tauri-apps/api';
+import { invoke, path, fs } from '@tauri-apps/api';
 
 import useData from '@/hooks/useData';
 import useChatModel, { useCacheModel } from '@/hooks/useChatModel';
@@ -10,7 +10,7 @@ import { CHAT_MODEL_JSON, chatRoot, readJSON, genCmd } from '@/utils';
 import { syncColumns, getPath } from './config';
 import SyncForm from './Form';
 
-const setTag = (data: Record<string, any>[]) => data.map((i) => ({ ...i, tags: ['user-sync'], enable: true }))
+const fmtData = (data: Record<string, any>[] = []) => (Array.isArray(data) ? data : []).map((i) => ({ ...i, cmd: i.cmd ? i.cmd : genCmd(i.act), tags: ['user-sync'], enable: true }));
 
 export default function SyncCustom() {
   const [isVisible, setVisible] = useState(false);
@@ -44,9 +44,16 @@ export default function SyncCustom() {
       setVisible(true);
     }
     if (['delete'].includes(opInfo.opType)) {
+      (async () => {
+        try {
+          const file = await path.join(await chatRoot(), 'cache_model', `${opInfo?.opRecord?.id}.json`);
+          await fs.removeFile(file);
+        } catch(e) {}
         const data = opRemove(opInfo?.opRecord?.[opSafeKey]);
         modelSet(data);
         opInfo.resetRecord();
+        modelCacheCmd();
+      })();
     }
   }, [opInfo.opType, formRef]);
 
@@ -58,20 +65,9 @@ export default function SyncCustom() {
 
     // https or http
     if (/^http/.test(record?.protocol)) {
-      const res = await http.fetch(filePath, {
-        method: 'GET',
-        responseType: isJson ? 1 : 2,
-      });
-      if (res.ok) {
-        if (isJson) {
-          // parse json
-          await modelCacheSet(setTag(Array.isArray(res?.data) ? res?.data : []), file);
-        } else {
-          // parse csv
-          const list: Record<string, string>[] = await invoke('parse_prompt', { data: res?.data });
-          const fmtList = list.map(i => ({ ...i, cmd: i.cmd ? i.cmd : genCmd(i.act), enable: true, tags: ['user-sync'] }));
-          await modelCacheSet(fmtList, file);
-        }
+      const data = await invoke('sync_user_prompts', { url: filePath, dataType: record?.ext });
+      if (data) {
+        await modelCacheSet(data as [], file);
         await modelCacheCmd();
         message.success('ChatGPT Prompts data has been synchronized!');
       } else {
@@ -83,13 +79,12 @@ export default function SyncCustom() {
     if (isJson) {
       // parse json
       const data = await readJSON(filePath, { isRoot: true });
-      await modelCacheSet(setTag(Array.isArray(data) ? data : []), file);
+      await modelCacheSet(fmtData(data), file);
     } else {
       // parse csv
       const data = await fs.readTextFile(filePath);
       const list: Record<string, string>[] = await invoke('parse_prompt', { data });
-      const fmtList = list.map(i => ({ ...i, cmd: i.cmd ? i.cmd : genCmd(i.act), enable: true, tags: ['user-sync'] }));
-      await modelCacheSet(fmtList, file);
+      await modelCacheSet(fmtData(list), file);
     }
     await modelCacheCmd();
   };

src/view/model/SyncPrompts/index.tsx

@@ -1,13 +1,13 @@
 import { useEffect, useState } from 'react';
-import { Table, Button, message, Popconfirm } from 'antd';
-import { invoke, http, path, shell } from '@tauri-apps/api';
+import { Table, Button, Popconfirm } from 'antd';
+import { invoke, path, shell } from '@tauri-apps/api';
 
 import useInit from '@/hooks/useInit';
 import useData from '@/hooks/useData';
 import useColumns from '@/hooks/useColumns';
 import useChatModel, { useCacheModel } from '@/hooks/useChatModel';
 import useTable, { TABLE_PAGINATION } from '@/hooks/useTable';
-import { fmtDate, chatRoot, GITHUB_PROMPTS_CSV_URL, genCmd } from '@/utils';
+import { fmtDate, chatRoot } from '@/utils';
 import { syncColumns } from './config';
 import './index.scss';
 
@@ -33,24 +33,13 @@ export default function SyncPrompts() {
   }, [modelCacheJson.length]);
 
   const handleSync = async () => {
-    const res = await http.fetch(GITHUB_PROMPTS_CSV_URL, {
-      method: 'GET',
-      responseType: http.ResponseType.Text,
-    });
-    const data = (res.data || '') as string;
-    if (res.ok) {
-      // const content = data.replace(/"(\s+)?,(\s+)?"/g, '","');
-      const list: Record<string, string>[] = await invoke('parse_prompt', { data });
-      const fmtList = list.map(i => ({ ...i, cmd: i.cmd ? i.cmd : genCmd(i.act), enable: true, tags: ['chatgpt-prompts'] }));
-      await modelCacheSet(fmtList);
-      opInit(fmtList);
+    const data = await invoke('sync_prompts', { time: Date.now() });
+    if (data) {
+      opInit(data as any[]);
       modelSet({
         id: 'chatgpt_prompts',
         last_updated: Date.now(),
       });
-      message.success('ChatGPT Prompts data has been synchronized!');
-    } else {
-      message.error('ChatGPT Prompts data sync failed, please try again!');
     }
   };
 

src/view/model/SyncRecord/config.tsx

@@ -24,7 +24,9 @@ export const syncColumns = () => [
     dataIndex: 'tags',
     key: 'tags',
     // width: 150,
-    render: () => <Tag>chatgpt-prompts</Tag>,
+    render: (v: string[]) => (
+      <span className="chat-prompts-tags">{v?.map(i => <Tag key={i}>{i}</Tag>)}</span>
+    ),
   },
   {
     title: 'Enable',