From 2ad51022974c895d68f850bb948bef8b4a51ac75 Mon Sep 17 00:00:00 2001 From: Yiwen-Chan Date: Sat, 13 Feb 2021 20:42:53 +0800 Subject: [PATCH] :art: improve code --- GroupManager.go | 37 ----- go.mod | 8 +- go.sum | 38 +++-- groupmanager/group_manager.go | 256 ++++++++++++++++++++++++++++++++++ groupmanager/utils/utils.go | 15 ++ main.go | 38 +++++ modules/main.go | 50 ------- modules/manage.go | 70 ---------- modules/mute.go | 77 ---------- modules/notice.go | 61 -------- modules/request.go | 53 ------- modules/sender.go | 44 ------ modules/title.go | 55 -------- modules/utils.go | 19 --- utils/utils.go | 47 ------- utils/yaml.go | 58 -------- 16 files changed, 340 insertions(+), 586 deletions(-) delete mode 100644 GroupManager.go create mode 100644 groupmanager/group_manager.go create mode 100644 groupmanager/utils/utils.go create mode 100644 main.go delete mode 100644 modules/main.go delete mode 100644 modules/manage.go delete mode 100644 modules/mute.go delete mode 100644 modules/notice.go delete mode 100644 modules/request.go delete mode 100644 modules/sender.go delete mode 100644 modules/title.go delete mode 100644 modules/utils.go delete mode 100644 utils/utils.go delete mode 100644 utils/yaml.go diff --git a/GroupManager.go b/GroupManager.go deleted file mode 100644 index c1d3257f..00000000 --- a/GroupManager.go +++ /dev/null @@ -1,37 +0,0 @@ -package main - -import ( - "fmt" - - m "gm/modules" - "gm/utils" - - zero "github.com/wdvxdr1123/ZeroBot" -) - -var Conf = &utils.YamlConfig{} - -func main() { - zero.Run(zero.Option{ - Host: Conf.Host, - Port: Conf.Port, - AccessToken: Conf.AccessToken, - NickName: []string{"GroupManager"}, - CommandPrefix: "/", - SuperUsers: Conf.Master, - }) - select {} -} - -func init() { - fmt.Println(` -====================[GroupManager]==================== -* OneBot + ZeroBot + Golang -* Copyright © 2018-2020 Kanri, All Rights Reserved -* Project: https://github.com/Yiwen-Chan/GroupManager -======================================================= -`) - Conf = utils.Load("./GroupManager.yml") - m.Conf = Conf - fmt.Println("[GroupManager] 有需要请按 GitHub 项目上描述的方法修改配置文件") -} diff --git a/go.mod b/go.mod index b4b75dcc..ca348d9a 100644 --- a/go.mod +++ b/go.mod @@ -1,8 +1,8 @@ -module gm +module bot -go 1.14 +go 1.15 require ( - github.com/wdvxdr1123/ZeroBot v0.0.0-20201209121757-2a04edb62bac - gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776 + github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e + gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b ) diff --git a/go.sum b/go.sum index de39f835..3b21cf0b 100644 --- a/go.sum +++ b/go.sum @@ -1,15 +1,23 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc= github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= github.com/json-iterator/go v1.1.10 h1:Kz6Cvnvv2wGdaG/V8yMvfkmNiXq9Ya2KUv4rouJJr68= github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4= github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= -github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLDQ0W1YjYsBW+p8U2u7vzgW2SQVmlNazg= github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.2-0.20210109003243-333559e1834b h1:6Xjqolv/0DDdUqlpnsTomXQvjvvkz7Ux7TcMALvozEw= +github.com/modern-go/reflect2 v1.0.2-0.20210109003243-333559e1834b/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE= github.com/sirupsen/logrus v1.7.0 h1:ShrD1U9pZB12TX0cVy0DtePoCH97K8EtX+mg7ZARUtM= @@ -18,20 +26,28 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+ github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ= github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816 h1:J6v8awz+me+xeb/cUTotKgceAYouhIB3pjzgRd6IlGk= github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816/go.mod h1:tzym/CEb5jnFI+Q0k4Qq3+LvRF4gO3E2pxS8fHP8jcA= -github.com/tidwall/gjson v1.6.1 h1:LRbvNuNuvAiISWg6gxLEFuCe72UKy5hDqhxW/8183ws= -github.com/tidwall/gjson v1.6.1/go.mod h1:BaHyNc5bjzYkPqgLq7mdVzeiRtULKULXLgZFKsxEHI0= -github.com/tidwall/match v1.0.1 h1:PnKP62LPNxHKTwvHHZZzdOAOCtsJTjo6dZLCwpKm5xc= -github.com/tidwall/match v1.0.1/go.mod h1:LujAq0jyVjBy028G1WhWfIzbpQfMO8bBZ6Tyb0+pL9E= +github.com/tidwall/gjson v1.6.8 h1:CTmXMClGYPAmln7652e69B7OLXfTi5ABcPPwjIWUv7w= +github.com/tidwall/gjson v1.6.8/go.mod h1:zeFuBCIqD4sN/gmqBzZ4j7Jd6UcA2Fc56x7QFsv+8fI= +github.com/tidwall/match v1.0.3 h1:FQUVvBImDutD8wJLN6c5eMzWtjgONK9MwIBCOrUJKeE= +github.com/tidwall/match v1.0.3/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM= github.com/tidwall/pretty v1.0.2 h1:Z7S3cePv9Jwm1KwS0513MRaoUe3S01WPbLNV40pwWZU= github.com/tidwall/pretty v1.0.2/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk= -github.com/wdvxdr1123/ZeroBot v0.0.0-20201209121757-2a04edb62bac h1:foT1gSWW3Ovkecyqoircrp2kP5pajSTUawwom9GLSlM= -github.com/wdvxdr1123/ZeroBot v0.0.0-20201209121757-2a04edb62bac/go.mod h1:e92qGONnOry8XSfaUUlG7OhT9gyD839P8oZ2wfPe60w= +github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e h1:BaUnQOFcCU4l+uf+mifbPdzrBtYfGeDM9RgIFhMBRLs= +github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e/go.mod h1:MC6+vOkILANgHlEvZpDMKLNeOWSde5eRgiGFtNGix/4= +golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= -golang.org/x/sys v0.0.0-20201029080932-201ba4db2418 h1:HlFl4V6pEMziuLXyRkm5BIYq1y1GAbb02pRlWvI54OM= -golang.org/x/sys v0.0.0-20201029080932-201ba4db2418/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c h1:VwygUrnw9jn88c4u8GD3rZQbqrP/tgas88tPUbBxQrk= +golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= -gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776 h1:tQIYjPdBoyREyB9XMu+nnTclpTYkz2zFM+lzLJFO4gQ= -gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw= +gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/groupmanager/group_manager.go b/groupmanager/group_manager.go new file mode 100644 index 00000000..08873b46 --- /dev/null +++ b/groupmanager/group_manager.go @@ -0,0 +1,256 @@ +package groupmanager + +import ( + "bot/groupmanager/utils" + "time" + + zero "github.com/wdvxdr1123/ZeroBot" +) + +func init() { + zero.RegisterPlugin(groupManager{}) // 注册插件 +} + +type groupManager struct{} // pixivSearch 搜索P站插图 + +func (_ groupManager) GetPluginInfo() zero.PluginInfo { // 返回插件信息 + return zero.PluginInfo{ + Author: "kanri", + PluginName: "GroupManager", + Version: "0.0.1", + Details: "群管", + } +} + +func (_ groupManager) Start() { // 插件主体 + // TODO 菜单 + zero.OnFullMatch("群管系统").SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.Send(event, `====群管==== +- 禁言@QQ 1 +- 解除禁言 @QQ +- 我要自闭 1分钟 +- 开启全员禁言 +- 解除全员禁言 +- 升为管理@QQ +- 取消管理@QQ +- 修改名片@QQ XXX +- 修改头衔@QQ XXX +- 申请头衔 XXX +- 群聊转发 1234 XXX +- 私聊转发 0000 XXX`) + return zero.SuccessResponse + }) + // TODO 升为管理 + zero.OnRegex(`^升为管理.*?(\d+)`, zero.OnlyGroup, zero.SuperUserPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupAdmin( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被升为管理的人的qq + true, + ) + nickname := zero.GetGroupMemberInfo( // 被升为管理的人的昵称 + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被升为管理的人的qq + false, + ).Get("nickname").Str + zero.Send( + event, + nickname+" 升为了管理~", + ) + return zero.SuccessResponse + }) + // TODO 取消管理 + zero.OnRegex(`^取消管理.*?(\d+)`, zero.OnlyGroup, zero.SuperUserPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupAdmin( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被取消管理的人的qq + false, + ) + nickname := zero.GetGroupMemberInfo( // 被取消管理的人的昵称 + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被取消管理的人的qq + false, + ).Get("nickname").Str + zero.Send( + event, + "残念~ "+nickname+" 暂时失去了管理员的资格", + ) + return zero.SuccessResponse + }) + // TODO 踢出群聊 + zero.OnRegex(`^踢出群聊.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupAdmin( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq + false, + ) + nickname := zero.GetGroupMemberInfo( // 被踢出群聊的人的昵称 + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq + false, + ).Get("nickname").Str + zero.Send( + event, + "残念~ "+nickname+" 被放逐", + ) + return zero.SuccessResponse + }) + // TODO 退出群聊 + zero.OnRegex(`^退出群聊.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupLeave( + utils.Str2Int(state["regex_matched"].([]string)[1]), // 要退出的群的群号 + true, + ) + return zero.SuccessResponse + }) + // TODO 开启全体禁言 + zero.OnRegex(`^开启全员禁言$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupWholeBan( + event.GroupID, + true, + ) + zero.Send(event, "全员自闭开始~") + return zero.SuccessResponse + }) + // TODO 解除全体禁言 + zero.OnRegex(`^解除全体禁言$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupWholeBan( + event.GroupID, + false, + ) + zero.Send(event, "全员自闭结束~") + return zero.SuccessResponse + }) + // TODO 禁言 + zero.OnRegex(`^禁言.*?(\d+).*?\s(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupBan( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 要禁言的人的qq + utils.Str2Int(state["regex_matched"].([]string)[2])*60, // 要禁言的时间（分钟） + ) + zero.Send(event, "小黑屋收留成功~") + return zero.SuccessResponse + }) + // TODO 解除禁言 + zero.OnRegex(`^解除禁言.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupBan( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 要解除禁言的人的qq + 0, + ) + zero.Send(event, "小黑屋释放成功~") + return zero.SuccessResponse + }) + // TODO 自闭禁言 + zero.OnRegex(`^我要自闭.*?(\d+)分钟`, zero.OnlyGroup).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupBan( + event.GroupID, + event.UserID, + utils.Str2Int(state["regex_matched"].([]string)[1])*60, // 要自闭的时间（分钟） + ) + zero.Send(event, "那我就不手下留情了~") + return zero.SuccessResponse + }) + // TODO 修改名片 + zero.OnRegex(`^修改名片.*?(\d+).*?\s(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupCard( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被修改群名片的人 + state["regex_matched"].([]string)[2], // 修改成的群名片 + ) + zero.Send( + event, + "嗯！已经修改了", + ) + return zero.SuccessResponse + }) + // TODO 修改头衔 + zero.OnRegex(`^修改头衔.*?(\d+).*?\s(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupSpecialTitle( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被修改群头衔的人 + state["regex_matched"].([]string)[2], // 修改成的群头衔 + ) + zero.Send( + event, + "嗯！已经修改了", + ) + return zero.SuccessResponse + }) + // TODO 申请头衔 + zero.OnRegex(`^申请头衔(.*)`, zero.OnlyGroup).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SetGroupSpecialTitle( + event.GroupID, + utils.Str2Int(state["regex_matched"].([]string)[1]), // 被修改群头衔的人 + state["regex_matched"].([]string)[2], // 修改成的群头衔 + ) + zero.Send( + event, + "嗯！不错的头衔呢~", + ) + return zero.SuccessResponse + }) + // TODO 群聊转发 + zero.OnRegex(`^群聊转发.*?(\d+)\s(.*)`, zero.SuperUserPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SendGroupMessage( + utils.Str2Int(state["regex_matched"].([]string)[1]), // 需要发送的群 + state["regex_matched"].([]string)[1], // 需要发送的信息 + ) + zero.Send( + event, + "📧 --> "+state["regex_matched"].([]string)[1], + ) + return zero.SuccessResponse + }) + // TODO 私聊转发 + zero.OnRegex(`^私聊转发.*?(\d+)\s(.*)`, zero.SuperUserPermission).SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + zero.SendPrivateMessage( + utils.Str2Int(state["regex_matched"].([]string)[1]), // 需要发送的人的qq + state["regex_matched"].([]string)[1], // 需要发送的信息 + ) + zero.Send( + event, + "📧 --> "+state["regex_matched"].([]string)[1], + ) + return zero.SuccessResponse + }) + // TODO 戳一戳 + zero.OnNotice().SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + if event.NoticeType == "notify" { + time.Sleep(time.Second * 1) + zero.Send(event, "请不要戳我 >_<") + } + return zero.SuccessResponse + }) + // TODO 入群欢迎 + zero.OnNotice().SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + if event.NoticeType == "group_increase" { + zero.Send(event, "欢迎~") + } + return zero.SuccessResponse + }) + // TODO 退群提醒 + zero.OnNotice().SetBlock(true).SetPriority(40). + Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { + if event.NoticeType == "group_increase" { + zero.Send(event, "有人跑路了~") + } + return zero.SuccessResponse + }) +} diff --git a/groupmanager/utils/utils.go b/groupmanager/utils/utils.go new file mode 100644 index 00000000..b55ffa7b --- /dev/null +++ b/groupmanager/utils/utils.go @@ -0,0 +1,15 @@ +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/main.go b/main.go new file mode 100644 index 00000000..0f0d6589 --- /dev/null +++ b/main.go @@ -0,0 +1,38 @@ +package main + +import ( + "fmt" + + log "github.com/sirupsen/logrus" + easy "github.com/t-tomalak/logrus-easy-formatter" + zero "github.com/wdvxdr1123/ZeroBot" + + _ "bot/groupmanager" +) + +func init() { + log.SetFormatter(&easy.Formatter{ + TimestampFormat: "2006-01-02 15:04:05", + LogFormat: "[zero][%time%][%lvl%]: %msg% \n", + }) + log.SetLevel(log.DebugLevel) +} + +func main() { + fmt.Printf(` +====================[GroupManager]==================== +* OneBot + ZeroBot + Golang +* Copyright © 2018-2020 Kanri, All Rights Reserved +* Project: https://github.com/Yiwen-Chan/GroupManager +======================================================= +`) + zero.Run(zero.Option{ + Host: "127.0.0.1", + Port: "6700", + AccessToken: "", + NickName: []string{"椛椛"}, + CommandPrefix: "/", + SuperUsers: []string{"825111790"}, + }) + select {} +} diff --git a/modules/main.go b/modules/main.go deleted file mode 100644 index bbc5cefb..00000000 --- a/modules/main.go +++ /dev/null @@ -1,50 +0,0 @@ -package modules - -import ( - "fmt" - "gm/utils" - - zero "github.com/wdvxdr1123/ZeroBot" -) - -var Conf = &utils.YamlConfig{} - -func init() { - zero.RegisterPlugin(menu{}) -} - -type menu struct{} - -func (menu) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "menu", - Version: "0.0.1", - Details: "菜单", - } -} - -func (menu) Start() { // 插件主体 - gmmenu := zero.OnRegex("^群管系统$"). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - menuText := `====群管==== -- 禁言@QQ 1 -- 解除禁言 @QQ -- 我要自闭 1分钟 -- 开启全员禁言 -- 解除全员禁言 -- 升为管理@QQ -- 取消管理@QQ -- 修改名片@QQ XXX -- 修改头衔@QQ XXX -- 申请头衔 XXX -- 群聊转发 1234 XXX -- 私聊转发 0000 XXX` - zero.Send(event, fmt.Sprintf("%v", menuText)) - return zero.SuccessResponse - }, - ) - gmmenu.Priority = 60 - gmmenu.Block = true -} diff --git a/modules/manage.go b/modules/manage.go deleted file mode 100644 index 1b5b3af5..00000000 --- a/modules/manage.go +++ /dev/null @@ -1,70 +0,0 @@ -package modules - -import ( - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(manage{}) -} - -type manage struct{} - -func (manage) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "manage", - Version: "0.0.1", - Details: "管理任命 踢出退出群聊", - } -} - -func (manage) Start() { // 插件主体 - promoteManager := zero.OnRegex("^升为管理.*?(\\d+)", zero.OnlyGroup, zero.SuperUserPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupAdmin(event.GroupID, GetInt(state, 1), true) - nickname := GetNickname(event.GroupID, GetInt(state, 1)) - zero.Send(event, nickname+" 升为了管理~") - return zero.SuccessResponse - }, - ) - promoteManager.Priority = 20 - promoteManager.Block = true - - cancleManager := zero.OnRegex("^取消管理.*?(\\d+)", zero.OnlyGroup, zero.SuperUserPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupAdmin(event.GroupID, GetInt(state, 1), false) - nickname := GetNickname(event.GroupID, GetInt(state, 1)) - zero.Send(event, "残念~ "+nickname+" 暂时失去了管理员的资格") - return zero.SuccessResponse - }, - ) - cancleManager.Priority = 21 - cancleManager.Block = true - - kick := zero.OnRegex("^踢出群聊.*?(\\d+)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupKick(event.GroupID, GetInt(state, 1), false) - nickname := GetNickname(event.GroupID, GetInt(state, 1)) - zero.Send(event, "残念~ "+nickname+"("+GetStr(state, 1)+") 被放逐") - return zero.SuccessResponse - }, - ) - kick.Priority = 22 - kick.Block = true - - leave := zero.OnRegex("^退出群聊.*?(\\d+)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupLeave(GetInt(state, 1), true) - zero.Send(event, "姬气人已主动离开群聊 "+GetStr(state, 1)) - return zero.SuccessResponse - }, - ) - leave.Priority = 23 - leave.Block = true - -} diff --git a/modules/mute.go b/modules/mute.go deleted file mode 100644 index b4a6a4b1..00000000 --- a/modules/mute.go +++ /dev/null @@ -1,77 +0,0 @@ -package modules - -import ( - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(mute{}) -} - -type mute struct{} - -func (mute) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "mute", - Version: "0.0.1", - Details: "禁言", - } -} - -func (mute) Start() { // 插件主体 - unWholeBan := zero.OnRegex("^解除全员禁言$", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupWholeBan(event.GroupID, false) - zero.Send(event, "全员自闭结束~") - return zero.SuccessResponse - }, - ) - unWholeBan.Priority = 1 - unWholeBan.Block = true - - wholeBan := zero.OnRegex("^开启全员禁言$", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupWholeBan(event.GroupID, true) - zero.Send(event, "全员自闭开始~") - return zero.SuccessResponse - }, - ) - wholeBan.Priority = 2 - wholeBan.Block = true - - unban := zero.OnRegex("^解除禁言.*?(\\d+)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupBan(event.GroupID, GetInt(state, 1), 0) - zero.Send(event, "小黑屋释放成功~") - return zero.SuccessResponse - }, - ) - unban.Priority = 3 - unban.Block = true - - ban := zero.OnRegex("^禁言.*?(\\d+).*?\\s(\\d+)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupBan(event.GroupID, GetInt(state, 1), GetInt(state, 2)*60) - zero.Send(event, "小黑屋收留成功~") - return zero.SuccessResponse - }, - ) - ban.Priority = 4 - ban.Block = true - - selfBan := zero.OnRegex("^我要自闭.*?(\\d+)分钟", zero.OnlyGroup). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupBan(event.GroupID, event.UserID, GetInt(state, 1)*60) - zero.Send(event, "那我就不手下留情了") - return zero.SuccessResponse - }, - ) - selfBan.Priority = 5 - selfBan.Block = true -} diff --git a/modules/notice.go b/modules/notice.go deleted file mode 100644 index 66d78d0d..00000000 --- a/modules/notice.go +++ /dev/null @@ -1,61 +0,0 @@ -package modules - -import ( - "time" - - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(notice{}) -} - -type notice struct{} - -func (notice) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "notice", - Version: "0.0.1", - Details: "进群退群提醒 戳一戳", - } -} - -func (notice) Start() { // 插件主体 - notify := zero.OnNotice(). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - if event.NoticeType == "notify" { - time.Sleep(time.Second * 1) - zero.Send(event, "请不要戳我 >_<") - } - return zero.SuccessResponse - }, - ) - notify.Priority = 40 - notify.Block = true - - increase := zero.OnNotice(). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - if event.NoticeType == "group_increase" { - zero.Send(event, "欢迎~") - } - return zero.SuccessResponse - }, - ) - increase.Priority = 41 - increase.Block = true - - decrease := zero.OnNotice(). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - if event.NoticeType == "group_decrease" { - zero.Send(event, "有人跑路了") - } - return zero.SuccessResponse - }, - ) - decrease.Priority = 42 - decrease.Block = true -} diff --git a/modules/request.go b/modules/request.go deleted file mode 100644 index d2f4b6dd..00000000 --- a/modules/request.go +++ /dev/null @@ -1,53 +0,0 @@ -package modules - -import ( - "gm/utils" - - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(request{}) -} - -type request struct{} - -func (request) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "request", - Version: "0.0.1", - Details: "申请添加好友 加入群聊 邀请群聊", - } -} - -func (request) Start() { // 插件主体 - friendAdd := zero.OnNotice(). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - if event.RequestType == "friend" { - zero.SendPrivateMessage(utils.Str2Int(Conf.Master[0]), "有人想加我") - } - return zero.SuccessResponse - }, - ) - friendAdd.Priority = 50 - friendAdd.Block = true - - groupAdd := zero.OnNotice(). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - if event.RequestType == "group" { - nickname := GetNickname(event.GroupID, event.UserID) - if event.SubType == "add" { - zero.Send(event, nickname+"("+utils.Int2Str(event.UserID)+") 申请加群") - } else { - zero.SendPrivateMessage(utils.Str2Int(Conf.Master[0]), nickname+"("+utils.Int2Str(event.UserID)+") 邀请我加入群 "+utils.Int2Str(event.GroupID)) - } - } - return zero.SuccessResponse - }, - ) - groupAdd.Priority = 51 - groupAdd.Block = true -} diff --git a/modules/sender.go b/modules/sender.go deleted file mode 100644 index f61a6d5a..00000000 --- a/modules/sender.go +++ /dev/null @@ -1,44 +0,0 @@ -package modules - -import ( - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(sender{}) -} - -type sender struct{} - -func (sender) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "sender", - Version: "0.0.1", - Details: "转发信息", - } -} - -func (sender) Start() { // 插件主体 - promoteManager := zero.OnRegex("^群聊转发.*?(\\d+)\\s(.*)", zero.SuperUserPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SendGroupMessage(GetInt(state, 1), GetStr(state, 2)) - zero.Send(event, "📧 --> "+GetStr(state, 1)) - return zero.SuccessResponse - }, - ) - promoteManager.Priority = 30 - promoteManager.Block = true - - cancleManager := zero.OnRegex("^私聊转发.*?(\\d+)\\s(.*)", zero.SuperUserPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SendPrivateMessage(GetInt(state, 1), GetStr(state, 2)) - zero.Send(event, "📧 --> "+GetStr(state, 1)) - return zero.SuccessResponse - }, - ) - cancleManager.Priority = 31 - cancleManager.Block = true -} diff --git a/modules/title.go b/modules/title.go deleted file mode 100644 index 4ec9cc05..00000000 --- a/modules/title.go +++ /dev/null @@ -1,55 +0,0 @@ -package modules - -import ( - zero "github.com/wdvxdr1123/ZeroBot" -) - -func init() { - zero.RegisterPlugin(title{}) -} - -type title struct{} - -func (title) GetPluginInfo() zero.PluginInfo { // 返回插件信息 - return zero.PluginInfo{ - Author: "kanri", - PluginName: "title", - Version: "0.0.1", - Details: "设置群名片群头衔", - } -} - -func (title) Start() { // 插件主体 - setCard := zero.OnRegex("^修改名片.*?(\\d+).*?\\s(.*)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupCard(event.GroupID, GetInt(state, 1), GetStr(state, 2)) - zero.Send(event, "嗯！已经修改了") - return zero.SuccessResponse - }, - ) - setCard.Priority = 10 - setCard.Block = true - - setTitle := zero.OnRegex("^修改头衔.*?(\\d+).*?\\s(.*)", zero.OnlyGroup, zero.AdminPermission). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupSpecialTitle(event.GroupID, GetInt(state, 1), GetStr(state, 2)) - zero.Send(event, "嗯！已经修改了") - return zero.SuccessResponse - }, - ) - setTitle.Priority = 11 - setTitle.Block = true - - setSelfTitle := zero.OnRegex("^申请头衔(.*)", zero.OnlyGroup). - Handle( - func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response { - zero.SetGroupSpecialTitle(event.GroupID, event.UserID, GetStr(state, 1)) - zero.Send(event, "嗯！不错的头衔呢~") - return zero.SuccessResponse - }, - ) - setSelfTitle.Priority = 12 - setSelfTitle.Block = true -} diff --git a/modules/utils.go b/modules/utils.go deleted file mode 100644 index b01e1347..00000000 --- a/modules/utils.go +++ /dev/null @@ -1,19 +0,0 @@ -package modules - -import ( - "gm/utils" - - zero "github.com/wdvxdr1123/ZeroBot" -) - -func GetInt(state zero.State, index int64) int64 { - return utils.Str2Int(state["regex_matched"].([]string)[index]) -} - -func GetStr(state zero.State, index int64) string { - return state["regex_matched"].([]string)[index] -} - -func GetNickname(groupID int64, userID int64) string { - return zero.GetGroupMemberInfo(groupID, userID, false).Get("nickname").Str -} diff --git a/utils/utils.go b/utils/utils.go deleted file mode 100644 index b6f2e812..00000000 --- a/utils/utils.go +++ /dev/null @@ -1,47 +0,0 @@ -package utils - -import ( - "io/ioutil" - "os" - "strconv" -) - -func PathExecute() string { - dir, _ := os.Getwd() - return dir + "\\" -} - -func PathExists(path string) bool { - _, err := os.Stat(path) - return err == nil || os.IsExist(err) -} - -func ReadAllText(path string) string { - b, err := ioutil.ReadFile(path) - if err != nil { - return "" - } - return string(b) -} - -func WriteAllText(path, text string) { - _ = ioutil.WriteFile(path, []byte(text), 0644) -} - -func CreatePath(path string) error { - err := os.MkdirAll(path, 0644) - if err != nil { - return err - } - return nil -} - -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/utils/yaml.go b/utils/yaml.go deleted file mode 100644 index 58df5677..00000000 --- a/utils/yaml.go +++ /dev/null @@ -1,58 +0,0 @@ -package utils - -import ( - "fmt" - "os" - "strconv" - "time" - - "gopkg.in/yaml.v3" -) - -type YamlConfig struct { - Version string `yaml:"插件版本"` - Host string `yaml:"监听地址"` - Port string `yaml:"监听端口"` - AccessToken string `yaml:"Token"` - Master []string `yaml:"主人QQ"` -} - -func DefaultConfig() *YamlConfig { - return &YamlConfig{ - Version: "1", - Host: "127.0.0.1", - Port: "8080", - AccessToken: "", - Master: []string{ - "66666666", - "88888888", - }, - } -} - -func Load(p string) *YamlConfig { - if !PathExists(p) { - DefaultConfig().Save(p) - } - c := YamlConfig{} - err := yaml.Unmarshal([]byte(ReadAllText(p)), &c) - if err != nil { - fmt.Println("[GroupManager] 尝试加载配置文件失败: 读取文件失败") - fmt.Println("[GroupManager] 原配置文件已备份") - os.Rename(p, p+".backup"+strconv.FormatInt(time.Now().Unix(), 10)) - DefaultConfig().Save(p) - } - c = YamlConfig{} - yaml.Unmarshal([]byte(ReadAllText(p)), &c) - return &c -} - -func (c *YamlConfig) Save(p string) error { - data, err := yaml.Marshal(c) - if err != nil { - fmt.Println("[GroupManager] 写入新的配置文件失败") - return err - } - WriteAllText(p, string(data)) - return nil -}