[feat] charajs apis

2023-11-08 16:33:53 +09:00
parent c31f3dffe6
commit 406c2fc3c1
7 changed files with 359 additions and 31 deletions
--- a/src/ts/plugins/embedscript.ts
+++ b/src/ts/plugins/embedscript.ts
@@ -1,11 +1,14 @@
 import { get } from 'svelte/store'
 import type { ScriptMode } from '../process/scripts'
 import myWorkerUrl from './embedworker?worker&url'
-import { DataBase } from '../storage/database'
+import { DataBase, type Chat, type character, type Message } from '../storage/database'
 import { selectedCharID } from '../stores'
-import { cloneDeep } from 'lodash'
+import { add, cloneDeep } from 'lodash'
+import { sleep } from '../util'
+import { characterFormatUpdate } from '../characters'
+import { setDatabase } from '../storage/database'

-let worker = new Worker(new URL(myWorkerUrl), {type: 'module'})
+let worker = new Worker(myWorkerUrl, {type: 'module'})

 let results:{
    id: string,
@@ -49,7 +52,6 @@ function runVirtualJS(code:string){
  return new Promise((resolve,reject)=>{
    const interval = setInterval(()=>{
      const result = results.find(r=>r.id === id)
-      console.log(performance.now() - startTime )
      if(result){
        clearInterval(interval)
        resolve(result.result)
@@ -61,30 +63,184 @@ function runVirtualJS(code:string){
        worker = new Worker(new URL('./worker.ts', import.meta.url), {type: 'module'})
        reject('timeout')
      }
-    },100)
+    },10)
  })
 }

