Northern_Lights/ZeroBot-Plugin
0
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-03-01 18:40:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化代码结构

Browse Source
This commit is contained in:
源文雨
2023-03-04 23:30:51 +08:00
parent d41ae01f01
commit 5226548cec
3 changed files with 41 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
plugin/warframeapi/main.go
View File

@@ -5,26 +5,17 @@ import (
"fmt"
"strconv"
"strings"
"sync"
"time"
ctrl "github.com/FloatTech/zbpctrl"
"github.com/FloatTech/zbputils/control"
"github.com/FloatTech/zbputils/ctxext"
"github.com/RomiChan/syncx"
"github.com/lithammer/fuzzysearch/fuzzy"
zero "github.com/wdvxdr1123/ZeroBot"
"github.com/wdvxdr1123/ZeroBot/message"
)
var wmdr sync.RWMutex
type wmdata struct {
wmitems map[string]items
itemNames []string
}
var wd, _ = newwm()
func init() {
eng := control.Register("warframeapi", &ctrl.Options[*zero.Ctx]{
DisableOnDefault: false,
@@ -242,12 +233,24 @@ func init() {
ctx.SendChain(message.Text("已拉取服务器时间并同步到本地模拟"))
})
// 根据名称从Warframe市场查询物品售价
eng.OnPrefix(".wm ", checknwm).SetBlock(true).
eng.OnPrefix(".wm ", func(ctx *zero.Ctx) bool {
if wd.Get().wmitems == nil || wd.Get().itemNames == nil {
if wderr != nil { // 获取失败
ctx.SendChain(message.Text("ERROR: 获取Warframe市场物品列表失败: ", wderr))
} else {
ctx.SendChain(message.Text("ERROR: Warframe市场物品列表为空!"))
}
wd = syncx.Lazy[*wmdata]{Init: func() (d *wmdata) {
d, wderr = newwm()
return
}}
return false
}
return true
}).SetBlock(true).
Handle(func(ctx *zero.Ctx) {
// 根据输入的名称, 从游戏物品名称列表中进行模糊搜索
wmdr.RLock()
sol := fuzzy.FindNormalizedFold(ctx.State["args"].(string), wd.itemNames)
wmdr.RUnlock()
sol := fuzzy.FindNormalizedFold(ctx.State["args"].(string), wd.Get().itemNames)
// 物品名称
var name string
@@ -293,19 +296,17 @@ func init() {
msgs = msgs[:0]
}
sells, iteminfo, txt, err := getitemsorder(wd.wmitems[name].URLName, onlymaxrank)
sells, iteminfo, txt, err := getitemsorder(wd.Get().wmitems[name].URLName, onlymaxrank)
if !onlymaxrank {
wmdr.RLock()
if iteminfo.ZhHans.WikiLink == "" {
msgs = append(msgs, ctxext.FakeSenderForwardNode(ctx,
message.Image("https://warframe.market/static/assets/"+wd.wmitems[name].Thumb),
message.Text("\n", wd.wmitems[name].ItemName)))
message.Image("https://warframe.market/static/assets/"+wd.Get().wmitems[name].Thumb),
message.Text("\n", wd.Get().wmitems[name].ItemName)))
} else {
msgs = append(msgs, ctxext.FakeSenderForwardNode(ctx,
message.Image("https://warframe.market/static/assets/"+wd.wmitems[name].Thumb),
message.Text("\n", wd.wmitems[name].ItemName, "\nwiki: ", iteminfo.ZhHans.WikiLink)))
message.Image("https://warframe.market/static/assets/"+wd.Get().wmitems[name].Thumb),
message.Text("\n", wd.Get().wmitems[name].ItemName, "\nwiki: ", iteminfo.ZhHans.WikiLink)))
}
wmdr.RUnlock()
}
if err != nil {