107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package models
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
|
||
tb "gopkg.in/tucnak/telebot.v2"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type User struct {
|
||
ID int
|
||
Number int `gorm:"unique"`
|
||
Class string
|
||
ActiveAt time.Time
|
||
Coin int
|
||
}
|
||
|
||
func NewActiveUser(class string, uid int, msgs ...interface{}) {
|
||
msg := ""
|
||
if class == "tgg" {
|
||
sender := msgs[4].(*tb.User)
|
||
last := ""
|
||
if sender.LastName != "" {
|
||
last = " " + sender.LastName
|
||
}
|
||
if sender.Username == "" {
|
||
msg = fmt.Sprintf(`@%s%s `, sender.FirstName, last)
|
||
} else {
|
||
msg = fmt.Sprintf(`@%s `, sender.Username)
|
||
}
|
||
|
||
class = "tg"
|
||
}
|
||
if class == "qqg" {
|
||
class = "qq"
|
||
}
|
||
zero, _ := time.ParseInLocation("2006-01-02", time.Now().Local().Format("2006-01-02"), time.Local)
|
||
var u User
|
||
var ntime = time.Now()
|
||
var first = false
|
||
total := []int{}
|
||
err := db.Where("class = ? and number = ?", class, uid).First(&u).Error
|
||
if err != nil {
|
||
first = true
|
||
u = User{
|
||
Class: class,
|
||
Number: uid,
|
||
Coin: 1,
|
||
ActiveAt: ntime,
|
||
}
|
||
if err := db.Create(&u).Error; err != nil {
|
||
return
|
||
}
|
||
} else {
|
||
if zero.Unix() > u.ActiveAt.Unix() {
|
||
first = true
|
||
} else {
|
||
msg += fmt.Sprintf("你打过卡了，许愿币余额%d。", u.Coin)
|
||
sendMessagee(msg, msgs...)
|
||
}
|
||
}
|
||
if first {
|
||
db.Model(User{}).Select("count(id) as total").Where("active_at > ?", zero).Pluck("total", &total)
|
||
coin := 1
|
||
if total[0]%3 == 0 {
|
||
coin = 2
|
||
}
|
||
if total[0]%13 == 0 {
|
||
coin = 8
|
||
}
|
||
db.Model(&u).Updates(map[string]interface{}{
|
||
"active_at": ntime,
|
||
"coin": gorm.Expr(fmt.Sprintf("coin+%d", coin)),
|
||
})
|
||
u.Coin += coin
|
||
msg += fmt.Sprintf("你是打卡第%d人，奖励%d个许愿币，许愿币余额%d。", total[0]+1, coin, u.Coin)
|
||
sendMessagee(msg, msgs...)
|
||
}
|
||
}
|
||
|
||
func AddCoin(uid int) int {
|
||
var u User
|
||
db.Where("number = ?", uid).First(&u)
|
||
db.Model(u).Updates(map[string]interface{}{
|
||
"coin": gorm.Expr("coin+1"),
|
||
})
|
||
u.Coin++
|
||
return u.Coin
|
||
}
|
||
|
||
func RemCoin(uid int, num int) int {
|
||
var u User
|
||
db.Where("number = ?", uid).First(&u)
|
||
db.Model(u).Updates(map[string]interface{}{
|
||
"coin": gorm.Expr(fmt.Sprintf("coin-%d", num)),
|
||
})
|
||
u.Coin--
|
||
return u.Coin
|
||
}
|
||
|
||
func GetCoin(uid int) int {
|
||
var u User
|
||
db.Where("number = ?", uid).First(&u)
|
||
return u.Coin
|
||
}
|