risuai/src/ts/stores.ts

import { get, writable, type Writable } from "svelte/store";
import { DataBase, type Chat, type character, type groupChat } from "./storage/database";
import { isEqual } from "lodash";
import type { simpleCharacterArgument } from "./parser";
import { sleep } from "./util";

function updateSize(){
    SizeStore.set({
        w: window.innerWidth,
        h: window.innerHeight
    })
    DynamicGUI.set(window.innerWidth <= 1024)
}

export const SizeStore = writable({
    w: 0,
    h: 0
})
export const DynamicGUI = writable(false)
export const sideBarClosing = writable(false)
export const sideBarStore = writable(window.innerWidth > 1024)
export const selectedCharID = writable(-1)
export const CharEmotion = writable({} as {[key:string]: [string, string, number][]})
export const ViewBoxsize = writable({ width: 12 * 16, height: 12 * 16 }); // Default width and height in pixels
export const settingsOpen = writable(false)
export const botMakerMode = writable(false)

//optimization

export const CurrentCharacter = writable(null) as Writable<character | groupChat>
export const CurrentSimpleCharacter = writable(null) as Writable<simpleCharacterArgument>
export const CurrentChat = writable(null) as Writable<Chat>
export const CurrentUsername = writable('') as Writable<string>
export const CurrentUserIcon = writable('') as Writable<string>
export const CurrentShowMemoryLimit = writable(false) as Writable<boolean>
export const ShowVN = writable(false)
export const SettingsMenuIndex = writable(-1)
export const CurrentVariablePointer = writable({} as {[key:string]: string|number|boolean})
export const OpenRealmStore = writable(false)
export const ShowRealmFrameStore = writable('')
export const PlaygroundStore = writable(0)

function createSimpleCharacter(char:character|groupChat){
    if((!char) || char.type === 'group'){
        return null
    }

    const simpleChar:simpleCharacterArgument = {
        type: "simple",
        customscript: structuredClone(char.customscript),
        chaId: char.chaId,
        additionalAssets: char.additionalAssets,
        virtualscript: char.virtualscript,
        emotionImages: char.emotionImages,
    }

    return simpleChar

}

function trySync(){
    try {
        let db = get(DataBase)
        let currentChar = get(selectedCharID)
        let currentCharacter = db.characters ? (db.characters[currentChar]) : null
        let currentChat = currentCharacter ? (currentCharacter.chats[currentCharacter.chatPage]) : null
        CurrentCharacter.set(structuredClone(currentCharacter))
        CurrentSimpleCharacter.set(createSimpleCharacter(currentCharacter))
        CurrentChat.set(structuredClone(currentChat))
        CurrentUsername.set(db.username)
        CurrentUserIcon.set(db.userIcon)
        CurrentShowMemoryLimit.set(db.showMemoryLimit)
    } catch (error) {}
}

trySync()

async function preInit(){
    await sleep(1)
    trySync()
    function updateCurrentCharacter(){
        
        const db = get(DataBase)
        if(!db.characters){
            CurrentCharacter.set(null)
            updateCurrentChat()
            return
        }

        const currentCharId = get(selectedCharID)
        const currentChar = db.characters[currentCharId]
        const gotCharacter = get(CurrentCharacter)
        if(isEqual(gotCharacter, currentChar)){
            return
        }
        if((currentChar?.viewScreen === 'vn') !== get(ShowVN)){
            ShowVN.set(currentChar?.viewScreen === 'vn')   
        }

        CurrentCharacter.set(structuredClone(currentChar))
        const simp = createSimpleCharacter(currentChar)
        
        if(!isEqual(get(CurrentSimpleCharacter), simp)){
            CurrentSimpleCharacter.set(simp)
        }

        updateCurrentChat()
    }

    function updateCurrentChat(){
        const currentChar = get(CurrentCharacter)
        if(!currentChar){
            CurrentChat.set(null)
            return
        }
        const chat = (currentChar.chats[currentChar.chatPage])
        const gotChat = get(CurrentChat)
        if(isEqual(gotChat, chat)){
            return
        }
        CurrentChat.set(structuredClone(chat))
    }

    DataBase.subscribe((data) => {
        updateCurrentCharacter()
        if(data.username !== get(CurrentUsername)){
            CurrentUsername.set(data.username)
        }
        if(data.userIcon !== get(CurrentUserIcon)){
            CurrentUserIcon.set(data.userIcon)
        }
        if(data.showMemoryLimit !== get(CurrentShowMemoryLimit)){
            CurrentShowMemoryLimit.set(data.showMemoryLimit)
        }
    })

    selectedCharID.subscribe((id) => {

        updateCurrentCharacter()
    })

    CurrentCharacter.subscribe((char) => {
        updateCurrentChat()
        let db = get(DataBase)
        let charId = get(selectedCharID)
        if(charId === -1 || charId > db.characters.length){
            return
        }
        let cha = db.characters[charId]
        if(isEqual(cha, char)){
            return
        }
        db.characters[charId] = structuredClone(char)
        DataBase.set(db)
    })

    CurrentChat.subscribe((chat) => {
        let currentChar = get(CurrentCharacter)

        if(currentChar){
            if(!isEqual(currentChar.chats[currentChar.chatPage], chat)){
                currentChar.chats[currentChar.chatPage] = structuredClone(chat)
                CurrentCharacter.set(currentChar)
            }
        }

        const variablePointer = get(CurrentVariablePointer)
        const currentState = structuredClone(chat?.scriptstate)

        if(!isEqual(variablePointer, currentState)){
            CurrentVariablePointer.set(currentState)
        }
    })
}


updateSize()
window.addEventListener("resize", updateSize);
preInit()