✨ 增加-u参数说明

README.md

@@ -20,13 +20,20 @@
 
 ## 命令行参数
 ```bash
-zerobot [-d] [-g] qq1 qq2 qq3 ...
+zerobot -h -t token -u url [-d|w] [-g] qq1 qq2 qq3 ...
 ```
-- **-d**: 开启 debug 级别日志输出
+- **-h**: 显示帮助
+- **-t token**: 设置`AccessToken`，默认为空
+- **-u url**: 设置`Url`，默认为`ws://127.0.0.1:6700`
+- **-d|w**: 开启 debug | warning 级别及以上日志输出
 - **-g**: 开启 [webgui](https://github.com/FloatTech/bot-manager)
 - **qqs**: superusers 的 qq 号
 
 ## 功能
+> 在编译时，以下功能除插件控制外，均可通过注释`main.go`中的相应`import`而物理禁用，减小插件体积。
+> 通过插件控制，还可动态管理某个功能在某个群的打开/关闭。
+- **web管理**
+    - 需要配合 [webgui](https://github.com/FloatTech/bot-manager) 使用
 - **插件控制**
     - [x] /启用 xxx
     - [x] /禁用 xxx
@@ -179,7 +186,7 @@ zerobot [-d] [-g] qq1 qq2 qq3 ...
 1. 点击右上角 Fork 本项目，并转跳到自己 Fork 的仓库
 2. 点击仓库上方的 Actions 按钮，确认使用 Actions
 3. 编辑 main.go 文件，内容按需修改
-4. 前往 Release 页面发布一个 Release，`tag`形如`vx.y.z`，以触发稳定版编译流程
+4. 前往 Release 页面发布一个 Release，`tag`形如`v1.2.3`，以触发稳定版编译流程
 5. 点击 Actions 按钮，等待编译完成，回到 Release 页面下载编译好的文件
 6. 运行 OneBot 框架，并同时运行本插件
 7. 啾咪~

control/rule.go

@@ -49,8 +49,8 @@ func newctrl(service string, o *Options) *Control {
 	return m
 }
 
-// enable enables a group to pass the Manager.
-func (m *Control) enable(groupID int64) {
+// Enable enables a group to pass the Manager.
+func (m *Control) Enable(groupID int64) {
 	m.Lock()
 	err := db.Insert(m.service, &grpcfg{groupID, 0})
 	if err != nil {
@@ -59,8 +59,8 @@ func (m *Control) enable(groupID int64) {
 	m.Unlock()
 }
 
-// disable disables a group to pass the Manager.
-func (m *Control) disable(groupID int64) {
+// Disable disables a group to pass the Manager.
+func (m *Control) Disable(groupID int64) {
 	m.Lock()
 	err := db.Insert(m.service, &grpcfg{groupID, 1})
 	if err != nil {
@@ -69,7 +69,7 @@ func (m *Control) disable(groupID int64) {
 	m.Unlock()
 }
 
-func (m *Control) isEnabledIn(gid int64) bool {
+func (m *Control) IsEnabledIn(gid int64) bool {
 	m.RLock()
 	var c grpcfg
 	err := db.Find(m.service, &c, "WHERE gid = "+strconv.FormatInt(gid, 10))
@@ -81,9 +81,9 @@ func (m *Control) isEnabledIn(gid int64) bool {
 	logrus.Errorf("[control] %v", err)
 	m.RUnlock()
 	if m.options.DisableOnDefault {
-		m.disable(gid)
+		m.Disable(gid)
 	} else {
-		m.enable(gid)
+		m.Enable(gid)
 	}
 	return !m.options.DisableOnDefault
 }
@@ -92,21 +92,21 @@ func (m *Control) isEnabledIn(gid int64) bool {
 func (m *Control) Handler() zero.Rule {
 	return func(ctx *zero.Ctx) bool {
 		ctx.State["manager"] = m
-		return m.isEnabledIn(ctx.Event.GroupID)
+		return m.IsEnabledIn(ctx.Event.GroupID)
 	}
 }
 
-// lookup returns a Manager by the service name, if
+// Lookup returns a Manager by the service name, if
 // not exist, it will returns nil.
-func lookup(service string) (*Control, bool) {
+func Lookup(service string) (*Control, bool) {
 	mu.RLock()
 	defer mu.RUnlock()
 	m, ok := managers[service]
 	return m, ok
 }
 
-// forEach iterates through managers.
-func forEach(iterator func(key string, manager *Control) bool) {
+// ForEach iterates through managers.
+func ForEach(iterator func(key string, manager *Control) bool) {
 	mu.RLock()
 	m := copyMap(managers)
 	mu.RUnlock()
@@ -138,11 +138,11 @@ func init() {
 					Handle(func(ctx *zero.Ctx) {
 						model := extension.CommandModel{}
 						_ = ctx.Parse(&model)
-						service, ok := lookup(model.Args)
+						service, ok := Lookup(model.Args)
 						if !ok {
 							ctx.Send("没有找到指定服务!")
 						}
-						service.enable(ctx.Event.GroupID)
+						service.Enable(ctx.Event.GroupID)
 						ctx.Send(message.Text("已启用服务: " + model.Args))
 					})
 
@@ -150,11 +150,11 @@ func init() {
 					Handle(func(ctx *zero.Ctx) {
 						model := extension.CommandModel{}
 						_ = ctx.Parse(&model)
-						service, ok := lookup(model.Args)
+						service, ok := Lookup(model.Args)
 						if !ok {
 							ctx.Send("没有找到指定服务!")
 						}
-						service.disable(ctx.Event.GroupID)
+						service.Disable(ctx.Event.GroupID)
 						ctx.Send(message.Text("已关闭服务: " + model.Args))
 					})
 
@@ -162,7 +162,7 @@ func init() {
 					Handle(func(ctx *zero.Ctx) {
 						model := extension.CommandModel{}
 						_ = ctx.Parse(&model)
-						service, ok := lookup(model.Args)
+						service, ok := Lookup(model.Args)
 						if !ok {
 							ctx.Send("没有找到指定服务!")
 						}
@@ -177,10 +177,10 @@ func init() {
 					Handle(func(ctx *zero.Ctx) {
 						msg := `---服务列表---`
 						i := 0
-						forEach(func(key string, manager *Control) bool {
+						ForEach(func(key string, manager *Control) bool {
 							i++
 							msg += "\n" + strconv.Itoa(i) + `: `
-							if manager.isEnabledIn(ctx.Event.GroupID) {
+							if manager.IsEnabledIn(ctx.Event.GroupID) {
 								msg += "●" + key
 							} else {
 								msg += "○" + key

control/web/gui.go

@@ -1,4 +1,4 @@
-package control
+package web
 
 import (
 	"encoding/json"
@@ -17,6 +17,8 @@ import (
 	log "github.com/sirupsen/logrus"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
+
+	ctrl "github.com/FloatTech/ZeroBot-Plugin/control"
 )
 
 var (
@@ -34,8 +36,8 @@ var (
 type logWriter struct {
 }
 
-// InitGui 初始化gui
-func InitGui() {
+// initGui 初始化gui
+func initGui() {
 	// 将日志重定向到前端hook
 	writer := io.MultiWriter(l, os.Stderr)
 	log.SetOutput(writer)
@@ -86,8 +88,8 @@ func controller() {
 	// 获取插件列表
 	engine.POST("/get_plugins", func(context *gin.Context) {
 		var datas []map[string]interface{}
-		forEach(func(key string, manager *Control) bool {
-			datas = append(datas, map[string]interface{}{"id": 1, "handle_type": "", "name": key, "enable": manager.isEnabledIn(0)})
+		ctrl.ForEach(func(key string, manager *ctrl.Control) bool {
+			datas = append(datas, map[string]interface{}{"id": 1, "handle_type": "", "name": key, "enable": manager.IsEnabledIn(0)})
 			return true
 		})
 		context.JSON(200, datas)
@@ -134,14 +136,14 @@ func updateAllPluginStatus(context *gin.Context) {
 		return true
 	})
 
-	forEach(func(key string, manager *Control) bool {
+	ctrl.ForEach(func(key string, manager *ctrl.Control) bool {
 		if enable {
 			for _, group := range groups {
-				manager.enable(group)
+				manager.Enable(group)
 			}
 		} else {
 			for _, group := range groups {
-				manager.disable(group)
+				manager.Disable(group)
 			}
 		}
 		return true
@@ -168,7 +170,7 @@ func updatePluginAllGroupStatus(context *gin.Context) {
 		name = parse["name"].(string)
 		enable = parse["enable"].(bool)
 	}
-	control, b := lookup(name)
+	control, b := ctrl.Lookup(name)
 	if !b {
 		context.JSON(404, nil)
 		return
@@ -176,9 +178,9 @@ func updatePluginAllGroupStatus(context *gin.Context) {
 	zero.RangeBot(func(id int64, ctx *zero.Ctx) bool {
 		for _, group := range ctx.GetGroupList().Array() {
 			if enable {
-				control.enable(group.Get("group_id").Int())
+				control.Enable(group.Get("group_id").Int())
 			} else {
-				control.disable(group.Get("group_id").Int())
+				control.Disable(group.Get("group_id").Int())
 			}
 		}
 
@@ -205,15 +207,15 @@ func updatePluginStatus(context *gin.Context) {
 	name := parse["name"].(string)
 	enable := parse["enable"].(bool)
 	fmt.Println(name)
-	control, b := lookup(name)
+	control, b := ctrl.Lookup(name)
 	if !b {
 		context.JSON(404, "服务不存在")
 		return
 	}
 	if enable {
-		control.enable(groupID)
+		control.Enable(groupID)
 	} else {
-		control.disable(groupID)
+		control.Disable(groupID)
 	}
 	context.JSON(200, nil)
 }
@@ -237,12 +239,12 @@ func getPluginStatus(context *gin.Context) {
 		groupID = int64(parse["group_id"].(float64))
 		name = parse["name"].(string)
 	}
-	control, b := lookup(name)
+	control, b := ctrl.Lookup(name)
 	if !b {
 		context.JSON(404, "服务不存在")
 		return
 	}
-	context.JSON(200, gin.H{"enable": control.isEnabledIn(groupID)})
+	context.JSON(200, gin.H{"enable": control.IsEnabledIn(groupID)})
 }
 
 // getPluginsStatus
@@ -263,8 +265,8 @@ func getPluginsStatus(context *gin.Context) {
 		groupID = int64(parse["group_id"].(float64))
 	}
 	var datas []map[string]interface{}
-	forEach(func(key string, manager *Control) bool {
-		enable := manager.isEnabledIn(groupID)
+	ctrl.ForEach(func(key string, manager *ctrl.Control) bool {
+		enable := manager.IsEnabledIn(groupID)
 		datas = append(datas, map[string]interface{}{"name": key, "enable": enable})
 		return true
 	})

control/web/init.go

@@ -0,0 +1,11 @@
+// Package web 网页管理后端
+package web
+
+import "flag"
+
+func init() {
+	// 解析命令行参数，输入 `-g` 即可启用 gui
+	if *flag.Bool("g", false, "Enable web gui.") {
+		initGui()
+	}
+}

main.go

@@ -3,13 +3,16 @@ package main
 import (
 	"flag"
 	"fmt"
+	"os"
 	"strings"
 
 	// 注：以下插件均可通过前面加 // 注释，注释后停用并不加载插件
 	// 下列插件可与 wdvxdr1123/ZeroBot v1.1.2 以上配合单独使用
-	// 词库类
-	"github.com/sirupsen/logrus"
 
+	// 插件控制
+	_ "github.com/FloatTech/ZeroBot-Plugin/control/web" // web 后端控制
+
+	// 词库类
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_atri"      // ATRI词库
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_chat"      // 基础词库
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_qingyunke" // 青云客
@@ -45,11 +48,9 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_setutime"     // 来份涩图
 
 	// 以下为内置依赖，勿动
-
+	"github.com/sirupsen/logrus"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/driver"
-
-	"github.com/FloatTech/ZeroBot-Plugin/control"
 )
 
 var (
@@ -60,30 +61,45 @@ var (
 		"* Project: https://github.com/FloatTech/ZeroBot-Plugin",
 	}
 	banner = strings.Join(contents, "\n")
+	token  *string
+	url    *string
 )
 
 func init() {
-	var en bool
-	var debg bool
-	// 解析命令行参数，输入 `-g` 即可启用 gui
-	flag.BoolVar(&en, "g", false, "Enable web gui.")
-	// 解析命令行参数，输入 `-d` 即可开启 debug log
-	flag.BoolVar(&debg, "d", false, "Enable debug log.")
+	// 解析命令行参数
+	d := flag.Bool("d", false, "Enable debug level log and higher.")
+	w := flag.Bool("w", false, "Enable warning level log and higher.")
+	h := flag.Bool("h", false, "Display this help.")
+	// 直接写死 AccessToken 时，请更改下面第二个参数
+	token = flag.String("t", "", "Set AccessToken of WSClient.")
+	// 直接写死 URL 时，请更改下面第二个参数
+	url = flag.String("u", "ws://127.0.0.1:6700", "Set Url of WSClient.")
 	flag.Parse()
-	if en {
-		control.InitGui()
-	}
-	if debg {
-		logrus.SetLevel(logrus.DebugLevel)
+	if *h {
+		printBanner()
+		fmt.Println("Usage:")
+		flag.PrintDefaults()
+		os.Exit(0)
+	} else {
+		if *d && !*w {
+			logrus.SetLevel(logrus.DebugLevel)
+		}
+		if *w {
+			logrus.SetLevel(logrus.WarnLevel)
+		}
 	}
 }
 
-func main() {
+func printBanner() {
 	fmt.Print(
 		"\n======================[ZeroBot-Plugin]======================",
 		"\n", banner, "\n",
 		"============================================================\n",
-	) // 启动打印
+	)
+}
+
+func main() {
+	printBanner()
 	zero.Run(zero.Config{
 		NickName:      []string{"椛椛", "ATRI", "atri", "亚托莉", "アトリ"},
 		CommandPrefix: "/",
@@ -96,8 +112,8 @@ func main() {
 		Driver: []zero.Driver{
 			&driver.WSClient{
 				// OneBot 正向WS 默认使用 6700 端口
-				Url:         "ws://127.0.0.1:6700",
-				AccessToken: "",
+				Url:         *url,
+				AccessToken: *token,
 			},
 		},
 	})

plugin_fortune/fortune.go

@@ -53,14 +53,20 @@ func init() {
 			"- 运势|抽签\n" +
 			"- 设置底图[车万 DC4 爱因斯坦 星空列车 樱云之恋 富婆妹 李清歌 公主连结 原神 明日方舟 碧蓝航线 碧蓝幻想 战双 阴阳师]",
 	})
-	en.OnRegex(`^设置底图(.*)`, zero.OnlyGroup).SetBlock(true).SecondPriority().
+	en.OnRegex(`^设置底图(.*)`).SetBlock(true).SecondPriority().
 		Handle(func(ctx *zero.Ctx) {
+			gid := ctx.Event.GroupID
+			if gid <= 0 {
+				// 个人用户设为负数
+				gid = -ctx.Event.UserID
+			}
 			i, ok := index[ctx.State["regex_matched"].([]string)[1]]
 			if ok {
-				conf.Kind[ctx.Event.GroupID] = i
+				conf.Kind[gid] = i
 				savecfg("cfg.pb")
+				ctx.SendChain(message.Text("设置成功~"))
 			} else {
-				ctx.Send("没有这个底图哦～")
+				ctx.SendChain(message.Text("没有这个底图哦～"))
 			}
 		})
 	en.OnFullMatchGroup([]string{"运势", "抽签"}).SetBlock(true).SecondPriority().
@@ -89,7 +95,12 @@ func init() {
 			}
 			// 获取该群背景类型，默认车万
 			kind := "车万"
-			if v, ok := conf.Kind[ctx.Event.GroupID]; ok {
+			gid := ctx.Event.GroupID
+			if gid <= 0 {
+				// 个人用户设为负数
+				gid = -ctx.Event.UserID
+			}
+			if v, ok := conf.Kind[gid]; ok {
 				kind = table[v]
 			}
 			// 检查背景图片是否存在
@@ -116,13 +127,13 @@ func init() {
 			t, _ := strconv.ParseInt(time.Now().Format("20060102"), 10, 64)
 			seed := ctx.Event.UserID + t
 			// 随机获取背景
-			background, err := randimage(base+kind+"/", seed)
+			background, err := randimage(folder+"/", seed)
 			if err != nil {
 				ctx.SendChain(message.Text("ERROR: ", err))
 				return
 			}
 			// 随机获取签文
-			title, text, err := randtext(base+"运势签文.json", seed)
+			title, text, err := randtext(mikuji, seed)
 			if err != nil {
 				ctx.SendChain(message.Text("ERROR: ", err))
 				return