mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
// Package baidu 百度百科
|
||
package baidu
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"net/url"
|
||
|
||
"github.com/FloatTech/floatbox/web"
|
||
ctrl "github.com/FloatTech/zbpctrl"
|
||
"github.com/FloatTech/zbputils/control"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
)
|
||
|
||
const (
|
||
duURL = "https://api.a20safe.com/api.php?api=21&key=%s&text=%s" // api地址
|
||
wikiURL = "https://api.a20safe.com/api.php?api=23&key=%s&text=%s"
|
||
key = "7d06a110e9e20a684e02934549db1d3d"
|
||
)
|
||
|
||
type result struct {
|
||
Code int `json:"code"`
|
||
Msg string `json:"msg"`
|
||
Data []struct {
|
||
Content string `json:"content"`
|
||
} `json:"data"`
|
||
}
|
||
|
||
func init() { // 主函数
|
||
en := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
||
DisableOnDefault: false,
|
||
Help: "百科\n" +
|
||
"- 百度/百科/维基/wiki[关键字]",
|
||
})
|
||
en.OnRegex(`^(百度|维基|百科|wiki)\s*(.+)$`).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
var es []byte
|
||
var err error
|
||
switch ctx.State["regex_matched"].([]string)[1] {
|
||
case "百度", "百科":
|
||
es, err = web.GetData(fmt.Sprintf(duURL, key, url.QueryEscape(ctx.State["regex_matched"].([]string)[2]))) // 将网站返回结果赋值
|
||
case "wiki", "维基":
|
||
es, err = web.GetData(fmt.Sprintf(wikiURL, key, url.QueryEscape(ctx.State["regex_matched"].([]string)[2]))) // 将网站返回结果赋值
|
||
}
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("出现错误捏：", err))
|
||
return
|
||
}
|
||
var r result // r数组
|
||
err = json.Unmarshal(es, &r) // 填api返回结果，struct地址
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("出现错误捏：", err))
|
||
return
|
||
}
|
||
if r.Code == 0 && len(r.Data) > 0 {
|
||
ctx.SendChain(message.Text(r.Data[0].Content)) // 输出提取后的结果
|
||
} else {
|
||
ctx.SendChain(message.Text("API访问错误"))
|
||
}
|
||
})
|
||
}
|