mirror of
https://github.com/purofle/remake_bot.git
synced 2025-06-17 23:03:30 +08:00
321 lines
7.1 KiB
Go
321 lines
7.1 KiB
Go
package bot
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"github.com/purofle/remake_bot/bot/quotely"
|
||
"go.uber.org/zap"
|
||
tele "gopkg.in/telebot.v3"
|
||
"math/rand"
|
||
"strconv"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
type Handler struct {
|
||
bot *tele.Bot
|
||
logger *zap.Logger
|
||
database *sql.DB
|
||
|
||
mutex sync.Mutex
|
||
|
||
remake *Remake
|
||
messageCounter *MessageCounter
|
||
}
|
||
|
||
func NewHandler(bot *tele.Bot, logger *zap.Logger, remake *Remake, counter *MessageCounter) *Handler {
|
||
return &Handler{
|
||
bot: bot,
|
||
logger: logger,
|
||
remake: remake,
|
||
messageCounter: counter,
|
||
}
|
||
}
|
||
|
||
func (h *Handler) RegisterAll() {
|
||
h.bot.Handle(tele.OnQuery, h.InlineQuery)
|
||
h.bot.Handle("/remake", h.CommandRemake)
|
||
h.bot.Handle("/remake_data", h.CommandRemakeData)
|
||
h.bot.Handle("/msg_stats", h.CommandMsgStats)
|
||
h.bot.Handle(tele.OnText, h.CommandOnText)
|
||
h.bot.Handle(tele.OnSticker, h.CommandOnSticker)
|
||
}
|
||
|
||
func (h *Handler) getRandomCountry() Country {
|
||
// 生成随机数
|
||
randomNum := rand.Int63n(h.remake.TotalPopulation)
|
||
|
||
// 根据随机数获取对应的国家
|
||
index := 0
|
||
for i, country := range h.remake.CountryList {
|
||
if randomNum < country.Population {
|
||
index = i
|
||
break
|
||
}
|
||
randomNum -= country.Population
|
||
}
|
||
|
||
return h.remake.CountryList[index]
|
||
}
|
||
|
||
func (h *Handler) CommandRemake(c tele.Context) error {
|
||
msg := c.Message()
|
||
|
||
remakeData := []string{"男孩子", "女孩子", "MtF", "FtM", "MtC", "萝莉", "正太", "武装直升机", "沃尔玛购物袋", "星巴克", "无性别", "扶她", "死胎"}
|
||
|
||
remakeResult := rand.Intn(len(remakeData))
|
||
randomCountry := h.getRandomCountry()
|
||
|
||
func() {
|
||
h.mutex.Lock()
|
||
defer h.mutex.Unlock()
|
||
_, hasKey := h.remake.RemakeCount[c.Sender().ID]
|
||
if !hasKey {
|
||
h.remake.RemakeCount[c.Sender().ID] = new(RemakeData)
|
||
}
|
||
oldGender := h.remake.RemakeCount[c.Sender().ID].count
|
||
h.remake.RemakeCount[c.Sender().ID] = &RemakeData{
|
||
country: randomCountry.CountryName,
|
||
gender: remakeData[remakeResult],
|
||
count: oldGender + 1,
|
||
}
|
||
}()
|
||
|
||
text := fmt.Sprintf("转生成功！您现在是 %s 的 %s 了。", randomCountry.CountryName, remakeData[remakeResult])
|
||
|
||
reply, err := c.Bot().Reply(msg, text)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if c.Chat().Type == tele.ChatPrivate {
|
||
return nil
|
||
}
|
||
|
||
time.AfterFunc(5*time.Second, func() {
|
||
err = c.Bot().Delete(reply)
|
||
err = c.Bot().Delete(msg)
|
||
if err != nil {
|
||
return
|
||
}
|
||
})
|
||
return nil
|
||
}
|
||
|
||
func (h *Handler) CommandRemakeData(c tele.Context) error {
|
||
|
||
msg := c.Message()
|
||
|
||
var text string
|
||
userData, hasKey := h.remake.RemakeCount[c.Sender().ID]
|
||
if hasKey {
|
||
text = fmt.Sprintf("您现在是 %s 的 %s，共 remake 了 %d 次", userData.country, userData.gender, userData.count)
|
||
} else {
|
||
text = "您还没有 remake 过呢，快 /remake 吧"
|
||
}
|
||
|
||
reply, err := c.Bot().Reply(msg, text)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if c.Chat().Type == tele.ChatPrivate {
|
||
return nil
|
||
}
|
||
|
||
time.AfterFunc(10*time.Second, func() {
|
||
err = c.Bot().Delete(reply)
|
||
err = c.Bot().Delete(msg)
|
||
if err != nil {
|
||
return
|
||
}
|
||
})
|
||
return nil
|
||
}
|
||
|
||
func (h *Handler) CommandOnSticker(c tele.Context) error {
|
||
if c.Chat().ID != -1001965344356 {
|
||
return nil
|
||
}
|
||
|
||
err := h.messageCounter.Increment(c.Sender().ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func (h *Handler) CommandMsgStats(c tele.Context) error {
|
||
if c.Chat().ID != -1001965344356 {
|
||
return nil
|
||
}
|
||
|
||
count, err := h.messageCounter.GetCount(c.Sender().ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
topID, topCounter, err := h.messageCounter.GetTopUserInLast24Hours()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 获取用户信息
|
||
member, err := c.Bot().ChatMemberOf(c.Chat(), &tele.User{ID: topID})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
text := fmt.Sprintf("您今天发了 %d 条消息\n最近 24 小时内的龙王是 %s！一共水了 %d 条消息", count, member.User.FirstName, topCounter)
|
||
reply, err := c.Bot().Reply(c.Message(), text)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if c.Chat().Type == tele.ChatPrivate {
|
||
return nil
|
||
}
|
||
|
||
time.AfterFunc(10*time.Second, func() {
|
||
err = c.Bot().Delete(reply)
|
||
err = c.Bot().Delete(c.Message())
|
||
if err != nil {
|
||
return
|
||
}
|
||
})
|
||
return nil
|
||
}
|
||
|
||
func (h *Handler) CommandOnText(c tele.Context) error {
|
||
if c.Chat().ID != -1001965344356 {
|
||
return nil
|
||
}
|
||
|
||
err := h.messageCounter.Increment(c.Sender().ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if c.Message().ReplyTo != nil {
|
||
text := quotely.QuoteReply(c.Bot(), c.Message())
|
||
if text != "" {
|
||
return c.Reply(text, tele.ModeMarkdownV2)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (h *Handler) getQuote(text string) (error, []string, []string) {
|
||
var rows *sql.Rows
|
||
var err error
|
||
if text == "" {
|
||
query := "select text, \"from\" from result_new where from_id not like 'channel%' order by random() limit 50"
|
||
rows, err = h.database.Query(query)
|
||
} else {
|
||
query := "select text, \"from\" from result_new where from_id not like 'channel%' AND text like '%' || $1 || '%' order by random() limit 50"
|
||
rows, err = h.database.Query(query, text)
|
||
}
|
||
if err != nil {
|
||
return err, nil, nil
|
||
}
|
||
|
||
defer func(rows *sql.Rows) {
|
||
err := rows.Close()
|
||
if err != nil {
|
||
return
|
||
}
|
||
}(rows)
|
||
|
||
var resultText []string
|
||
var from []string
|
||
for rows.Next() {
|
||
var t, f string
|
||
if err := rows.Scan(&t, &f); err != nil {
|
||
return err, nil, nil
|
||
}
|
||
resultText = append(resultText, t)
|
||
from = append(from, f)
|
||
}
|
||
|
||
return nil, resultText, from
|
||
}
|
||
|
||
func (h *Handler) InlineQuery(c tele.Context) error {
|
||
member, err := c.Bot().ChatMemberOf(
|
||
&tele.Chat{ID: -1001965344356},
|
||
c.Sender(),
|
||
)
|
||
if err != nil {
|
||
fmt.Println(fmt.Sprintf("sender: %s 不在群", c.Sender().FirstName))
|
||
return err
|
||
}
|
||
if member.Role != "creator" {
|
||
fmt.Println(fmt.Sprintf("sender: %s 不在群", c.Sender().FirstName))
|
||
return nil
|
||
}
|
||
|
||
var resultText []string
|
||
var from []string
|
||
|
||
if c.Query().Text == "" {
|
||
err, resultText, from = h.getQuote("")
|
||
} else {
|
||
err, resultText, from = h.getQuote(c.Query().Text)
|
||
}
|
||
results := make(tele.Results, len(resultText))
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
for i, text := range resultText {
|
||
results[i] = &tele.ArticleResult{
|
||
Title: text,
|
||
Text: fmt.Sprintf("%s: %s", from[i], text),
|
||
Description: fmt.Sprintf("来自 %s", from[i]),
|
||
}
|
||
results[i].SetResultID(strconv.Itoa(i))
|
||
}
|
||
|
||
return c.Answer(&tele.QueryResponse{
|
||
Results: results,
|
||
CacheTime: 0,
|
||
})
|
||
}
|
||
|
||
//func CommandEat(c tele.Context) error {
|
||
// if !(c.Chat().Type == tele.ChatPrivate || c.Chat().ID == -1001965344356) {
|
||
// fmt.Println(c.Chat().ID)
|
||
// return nil
|
||
// }
|
||
//
|
||
// method := []string{"炒", "蒸", "煮", "红烧", "爆炒", "烤", "炸", "煎", "炖", "焖", "炖", "卤"}
|
||
//
|
||
// // 获取时间段
|
||
// hour := time.Now().Hour()
|
||
// var hourText string
|
||
// switch {
|
||
// case hour > 6 && hour <= 10:
|
||
// hourText = "早上"
|
||
// case hour > 10 && hour <= 14:
|
||
// hourText = "中午"
|
||
// case hour > 14 && hour <= 17:
|
||
// hourText = "下午"
|
||
// case hour > 18 && hour <= 21:
|
||
// hourText = "晚上"
|
||
// default:
|
||
// hourText = "宵夜"
|
||
// }
|
||
//
|
||
// var name string
|
||
// if strings.Contains(c.Sender().FirstName, " | ") {
|
||
// name = strings.Split(c.Sender().FirstName, " | ")[0]
|
||
// } else {
|
||
// name = c.Sender().FirstName
|
||
// }
|
||
//
|
||
// result := fmt.Sprintf("今天%s吃 %s %s %s", hourText, name, method[rand.Intn(len(method))], userList[crand.Intn(len(userList))])
|
||
// return c.Reply(result)
|
||
//}
|