mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-25 19:20:05 +08:00
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package setutime
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/tidwall/gjson"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
|
)
|
|
|
|
var limit = rate.NewManager(time.Minute*1, 3)
|
|
|
|
func init() { // 插件主体
|
|
zero.OnRegex(`^点歌(.*)$`).SetBlock(true).SetPriority(50).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
if limit.Load(ctx.Event.UserID).Acquire() == false {
|
|
ctx.Send("请稍后重试0x0...")
|
|
return
|
|
}
|
|
// 调用网易云 API
|
|
var api = "http://music.163.com/api/search/pc"
|
|
|
|
client := &http.Client{}
|
|
|
|
// 包装请求参数
|
|
data := url.Values{}
|
|
data.Set("offset", "0")
|
|
data.Set("total", "true")
|
|
data.Set("limit", "9")
|
|
data.Set("type", "1")
|
|
data.Set("s", ctx.State["regex_matched"].([]string)[1])
|
|
fromData := strings.NewReader(data.Encode())
|
|
|
|
// 网络请求
|
|
req, err := http.NewRequest("POST", api, fromData)
|
|
if err != nil {
|
|
ctx.Send(fmt.Sprintf("ERROR: %v", err))
|
|
return
|
|
}
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.182 Safari/537.36")
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
ctx.Send(fmt.Sprintf("ERROR: %v", err))
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
ctx.Send(fmt.Sprintf("ERROR: %v", err))
|
|
return
|
|
}
|
|
|
|
if code := resp.StatusCode; code != 200 {
|
|
// 如果返回不是200则立刻抛出错误
|
|
ctx.Send(fmt.Sprintf("ERROR: code %d", code))
|
|
return
|
|
}
|
|
content := gjson.ParseBytes(body).Get("result.songs.0")
|
|
|
|
// 发送搜索结果
|
|
ctx.Send(
|
|
fmt.Sprintf(
|
|
"[CQ:music,type=%s,url=%s,audio=%s,title=%s,content=%s,image=%s]",
|
|
"custom",
|
|
fmt.Sprintf("http://y.music.163.com/m/song?id=%d", content.Get("id").Int()),
|
|
fmt.Sprintf("http://music.163.com/song/media/outer/url?id=%d.mp3", content.Get("id").Int()),
|
|
content.Get("name").Str,
|
|
content.Get("artists.0.name").Str,
|
|
content.Get("album.blurPicUrl").Str,
|
|
),
|
|
)
|
|
return
|
|
})
|
|
}
|