diff --git a/chat/chat.go b/chat/chat.go index 259fa24e..d7a1e6b6 100644 --- a/chat/chat.go +++ b/chat/chat.go @@ -1,15 +1,31 @@ package chat import ( + "math/rand" "time" zero "github.com/wdvxdr1123/ZeroBot" "github.com/wdvxdr1123/ZeroBot/extension/rate" + "github.com/wdvxdr1123/ZeroBot/message" ) var poke = rate.NewManager(time.Minute*5, 8) // 戳一戳 func init() { // 插件主体 + var NICKNAME = zero.BotConfig.NickName[0] + // 被喊名字 + zero.OnFullMatchGroup(zero.BotConfig.NickName).SetBlock(false).FirstPriority(). + Handle(func(ctx *zero.Ctx) { + time.Sleep(time.Second * 1) + ctx.SendChain(message.Text( + []string{ + NICKNAME + "在此，有何贵干~", + "(っ●ω●)っ在~", + "这里是" + NICKNAME + "(っ●ω●)っ", + NICKNAME + "不在呢~", + }[rand.Intn(4)], + )) + }) // 戳一戳 zero.On("notice/notify/poke", zero.OnlyToMe).SetBlock(false).FirstPriority(). Handle(func(ctx *zero.Ctx) { @@ -17,11 +33,11 @@ func init() { // 插件主体 case poke.Load(ctx.Event.UserID).AcquireN(3): // 5分钟共8块命令牌 一次消耗3块命令牌 time.Sleep(time.Second * 1) - ctx.Send("请不要戳我 >_<") + ctx.SendChain(message.Text("请不要戳", NICKNAME, " >_<")) case poke.Load(ctx.Event.UserID).Acquire(): // 5分钟共8块命令牌 一次消耗1块命令牌 time.Sleep(time.Second * 1) - ctx.Send("喂(#`O′) 戳我干嘛！") + ctx.SendChain(message.Text("喂(#`O′) 戳", NICKNAME, "干嘛！")) default: // 频繁触发，不回复 } diff --git a/github/repo_searcher.go b/github/repo_searcher.go index 94273acd..bca28e85 100644 --- a/github/repo_searcher.go +++ b/github/repo_searcher.go @@ -1,6 +1,7 @@ package github import ( + "errors" "fmt" "io/ioutil" "net/http" @@ -9,71 +10,81 @@ import ( "github.com/tidwall/gjson" zero "github.com/wdvxdr1123/ZeroBot" + "github.com/wdvxdr1123/ZeroBot/message" ) func init() { // 插件主体 zero.OnRegex(`>G\s(.*)`).SetBlock(true).FirstPriority(). Handle(func(ctx *zero.Ctx) { + // 发送请求 + header := http.Header{ + "User-Agent": []string{"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"}, + } api, _ := url.Parse("https://api.github.com/search/repositories") - params := url.Values{ + api.RawQuery = url.Values{ "q": []string{ctx.State["regex_matched"].([]string)[1]}, - } - api.RawQuery = params.Encode() - link := api.String() - - client := &http.Client{} - - req, err := http.NewRequest("GET", link, nil) + }.Encode() + body, err := netGet(api.String(), header) if err != nil { - ctx.Send(fmt.Sprintf("ERROR: %v", err)) + ctx.SendChain(message.Text("ERROR: ", err)) + } + // 解析请求 + info := gjson.ParseBytes(body) + if info.Get("total_count").Int() == 0 { + ctx.SendChain(message.Text("ERROR: 没有找到这样的仓库")) return } - 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 - } - count := gjson.ParseBytes(body).Get("total_count").Int() - if count == 0 { - ctx.Send("没有找到这样的仓库") - return - } - repo := gjson.ParseBytes(body).Get("items.0") - language := repo.Get("language").Str - if language == "" { - language = "None" - } - license := strings.ToUpper(repo.Get("license.key").Str) - if license == "" { - license = "None" - } - id := ctx.Send(fmt.Sprintf( - "%s: \nDescription: %s\nStar/Fork/Issue: %d/%d/%d\nLanguage: %s\nLicense: %s\nLast pushed: %s\nJump: %s", - repo.Get("full_name").Str, - repo.Get("description").Str, - repo.Get("watchers").Int(), - repo.Get("forks").Int(), - repo.Get("open_issues").Int(), - language, - license, - repo.Get("updated_at").Str, - repo.Get("html_url").Str, - )) - if id == 0 { - ctx.Send("ERROR: 可能被风控，发送失败") - } + repo := info.Get("items.0") + // 发送结果 + ctx.SendChain( + message.Text( + repo.Get("full_name").Str, "\n", + "Description: ", + repo.Get("description").Str, "\n", + "Star/Fork/Issue: ", + repo.Get("watchers").Int(), "/", repo.Get("forks").Int(), "/", repo.Get("open_issues").Int(), "\n", + "Language: ", + notnull(repo.Get("language").Str, "None"), "\n", + "License: ", + notnull(strings.ToUpper(repo.Get("license.key").Str), "None"), "\n", + "Last pushed: ", + repo.Get("pushed_at").Str, "\n", + "Jump: ", + repo.Get("html_url").Str, "\n", + ), + ) }) } + +// notnull 如果传入文本为空，则返回默认值 +func notnull(text, default_ string) string { + if text == "" { + return default_ + } + return text +} + +// netGet 返回请求结果 +func netGet(dest string, header http.Header) ([]byte, error) { + client := &http.Client{} + + req, err := http.NewRequest("GET", dest, nil) + if err != nil { + return nil, err + } + req.Header = header + resp, err := client.Do(req) + if err != nil { + return nil, err + } + defer resp.Body.Close() + body, err := ioutil.ReadAll(resp.Body) + if err != nil { + return nil, err + } + if code := resp.StatusCode; code != 200 { + // 如果返回不是200则立刻抛出错误 + return nil, errors.New(fmt.Sprintf("code %d", code)) + } + return body, nil +} diff --git a/main.go b/main.go index 3fca8b51..c518321b 100644 --- a/main.go +++ b/main.go @@ -35,7 +35,7 @@ func main() { ======================================================== `) zero.Run(zero.Config{ - NickName: []string{"bot"}, + NickName: []string{"椛椛"}, CommandPrefix: "/", SuperUsers: []string{"825111790", "213864964"}, // 必须修改，否则无权限 Driver: []zero.Driver{ diff --git a/manager/manager.go b/manager/manager.go index 7c6bda46..bad8edf8 100644 --- a/manager/manager.go +++ b/manager/manager.go @@ -1,20 +1,20 @@ package manager import ( + "strconv" "strings" - "github.com/Yiwen-Chan/ZeroBot-Plugin/manager/utils" zero "github.com/wdvxdr1123/ZeroBot" ) func init() { // 插件主体 // 菜单 - zero.OnFullMatch("群管系统").SetBlock(true).SetPriority(40). + zero.OnFullMatch("群管系统", zero.AdminPermission).SetBlock(true).FirstPriority(). Handle(func(ctx *zero.Ctx) { ctx.Send(`====群管==== -- 禁言@QQ 1 +- 禁言@QQ 1分钟 - 解除禁言 @QQ -- 我要自闭 1 +- 我要自闭 1分钟 - 开启全员禁言 - 解除全员禁言 - 升为管理@QQ @@ -33,12 +33,12 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupAdmin( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq true, ) nickname := ctx.GetGroupMemberInfo( // 被升为管理的人的昵称 ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq false, ).Get("nickname").Str ctx.Send(nickname + " 升为了管理~") @@ -49,12 +49,12 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupAdmin( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq false, ) nickname := ctx.GetGroupMemberInfo( // 被取消管理的人的昵称 ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq false, ).Get("nickname").Str ctx.Send("残念~ " + nickname + " 暂时失去了管理员的资格") @@ -65,12 +65,12 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupKick( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq false, ) nickname := ctx.GetGroupMemberInfo( // 被踢出群聊的人的昵称 ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq false, ).Get("nickname").Str ctx.Send("残念~ " + nickname + " 被放逐") @@ -80,7 +80,7 @@ func init() { // 插件主体 zero.OnRegex(`^退出群聊.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). Handle(func(ctx *zero.Ctx) { ctx.SetGroupLeave( - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 要退出的群的群号 + strToInt(ctx.State["regex_matched"].([]string)[1]), // 要退出的群的群号 true, ) return @@ -108,7 +108,7 @@ func init() { // 插件主体 // 禁言 zero.OnRegex(`^禁言.*?(\d+).*?\s(\d+)(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). Handle(func(ctx *zero.Ctx) { - duration := utils.Str2Int(ctx.State["regex_matched"].([]string)[2]) + duration := strToInt(ctx.State["regex_matched"].([]string)[2]) switch ctx.State["regex_matched"].([]string)[3] { case "分钟": // @@ -124,7 +124,7 @@ func init() { // 插件主体 } ctx.SetGroupBan( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 要禁言的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 要禁言的人的qq duration*60, // 要禁言的时间（分钟） ) ctx.Send("小黑屋收留成功~") @@ -135,7 +135,7 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupBan( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 要解除禁言的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 要解除禁言的人的qq 0, ) ctx.Send("小黑屋释放成功~") @@ -144,7 +144,7 @@ func init() { // 插件主体 // 自闭禁言 zero.OnRegex(`^我要自闭.*?(\d+)(.*)`, zero.OnlyGroup).SetBlock(true).SetPriority(40). Handle(func(ctx *zero.Ctx) { - duration := utils.Str2Int(ctx.State["regex_matched"].([]string)[1]) + duration := strToInt(ctx.State["regex_matched"].([]string)[1]) switch ctx.State["regex_matched"].([]string)[2] { case "分钟": // @@ -171,8 +171,8 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupCard( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被修改群名片的人 - ctx.State["regex_matched"].([]string)[2], // 修改成的群名片 + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被修改群名片的人 + ctx.State["regex_matched"].([]string)[2], // 修改成的群名片 ) ctx.Send("嗯！已经修改了") return @@ -182,8 +182,8 @@ func init() { // 插件主体 Handle(func(ctx *zero.Ctx) { ctx.SetGroupSpecialTitle( ctx.Event.GroupID, - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 被修改群头衔的人 - ctx.State["regex_matched"].([]string)[2], // 修改成的群头衔 + strToInt(ctx.State["regex_matched"].([]string)[1]), // 被修改群头衔的人 + ctx.State["regex_matched"].([]string)[2], // 修改成的群头衔 ) ctx.Send("嗯！已经修改了") return @@ -207,7 +207,7 @@ func init() { // 插件主体 content = strings.ReplaceAll(content, "[", "[") content = strings.ReplaceAll(content, "]", "]") ctx.SendGroupMessage( - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 需要发送的群 + strToInt(ctx.State["regex_matched"].([]string)[1]), // 需要发送的群 content, // 需要发送的信息 ) ctx.Send("📧 --> " + ctx.State["regex_matched"].([]string)[1]) @@ -221,7 +221,7 @@ func init() { // 插件主体 content = strings.ReplaceAll(content, "[", "[") content = strings.ReplaceAll(content, "]", "]") ctx.SendPrivateMessage( - utils.Str2Int(ctx.State["regex_matched"].([]string)[1]), // 需要发送的人的qq + strToInt(ctx.State["regex_matched"].([]string)[1]), // 需要发送的人的qq content, // 需要发送的信息 ) ctx.Send("📧 --> " + ctx.State["regex_matched"].([]string)[1]) @@ -252,3 +252,8 @@ func init() { // 插件主体 ctx.Send(cmd) }) } + +func strToInt(str string) int64 { + val, _ := strconv.ParseInt(str, 10, 64) + return val +} diff --git a/manager/utils/utils.go b/manager/utils/utils.go deleted file mode 100644 index b55ffa7b..00000000 --- a/manager/utils/utils.go +++ /dev/null @@ -1,15 +0,0 @@ -package utils - -import ( - "strconv" -) - -func Int2Str(val int64) string { - str := strconv.FormatInt(val, 10) - return str -} - -func Str2Int(str string) int64 { - val, _ := strconv.ParseInt(str, 10, 64) - return val -} diff --git a/music/music_selecter.go b/music/music_selecter.go index a012d0ee..e7f387c3 100644 --- a/music/music_selecter.go +++ b/music/music_selecter.go @@ -12,31 +12,30 @@ import ( "github.com/tidwall/gjson" zero "github.com/wdvxdr1123/ZeroBot" + "github.com/wdvxdr1123/ZeroBot/extension/rate" "github.com/wdvxdr1123/ZeroBot/message" ) +var limit = rate.NewManager(time.Minute*3, 5) + func init() { - zero.OnRegex("^酷我点歌(.+?)$").SetBlock(true).FirstPriority(). + zero.OnRegex("^(.{0,2})点歌(.{1,25})$").SetBlock(true).FirstPriority(). Handle(func(ctx *zero.Ctx) { - ctx.SendChain(kuwo(ctx.State["regex_matched"].([]string)[1])) - return - }) - - zero.OnRegex("^酷狗点歌(.+?)$").SetBlock(true).FirstPriority(). - Handle(func(ctx *zero.Ctx) { - ctx.SendChain(kugou(ctx.State["regex_matched"].([]string)[1])) - return - }) - - zero.OnRegex("^网易点歌(.+?)$").SetBlock(true).FirstPriority(). - Handle(func(ctx *zero.Ctx) { - ctx.SendChain(cloud163(ctx.State["regex_matched"].([]string)[1])) - return - }) - - zero.OnRegex("^点歌(.+?)$").SetBlock(true).FirstPriority(). - Handle(func(ctx *zero.Ctx) { - ctx.SendChain(qqmusic(ctx.State["regex_matched"].([]string)[1])) + if !limit.Load(ctx.Event.UserID).Acquire() { + ctx.Send("请稍后重试0x0...") + return + } + // switch 平台 + switch ctx.State["regex_matched"].([]string)[1] { + case "酷我": + ctx.SendChain(kuwo(ctx.State["regex_matched"].([]string)[2])) + case "酷狗": + ctx.SendChain(kugou(ctx.State["regex_matched"].([]string)[2])) + case "网易": + ctx.SendChain(cloud163(ctx.State["regex_matched"].([]string)[2])) + default: // 默认 QQ音乐 + ctx.SendChain(qqmusic(ctx.State["regex_matched"].([]string)[2])) + } return }) } diff --git a/setutime/setu_geter.go b/setutime/setu_geter.go index fb94e254..0d9d3d40 100644 --- a/setutime/setu_geter.go +++ b/setutime/setu_geter.go @@ -45,7 +45,7 @@ func init() { func init() { // 插件主体 zero.OnRegex(`^来份(.*)$`, FirstValueInList(PoolList)).SetBlock(true).SetPriority(20). Handle(func(ctx *zero.Ctx) { - if limit.Load(ctx.Event.UserID).Acquire() == false { + if !limit.Load(ctx.Event.UserID).Acquire() { ctx.Send("请稍后重试0x0...") return }