新增设置管理员功能 更新菜单

conf/demo_reply.php

@@ -61,6 +61,7 @@ return reply(
             回复  丨 屏蔽
             更新账号 |  清理过期账号
             取消屏蔽 |  任务列表
+            设置管理员
       ———————————
               京东机器人
         ]

models/bot.go

@@ -70,6 +70,11 @@ var handleMessage = func(msgs ...interface{}) interface{} {
 	if sender.UserID == Config.TelegramUserID || sender.UserID == int(Config.QQID) {
 		sender.IsAdmin = true
 	}
+
+	if IsUserAdmin(string(sender.UserID)) {
+		sender.IsAdmin = true
+	}
+
 	for i := range codeSignals {
 		for j := range codeSignals[i].Command {
 			if codeSignals[i].Command[j] == head {

models/command.go

@@ -302,6 +302,15 @@ var codeSignals = []CodeSignal{
 			return nil
 		},
 	},
+	{
+		Command: []string{"设置管理员"},
+		Admin:   true,
+		Handle: func(sender *Sender) interface{} {
+			ctt := sender.JoinContens()
+			db.Create(&UserAdmin{Content: ctt})
+			return "已设置管理员"
+		},
+	},
 	{
 		Command: []string{"赌一把"},
 		Handle: func(sender *Sender) interface{} {

models/db.go

@@ -129,6 +129,7 @@ var Available = "Available"
 var UnAvailable = "UnAvailable"
 var PtKey = "PtKey"
 var PtPin = "PtPin"
+var Content = "Content"
 var WsKey = "WsKey"
 var Address = "Address"
 var Priority = "Priority"

models/useradmin.go

@@ -0,0 +1,15 @@
+package models
+
+type UserAdmin struct {
+	ID      int
+	Content string
+}
+
+func IsUserAdmin(id string) bool {
+	user := UserAdmin{}
+	db.Model(UserAdmin{}).Where(Content+" = ?", id).First(user)
+	if len(user.Content) > 0 {
+		return true
+	}
+	return false
+}