-addWorkerFunction('getCharacter', async () => {
+addWorkerFunction('getChat', async () => {
    const db = get(DataBase)
    const selectedChar = get(selectedCharID)
-    return cloneDeep(db.characters[selectedChar])
+    const char = db.characters[selectedChar]
+    return cloneDeep(char.chats[char.chatPage].message)
 })

+addWorkerFunction('setChat', async (data:Message[]) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    let newChat:Message[] = []
+    for(const dat of data){
+        if(dat.role !== 'char' && dat.role !== 'user'){
+            return false
+        }
+        if(typeof dat.data !== 'string'){
+            return false
+        }
+        if(typeof dat.saying !== 'string'){
+            return false
+        }
+        if(typeof dat.time !== 'number'){
+            return false
+        }
+        if(typeof dat.chatId !== 'string'){
+            return false
+        }
+        newChat.push({
+            role: dat.role,
+            data: dat.data,
+            saying: dat.saying,
+            time: dat.time,
+            chatId: dat.chatId
+        })
+    }
+    db.characters[selectedChar].chats[db.characters[selectedChar].chatPage].message = newChat
+    setDatabase(db)
+    return true
+})
+
+addWorkerFunction('getName', async () => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    return char.name
+})
+
+addWorkerFunction('setName', async (data:string) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    if(typeof data !== 'string'){
+        return false
+    }
+    db.characters[selectedChar].name = data
+    setDatabase(db)
+    return true
+})
+
+addWorkerFunction('getDescription', async () => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    if(char.type === 'group'){
+        return ''
+    }
+    return char.desc
+})
+
+addWorkerFunction('setDescription', async (data:string) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char =db.characters[selectedChar]
+    if(typeof data !== 'string'){
+        return false
+    }
+    if(char.type === 'group'){
+        return false
+    }
+    char.desc = data
+    db.characters[selectedChar] = char
+    setDatabase(db)
+    return true
+})
+
+addWorkerFunction('getCharacterFirstMessage', async () => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    return char.firstMessage
+})
+
+addWorkerFunction('setCharacterFirstMessage', async (data:string) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    if(typeof data !== 'string'){
+        return false
+    }
+    char.firstMessage = data
+    db.characters[selectedChar] = char
+    setDatabase(db)
+    return true
+})
+
+addWorkerFunction('getState', async (statename) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    const chat = char.chats[char.chatPage]
+    return (chat.scriptstate ?? {})[statename]
+})
+
+addWorkerFunction('setState', async (statename, data) => {
+    const db = get(DataBase)
+    const selectedChar = get(selectedCharID)
+    const char = db.characters[selectedChar]
+    const chat = char.chats[char.chatPage]
+    if(typeof statename !== 'string'){
+        return false
+    }
+    if(typeof data !== 'string' && typeof data !== 'number' && typeof data !== 'boolean'){
+        return false
+    }
+    if(!chat.scriptstate){
+        chat.scriptstate = {}
+    }
+    chat.scriptstate[statename] = data
+    char.chats[char.chatPage] = chat
+    db.characters[selectedChar] = char
+    setDatabase(db)
+    return true
+})
+
+
+
+let lastCode = ''
+let lastModeList:string[] = []
+
 export async function runCharacterJS(arg:{
-    code: string,
-    mode: ScriptMode
+    code: string|null,
+    mode: ScriptMode|'onButtonClick'
    data: string
 }):Promise<string>{
    try {
+        if(arg.code === null){
+            const db = get(DataBase)
+            const selectedChar = get(selectedCharID)
+            arg.code = db.characters[selectedChar].virtualscript
+        }
        const codes = {
            "editinput": 'editInput',
            "editoutput": 'editOutput',
            "editprocess": 'editProcess',
            "editdisplay": 'editDisplay',
+            'onButtonClick': "onButtonClick"
        } as const

+        if(lastCode !== arg.code){
+            lastModeList = []
+            const codesplit = arg.code.split('\n')
+            for(let i = 0; i < codesplit.length; i++){
+                const line = codesplit[i]
+                if(line.startsWith('//@use')){
+                    lastModeList.push(line.replace('//@use','').trim())
+                }
+            }
+            lastCode = arg.code
+        }
+
        const runCode = codes[arg.mode]
+
+        if(!lastModeList.includes(runCode)){
+            return arg.data
+        }
        const result = await runVirtualJS(`${arg.code}\n${runCode}(${JSON.stringify(arg.data)})`)

        if(!result){
@@ -99,4 +255,23 @@ export async function runCharacterJS(arg:{
        return arg.data
    }

+}
+
+export async function watchParamButton() {
+    while(true){
+        const qs = document.querySelectorAll('*[risu-btn]:not([risu-btn-run="true"])')
+        for(let i = 0; i < qs.length; i++){
+            const q = qs[i]
+            const code = q.getAttribute('risu-btn')
+            q.setAttribute('risu-btn-run','true')
+            q.addEventListener('click',async ()=>{
+                await runCharacterJS({
+                    code: null,
+                    mode: 'onButtonClick',
+                    data: code
+                })
+            })
+        }
+        await sleep(100)
+    }
 }
--- a/src/ts/plugins/embedworker.ts
+++ b/src/ts/plugins/embedworker.ts
@@ -71,20 +71,27 @@ const whitelist = [
    "Request",
    "Response",
    "Blob",
-    "postMessage"
+    "postMessage",
+    "Node",
+    "Element",
+    "Text",
+    "Comment",
 ]

-const evaluation = global.eval
+const evaluation = globaly.eval

-Object.getOwnPropertyNames( global ).forEach( function( prop ) {
-    if( !whitelist.includes(prop) ) {
-        Object.defineProperty( global, prop, {
-            get : function() {
-                throw "Security Exception: cannot access "+prop;
-                return 1;
-            }, 
-            configurable : false
-        });    
+Object.getOwnPropertyNames( globaly ).forEach( function( prop ) {
+    if( (!whitelist.includes(prop)) && (!prop.startsWith('HTML')) && (!prop.startsWith('XML')) ) {
+        try {
+            Object.defineProperty( globaly, prop, {
+                get : function() {
+                    throw "Security Exception: cannot access "+prop;
+                    return 1;
+                }, 
+                configurable : false
+            });     
+        } catch (error) {
+        }  
    }
 });

@@ -93,6 +100,7 @@ let workerResults:{
    result: any
 }[] = []

+
 self.onmessage = async (event) => {
    const da = event.data
    if(da.type === 'result'){
@@ -101,7 +109,7 @@ self.onmessage = async (event) => {
    }
    if(da.type === 'api'){
        //add api
-        Object.defineProperty( global, da.name, {
+        Object.defineProperty( globaly, da.name, {
            get : function() {
                return function (...args:any[]) {
                    return new Promise((resolve)=>{
@@ -118,7 +126,7 @@ self.onmessage = async (event) => {
                                clearInterval(interval)
                                resolve(result.result)
                            }
-                        },100)
+                        },10)
                    })
                }
            }