自定义源列显示源版本号、作者名字，优化自定义源脚本数据存储

src/main/modules/userApi/config/index.ts

@@ -1,2 +1,2 @@
 
-export const userApis: LX.UserApi.UserApiInfo[] = []
+export const userApis: LX.UserApi.UserApiInfoFull[] = []

src/main/modules/userApi/index.ts

@@ -6,9 +6,9 @@ let userApiId: string | null
 
 export const getApiList = getUserApis
 
-export const importApi = (script: string): LX.UserApi.ImportUserApi => {
+export const importApi = async(script: string): Promise<LX.UserApi.ImportUserApi> => {
   return {
-    apiInfo: handleImportApi(script),
+    apiInfo: await handleImportApi(script),
     apiList: getUserApis(),
   }
 }

src/main/modules/userApi/main.ts

@@ -5,6 +5,7 @@ import path from 'node:path'
 import { openDevTools as handleOpenDevTools } from '@main/utils'
 import { encodePath } from '@common/utils/electron'
 import USER_API_RENDERER_EVENT_NAME from './rendererEvent/name'
+import { getScript } from './utils'
 
 let browserWindow: Electron.BrowserWindow | null = null
 
@@ -91,8 +92,8 @@ export const createWindow = async(userApi: LX.UserApi.UserApiInfo) => {
   // const randomNum = Math.random().toString().substring(2, 10)
   await browserWindow.loadURL('data:text/html;charset=UTF-8,' + encodeURIComponent(html))
 
-  browserWindow.on('ready-to-show', () => {
-    sendEvent(USER_API_RENDERER_EVENT_NAME.initEnv, userApi)
+  browserWindow.on('ready-to-show', async() => {
+    sendEvent(USER_API_RENDERER_EVENT_NAME.initEnv, { ...userApi, script: await getScript(userApi.id) })
   })
 
   // global.modules.userApiWindow.loadFile(join(dir, 'renderer/user-api.html'))
@@ -106,7 +107,7 @@ export const closeWindow = async() => {
     browserWindow.webContents.session.clearStorageData(),
     browserWindow.webContents.session.clearCache(),
   ])
-  browserWindow.destroy()
+  browserWindow?.destroy()
   browserWindow = null
 }
 

src/main/modules/userApi/utils.ts

@@ -1,20 +1,52 @@
 import { userApis as defaultUserApis } from './config'
 import { STORE_NAMES } from '@common/constants'
 import getStore from '@main/utils/store'
+import zlib from 'node:zlib'
 
 let userApis: LX.UserApi.UserApiInfo[] | null
+let scripts = new Map<string, string>()
+
+const saveData = () => {
+  getStore(STORE_NAMES.USER_API).set('userApis', userApis!.map(api => {
+    return {
+      ...api,
+      script: scripts.get(api.id),
+    }
+  }))
+}
 
 export const getUserApis = (): LX.UserApi.UserApiInfo[] => {
-  const electronStore_userApi = getStore(STORE_NAMES.USER_API)
   if (userApis) return userApis
-  userApis = electronStore_userApi.get('userApis') as LX.UserApi.UserApiInfo[]
-  if (!userApis) {
-    userApis = defaultUserApis
+
+  const electronStore_userApi = getStore(STORE_NAMES.USER_API)
+  let infoFull = electronStore_userApi.get('userApis') as LX.UserApi.UserApiInfoFull[]
+  let requiredUpdate = false
+  if (infoFull) {
+    for (let i = 0; i < infoFull.length; i++) {
+      const api = infoFull[i]
+      if (api.version != null) continue
+      requiredUpdate ||= true
+      try {
+        infoFull.splice(i, 1, {
+          ...parseScriptInfo(api.script),
+          ...api,
+        })
+      } catch (e) {
+        infoFull.splice(i, 1)
+        i--
+      }
+    }
+  } else {
+    infoFull = defaultUserApis
     electronStore_userApi.set('userApis', userApis)
   }
-  for (const api of userApis) {
+  userApis = infoFull.map(api => {
     if (api.allowShowUpdateAlert == null) api.allowShowUpdateAlert = false
-  }
+    const { script, ...info } = api
+    scripts.set(api.id, script)
+    return info
+  })
+  if (requiredUpdate) saveData()
   return userApis
 }
 
@@ -46,22 +78,47 @@ const matchInfo = (scriptInfo: string) => {
 
   return infos as Record<keyof typeof INFO_NAMES, string>
 }
-export const importApi = (script: string): LX.UserApi.UserApiInfo => {
+const parseScriptInfo = (script: string) => {
   const result = /^\/\*[\S|\s]+?\*\//.exec(script)
   if (!result) throw new Error('无效的自定义源文件')
 
   let scriptInfo = matchInfo(result[0])
 
   scriptInfo.name ||= `user_api_${new Date().toLocaleString()}`
+  return scriptInfo
+}
+const deflateScript = async(script: string) => new Promise<string>((resolve, reject) => {
+  zlib.deflate(Buffer.from(script, 'utf8'), (err, buf) => {
+    if (err) {
+      reject(err)
+      return
+    }
+    resolve('gz_' + buf.toString('base64'))
+  })
+})
+const inflateScript = async(script: string) => new Promise<string>((resolve, reject) => {
+  if (script.startsWith('gz_')) {
+    zlib.inflate(Buffer.from(script.substring(3), 'base64'), (err, buf) => {
+      if (err) {
+        reject(err)
+        return
+      }
+      resolve(buf.toString('utf8'))
+    })
+  } else resolve(script)
+})
+export const importApi = async(scriptRaw: string): Promise<LX.UserApi.UserApiInfo> => {
+  let scriptInfo = parseScriptInfo(scriptRaw)
   const apiInfo = {
     id: `user_api_${Math.random().toString().substring(2, 5)}_${Date.now()}`,
     ...scriptInfo,
-    script,
     allowShowUpdateAlert: true,
   }
   userApis ??= []
   userApis.push(apiInfo)
-  getStore(STORE_NAMES.USER_API).set('userApis', userApis)
+  const script = await deflateScript(scriptRaw)
+  scripts.set(apiInfo.id, script)
+  saveData()
   return apiInfo
 }
 
@@ -69,16 +126,21 @@ export const removeApi = (ids: string[]) => {
   if (!userApis) return
   for (let index = userApis.length - 1; index > -1; index--) {
     if (ids.includes(userApis[index].id)) {
+      scripts.delete(userApis[index].id)
       userApis.splice(index, 1)
       ids.splice(index, 1)
     }
   }
-  getStore(STORE_NAMES.USER_API).set('userApis', userApis)
+  saveData()
 }
 
 export const setAllowShowUpdateAlert = (id: string, enable: boolean) => {
   const targetApi = userApis?.find(api => api.id == id)
   if (!targetApi) return
   targetApi.allowShowUpdateAlert = enable
-  getStore(STORE_NAMES.USER_API).set('userApis', userApis)
+  saveData()
+}
+
+export const getScript = async(id: string) => {
+  return inflateScript(scripts.get(id) ?? '')
 }