mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
🎨 改进代码样式
This commit is contained in:
github-actions[bot]
parent
477ffa825b
commit
15a5b347e8
@ -1,475 +1,475 @@
|
|||||||
// Package guessmusic 基于zbp的猜歌插件
|
// Package guessmusic 基于zbp的猜歌插件
|
||||||
package guessmusic
|
package guessmusic
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
zero "github.com/wdvxdr1123/ZeroBot"
|
zero "github.com/wdvxdr1123/ZeroBot"
|
||||||
"github.com/wdvxdr1123/ZeroBot/message"
|
"github.com/wdvxdr1123/ZeroBot/message"
|
||||||
|
|
||||||
ctrl "github.com/FloatTech/zbpctrl"
|
ctrl "github.com/FloatTech/zbpctrl"
|
||||||
"github.com/FloatTech/zbputils/binary"
|
"github.com/FloatTech/zbputils/binary"
|
||||||
control "github.com/FloatTech/zbputils/control"
|
control "github.com/FloatTech/zbputils/control"
|
||||||
"github.com/FloatTech/zbputils/ctxext"
|
"github.com/FloatTech/zbputils/ctxext"
|
||||||
"github.com/FloatTech/zbputils/file"
|
"github.com/FloatTech/zbputils/file"
|
||||||
"github.com/FloatTech/zbputils/web"
|
"github.com/FloatTech/zbputils/web"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/tidwall/gjson"
|
"github.com/tidwall/gjson"
|
||||||
"github.com/wdvxdr1123/ZeroBot/extension/single"
|
"github.com/wdvxdr1123/ZeroBot/extension/single"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36 Edg/87.0.664.66"
|
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36 Edg/87.0.664.66"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
musicPath = file.BOTPATH + "/data/guessmusic/music/" // 绝对路径，歌库根目录,通过指令进行更改
|
musicPath = file.BOTPATH + "/data/guessmusic/music/" // 绝对路径，歌库根目录,通过指令进行更改
|
||||||
cuttime = [...]string{"00:00:05", "00:00:30", "00:01:00"} // 音乐切割时间点，可自行调节时间（时：分：秒）
|
cuttime = [...]string{"00:00:05", "00:00:30", "00:01:00"} // 音乐切割时间点，可自行调节时间（时：分：秒）
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() { // 插件主体
|
func init() { // 插件主体
|
||||||
engine := control.Register("guessmusic", &ctrl.Options[*zero.Ctx]{
|
engine := control.Register("guessmusic", &ctrl.Options[*zero.Ctx]{
|
||||||
DisableOnDefault: false,
|
DisableOnDefault: false,
|
||||||
Help: "猜歌插件（该插件依赖ffmpeg）\n" +
|
Help: "猜歌插件（该插件依赖ffmpeg）\n" +
|
||||||
"- 个人猜歌\n" +
|
"- 个人猜歌\n" +
|
||||||
"- 团队猜歌\n" +
|
"- 团队猜歌\n" +
|
||||||
"- 设置缓存歌库路径 [绝对路径]\n" +
|
"- 设置缓存歌库路径 [绝对路径]\n" +
|
||||||
"注：默认歌库为网易云热歌榜\n" +
|
"注：默认歌库为网易云热歌榜\n" +
|
||||||
"1.可在后面添加“-动漫”进行动漫歌猜歌\n-这个只能猜歌名和歌手\n" +
|
"1.可在后面添加“-动漫”进行动漫歌猜歌\n-这个只能猜歌名和歌手\n" +
|
||||||
"2.可在后面添加“-动漫2”进行动漫歌猜歌\n-这个可以猜番名，但歌手经常“未知”",
|
"2.可在后面添加“-动漫2”进行动漫歌猜歌\n-这个可以猜番名，但歌手经常“未知”",
|
||||||
PrivateDataFolder: "guessmusic",
|
PrivateDataFolder: "guessmusic",
|
||||||
}).ApplySingle(single.New(
|
}).ApplySingle(single.New(
|
||||||
single.WithKeyFn(func(ctx *zero.Ctx) int64 { return ctx.Event.GroupID }),
|
single.WithKeyFn(func(ctx *zero.Ctx) int64 { return ctx.Event.GroupID }),
|
||||||
single.WithPostFn[int64](func(ctx *zero.Ctx) {
|
single.WithPostFn[int64](func(ctx *zero.Ctx) {
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(ctx.Event.MessageID,
|
message.ReplyWithMessage(ctx.Event.MessageID,
|
||||||
message.Text("已经有正在进行的游戏..."),
|
message.Text("已经有正在进行的游戏..."),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
}),
|
}),
|
||||||
))
|
))
|
||||||
cachePath := engine.DataFolder() + "cache/"
|
cachePath := engine.DataFolder() + "cache/"
|
||||||
err := os.MkdirAll(cachePath, 0755)
|
err := os.MkdirAll(cachePath, 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
cfgfile := engine.DataFolder() + "setpath.txt"
|
cfgfile := engine.DataFolder() + "setpath.txt"
|
||||||
if file.IsExist(cfgfile) {
|
if file.IsExist(cfgfile) {
|
||||||
b, err := os.ReadFile(cfgfile)
|
b, err := os.ReadFile(cfgfile)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
musicPath = binary.BytesToString(b)
|
musicPath = binary.BytesToString(b)
|
||||||
logrus.Infoln("[guessmusic] set dir to", musicPath)
|
logrus.Infoln("[guessmusic] set dir to", musicPath)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
engine.OnRegex(`^设置缓存歌库路径(.*)$`, func(ctx *zero.Ctx) bool {
|
engine.OnRegex(`^设置缓存歌库路径(.*)$`, func(ctx *zero.Ctx) bool {
|
||||||
if !zero.SuperUserPermission(ctx) {
|
if !zero.SuperUserPermission(ctx) {
|
||||||
ctx.SendChain(message.Text("只有bot主人可以设置！"))
|
ctx.SendChain(message.Text("只有bot主人可以设置！"))
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return true
|
return true
|
||||||
}).SetBlock(true).
|
}).SetBlock(true).
|
||||||
Handle(func(ctx *zero.Ctx) {
|
Handle(func(ctx *zero.Ctx) {
|
||||||
musicPath = ctx.State["regex_matched"].([]string)[1]
|
musicPath = ctx.State["regex_matched"].([]string)[1]
|
||||||
if musicPath == "" {
|
if musicPath == "" {
|
||||||
ctx.SendChain(message.Text("请输入正确的路径!"))
|
ctx.SendChain(message.Text("请输入正确的路径!"))
|
||||||
}
|
}
|
||||||
musicPath = strings.ReplaceAll(musicPath, "\\", "/")
|
musicPath = strings.ReplaceAll(musicPath, "\\", "/")
|
||||||
if !strings.HasSuffix(musicPath, "/") {
|
if !strings.HasSuffix(musicPath, "/") {
|
||||||
musicPath += "/"
|
musicPath += "/"
|
||||||
}
|
}
|
||||||
err := os.WriteFile(cfgfile, binary.StringToBytes(musicPath), 0644)
|
err := os.WriteFile(cfgfile, binary.StringToBytes(musicPath), 0644)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ctx.SendChain(message.Text("成功！"))
|
ctx.SendChain(message.Text("成功！"))
|
||||||
} else {
|
} else {
|
||||||
ctx.SendChain(message.Text("ERROR:", err))
|
ctx.SendChain(message.Text("ERROR:", err))
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
engine.OnRegex(`^(个人|团队)猜歌(-动漫|-动漫2)?$`, zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByGroup).
|
engine.OnRegex(`^(个人|团队)猜歌(-动漫|-动漫2)?$`, zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByGroup).
|
||||||
Handle(func(ctx *zero.Ctx) {
|
Handle(func(ctx *zero.Ctx) {
|
||||||
mode := ctx.State["regex_matched"].([]string)[2]
|
mode := ctx.State["regex_matched"].([]string)[2]
|
||||||
gid := strconv.FormatInt(ctx.Event.GroupID, 10)
|
gid := strconv.FormatInt(ctx.Event.GroupID, 10)
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
ctx.SendChain(message.Text("正在准备歌曲,请稍等\n回答“-[歌曲名称|歌手|番剧|提示|取消]”\n一共3段语音，6次机会"))
|
ctx.SendChain(message.Text("正在准备歌曲,请稍等\n回答“-[歌曲名称|歌手|番剧|提示|取消]”\n一共3段语音，6次机会"))
|
||||||
} else {
|
} else {
|
||||||
ctx.SendChain(message.Text("正在准备歌曲,请稍等\n回答“-[歌曲名称|歌手|提示|取消]”\n一共3段语音，6次机会"))
|
ctx.SendChain(message.Text("正在准备歌曲,请稍等\n回答“-[歌曲名称|歌手|提示|取消]”\n一共3段语音，6次机会"))
|
||||||
}
|
}
|
||||||
// 随机抽歌
|
// 随机抽歌
|
||||||
musicname, pathofmusic, err := musiclottery(mode, musicPath)
|
musicname, pathofmusic, err := musiclottery(mode, musicPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.SendChain(message.Text(err))
|
ctx.SendChain(message.Text(err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 切割音频，生成3个10秒的音频
|
// 切割音频，生成3个10秒的音频
|
||||||
outputPath := cachePath + gid + "/"
|
outputPath := cachePath + gid + "/"
|
||||||
err = musiccut(musicname, pathofmusic, outputPath)
|
err = musiccut(musicname, pathofmusic, outputPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.SendChain(message.Text(err))
|
ctx.SendChain(message.Text(err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 进行猜歌环节
|
// 进行猜歌环节
|
||||||
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + "0.wav"))
|
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + "0.wav"))
|
||||||
answerstring := strings.Split(musicname, " - ")
|
answerstring := strings.Split(musicname, " - ")
|
||||||
var next *zero.FutureEvent
|
var next *zero.FutureEvent
|
||||||
if ctx.State["regex_matched"].([]string)[1] == "个人" {
|
if ctx.State["regex_matched"].([]string)[1] == "个人" {
|
||||||
next = zero.NewFutureEvent("message", 999, false, zero.OnlyGroup, zero.RegexRule(`^-\S{1,}`), ctx.CheckSession())
|
next = zero.NewFutureEvent("message", 999, false, zero.OnlyGroup, zero.RegexRule(`^-\S{1,}`), ctx.CheckSession())
|
||||||
} else {
|
} else {
|
||||||
next = zero.NewFutureEvent("message", 999, false, zero.OnlyGroup, zero.RegexRule(`^-\S{1,}`), zero.CheckGroup(ctx.Event.GroupID))
|
next = zero.NewFutureEvent("message", 999, false, zero.OnlyGroup, zero.RegexRule(`^-\S{1,}`), zero.CheckGroup(ctx.Event.GroupID))
|
||||||
}
|
}
|
||||||
var countofmusic = 0 // 音频数量
|
var countofmusic = 0 // 音频数量
|
||||||
var countofanswer = 0 // 问答次数
|
var countofanswer = 0 // 问答次数
|
||||||
recv, cancel := next.Repeat()
|
recv, cancel := next.Repeat()
|
||||||
defer cancel()
|
defer cancel()
|
||||||
wait := time.NewTimer(40 * time.Second)
|
wait := time.NewTimer(40 * time.Second)
|
||||||
tick := time.NewTimer(105 * time.Second)
|
tick := time.NewTimer(105 * time.Second)
|
||||||
after := time.NewTimer(120 * time.Second)
|
after := time.NewTimer(120 * time.Second)
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-tick.C:
|
case <-tick.C:
|
||||||
ctx.SendChain(message.Text("猜歌游戏，你还有15s作答时间"))
|
ctx.SendChain(message.Text("猜歌游戏，你还有15s作答时间"))
|
||||||
case <-after.C:
|
case <-after.C:
|
||||||
msg := make(message.Message, 0, 3)
|
msg := make(message.Message, 0, 3)
|
||||||
msg = append(msg, message.Reply(ctx.Event.MessageID))
|
msg = append(msg, message.Reply(ctx.Event.MessageID))
|
||||||
msg = append(msg, message.Text("猜歌超时，游戏结束\n答案是:",
|
msg = append(msg, message.Text("猜歌超时，游戏结束\n答案是:",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1]))
|
"\n歌手:", answerstring[1]))
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
||||||
}
|
}
|
||||||
ctx.Send(msg)
|
ctx.Send(msg)
|
||||||
return
|
return
|
||||||
case <-wait.C:
|
case <-wait.C:
|
||||||
wait.Reset(40 * time.Second)
|
wait.Reset(40 * time.Second)
|
||||||
countofmusic++
|
countofmusic++
|
||||||
if countofmusic > 2 {
|
if countofmusic > 2 {
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
ctx.SendChain(
|
ctx.SendChain(
|
||||||
message.Text("好像有些难度呢，再听这段音频，要仔细听哦"),
|
message.Text("好像有些难度呢，再听这段音频，要仔细听哦"),
|
||||||
)
|
)
|
||||||
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
||||||
case c := <-recv:
|
case c := <-recv:
|
||||||
wait.Reset(40 * time.Second)
|
wait.Reset(40 * time.Second)
|
||||||
tick.Reset(105 * time.Second)
|
tick.Reset(105 * time.Second)
|
||||||
after.Reset(120 * time.Second)
|
after.Reset(120 * time.Second)
|
||||||
answer := strings.Replace(c.Event.Message.String(), "-", "", 1)
|
answer := strings.Replace(c.Event.Message.String(), "-", "", 1)
|
||||||
switch {
|
switch {
|
||||||
case answer == "取消":
|
case answer == "取消":
|
||||||
if c.Event.UserID == ctx.Event.UserID {
|
if c.Event.UserID == ctx.Event.UserID {
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
after.Stop()
|
after.Stop()
|
||||||
msg := make(message.Message, 0, 3)
|
msg := make(message.Message, 0, 3)
|
||||||
msg = append(msg, message.Reply(c.Event.MessageID))
|
msg = append(msg, message.Reply(c.Event.MessageID))
|
||||||
msg = append(msg, message.Text("游戏已取消，猜歌答案是",
|
msg = append(msg, message.Text("游戏已取消，猜歌答案是",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1]))
|
"\n歌手:", answerstring[1]))
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
||||||
}
|
}
|
||||||
ctx.Send(msg)
|
ctx.Send(msg)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("你无权限取消"),
|
message.Text("你无权限取消"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
case answer == "提示":
|
case answer == "提示":
|
||||||
countofmusic++
|
countofmusic++
|
||||||
if countofmusic > 2 {
|
if countofmusic > 2 {
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("已经没有提示了哦"),
|
message.Text("已经没有提示了哦"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
wait.Reset(40 * time.Second)
|
wait.Reset(40 * time.Second)
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("再听这段音频，要仔细听哦"),
|
message.Text("再听这段音频，要仔细听哦"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
||||||
case strings.Contains(answerstring[0], answer) || strings.EqualFold(answerstring[0], answer):
|
case strings.Contains(answerstring[0], answer) || strings.EqualFold(answerstring[0], answer):
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
after.Stop()
|
after.Stop()
|
||||||
msg := make(message.Message, 0, 3)
|
msg := make(message.Message, 0, 3)
|
||||||
msg = append(msg, message.Reply(c.Event.MessageID))
|
msg = append(msg, message.Reply(c.Event.MessageID))
|
||||||
msg = append(msg, message.Text("太棒了，你猜对歌曲名了！答案是",
|
msg = append(msg, message.Text("太棒了，你猜对歌曲名了！答案是",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1]))
|
"\n歌手:", answerstring[1]))
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
||||||
}
|
}
|
||||||
ctx.Send(msg)
|
ctx.Send(msg)
|
||||||
return
|
return
|
||||||
case answerstring[1] == "未知" && answer == "未知":
|
case answerstring[1] == "未知" && answer == "未知":
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("该模式禁止回答“未知”"),
|
message.Text("该模式禁止回答“未知”"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
case strings.Contains(answerstring[1], answer) || strings.EqualFold(answerstring[1], answer):
|
case strings.Contains(answerstring[1], answer) || strings.EqualFold(answerstring[1], answer):
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
after.Stop()
|
after.Stop()
|
||||||
msg := make(message.Message, 0, 3)
|
msg := make(message.Message, 0, 3)
|
||||||
msg = append(msg, message.Reply(c.Event.MessageID))
|
msg = append(msg, message.Reply(c.Event.MessageID))
|
||||||
msg = append(msg, message.Text("太棒了，你猜对歌手名了！答案是",
|
msg = append(msg, message.Text("太棒了，你猜对歌手名了！答案是",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1]))
|
"\n歌手:", answerstring[1]))
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
||||||
}
|
}
|
||||||
ctx.Send(msg)
|
ctx.Send(msg)
|
||||||
return
|
return
|
||||||
default:
|
default:
|
||||||
if mode == "-动漫2" && (strings.Contains(answerstring[2], answer) || strings.EqualFold(answerstring[2], answer)) {
|
if mode == "-动漫2" && (strings.Contains(answerstring[2], answer) || strings.EqualFold(answerstring[2], answer)) {
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
after.Stop()
|
after.Stop()
|
||||||
ctx.Send(message.ReplyWithMessage(c.Event.MessageID,
|
ctx.Send(message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("太棒了，你猜对番剧名了！答案是:",
|
message.Text("太棒了，你猜对番剧名了！答案是:",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1],
|
"\n歌手:", answerstring[1],
|
||||||
"\n歌曲出自:", answerstring[2]),
|
"\n歌曲出自:", answerstring[2]),
|
||||||
))
|
))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
countofmusic++
|
countofmusic++
|
||||||
switch {
|
switch {
|
||||||
case countofmusic > 2 && countofanswer < 6:
|
case countofmusic > 2 && countofanswer < 6:
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
countofanswer++
|
countofanswer++
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("答案不对哦，加油啊~"),
|
message.Text("答案不对哦，加油啊~"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
case countofmusic > 2:
|
case countofmusic > 2:
|
||||||
wait.Stop()
|
wait.Stop()
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
after.Stop()
|
after.Stop()
|
||||||
msg := make(message.Message, 0, 3)
|
msg := make(message.Message, 0, 3)
|
||||||
msg = append(msg, message.Reply(c.Event.MessageID))
|
msg = append(msg, message.Reply(c.Event.MessageID))
|
||||||
msg = append(msg, message.Text("次数到了，你没能猜出来。\n答案是:",
|
msg = append(msg, message.Text("次数到了，你没能猜出来。\n答案是:",
|
||||||
"\n歌名:", answerstring[0],
|
"\n歌名:", answerstring[0],
|
||||||
"\n歌手:", answerstring[1]))
|
"\n歌手:", answerstring[1]))
|
||||||
if mode == "-动漫2" {
|
if mode == "-动漫2" {
|
||||||
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
msg = append(msg, message.Text("\n歌曲出自:", answerstring[2]))
|
||||||
}
|
}
|
||||||
ctx.Send(msg)
|
ctx.Send(msg)
|
||||||
return
|
return
|
||||||
default:
|
default:
|
||||||
wait.Reset(40 * time.Second)
|
wait.Reset(40 * time.Second)
|
||||||
countofanswer++
|
countofanswer++
|
||||||
ctx.Send(
|
ctx.Send(
|
||||||
message.ReplyWithMessage(c.Event.MessageID,
|
message.ReplyWithMessage(c.Event.MessageID,
|
||||||
message.Text("答案不对，再听这段音频，要仔细听哦"),
|
message.Text("答案不对，再听这段音频，要仔细听哦"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
ctx.SendChain(message.Record("file:///" + file.BOTPATH + "/" + outputPath + strconv.Itoa(countofmusic) + ".wav"))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// 随机抽取音乐
|
// 随机抽取音乐
|
||||||
func musiclottery(mode, musicPath string) (musicname, pathofmusic string, err error) {
|
func musiclottery(mode, musicPath string) (musicname, pathofmusic string, err error) {
|
||||||
switch mode {
|
switch mode {
|
||||||
case "-动漫":
|
case "-动漫":
|
||||||
pathofmusic = musicPath + "动漫/"
|
pathofmusic = musicPath + "动漫/"
|
||||||
case "-动漫2":
|
case "-动漫2":
|
||||||
pathofmusic = musicPath + "动漫2/"
|
pathofmusic = musicPath + "动漫2/"
|
||||||
default:
|
default:
|
||||||
pathofmusic = musicPath + "歌榜/"
|
pathofmusic = musicPath + "歌榜/"
|
||||||
}
|
}
|
||||||
err = os.MkdirAll(pathofmusic, 0755)
|
err = os.MkdirAll(pathofmusic, 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Errorf("[生成文件夹错误]ERROR:%s", err)
|
err = errors.Errorf("[生成文件夹错误]ERROR:%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
files, err := ioutil.ReadDir(pathofmusic)
|
files, err := ioutil.ReadDir(pathofmusic)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Errorf("[读取本地列表错误]ERROR:%s", err)
|
err = errors.Errorf("[读取本地列表错误]ERROR:%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// 随机抽取音乐从本地或者线上
|
// 随机抽取音乐从本地或者线上
|
||||||
switch {
|
switch {
|
||||||
case len(files) == 0:
|
case len(files) == 0:
|
||||||
// 如果没有任何本地就下载歌曲
|
// 如果没有任何本地就下载歌曲
|
||||||
switch mode {
|
switch mode {
|
||||||
case "-动漫":
|
case "-动漫":
|
||||||
musicname, err = getpaugramdata(pathofmusic)
|
musicname, err = getpaugramdata(pathofmusic)
|
||||||
case "-动漫2":
|
case "-动漫2":
|
||||||
musicname, err = getanimedata(pathofmusic)
|
musicname, err = getanimedata(pathofmusic)
|
||||||
default:
|
default:
|
||||||
musicname, err = getuomgdata(pathofmusic)
|
musicname, err = getuomgdata(pathofmusic)
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Errorf("[本地数据为0，歌曲下载错误]ERROR:%s", err)
|
err = errors.Errorf("[本地数据为0，歌曲下载错误]ERROR:%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
case rand.Intn(2) == 0:
|
case rand.Intn(2) == 0:
|
||||||
// [0,1)只会取到0，rand不允许的
|
// [0,1)只会取到0，rand不允许的
|
||||||
if len(files) > 1 {
|
if len(files) > 1 {
|
||||||
musicname = strings.Replace(files[rand.Intn(len(files))].Name(), ".mp3", "", 1)
|
musicname = strings.Replace(files[rand.Intn(len(files))].Name(), ".mp3", "", 1)
|
||||||
} else {
|
} else {
|
||||||
musicname = strings.Replace(files[0].Name(), ".mp3", "", 1)
|
musicname = strings.Replace(files[0].Name(), ".mp3", "", 1)
|
||||||
}
|
}
|
||||||
default:
|
default:
|
||||||
switch mode {
|
switch mode {
|
||||||
case "-动漫":
|
case "-动漫":
|
||||||
musicname, err = getpaugramdata(pathofmusic)
|
musicname, err = getpaugramdata(pathofmusic)
|
||||||
case "-动漫2":
|
case "-动漫2":
|
||||||
musicname, err = getanimedata(pathofmusic)
|
musicname, err = getanimedata(pathofmusic)
|
||||||
default:
|
default:
|
||||||
musicname, err = getuomgdata(pathofmusic)
|
musicname, err = getuomgdata(pathofmusic)
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// 如果下载失败就从本地抽一个歌曲
|
// 如果下载失败就从本地抽一个歌曲
|
||||||
if len(files) > 1 {
|
if len(files) > 1 {
|
||||||
musicname = strings.Replace(files[rand.Intn(len(files))].Name(), ".mp3", "", 1)
|
musicname = strings.Replace(files[rand.Intn(len(files))].Name(), ".mp3", "", 1)
|
||||||
} else {
|
} else {
|
||||||
musicname = strings.Replace(files[0].Name(), ".mp3", "", 1)
|
musicname = strings.Replace(files[0].Name(), ".mp3", "", 1)
|
||||||
}
|
}
|
||||||
err = nil
|
err = nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 下载保罗API的歌曲
|
// 下载保罗API的歌曲
|
||||||
func getpaugramdata(musicPath string) (musicname string, err error) {
|
func getpaugramdata(musicPath string) (musicname string, err error) {
|
||||||
api := "https://api.paugram.com/acgm/?list=1"
|
api := "https://api.paugram.com/acgm/?list=1"
|
||||||
referer := "https://api.paugram.com/"
|
referer := "https://api.paugram.com/"
|
||||||
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
name := gjson.Get(binary.BytesToString(data), "title").String()
|
name := gjson.Get(binary.BytesToString(data), "title").String()
|
||||||
artistsname := gjson.Get(binary.BytesToString(data), "artist").String()
|
artistsname := gjson.Get(binary.BytesToString(data), "artist").String()
|
||||||
musicurl := gjson.Get(binary.BytesToString(data), "link").String()
|
musicurl := gjson.Get(binary.BytesToString(data), "link").String()
|
||||||
if name == "" || artistsname == "" {
|
if name == "" || artistsname == "" {
|
||||||
err = errors.Errorf("the music is missed")
|
err = errors.Errorf("the music is missed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
musicname = name + " - " + artistsname
|
musicname = name + " - " + artistsname
|
||||||
downmusic := musicPath + "/" + musicname + ".mp3"
|
downmusic := musicPath + "/" + musicname + ".mp3"
|
||||||
response, err := http.Head(musicurl)
|
response, err := http.Head(musicurl)
|
||||||
if err != nil || response.StatusCode != 200 {
|
if err != nil || response.StatusCode != 200 {
|
||||||
err = errors.Errorf("the music is missed")
|
err = errors.Errorf("the music is missed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if file.IsNotExist(downmusic) {
|
if file.IsNotExist(downmusic) {
|
||||||
data, err = web.GetData(musicurl + ".mp3")
|
data, err = web.GetData(musicurl + ".mp3")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
err = os.WriteFile(downmusic, data, 0666)
|
err = os.WriteFile(downmusic, data, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 下载animeMusic API的歌曲
|
// 下载animeMusic API的歌曲
|
||||||
func getanimedata(musicPath string) (musicname string, err error) {
|
func getanimedata(musicPath string) (musicname string, err error) {
|
||||||
api := "https://anime-music.jijidown.com/api/v2/music"
|
api := "https://anime-music.jijidown.com/api/v2/music"
|
||||||
referer := "https://anime-music.jijidown.com/"
|
referer := "https://anime-music.jijidown.com/"
|
||||||
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
name := gjson.Get(binary.BytesToString(data), "res").Get("title").String()
|
name := gjson.Get(binary.BytesToString(data), "res").Get("title").String()
|
||||||
artistsname := gjson.Get(binary.BytesToString(data), "res").Get("author").String()
|
artistsname := gjson.Get(binary.BytesToString(data), "res").Get("author").String()
|
||||||
acgname := gjson.Get(binary.BytesToString(data), "res").Get("anime_info").Get("title").String()
|
acgname := gjson.Get(binary.BytesToString(data), "res").Get("anime_info").Get("title").String()
|
||||||
musicurl := gjson.Get(binary.BytesToString(data), "res").Get("play_url").String()
|
musicurl := gjson.Get(binary.BytesToString(data), "res").Get("play_url").String()
|
||||||
if name == "" || artistsname == "" {
|
if name == "" || artistsname == "" {
|
||||||
err = errors.Errorf("the music is missed")
|
err = errors.Errorf("the music is missed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
musicname = name + " - " + artistsname + " - " + acgname
|
musicname = name + " - " + artistsname + " - " + acgname
|
||||||
downmusic := musicPath + "/" + musicname + ".mp3"
|
downmusic := musicPath + "/" + musicname + ".mp3"
|
||||||
response, err := http.Head(musicurl)
|
response, err := http.Head(musicurl)
|
||||||
if err != nil || response.StatusCode != 200 {
|
if err != nil || response.StatusCode != 200 {
|
||||||
err = errors.Errorf("the music is missed")
|
err = errors.Errorf("the music is missed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if file.IsNotExist(downmusic) {
|
if file.IsNotExist(downmusic) {
|
||||||
data, err = web.GetData(musicurl + ".mp3")
|
data, err = web.GetData(musicurl + ".mp3")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
err = os.WriteFile(downmusic, data, 0666)
|
err = os.WriteFile(downmusic, data, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 下载网易云热歌榜音乐
|
// 下载网易云热歌榜音乐
|
||||||
func getuomgdata(musicPath string) (musicname string, err error) {
|
func getuomgdata(musicPath string) (musicname string, err error) {
|
||||||
api := "https://api.uomg.com/api/rand.music?sort=%E7%83%AD%E6%AD%8C%E6%A6%9C&format=json"
|
api := "https://api.uomg.com/api/rand.music?sort=%E7%83%AD%E6%AD%8C%E6%A6%9C&format=json"
|
||||||
referer := "https://api.uomg.com/api/rand.music"
|
referer := "https://api.uomg.com/api/rand.music"
|
||||||
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
data, err := web.RequestDataWith(web.NewDefaultClient(), api, "GET", referer, ua)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
musicdata := gjson.Get(binary.BytesToString(data), "data")
|
musicdata := gjson.Get(binary.BytesToString(data), "data")
|
||||||
name := musicdata.Get("name").String()
|
name := musicdata.Get("name").String()
|
||||||
musicurl := musicdata.Get("url").String()
|
musicurl := musicdata.Get("url").String()
|
||||||
artistsname := musicdata.Get("artistsname").String()
|
artistsname := musicdata.Get("artistsname").String()
|
||||||
musicname = name + " - " + artistsname
|
musicname = name + " - " + artistsname
|
||||||
downmusic := musicPath + "/" + musicname + ".mp3"
|
downmusic := musicPath + "/" + musicname + ".mp3"
|
||||||
if file.IsNotExist(downmusic) {
|
if file.IsNotExist(downmusic) {
|
||||||
data, err = web.GetData(musicurl + ".mp3")
|
data, err = web.GetData(musicurl + ".mp3")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
err = os.WriteFile(downmusic, data, 0666)
|
err = os.WriteFile(downmusic, data, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 切割音乐成三个10s音频
|
// 切割音乐成三个10s音频
|
||||||
func musiccut(musicname, pathofmusic, outputPath string) (err error) {
|
func musiccut(musicname, pathofmusic, outputPath string) (err error) {
|
||||||
err = os.MkdirAll(outputPath, 0755)
|
err = os.MkdirAll(outputPath, 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Errorf("[生成歌曲目录错误]ERROR:%s", err)
|
err = errors.Errorf("[生成歌曲目录错误]ERROR:%s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
var stderr bytes.Buffer
|
var stderr bytes.Buffer
|
||||||
cmdArguments := []string{"-y", "-i", pathofmusic + musicname + ".mp3",
|
cmdArguments := []string{"-y", "-i", pathofmusic + musicname + ".mp3",
|
||||||
"-ss", cuttime[0], "-t", "10", file.BOTPATH + "/" + outputPath + "0.wav",
|
"-ss", cuttime[0], "-t", "10", file.BOTPATH + "/" + outputPath + "0.wav",
|
||||||
"-ss", cuttime[1], "-t", "10", file.BOTPATH + "/" + outputPath + "1.wav",
|
"-ss", cuttime[1], "-t", "10", file.BOTPATH + "/" + outputPath + "1.wav",
|
||||||
"-ss", cuttime[2], "-t", "10", file.BOTPATH + "/" + outputPath + "2.wav", "-hide_banner"}
|
"-ss", cuttime[2], "-t", "10", file.BOTPATH + "/" + outputPath + "2.wav", "-hide_banner"}
|
||||||
cmd := exec.Command("ffmpeg", cmdArguments...)
|
cmd := exec.Command("ffmpeg", cmdArguments...)
|
||||||
cmd.Stderr = &stderr
|
cmd.Stderr = &stderr
|
||||||
err = cmd.Run()
|
err = cmd.Run()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.Errorf("[生成歌曲错误]ERROR:%s", stderr.String())
|
err = errors.Errorf("[生成歌曲错误]ERROR:%s", stderr.String())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user