113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package controllers
|
|
|
|
import (
|
|
"bot/src/models"
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"math/rand"
|
|
"time"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
type DataController struct{ DB *sql.DB }
|
|
|
|
func NewDB(config models.Config) *DataController {
|
|
dbaddr := fmt.Sprintf("%s:%s@tcp(%s)/%s", config.UserName, config.Password, config.HostName, config.DbName)
|
|
|
|
db, err := sql.Open("mysql", dbaddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
db.SetConnMaxLifetime(time.Minute * 3)
|
|
db.SetMaxOpenConns(10)
|
|
db.SetMaxIdleConns(10)
|
|
|
|
return &DataController{DB: db}
|
|
}
|
|
|
|
func RandRange() int {
|
|
rand.Seed(time.Now().UnixNano())
|
|
return rand.Intn(21) - 5
|
|
}
|
|
|
|
// UpdateUser updates user's dick size and checks if it's blocked
|
|
func (d *DataController) ClearAllBlocked() {
|
|
_, err := d.DB.Exec("UPDATE users SET blocked = false")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
log.Println("Clear block stats complete")
|
|
}
|
|
}
|
|
|
|
// CreateUser creates new user with random dick size and sets it to blocked
|
|
func (d *DataController) CreateUser(id int64, chatId int64, lname string, fname string) {
|
|
_, err := d.DB.Exec("INSERT INTO users (id, chat_id, fname, lname, dick_size, admin, blocked, trick) VALUES (?,?,?,?,?,?,?,?)", id, chatId, fname, lname, 0, 0, 0, 1)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
log.Printf("User %d created", id)
|
|
}
|
|
}
|
|
|
|
// GetUser gets user's data by id
|
|
func (d *DataController) GetUser(id int64, chatId int64) *models.User {
|
|
var user models.User
|
|
row, err := d.DB.Query("SELECT * FROM users WHERE id = ? AND chat_id = ?", id, chatId)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for row.Next() {
|
|
err := row.Scan(&user.ID, &user.ChatID, &user.FirstName, &user.LastName, &user.DickSize, &user.Blocked, &user.Admin, &user.Trick)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
return &user
|
|
}
|
|
|
|
// UpdateUser updates user's dick size and blocked status
|
|
func (d *DataController) UpdateDick(id int64, chatId int64, newDickSize int) {
|
|
_, err := d.DB.Exec("UPDATE users SET dick_size =?, blocked =? WHERE id =? AND chat_id = ?", newDickSize, 1, id, chatId)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
log.Printf("User %d dick size updated", id)
|
|
}
|
|
}
|
|
|
|
// AddTricks updates user's
|
|
func (d *DataController) AddTricks(id int64, chatId int64) {
|
|
_, err := d.DB.Exec("UPDATE users SET trick = trick + 1 WHERE id =? AND chat_id = ?", id, chatId)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
log.Printf("User %d tricks updated", id)
|
|
}
|
|
}
|
|
|
|
// GetTopUsers gets top 10 users by dick size
|
|
func (d *DataController) GetTopUsers(chatId int64) []models.User {
|
|
var users []models.User
|
|
rows, err := d.DB.Query("SELECT * FROM users WHERE chat_id = ? ORDER BY dick_size DESC LIMIT 10", chatId)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for rows.Next() {
|
|
var user models.User
|
|
err := rows.Scan(&user.ID, &user.ChatID, &user.FirstName, &user.LastName, &user.DickSize, &user.Blocked, &user.Admin, &user.Trick)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
users = append(users, user)
|
|
}
|
|
|
|
return users
|
|
}
|