[feat] persona import/export

src/ts/persona.ts

@@ -0,0 +1,118 @@
+import { get } from "svelte/store"
+import { DataBase, saveImage, setDatabase } from "./storage/database"
+import { selectSingleFile, sleep } from "./util"
+import { alertError, alertNormal, alertStore } from "./alert"
+import * as yuso from 'yuso'
+import { downloadFile, readImage } from "./storage/globalApi"
+import { language } from "src/lang"
+import { cloneDeep } from "lodash"
+
+export async function selectUserImg() {
+    const selected = await selectSingleFile(['png'])
+    if(!selected){
+        return
+    }
+    const img = selected.data
+    let db = get(DataBase)
+    const imgp = await saveImage(img)
+    db.userIcon = imgp
+    db.personas[db.selectedPersona] = {
+        name: db.username,
+        icon: db.userIcon,
+        personaPrompt: db.personaPrompt
+    }
+    setDatabase(db)
+}
+
+export function saveUserPersona() {
+    let db = get(DataBase)
+    db.personas[db.selectedPersona] = {
+        name: db.username,
+        icon: db.userIcon,
+        personaPrompt: db.personaPrompt
+    }
+    setDatabase(db)
+}
+
+export function changeUserPersona(id:number, save:'save'|'noSave' = 'save') {
+    if(save === 'save'){
+        saveUserPersona()
+    }
+    let db = get(DataBase)
+    const pr = db.personas[id]
+    db.personaPrompt = pr.personaPrompt
+    db.username = pr.name,
+    db.userIcon = pr.icon
+    db.selectedPersona = id
+    setDatabase(db)
+}
+
+
+
+interface PersonaCard {
+    name: string
+    personaPrompt: string
+}
+
+export async function exportUserPersona(){
+    let db = get(DataBase)
+    if(!db.userIcon){
+        alertError(language.errors.noUserIcon)
+        return
+    }
+    if((!db.username) || (!db.personaPrompt)){
+        alertError("username or persona prompt is empty")
+        return
+    }
+
+
+    let img = await readImage(db.userIcon)
+
+    let card:PersonaCard = cloneDeep({
+        name: db.username,
+        personaPrompt: db.personaPrompt,
+    })
+
+    alertStore.set({
+        type: 'wait',
+        msg: 'Loading... (Writing Exif)'
+    })
+
+    await sleep(10)
+
+    img = yuso.encode(yuso.trim(img), "persona",Buffer.from(JSON.stringify(card)).toString('base64'))
+
+    alertStore.set({
+        type: 'wait',
+        msg: 'Loading... (Writing)'
+    })
+    
+    await sleep(10)
+    await downloadFile(`${db.username.replace(/[<>:"/\\|?*\.\,]/g, "")}_export.png`, img)
+
+    alertNormal(language.successExport)
+}
+
+export async function importUserPersona(){
+
+    try {
+        const v = await selectSingleFile(['png'])
+        const data:PersonaCard = JSON.parse(Buffer.from(yuso.decode(v.data, "persona"), 'base64').toString('utf-8'))
+        if(data.name && data.personaPrompt){
+            let db = get(DataBase)
+            db.personas.push({
+                name: data.name,
+                icon: await saveImage(yuso.trim(v.data)),
+                personaPrompt: data.personaPrompt
+            })
+            setDatabase(db)
+            alertNormal(language.successImport)
+        }else{
+            alertError(language.errors.noData)
+        }
+
+    } catch (error) {
+        alertError(`${error}`)
+        return
+    }
+}