mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2026-02-13 02:30:26 +00:00
add WarframeAPI (#541)
This commit is contained in:
GenesisAN
507
plugin/warframeapi/main.go
Normal file
507
plugin/warframeapi/main.go
Normal file
@@ -0,0 +1,507 @@
|
||||
// Package warframeapi 百度内容审核
|
||||
package warframeapi
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/FloatTech/floatbox/binary"
|
||||
"github.com/FloatTech/floatbox/web"
|
||||
ctrl "github.com/FloatTech/zbpctrl"
|
||||
"github.com/FloatTech/zbputils/control"
|
||||
"github.com/FloatTech/zbputils/img/text"
|
||||
"github.com/lithammer/fuzzysearch/fuzzy"
|
||||
zero "github.com/wdvxdr1123/ZeroBot"
|
||||
"github.com/wdvxdr1123/ZeroBot/message"
|
||||
"net/http"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
wmitems map[string]items //WarFrame市场的中文名称对应的物品的字典
|
||||
itmeNames []string //物品名称列表
|
||||
rt runtime
|
||||
)
|
||||
|
||||
// 时间同步状态
|
||||
type runtime struct {
|
||||
rwm sync.RWMutex
|
||||
enable bool //是否启动
|
||||
}
|
||||
|
||||
const wfapiurl = "https://api.warframestat.us/pc" //星际战甲API
|
||||
const wfitemurl = "https://api.warframe.market/v1/items" //星际战甲游戏品信息列表URL
|
||||
|
||||
func init() {
|
||||
eng := control.Register("warframeapi", &ctrl.Options[*zero.Ctx]{
|
||||
DisableOnDefault: false,
|
||||
Help: "warframeapi\n" +
|
||||
"- wf时间同步\n" +
|
||||
"- [金星|地球|火卫二]平原时间\n" +
|
||||
"- .wm [物品名称]\n" +
|
||||
"- 仲裁\n" +
|
||||
"- 警报\n" +
|
||||
"- 每日特惠",
|
||||
PrivateDataFolder: "warframeapi",
|
||||
})
|
||||
updateWM()
|
||||
|
||||
//获取具体的平原时间，在触发后，会启动持续时间按5分钟的时间更新模拟，以此处理短时间内请求时，时间不会变化的问题
|
||||
eng.OnSuffix("平原时间").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
if !rt.enable { //没有进行同步,就拉取一次服务器状态
|
||||
wfapi, err := wfapiGetData()
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("Error:获取服务器时间失败"))
|
||||
}
|
||||
loadTime(wfapi)
|
||||
}
|
||||
switch ctx.State["args"].(string) {
|
||||
case "地球", "夜灵":
|
||||
ctx.SendChain(message.Text(gameTimes[0]))
|
||||
case "金星", "奥布山谷":
|
||||
ctx.SendChain(message.Text(gameTimes[1]))
|
||||
case "魔胎之境", "火卫二", "火卫":
|
||||
ctx.SendChain(message.Text(gameTimes[2]))
|
||||
default:
|
||||
ctx.SendChain(message.Text("ERROR: 平原不存在"))
|
||||
}
|
||||
// 是否正在进行同步,没有就开启同步,有就不开启
|
||||
if !rt.enable {
|
||||
// 设置标志位
|
||||
rt.rwm.Lock()
|
||||
if rt.enable { //预检测，防止其他线程同时进来
|
||||
return
|
||||
}
|
||||
rt.enable = true
|
||||
rt.rwm.Unlock()
|
||||
|
||||
go func() {
|
||||
//30*10=300=5分钟
|
||||
for i := 0; i < 30; i++ {
|
||||
time.Sleep(10 * time.Second)
|
||||
timeDet() //5分钟内每隔10秒更新一下时间
|
||||
}
|
||||
//5分钟时间同步结束
|
||||
rt.rwm.Lock()
|
||||
rt.enable = false
|
||||
rt.rwm.Unlock()
|
||||
}()
|
||||
|
||||
}
|
||||
})
|
||||
eng.OnFullMatch("警报").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
wfapi, err := wfapiGetData()
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("ERROR:", err.Error()))
|
||||
return
|
||||
}
|
||||
//如果返回的wfapi中，警报数量>0
|
||||
if len(wfapi.Alerts) > 0 {
|
||||
//遍历警报数据，打印警报信息
|
||||
for _, v := range wfapi.Alerts {
|
||||
//如果警报处于激活状态
|
||||
if v.Active {
|
||||
ctx.SendChain(stringArrayToImage([]string{
|
||||
"节点:" + v.Mission.Node,
|
||||
"类型:" + v.Mission.Type,
|
||||
"敌人Lv:" + fmt.Sprint(v.Mission.MinEnemyLevel) + "~" + fmt.Sprint(v.Mission.MaxEnemyLevel),
|
||||
"奖励:" + v.Mission.Reward.AsString,
|
||||
"剩余时间:" + v.Eta,
|
||||
}))
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
})
|
||||
//TODO:订阅功能-等待重做
|
||||
//eng.OnRegex(`^(订阅|取消订阅)(.*)平原(.*)$`).SetBlock(true).
|
||||
// Handle(func(ctx *zero.Ctx) {
|
||||
// args := ctx.State["regex_matched"].([]string)
|
||||
// var isEnable bool
|
||||
// if args[1] == "订阅" {
|
||||
// isEnable = true
|
||||
// }
|
||||
// updateWFAPI()
|
||||
// status := false
|
||||
// switch args[3] {
|
||||
// case "fass", "白天", "温暖":
|
||||
// status = true
|
||||
// }
|
||||
// switch args[2] {
|
||||
// case "金星", "奥布山谷":
|
||||
// //sublist = append(sublist, subList{ctx.Event.GroupID, ctx.Event.UserID, 1, status, false})
|
||||
// if isEnable {
|
||||
// addUseSub(ctx.Event.UserID, ctx.Event.GroupID, 1, status)
|
||||
// } else {
|
||||
// removeUseSub(ctx.Event.UserID, ctx.Event.GroupID, 1)
|
||||
// }
|
||||
// ctx.SendChain(
|
||||
// message.At(ctx.Event.UserID),
|
||||
// message.Text("已成功", args[1]),
|
||||
// message.Text(gameTimes[1].Name),
|
||||
// message.Text(status),
|
||||
// )
|
||||
// case "地球", "夜灵":
|
||||
// if isEnable {
|
||||
// addUseSub(ctx.Event.UserID, ctx.Event.GroupID, 0, status)
|
||||
// } else {
|
||||
// removeUseSub(ctx.Event.UserID, ctx.Event.GroupID, 0)
|
||||
// }
|
||||
// ctx.SendChain(
|
||||
// message.At(ctx.Event.UserID),
|
||||
// message.Text("已成功", args[1]),
|
||||
// message.Text(gameTimes[0].Name),
|
||||
// message.Text(status),
|
||||
// )
|
||||
// case "魔胎之境", "火卫", "火卫二":
|
||||
// if isEnable {
|
||||
// addUseSub(ctx.Event.UserID, ctx.Event.GroupID, 2, status)
|
||||
// } else {
|
||||
// removeUseSub(ctx.Event.UserID, ctx.Event.GroupID, 2)
|
||||
// }
|
||||
// ctx.SendChain(
|
||||
// message.At(ctx.Event.UserID),
|
||||
// message.Text("已成功", args[1]),
|
||||
// message.Text(gameTimes[2].Name),
|
||||
// message.Text(status),
|
||||
// )
|
||||
// default:
|
||||
// ctx.SendChain(message.Text("ERROR: 平原不存在"))
|
||||
// return
|
||||
// }
|
||||
// })
|
||||
//eng.OnFullMatch(`wf订阅检测`).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||||
// rwm.Lock()
|
||||
// var msg []message.MessageSegment
|
||||
// for i, v := range gameTimes {
|
||||
// nt := time.Until(v.NextTime).Seconds()
|
||||
// switch {
|
||||
// case nt < 0:
|
||||
// if v.Status {
|
||||
// v.NextTime = v.NextTime.Add(time.Duration(v.NightTime) * time.Second)
|
||||
// } else {
|
||||
// v.NextTime = v.NextTime.Add(time.Duration(v.DayTime) * time.Second)
|
||||
// }
|
||||
// v.Status = !v.Status
|
||||
//
|
||||
// msg = callUser(i, v.Status, 0)
|
||||
// case nt < float64(5)*60:
|
||||
// msg = callUser(i, !v.Status, 5)
|
||||
// case nt < float64(15)*60:
|
||||
// if i == 2 && !v.Status {
|
||||
// return
|
||||
// }
|
||||
// msg = callUser(i, !v.Status, 15)
|
||||
// }
|
||||
// }
|
||||
// rwm.Unlock()
|
||||
// if msg != nil && len(msg) > 0 {
|
||||
// ctx.SendChain(msg...)
|
||||
// }
|
||||
//})
|
||||
eng.OnFullMatch("仲裁").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
//通过wfapi获取仲裁信息
|
||||
wfapi, err := wfapiGetData()
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("ERROR:", err.Error()))
|
||||
return
|
||||
}
|
||||
ctx.SendChain(stringArrayToImage([]string{
|
||||
"节点:" + wfapi.Arbitration.Node,
|
||||
"类型:" + wfapi.Arbitration.Type,
|
||||
"阵营:" + wfapi.Arbitration.Enemy,
|
||||
"剩余时间:" + fmt.Sprint(int(wfapi.Arbitration.Expiry.Sub(time.Now().UTC()).Minutes())) + "m",
|
||||
}))
|
||||
})
|
||||
eng.OnFullMatch("每日特惠").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
wfapi, err := wfapiGetData()
|
||||
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("ERROR:", err.Error()))
|
||||
return
|
||||
}
|
||||
for _, dd := range wfapi.DailyDeals {
|
||||
ctx.SendChain(
|
||||
message.Text(
|
||||
"物品:", dd.Item, "\n",
|
||||
"价格:", dd.OriginalPrice, "→", dd.SalePrice, "\n",
|
||||
"数量:(", dd.Total, "/", dd.Sold, ")\n",
|
||||
"时间:", dd.Eta,
|
||||
),
|
||||
)
|
||||
}
|
||||
})
|
||||
// eng.OnRegex(`^入侵$`).SetBlock(true).
|
||||
// Handle(func(ctx *zero.Ctx) {
|
||||
// updateWFAPI(ctx)
|
||||
// for _, dd := range wfapi.dailyDeals {
|
||||
// imagebuild.DrawTextSend([]string{
|
||||
// "节点:" + wfapi.arbitration.Node,
|
||||
// "类型:" + wfapi.arbitration.Type,
|
||||
// "阵营:" + wfapi.arbitration.Enemy,
|
||||
// "剩余时间:" + fmt.Sprint(int(wfapi.arbitration.Expiry.Sub(time.Now().UTC()).Minutes())) + "m",
|
||||
// }, ctx)
|
||||
// }
|
||||
// })
|
||||
eng.OnFullMatch("wf时间同步").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
wfapi, err := wfapiGetData()
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("ERROR:", err.Error()))
|
||||
return
|
||||
}
|
||||
loadTime(wfapi)
|
||||
ctx.SendChain(message.Text("已拉取服务器时间并同步到本地模拟"))
|
||||
})
|
||||
// 根据名称从Warframe市场查询物品售价
|
||||
eng.OnPrefix(".wm ").SetBlock(true).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
//根据输入的名称，从游戏物品名称列表中进行模糊搜索
|
||||
sol := fuzzy.FindNormalizedFold(ctx.State["args"].(string), itmeNames)
|
||||
var msg []string
|
||||
//物品名称
|
||||
var name string
|
||||
|
||||
//根据搜搜结果，打印找到的物品
|
||||
switch len(sol) {
|
||||
case 0: //没有搜索到任何东西
|
||||
ctx.SendChain(message.Text("无法查询到该物品"))
|
||||
return
|
||||
case 1: //如果只搜索到了一个
|
||||
name = sol[0]
|
||||
default: //如果搜搜到了多个
|
||||
//遍历搜索结果，并打印为图片展出
|
||||
for i, v := range sol {
|
||||
msg = append(msg, fmt.Sprintf("[%d] %s", i, v))
|
||||
}
|
||||
msg = append(msg, "包含多个结果，请输入编号查看(15s内),输入c直接结束会话")
|
||||
ctx.SendChain(stringArrayToImage(msg))
|
||||
msg = []string{}
|
||||
|
||||
itemIndex := itemNameFutureEvent(ctx, 2)
|
||||
if itemIndex == -1 {
|
||||
return
|
||||
}
|
||||
name = sol[itemIndex]
|
||||
}
|
||||
Mf := false
|
||||
GETWM:
|
||||
if Mf {
|
||||
msg = []string{}
|
||||
}
|
||||
sells, itmeinfo, txt, err := wmItemOrders(wmitems[name].URLName, Mf)
|
||||
if !Mf {
|
||||
if itmeinfo.ZhHans.WikiLink == "" {
|
||||
ctx.Send([]message.MessageSegment{
|
||||
message.Image("https://warframe.market/static/assets/" + wmitems[name].Thumb),
|
||||
message.Text(wmitems[name].ItemName, "\n"),
|
||||
})
|
||||
} else {
|
||||
ctx.Send([]message.MessageSegment{
|
||||
message.Image("https://warframe.market/static/assets/" + wmitems[name].Thumb),
|
||||
message.Text(wmitems[name].ItemName, "\n"),
|
||||
message.Text("wiki:", itmeinfo.ZhHans.WikiLink),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
msg = append(msg, wmitems[name].ItemName)
|
||||
|
||||
if err != nil {
|
||||
ctx.Send(message.Text("Error:", err.Error()))
|
||||
return
|
||||
}
|
||||
if sells == nil {
|
||||
ctx.Send(message.Text("无可购买对象"))
|
||||
return
|
||||
}
|
||||
|
||||
ismod := false
|
||||
if itmeinfo.ModMaxRank != 0 {
|
||||
ismod = true
|
||||
}
|
||||
|
||||
max := 5
|
||||
if len(sells) <= max {
|
||||
max = len(sells)
|
||||
}
|
||||
for i := 0; i < max; i++ {
|
||||
if ismod {
|
||||
msg = append(msg, fmt.Sprintf("[%d](Rank:%d/%d) %dP - %s\n", i, sells[i].ModRank, itmeinfo.ModMaxRank, sells[i].Platinum, sells[i].User.IngameName))
|
||||
} else {
|
||||
msg = append(msg, fmt.Sprintf("[%d] %dP -%s\n", i, sells[i].Platinum, sells[i].User.IngameName))
|
||||
}
|
||||
}
|
||||
|
||||
if ismod && !Mf {
|
||||
msg = append(msg, "请输入编号选择，或输入r获取满级报价(30s内)\n输入c直接结束会话")
|
||||
} else {
|
||||
msg = append(msg, "请输入编号选择(30s内)\n输入c直接结束会话")
|
||||
}
|
||||
ctx.SendChain(stringArrayToImage(msg))
|
||||
|
||||
GETNUM3:
|
||||
next := zero.NewFutureEvent("message", 999, false, ctx.CheckSession()).Next()
|
||||
select {
|
||||
case <-time.After(time.Second * 30):
|
||||
ctx.SendChain(message.Text("会话已结束!"))
|
||||
return
|
||||
case e := <-next:
|
||||
msg := e.Event.Message.ExtractPlainText()
|
||||
// 重新获取报价
|
||||
if msg == "r" {
|
||||
Mf = true
|
||||
goto GETWM
|
||||
}
|
||||
// 主动结束会话
|
||||
if msg == "c" {
|
||||
ctx.SendChain(message.Text("会话已结束!"))
|
||||
return
|
||||
}
|
||||
i, err := strconv.Atoi(msg)
|
||||
if err != nil {
|
||||
ctx.SendChain(message.Text("请输入数字!(输入c结束会话)"))
|
||||
goto GETNUM3
|
||||
}
|
||||
if err == nil {
|
||||
if ismod {
|
||||
ctx.Send(message.Text("/w ", sells[i].User.IngameName, " Hi! I want to buy: ", txt, "(Rank:", sells[i].ModRank, ") for ", sells[i].Platinum, " platinum. (warframe.market)"))
|
||||
} else {
|
||||
ctx.Send(message.Text("/w ", sells[i].User.IngameName, " Hi! I want to buy: ", txt, " for ", sells[i].Platinum, " platinum. (warframe.market)"))
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// 获取搜索结果中的物品具体名称index的FutureEvent,传入ctx和一个递归次数上限,返回一个int，如果为返回内容为-1，说明会话超时，或主动结束，或超出递归
|
||||
func itemNameFutureEvent(ctx *zero.Ctx, count int) int {
|
||||
next := zero.NewFutureEvent("message", 999, false, ctx.CheckSession()).Next()
|
||||
select {
|
||||
case <-time.After(time.Second * 15):
|
||||
//超时15秒处理
|
||||
ctx.SendChain(message.Text("会话已超时!"))
|
||||
return -1
|
||||
case e := <-next:
|
||||
msg := e.Event.Message.ExtractPlainText()
|
||||
//输入c主动结束的处理
|
||||
if msg == "c" {
|
||||
ctx.SendChain(message.Text("会话已结束!"))
|
||||
return -1
|
||||
}
|
||||
//尝试对输入进行数字转换
|
||||
num, err := strconv.Atoi(msg)
|
||||
//如果出错，说明输入的并非数字，则重新触发该内容
|
||||
if err != nil {
|
||||
//查看是否超时
|
||||
if count == 0 {
|
||||
ctx.SendChain(message.Text("连续输入错误，会话已结束!"))
|
||||
return -1
|
||||
}
|
||||
ctx.SendChain(message.Text("请输入数字!(输入c结束会话)[", count, "]"))
|
||||
count--
|
||||
return itemNameFutureEvent(ctx, count)
|
||||
|
||||
}
|
||||
return num
|
||||
}
|
||||
}
|
||||
|
||||
// 数组字符串转图片
|
||||
func stringArrayToImage(texts []string) message.MessageSegment {
|
||||
b, err := text.RenderToBase64(strings.Join(texts, "\n"), text.FontFile, 400, 20)
|
||||
if err != nil {
|
||||
return message.Text("ERROR: ", err)
|
||||
}
|
||||
return message.Image("base64://" + binary.BytesToString(b))
|
||||
}
|
||||
|
||||
// 从WFapi获取数据
|
||||
func wfapiGetData() (wfAPI, error) {
|
||||
var wfapi wfAPI //WarFrameAPI的数据实例
|
||||
var data []byte
|
||||
var err error
|
||||
data, err = web.GetData(wfapiurl)
|
||||
if err != nil {
|
||||
return wfapi, err
|
||||
}
|
||||
err = json.Unmarshal(data, &wfapi)
|
||||
if err != nil {
|
||||
return wfapi, err
|
||||
}
|
||||
return wfapi, nil
|
||||
}
|
||||
|
||||
// 从WF市场获取物品数据信息
|
||||
func updateWM() {
|
||||
var itmeapi wfAPIItem //WarFrame市场的数据实例
|
||||
|
||||
data, err := web.RequestDataWithHeaders(&http.Client{}, wfitemurl, "GET", func(request *http.Request) error {
|
||||
request.Header.Add("Accept", "application/json")
|
||||
request.Header.Add("Language", "zh-hans")
|
||||
return nil
|
||||
}, nil)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = json.Unmarshal(data, &itmeapi)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
loadToFuzzy(itmeapi)
|
||||
}
|
||||
|
||||
// 获取Warframe市场的售价表，并进行排序,cn_name为物品中文名称，onlyMaxRank表示只取最高等级的物品，返回物品售价表，物品信息，物品英文
|
||||
func wmItemOrders(cnName string, onlyMaxRank bool) (orders, itemsInSet, string, error) {
|
||||
|
||||
var wfapiio wfAPIItemsOrders
|
||||
data, err := web.RequestDataWithHeaders(&http.Client{}, fmt.Sprintf("https://api.warframe.market/v1/items/%s/orders?include=item", cnName), "GET", func(request *http.Request) error {
|
||||
request.Header.Add("Accept", "application/json")
|
||||
request.Header.Add("Platform", "pc")
|
||||
return nil
|
||||
}, nil)
|
||||
if err != nil {
|
||||
return nil, itemsInSet{}, "", err
|
||||
}
|
||||
err = json.Unmarshal(data, &wfapiio)
|
||||
var sellOrders orders
|
||||
//遍历市场物品列表
|
||||
for _, v := range wfapiio.Payload.Orders {
|
||||
//取其中类型为售卖，且去掉不在线的玩家
|
||||
if v.OrderType == "sell" && v.User.Status != "offline" {
|
||||
//如果需要满级报价
|
||||
if onlyMaxRank && v.ModRank == wfapiio.Include.Item.ItemsInSet[0].ModMaxRank {
|
||||
sellOrders = append(sellOrders, v)
|
||||
} else if !onlyMaxRank {
|
||||
sellOrders = append(sellOrders, v)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
//对报价表进行排序，由低到高
|
||||
sort.Sort(sellOrders)
|
||||
//获取物品信息
|
||||
for i, v := range wfapiio.Include.Item.ItemsInSet {
|
||||
if v.URLName == cnName {
|
||||
return sellOrders, wfapiio.Include.Item.ItemsInSet[i], wfapiio.Include.Item.ItemsInSet[i].En.ItemName, err
|
||||
}
|
||||
}
|
||||
return sellOrders, wfapiio.Include.Item.ItemsInSet[0], wfapiio.Include.Item.ItemsInSet[0].En.ItemName, err
|
||||
}
|
||||
|
||||
func loadToFuzzy(wminfo wfAPIItem) {
|
||||
wmitems = make(map[string]items)
|
||||
itmeNames = []string{}
|
||||
for _, v := range wminfo.Payload.Items {
|
||||
wmitems[v.ItemName] = v
|
||||
itmeNames = append(itmeNames, v.ItemName)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user