diff --git a/src/controllers/dataController.go b/src/controllers/dataController.go index 9b333b9..f3e8f0f 100644 --- a/src/controllers/dataController.go +++ b/src/controllers/dataController.go @@ -81,6 +81,16 @@ 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) + 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() []models.User { var users []models.User diff --git a/src/controllers/dickController.go b/src/controllers/dickController.go index 1ebe471..0bb5da9 100644 --- a/src/controllers/dickController.go +++ b/src/controllers/dickController.go @@ -35,6 +35,7 @@ func (d *DickController) Dick(c tele.Context) error { retMessage = fmt.Sprintf("%s %s, ты уже отращивал свой писюн, следющая попытка будет после 22:00 по МСК.", fName, lName) } else { d.DC.UpdateDick(id, newSize) + d.DC.AddTricks(id) if newRange == 0 { retMessage = fmt.Sprintf("%s %s, твой писюн сегодня не подрос. Не переживай, просто он, видимо, решил взять выходной — пусть лучше отдохнет перед завтрашними подвигами!", fName, lName)