forked from ProjectWIND/ProjectWIND
解决了提交带来的bug
This commit is contained in:
parent
2452df4da6
commit
cdc0316c52
@ -1,256 +0,0 @@
|
||||
//go:build windows
|
||||
// +build windows
|
||||
|
||||
package core
|
||||
|
||||
import (
|
||||
"ProjectWIND/LOG"
|
||||
"ProjectWIND/wba"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"syscall"
|
||||
"unsafe"
|
||||
"github.com/dop251/goja"
|
||||
)
|
||||
|
||||
var CmdMap = make(map[string]wba.Cmd)
|
||||
|
||||
func ReloadApps() (total int, success int) {
|
||||
appsDir := "./data/app/"
|
||||
appFiles, err := os.ReadDir(appsDir)
|
||||
total = 0
|
||||
success = 0
|
||||
if err != nil {
|
||||
LOG.Error("加载应用所在目录失败:%v", err)
|
||||
return
|
||||
}
|
||||
|
||||
for _, file := range appFiles {
|
||||
totalDelta, successDelta := reloadAPP(file, appsDir)
|
||||
total += totalDelta
|
||||
success += successDelta
|
||||
}
|
||||
CmdMap = mergeMaps(CmdMap, AppCore.CmdMap)
|
||||
return total, success
|
||||
}
|
||||
|
||||
func reloadAPP(file os.DirEntry, appsDir string) (totalDelta int, successDelta int) {
|
||||
if file.IsDir() {
|
||||
return 0, 0
|
||||
}
|
||||
ext := filepath.Ext(file.Name())
|
||||
if ext == ".dll" {
|
||||
pluginPath := filepath.Join(appsDir, file.Name())
|
||||
lib, err := syscall.LoadLibrary(pluginPath)
|
||||
if err != nil {
|
||||
LOG.Error("加载应用 %s 失败: %v", pluginPath, err)
|
||||
return 1, 0
|
||||
}
|
||||
defer func(handle syscall.Handle) {
|
||||
err := syscall.FreeLibrary(handle)
|
||||
if err != nil {
|
||||
LOG.Error("释放应用 %s 时发生错误: %v", pluginPath, err)
|
||||
}
|
||||
}(lib)
|
||||
|
||||
// 获取函数地址
|
||||
sym, err := syscall.GetProcAddress(lib, "AppInit")
|
||||
if err != nil {
|
||||
fmt.Println("找不到应用 %s 提供的 AppInit 接口: %v", err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 定义函数类型
|
||||
AppInitPtr := (*func() wba.AppInfo)(unsafe.Pointer(&sym))
|
||||
AppInit := *AppInitPtr
|
||||
|
||||
app := AppInit()
|
||||
|
||||
err = app.Init(&AppApi)
|
||||
if err != nil {
|
||||
LOG.Error("初始化应用 %s 失败: %v", pluginPath, err)
|
||||
}
|
||||
|
||||
err = app.InitWSD(&DatabaseApi)
|
||||
if err != nil {
|
||||
LOG.Error("初始化应用 %s 数据库失败: %v", pluginPath, err)
|
||||
}
|
||||
|
||||
CmdMap = mergeMaps(CmdMap, app.Get().CmdMap)
|
||||
LOG.Info("应用 %s 加载成功", pluginPath)
|
||||
return 1, 1
|
||||
|
||||
}
|
||||
if ext == ".js" {
|
||||
pluginPath := filepath.Join(appsDir, file.Name())
|
||||
jsCode, err := os.ReadFile(pluginPath)
|
||||
if err != nil {
|
||||
LOG.Error("读取应用 %s 时发生错误: %v", pluginPath, err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
runtime := goja.New()
|
||||
_, err = runtime.RunString(string(jsCode))
|
||||
if err != nil {
|
||||
LOG.Error("执行应用 %s 失败: %v", pluginPath, err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 创建JS可用的wbaObj对象
|
||||
wbaObj := runtime.NewObject()
|
||||
wsp := runtime.NewObject()
|
||||
wsd := runtime.NewObject()
|
||||
_ = runtime.Set("wba", wbaObj)
|
||||
_ = wbaObj.Set("NewApp", wba.NewApp)
|
||||
_ = wbaObj.Set("NewCmd", wba.NewCmd)
|
||||
_ = wbaObj.Set("NewScheduledTask", wba.NewScheduledTask)
|
||||
_ = wbaObj.Set("WithName", wba.WithName)
|
||||
_ = wbaObj.Set("WithAuthor", wba.WithAuthor)
|
||||
_ = wbaObj.Set("WithVersion", wba.WithVersion)
|
||||
_ = wbaObj.Set("WithDescription", wba.WithDescription)
|
||||
_ = wbaObj.Set("WithWebUrl", wba.WithWebUrl)
|
||||
_ = wbaObj.Set("WithLicense", wba.WithLicense)
|
||||
_ = wbaObj.Set("WithAppType", wba.WithAppType)
|
||||
_ = wbaObj.Set("WithRule", wba.WithRule)
|
||||
_ = wbaObj.Set("WSP", wsp)
|
||||
_ = wbaObj.Set("WSD", wsd)
|
||||
_ = wsp.Set("UnsafelySendMsg", AppApi.UnsafelySendMsg)
|
||||
_ = wsp.Set("UnsafelySendPrivateMsg", AppApi.UnsafelySendPrivateMsg)
|
||||
_ = wsp.Set("UnsafelySendGroupMsg", AppApi.UnsafelySendGroupMsg)
|
||||
_ = wsp.Set("SendMsg", AppApi.SendMsg)
|
||||
_ = wsp.Set("SendPrivateMsg", AppApi.SendPrivateMsg)
|
||||
_ = wsp.Set("SendGroupMsg", AppApi.SendGroupMsg)
|
||||
_ = wsp.Set("UnsafelyDeleteMsg", AppApi.UnsafelyDeleteMsg)
|
||||
_ = wsp.Set("DeleteMsg", AppApi.DeleteMsg)
|
||||
_ = wsp.Set("SendLike", AppApi.SendLike)
|
||||
_ = wsp.Set("SetGroupKick", AppApi.SetGroupKick)
|
||||
_ = wsp.Set("SetGroupBan", AppApi.SetGroupBan)
|
||||
_ = wsp.Set("SetGroupWholeBan", AppApi.SetGroupWholeBan)
|
||||
_ = wsp.Set("SetGroupAdmin", AppApi.SetGroupAdmin)
|
||||
_ = wsp.Set("SetGroupLeave", AppApi.SetGroupLeave)
|
||||
_ = wsp.Set("SetGroupCard", AppApi.SetGroupCard)
|
||||
_ = wsp.Set("SetGroupName", AppApi.SetGroupName)
|
||||
_ = wsp.Set("SetGroupSpecialTitle", AppApi.SetGroupSpecialTitle)
|
||||
_ = wsp.Set("SetFriendAddRequest", AppApi.SetFriendAddRequest)
|
||||
_ = wsp.Set("SetGroupAddRequest", AppApi.SetGroupAddRequest)
|
||||
_ = wsp.Set("GetLoginInfo", AppApi.GetLoginInfo)
|
||||
_ = wsp.Set("GetVersionInfo", AppApi.GetVersionInfo)
|
||||
_ = wsp.Set("GetMsg", AppApi.GetMsg)
|
||||
_ = wsp.Set("GetGroupInfo", AppApi.GetGroupInfo)
|
||||
_ = wsp.Set("GetForwardMsg", AppApi.GetForwardMsg)
|
||||
_ = wsp.Set("GetStrangerInfo", AppApi.GetStrangerInfo)
|
||||
_ = wsp.Set("GetGroupList", AppApi.GetGroupList)
|
||||
_ = wsp.Set("GetGroupMemberList", AppApi.GetGroupMemberList)
|
||||
_ = wsp.Set("GetFriendList", AppApi.GetFriendList)
|
||||
_ = wsp.Set("GetGroupMemberInfo", AppApi.GetGroupMemberInfo)
|
||||
_ = wsp.Set("GetGroupHonorInfo", AppApi.GetGroupHonorInfo)
|
||||
_ = wsp.Set("GetStatus", AppApi.GetStatus)
|
||||
_ = wsp.Set("GetCookies", AppApi.GetCookies)
|
||||
_ = wsp.Set("GetCSRFToken", AppApi.GetCSRFToken)
|
||||
_ = wsp.Set("GetCredentials", AppApi.GetCredentials)
|
||||
_ = wsp.Set("GetImage", AppApi.GetImage)
|
||||
_ = wsp.Set("GetRecord", AppApi.GetRecord)
|
||||
_ = wsp.Set("CanSendImage", AppApi.CanSendImage)
|
||||
_ = wsp.Set("CanSendRecord", AppApi.CanSendRecord)
|
||||
_ = wsp.Set("SetRestart", AppApi.SetRestart)
|
||||
_ = wsp.Set("CleanCache", AppApi.CleanCache)
|
||||
_ = wsp.Set("GetLoginInfo", AppApi.LogWith)
|
||||
_ = wsp.Set("GetVersionInfo", AppApi.GetVersionInfo)
|
||||
_ = wsd.Set("SetUserVariable", DatabaseApi.SetUserVariable)
|
||||
_ = wsd.Set("SetGroupVariable", DatabaseApi.SetGroupVariable)
|
||||
_ = wsd.Set("SetGlobalVariable", DatabaseApi.SetGlobalVariable)
|
||||
_ = wsd.Set("SetOutUserVariable", DatabaseApi.SetOutUserVariable)
|
||||
_ = wsd.Set("SetOutGroupVariable", DatabaseApi.SetOutGroupVariable)
|
||||
_ = wsd.Set("SetOutGlobalVariable", DatabaseApi.SetOutGlobalVariable)
|
||||
_ = wsd.Set("GetUserVariable", DatabaseApi.GetUserVariable)
|
||||
_ = wsd.Set("GetGroupVariable", DatabaseApi.GetGroupVariable)
|
||||
_ = wsd.Set("GetGlobalVariable", DatabaseApi.GetGlobalVariable)
|
||||
_ = wsd.Set("GetOutUserVariable", DatabaseApi.GetOutUserVariable)
|
||||
_ = wsd.Set("GetOutGroupVariable", DatabaseApi.GetOutGroupVariable)
|
||||
_ = wsd.Set("GetOutGlobalVariable", DatabaseApi.GetOutGlobalVariable)
|
||||
_ = wsd.Set("GetIntConfig", DatabaseApi.GetIntConfig)
|
||||
_ = wsd.Set("GetFloatConfig", DatabaseApi.GetFloatConfig)
|
||||
_ = wsd.Set("GetStringConfig", DatabaseApi.GetStringConfig)
|
||||
_ = wsd.Set("GetIntSliceConfig", DatabaseApi.GetIntSliceConfig)
|
||||
_ = wsd.Set("GetStringSliceConfig", DatabaseApi.GetStringSliceConfig)
|
||||
|
||||
// 获取AppInit函数
|
||||
appInitVal := runtime.Get("AppInit")
|
||||
if appInitVal == nil || goja.IsUndefined(appInitVal) {
|
||||
LOG.Error("应用 %s 缺少AppInit函数", pluginPath)
|
||||
return 1, 0
|
||||
}
|
||||
appInitFunc, ok := goja.AssertFunction(appInitVal)
|
||||
if !ok {
|
||||
LOG.Error("应用 %s 的AppInit不是有效函数", pluginPath)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 调用AppInit获取应用实例
|
||||
jsApp, err := appInitFunc(goja.Undefined())
|
||||
if err != nil {
|
||||
LOG.Error("初始化应用实例失败: %v", err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 调用Init方法
|
||||
initVal := jsApp.ToObject(runtime).Get("Init")
|
||||
initFunc, ok := goja.AssertFunction(initVal)
|
||||
if !ok {
|
||||
LOG.Error("应用 %s 缺少有效的Init方法", pluginPath)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
_, err = initFunc(wbaObj)
|
||||
if err != nil {
|
||||
LOG.Trace("应用初始化失败: %v", err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 调用Get方法
|
||||
getVal := jsApp.ToObject(runtime).Get("Get")
|
||||
getFunc, ok := goja.AssertFunction(getVal)
|
||||
if !ok {
|
||||
LOG.Error("应用 %s 缺少有效的Get方法", pluginPath)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
appInfoVal, err := getFunc(jsApp)
|
||||
if err != nil {
|
||||
LOG.Error("获取应用信息失败: %v", err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 转换应用信息
|
||||
var appInfo wba.AppInfo
|
||||
if err := runtime.ExportTo(appInfoVal, &appInfo); err != nil {
|
||||
LOG.Error("应用信息转换失败: %v", err)
|
||||
return 1, 0
|
||||
}
|
||||
|
||||
// 合并命令
|
||||
CmdMap = mergeMaps(CmdMap, appInfo.CmdMap)
|
||||
|
||||
// 注册定时任务
|
||||
for _, task := range appInfo.ScheduledTasks {
|
||||
RegisterCron(appInfo.Name, task)
|
||||
}
|
||||
|
||||
LOG.Info("JS应用 %s 加载成功", pluginPath)
|
||||
return 1, 1
|
||||
}
|
||||
return 0, 0
|
||||
}
|
||||
|
||||
func mergeMaps(map1, map2 map[string]wba.Cmd) map[string]wba.Cmd {
|
||||
// 合并map1和map2到map3中
|
||||
map3 := make(map[string]wba.Cmd)
|
||||
for key, value := range map1 {
|
||||
map3[key] = value
|
||||
}
|
||||
for key, value := range map2 {
|
||||
map3[key] = value
|
||||
}
|
||||
return map3
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user