diff --git a/conf/demo_reply.php b/conf/demo_reply.php index bce055f..cee2b3e 100644 --- a/conf/demo_reply.php +++ b/conf/demo_reply.php @@ -61,6 +61,7 @@ return reply( 回复 丨 屏蔽 更新账号 | 清理过期账号 取消屏蔽 | 任务列表 + 设置管理员 ——————————— 京东机器人 ] diff --git a/models/bot.go b/models/bot.go index fe8ec3b..35fb239 100644 --- a/models/bot.go +++ b/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 { diff --git a/models/command.go b/models/command.go index 5caae23..101dbfb 100644 --- a/models/command.go +++ b/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{} { diff --git a/models/db.go b/models/db.go index 69bb6e9..49ea371 100644 --- a/models/db.go +++ b/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" diff --git a/models/useradmin.go b/models/useradmin.go new file mode 100644 index 0000000..dc1749d --- /dev/null +++ b/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 +}