From 501fead589ca7bd8aaae2f954a392b7f35bbebb8 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Sat, 24 Jun 2023 05:41:00 +0900
Subject: [PATCH] [feat] added preset import/export

---
 src/lib/Setting/botpreset.svelte | 16 ++++++++++++++--
 src/ts/storage/database.ts       | 30 +++++++++++++++++++++++++++---
 2 files changed, 41 insertions(+), 5 deletions(-)

diff --git a/src/lib/Setting/botpreset.svelte b/src/lib/Setting/botpreset.svelte
index 12449d27..41d0a9b4 100644
--- a/src/lib/Setting/botpreset.svelte
+++ b/src/lib/Setting/botpreset.svelte
@@ -1,8 +1,8 @@
 <script>
     import { alertConfirm, alertError } from "../../ts/alert";
     import { language } from "../../lang";
-    import { DataBase, changeToPreset, copyPreset, presetTemplate } from "../../ts/storage/database";
-    import { CopyIcon, EditIcon, PlusIcon, TrashIcon, XIcon } from "lucide-svelte";
+    import { DataBase, changeToPreset, copyPreset, downloadPreset, importPreset, presetTemplate } from "../../ts/storage/database";
+    import { CopyIcon, DownloadIcon, EditIcon, FolderUpIcon, PlusIcon, TrashIcon, XIcon } from "lucide-svelte";
 
     let editMode = false
     export let close = () => {}
@@ -41,6 +41,13 @@
                     }}>
                         <CopyIcon size={18}/>
                     </button>
+                    <button class="text-gray-500 hover:text-green-500 cursor-pointer mr-2" on:click={(e) => {
+                        e.stopPropagation()
+                        downloadPreset(i)
+                    }}>
+
+                        <DownloadIcon size={18} />
+                    </button>
                     <button class="text-gray-500 hover:text-green-500 cursor-pointer" on:click={async (e) => {
                         e.stopPropagation()
                         if($DataBase.botPresets.length === 1){
@@ -71,6 +78,11 @@
             }}>
                 <PlusIcon/>
             </button>
+            <button class="text-gray-500 hover:text-green-500 mr-2 cursor-pointer" on:click={() => {
+                importPreset()
+            }}>
+                <FolderUpIcon size={18}/>
+            </button>
             <button class="text-gray-500 hover:text-green-500 cursor-pointer" on:click={() => {
                 editMode = !editMode
             }}>
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 3e72eeb3..72d1196b 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -1,10 +1,11 @@
 import { get, writable } from 'svelte/store';
-import { checkNullish } from '../util';
-import { changeLanguage } from '../../lang';
+import { checkNullish, selectSingleFile } from '../util';
+import { changeLanguage, language } from '../../lang';
 import type { RisuPlugin } from '../process/plugins';
-import { saveAsset as saveImageGlobal } from './globalApi';
+import { downloadFile, saveAsset as saveImageGlobal } from './globalApi';
 import { cloneDeep } from 'lodash';
 import { defaultAutoSuggestPrompt, defaultJailbreak, defaultMainPrompt } from './defaultPrompts';
+import { alertNormal } from '../alert';
 
 export const DataBase = writable({} as any as Database)
 export const loadedStore = writable(false)
@@ -711,4 +712,27 @@ export function changeToPreset(id =0){
     db.bias = newPres.bias ?? db.bias
     db.koboldURL = newPres.koboldURL ?? db.koboldURL
     DataBase.set(db)
+}
+
+export function downloadPreset(id:number){
+    saveCurrentPreset()
+    let db = get(DataBase)
+    let pres = cloneDeep(db.botPresets[id])
+    pres.openAIKey = ''
+    pres.forceReplaceUrl = ''
+    pres.forceReplaceUrl2 = ''
+    downloadFile(pres.name + "_preset.json", Buffer.from(JSON.stringify(pres, null, 2)))
+    alertNormal(language.successExport)
+}
+
+export async function importPreset(){
+    const f = await selectSingleFile(["json"])
+    if(!f){
+        return
+    }
+    let db = get(DataBase)
+    const pre = (JSON.parse(Buffer.from(f.data).toString('utf-8')))
+    pre.name ??= "Imported"
+    db.botPresets.push(pre)
+    DataBase.set(db)
 }
\ No newline at end of file