mirror of
https://github.com/purofle/remake_bot.git
synced 2025-02-06 17:13:29 +08:00
146 lines
3.2 KiB
Go
146 lines
3.2 KiB
Go
package main
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/purofle/remake_bot/quotely"
|
||
tele "gopkg.in/telebot.v3"
|
||
"math/rand"
|
||
crand "math/rand"
|
||
"os"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
type Country struct {
|
||
CountryName string `json:"country"`
|
||
Population int64 `json:"population"`
|
||
}
|
||
|
||
var (
|
||
countryList []Country
|
||
userList []string
|
||
totalPopulation int64
|
||
mutex sync.Mutex
|
||
)
|
||
|
||
func initList() error {
|
||
rawJson, err := os.ReadFile("countries.json")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if err = json.Unmarshal(rawJson, &countryList); err != nil {
|
||
return err
|
||
}
|
||
|
||
totalPopulation = int64(0)
|
||
for _, country := range countryList {
|
||
totalPopulation += country.Population
|
||
}
|
||
|
||
rawJson, err = os.ReadFile("user_list.json")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err = json.Unmarshal(rawJson, &userList); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func getRandomCountry() Country {
|
||
// 生成随机数
|
||
randomNum := rand.Int63n(totalPopulation)
|
||
|
||
// 根据随机数获取对应的国家
|
||
index := 0
|
||
for i, country := range countryList {
|
||
if randomNum < country.Population {
|
||
index = i
|
||
break
|
||
}
|
||
randomNum -= country.Population
|
||
}
|
||
|
||
return countryList[index]
|
||
}
|
||
|
||
func CommandRemake(c tele.Context) error {
|
||
|
||
remakeData := []string{"男孩子", "女孩子", "MtF", "FtM", "MtC", "萝莉", "正太", "武装直升机", "沃尔玛购物袋", "星巴克", "太监", "无性别", "扶她", "死胎"}
|
||
|
||
remakeResult := rand.Intn(len(remakeData))
|
||
randomCountry := getRandomCountry()
|
||
|
||
mutex.Lock()
|
||
_, hasKey := remakeCount[c.Sender().ID]
|
||
if !hasKey {
|
||
remakeCount[c.Sender().ID] = new(RemakeData)
|
||
}
|
||
oldGender := remakeCount[c.Sender().ID].count
|
||
remakeCount[c.Sender().ID] = &RemakeData{
|
||
country: randomCountry.CountryName,
|
||
gender: remakeData[remakeResult],
|
||
count: oldGender + 1,
|
||
}
|
||
mutex.Unlock()
|
||
|
||
text := fmt.Sprintf("转生成功！您现在是 %s 的 %s 了。", randomCountry.CountryName, remakeData[remakeResult])
|
||
|
||
return c.Reply(text)
|
||
}
|
||
|
||
func CommandRemakeData(c tele.Context) error {
|
||
var text string
|
||
userData, hasKey := remakeCount[c.Sender().ID]
|
||
if hasKey {
|
||
text = fmt.Sprintf("您现在是 %s 的 %s，共 remake 了 %d 次", userData.country, userData.gender, userData.count)
|
||
} else {
|
||
text = "您还没有 remake 过呢，快 /remake 吧"
|
||
}
|
||
|
||
return c.Reply(text)
|
||
}
|
||
|
||
func CommandEat(c tele.Context) error {
|
||
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)
|
||
}
|
||
|
||
func CommandOnText(c tele.Context) error {
|
||
if c.Message().ReplyTo != nil {
|
||
text := quotely.QuoteReply(c.Bot(), c.Message())
|
||
if text != "" {
|
||
return c.Reply(text, tele.ModeMarkdownV2)
|
||
}
|
||
}
|
||
return nil
|
||
}
|