Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-02-12 10:10:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🎨 调整代码结构

Browse Source
This commit is contained in:
Yiwen-Chan
2021-04-01 14:25:54 +08:00
parent 14d07bf963
commit 9a47bc93b0
10 changed files with 75 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
music/music_selecter.go
View File

@@ -2,30 +2,77 @@ package setutime
import (
"fmt"
"io/ioutil"
"net/http"
"net/url"
"strings"
"time"
"bot/music/utils"
"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) {
music, err := utils.CloudMusic(ctx.State["regex_matched"].([]string)[1])
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]",
music.Type,
music.Url,
music.Audio,
music.Title,
music.Content,
music.Image,
"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

57
music/utils/cloud_music.go
View File

@@ -1,57 +0,0 @@
package utils
import (
"errors"
"fmt"
"io/ioutil"
"net/http"
"net/url"
"strings"
"github.com/tidwall/gjson"
)
func CloudMusic(name string) (music CQMusic, err error) {
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", name)
fromData := strings.NewReader(data.Encode())
// 网络请求
req, err := http.NewRequest("POST", api, fromData)
if err != nil {
return music, err
}
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 {
return music, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return music, err
}
if code := resp.StatusCode; code != 200 {
// 如果返回不是200则立刻抛出错误
return music, errors.New(fmt.Sprintf("CloudMusic not found, code %d", code))
}
content := gjson.ParseBytes(body).Get("result.songs.0")
music.Type = "custom"
music.Url = fmt.Sprintf("http://y.music.163.com/m/song?id=%d", content.Get("id").Int())
music.Audio = fmt.Sprintf("http://music.163.com/song/media/outer/url?id=%d.mp3", content.Get("id").Int())
music.Title = content.Get("name").Str
music.Content = content.Get("artists.0.name").Str
music.Image = content.Get("album.blurPicUrl").Str
return music, nil
}

10
music/utils/utils.go
View File

@@ -1,10 +0,0 @@
package utils
type CQMusic struct {
Type string
Url string
Audio string
Title string
Content string
Image string
}