78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
|
package reborn
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"math/rand"
|
|||
|
|
"os"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
tele "gopkg.in/telebot.v3"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Country struct {
|
|||
|
|
Name string `json:"country"`
|
|||
|
|
Pop int64 `json:"population"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type RebornData struct {
|
|||
|
|
List []Country
|
|||
|
|
TotalPop int64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func InitRebornList(filePath string) (*RebornData, error) {
|
|||
|
|
rawJson, err := os.ReadFile(filePath)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to read file: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var countries []Country
|
|||
|
|
if err = json.Unmarshal(rawJson, &countries); err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to parse JSON: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
totalPop := int64(0)
|
|||
|
|
for _, country := range countries {
|
|||
|
|
totalPop += country.Pop
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &RebornData{
|
|||
|
|
List: countries,
|
|||
|
|
TotalPop: totalPop,
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (data *RebornData) randCountry(r *rand.Rand) (Country, error) {
|
|||
|
|
randNum := r.Int63n(data.TotalPop)
|
|||
|
|
|
|||
|
|
for _, country := range data.List {
|
|||
|
|
if randNum < country.Pop {
|
|||
|
|
return country, nil
|
|||
|
|
}
|
|||
|
|
randNum -= country.Pop
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return Country{}, fmt.Errorf("failed to select a country")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func randGender(r *rand.Rand) string {
|
|||
|
|
genders := []string{"男孩子", "女孩子", " MtF", " FtM", "萝莉", "正太", "武装直升机", "沃尔玛购物袋", "狗狗", "猫猫"}
|
|||
|
|
return genders[r.Intn(len(genders))]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Execute(c tele.Context, data *RebornData) error {
|
|||
|
|
source := rand.NewSource(time.Now().UnixNano())
|
|||
|
|
r := rand.New(source)
|
|||
|
|
|
|||
|
|
country, err := data.randCountry(r)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
gender := randGender(r)
|
|||
|
|
|
|||
|
|
outputText := fmt.Sprintf("投胎成功！\n你出生在%s，是%s。", country.Name, gender)
|
|||
|
|
|
|||
|
|
return c.Reply(outputText)
|
|||
|
|
}
|