Merge pull request #130 from tk103331/main

src-tauri/src/app/cmd.rs

@@ -38,6 +38,11 @@ pub fn get_chat_conf() -> ChatConfJson {
     ChatConfJson::get_chat_conf()
 }
 
+#[command]
+pub fn reset_chat_conf() -> ChatConfJson {
+    ChatConfJson::reset_chat_conf()
+}
+
 #[command]
 pub fn form_confirm(_app: AppHandle, data: serde_json::Value) {
     ChatConfJson::amend(&serde_json::json!(data), None).unwrap();

src-tauri/src/conf.rs

@@ -129,6 +129,17 @@ impl ChatConfJson {
         }
     }
 
+    pub fn reset_chat_conf() -> Self {
+        let conf_file = ChatConfJson::conf_path();
+        let content = if cfg!(target_os = "macos") {
+            DEFAULT_CHAT_CONF_MAC
+        } else {
+            DEFAULT_CHAT_CONF
+        };
+        fs::write(&conf_file, content).unwrap();
+        serde_json::from_str(content).unwrap()
+    }
+
     // https://users.rust-lang.org/t/updating-object-fields-given-dynamic-json/39049/3
     pub fn amend(new_rules: &Value, app: Option<tauri::AppHandle>) -> Result<()> {
         let config = ChatConfJson::get_chat_conf();

src-tauri/src/main.rs

@@ -56,6 +56,7 @@ async fn main() {
             cmd::download,
             cmd::open_link,
             cmd::get_chat_conf,
+            cmd::reset_chat_conf,
             cmd::form_cancel,
             cmd::form_confirm,
             cmd::form_msg,

src/view/General.tsx

@@ -58,6 +58,19 @@ export default function General() {
     form.setFieldsValue(chatConf);
   };
 
+  const onReset = async () => {
+    const chatData = await invoke('reset_chat_conf');
+    setChatConf(chatData);
+    const isOk = await ask(`Configuration reset successfully, whether to restart?`, {
+      title: 'ChatGPT Preferences'
+    });
+    if (isOk) {
+      relaunch();
+      return;
+    }
+    message.success('Configuration reset successfully');
+  };
+
   const onFinish = async (values: any) => {
     if (!isEqual(omit(chatConf, ['default_origin']), values)) {
       await invoke('form_confirm', { data: values, label: 'main' });
@@ -115,10 +128,13 @@ export default function General() {
         <Form.Item>
           <Space size={20}>
             <Button onClick={onCancel}>Cancel</Button>
+
             <Button type="primary" htmlType="submit">
               Submit
             </Button>
+            <a onClick={onReset}>Reset to defaults</a>
           </Space>
+
         </Form.Item>
       </Form>
     </>