72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package tgbot
|
||
|
||
import (
|
||
"fmt"
|
||
"math/rand"
|
||
"strconv"
|
||
"tg-mc/conf"
|
||
"tg-mc/models"
|
||
"tg-mc/services/utils"
|
||
commonUtils "tg-mc/utils"
|
||
"time"
|
||
|
||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
||
"github.com/samber/lo"
|
||
)
|
||
|
||
func SetHandler(m *tgbotapi.Message, i interface{}) {
|
||
var a []string
|
||
if t, ok := i.([]string); ok {
|
||
a = t
|
||
} else {
|
||
a = commonUtils.GetArgs(m.CommandArguments())
|
||
}
|
||
|
||
if !utils.IsAdmin(m) {
|
||
tm := tgbotapi.NewMessage(m.Chat.ID, "您不是管理员，没有该权限")
|
||
conf.Bot.Send(tm)
|
||
return
|
||
}
|
||
if len(a) != 3 {
|
||
tm := tgbotapi.NewMessage(m.Chat.ID, "参数错误，样例：\n```\n/set username tgid status\n```")
|
||
tm.ParseMode = "Markdown"
|
||
conf.Bot.Send(tm)
|
||
return
|
||
}
|
||
tgid, err := strconv.ParseInt(a[1], 10, 64)
|
||
if err != nil {
|
||
conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "ID错误，应该为int64"))
|
||
return
|
||
}
|
||
status, err := strconv.ParseInt(a[2], 10, 64)
|
||
if err != nil || !lo.Contains([]int64{0, 1, 2}, status) {
|
||
conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "Status错误，应该为0(Pending),1(Normal),2(Banned)"))
|
||
return
|
||
}
|
||
if err := models.CreateUser(&models.User{
|
||
TGID: tgid,
|
||
MCName: a[0],
|
||
Status: int(status),
|
||
}); err != nil {
|
||
tm := tgbotapi.NewMessage(m.Chat.ID, fmt.Sprintf("创建用户错误，err：\n```\n%+v\n```", err))
|
||
tm.ParseMode = "Markdown"
|
||
conf.Bot.Send(tm)
|
||
return
|
||
}
|
||
conf.Bot.Send(tgbotapi.NewMessage(m.Chat.ID, "设置用户成功"))
|
||
}
|
||
|
||
func BanHandler(m *tgbotapi.Message, i interface{}) {
|
||
rand.Seed(time.Now().UnixNano())
|
||
randomNumber := rand.Intn(11)
|
||
for {
|
||
_, err := models.GetUserByTGID(int64(randomNumber))
|
||
if err != nil {
|
||
break
|
||
} else {
|
||
randomNumber = rand.Intn(11)
|
||
}
|
||
}
|
||
SetHandler(m, []string{m.CommandArguments(), fmt.Sprintf("-%d", randomNumber), "2"})
|
||
}
|