chore: optim

src-tauri/src/app/cmd.rs

@@ -1,5 +1,5 @@
 use crate::{conf::ChatConfJson, utils};
-use std::{fs, path::PathBuf};
+use std::{collections::HashMap, fs, path::PathBuf};
 use tauri::{api, command, AppHandle, Manager};
 
 #[command]
@@ -72,7 +72,7 @@ pub fn get_chat_model_cmd() -> serde_json::Value {
     serde_json::from_str(&content).unwrap()
 }
 
-#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct PromptRecord {
     pub cmd: Option<String>,
     pub act: String,
@@ -99,9 +99,8 @@ pub fn window_reload(app: AppHandle, label: &str) {
         .unwrap();
 }
 
-
-use walkdir::WalkDir;
 use utils::chat_root;
+use walkdir::WalkDir;
 
 #[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
 pub struct ModelRecord {
@@ -115,12 +114,14 @@ pub struct ModelRecord {
 #[command]
 pub fn cmd_list() -> Vec<ModelRecord> {
     let mut list = vec![];
-    for entry in WalkDir::new(chat_root().join("cache_model")).into_iter().filter_map(|e| e.ok()) {
+    for entry in WalkDir::new(chat_root().join("cache_model"))
+        .into_iter()
+        .filter_map(|e| e.ok())
+    {
         let file = fs::read_to_string(entry.path().display().to_string());
         if let Ok(v) = file {
             let data: Vec<ModelRecord> = serde_json::from_str(&v).unwrap_or_else(|_| vec![]);
-            let enable_list = data.into_iter()
-                .filter(|v| v.enable);
+            let enable_list = data.into_iter().filter(|v| v.enable);
             list.extend(enable_list)
         }
     }
@@ -128,3 +129,65 @@ pub fn cmd_list() -> Vec<ModelRecord> {
     list.sort_by(|a, b| a.cmd.len().cmp(&b.cmd.len()));
     list
 }
+
+#[command]
+pub fn sync_prompts(app: AppHandle, data: String, time: u64) {
+    let data = parse_prompt(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!["chatgpt-prompts".to_string()],
+            enable: true,
+        })
+        .collect::<Vec<ModelRecord>>();
+
+    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");
+
+    // chatgpt_prompts.json
+    fs::write(
+        chatgpt_prompts,
+        serde_json::to_string_pretty(&data).unwrap(),
+    )
+    .unwrap();
+    let cmd_data = cmd_list();
+
+    // chat.model.cmd.json
+    fs::write(
+        model_cmd,
+        serde_json::to_string_pretty(&serde_json::json!({
+            "name": "ChatGPT CMD",
+            "last_updated": time,
+            "data": cmd_data,
+        }))
+        .unwrap(),
+    )
+    .unwrap();
+    let mut kv = HashMap::new();
+    kv.insert(
+        "sync_prompts".to_string(),
+        serde_json::json!({ "id": "chatgpt_prompts", "last_updated": time }),
+    );
+    let model_data = utils::merge(
+        &serde_json::from_str(&fs::read_to_string(&model).unwrap()).unwrap(),
+        &kv,
+    );
+
+    // chat.model.json
+    fs::write(model, serde_json::to_string_pretty(&model_data).unwrap()).unwrap();
+
+    // refresh window
+    api::dialog::message(
+        app.get_window("core").as_ref(),
+        "Sync Prompts",
+        "ChatGPT Prompts data has been synchronized!",
+    );
+    window_reload(app, "core");
+}