serverus/dickbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

split tg-groups

Browse Source
This commit is contained in:
Smile Rex
2025-11-21 15:49:10 +03:00
parent 8309a64b02
commit 75e01df0a9
6 changed files with 133 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/controllers/dataController.go
View File

@@ -44,8 +44,8 @@ func (d *DataController) ClearAllBlocked() {
}
// CreateUser creates new user with random dick size and sets it to blocked
func (d *DataController) CreateUser(id int64, lname string, fname string) {
_, err := d.DB.Exec("INSERT INTO users (id, fname, lname, dick_size, admin, blocked, trick) VALUES (?,?,?,?,?,?,?)", id, fname, lname, 0, 0, 0, 1)
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 {
@@ -54,15 +54,15 @@ func (d *DataController) CreateUser(id int64, lname string, fname string) {
}
// GetUser gets user's data by id
func (d *DataController) GetUser(id int64) *models.User {
func (d *DataController) GetUser(id int64, chatId int64) *models.User {
var user models.User
row, err := d.DB.Query("SELECT * FROM users WHERE id =?", id)
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.FirstName, &user.LastName, &user.DickSize, &user.Blocked, &user.Admin, &user.Trick)
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)
}
@@ -72,8 +72,8 @@ func (d *DataController) GetUser(id int64) *models.User {
}
// UpdateUser updates user's dick size and blocked status
func (d *DataController) UpdateDick(id int64, newDickSize int) {
_, err := d.DB.Exec("UPDATE users SET dick_size =?, blocked =? WHERE id =?", newDickSize, 1, id)
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 {
@@ -82,8 +82,8 @@ func (d *DataController) UpdateDick(id int64, newDickSize int) {
}
// AddTricks updates user's
func (d *DataController) AddTricks(id int64) {
_, err := d.DB.Exec("UPDATE users SET trick = trick + 1 WHERE id =?", id)
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 {
@@ -92,16 +92,16 @@ func (d *DataController) AddTricks(id int64) {
}
// GetTopUsers gets top 10 users by dick size
func (d *DataController) GetTopUsers() []models.User {
func (d *DataController) GetTopUsers(chatId int64) []models.User {
var users []models.User
rows, err := d.DB.Query("SELECT * FROM users ORDER BY dick_size DESC LIMIT 10")
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.FirstName, &user.LastName, &user.DickSize, &user.Blocked, &user.Admin, &user.Trick)
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)
}