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

add future

Browse Source
This commit is contained in:
SmileRex
2025-01-23 01:32:55 +03:00
parent e7573abcd0
commit ed8eb2431b
9 changed files with 301 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
src/controllers/dataController.go
View File

@@ -2,25 +2,40 @@ package controllers
import (
"bot/src/models"
"database/sql"
"fmt"
"log"
"math/rand"
"time"
"github.com/glebarez/sqlite"
"github.com/robfig/cron"
"gorm.io/gorm"
_ "github.com/go-sql-driver/mysql"
)
type DataController struct {
DB *gorm.DB
type DataConfig struct {
UserName string
Password string
HostName string
DbName string
}
func (d *DataConfig) Get() string {
return fmt.Sprintf("%s:%s@tcp(%s)/%s", d.UserName, d.Password, d.HostName, d.DbName)
}
type DataController struct{ DB *sql.DB }
func NewDB() *DataController {
db, err := gorm.Open(sqlite.Open("users.db"), &gorm.Config{})
config := DataConfig{UserName: "root", Password: "12641264", HostName: "217.12.40.237", DbName: "test"}
db, err := sql.Open("mysql", config.Get())
if err != nil {
panic("failed to connect database")
panic(err)
}
db.AutoMigrate(models.User{})
db.SetConnMaxLifetime(time.Minute * 3)
db.SetMaxOpenConns(10)
db.SetMaxIdleConns(10)
return &DataController{DB: db}
}
@@ -29,16 +44,70 @@ func RandRange() int {
return rand.Intn(21) - 5
}
// UpdateUser updates user's dick size and checks if it's blocked
func (d *DataController) ClearAllBlocked() {
d.DB.Model(&models.User{}).Where("is_blocked = ?", true).Update("is_blocked", false)
_, err := d.DB.Exec("UPDATE users SET blocked = false")
if err != nil {
log.Fatal(err)
} else {
log.Println("Clear block stats complete")
}
}
func (d *DataController) CronRun() {
c := cron.New()
c.AddFunc("0 20 * * *", func() {
d.ClearAllBlocked()
})
c.Start()
// 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)
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) *models.User {
var user models.User
row, err := d.DB.Query("SELECT * FROM users WHERE id =?", id)
if err != nil {
log.Fatal(err)
}
for row.Next() {
err := row.Scan(&user.ID, &user.FirstName, &user.LastName, &user.DickSize, &user.Admin, &user.Blocked, &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, newDickSize int) {
_, err := d.DB.Exec("UPDATE users SET dick_size =?, blocked =?, trick =? WHERE id =?", newDickSize, 1, +1, id)
if err != nil {
log.Fatal(err)
} else {
log.Printf("User %d dick size updated", id)
}
}
// GetTopUsers gets top 10 users by dick size
func (d *DataController) GetTopUsers() []models.User {
var users []models.User
rows, err := d.DB.Query("SELECT * FROM users ORDER BY dick_size DESC LIMIT 10")
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)
if err != nil {
log.Fatal(err)
}
users = append(users, user)
}
return users
}

170
src/controllers/duelController.go Normal file
View File

@@ -0,0 +1,170 @@
package controllers
import (
"fmt"
"math/rand"
"strconv"
"strings"
"time"
tele "gopkg.in/telebot.v4"
)
type DuelController struct {
PlayerOne struct {
ID int64
Counter int
FName string
}
PlayerTwo struct {
ID int64
Counter int
FName string
}
Summ int
Inited bool
Bot *tele.Bot
DC *DataController
}
// Controller Init
func NewDuel(bot *tele.Bot, dc *DataController) *DuelController {
return &DuelController{
Inited: false,
Bot: bot,
DC: dc,
}
}
// Start Match
func (s *DuelController) StartMatch(c tele.Context) error {
if s.Inited {
return c.Send(fmt.Sprintf("Дуэль уже началась!\nИгрок: %s\nСтавка: %d", s.PlayerOne.FName, s.Summ))
}
s.Inited = true
s.PlayerOne.ID = c.Sender().ID
s.PlayerOne.FName = c.Sender().FirstName
user := s.DC.GetUser(s.PlayerOne.ID)
if c.Text() == "/duel" {
return c.Send("Используйте /duel <сумма> для начала дуэли!")
}
a := strings.Split(c.Text(), "/duel ")[1]
summ, err := strconv.Atoi(a)
if err != nil {
return c.Send("Некорректная команда, используйте /duel <сумма>")
} else {
s.Summ = summ
}
if user.DickSize < summ {
return c.Send("У вас недостаточно длинный пенис для состязания!")
}
if summ <= 0 {
return c.Send("Некорректная сумма, используйте положительное значение!")
}
return c.Send(fmt.Sprintf("%s начал дуель, ставка: %d", s.PlayerOne.FName, summ))
}
// Accept Match
func (s *DuelController) AcceptMatch(c tele.Context) error {
if !s.Inited {
return c.Send("Дуэль еще не началась! Дождитесь старта матча или начните новую дуель с помощью /duel <сумма>")
}
s.PlayerTwo.ID = c.Sender().ID
s.PlayerTwo.FName = c.Sender().FirstName
user := s.DC.GetUser(s.PlayerTwo.ID)
if user.DickSize < s.Summ {
return c.Send("У вас недостаточно длинный пенис для состязания!")
}
if s.PlayerTwo.ID == s.PlayerOne.ID {
return c.Send("Вы не можете принять участие в своей же дуэли!")
}
s.Summ *= 2
go s.gameProccess(c.Chat())
return c.Send(fmt.Sprintf("%s принимает участие в дуэли.", s.PlayerTwo.FName))
}
// Start Game
func (s *DuelController) gameProccess(r tele.Recipient) {
time.Sleep(time.Second * 3)
s.Bot.Send(r, fmt.Sprintf("Дуэль началась!\nНа кону: %d", s.Summ))
s.RoundOne(r)
}
// Round One
func (s *DuelController) RoundOne(r tele.Recipient) {
time.Sleep(time.Second * 3)
s.PlayerOne.Counter += rand.Intn(11) + 1
s.PlayerTwo.Counter += rand.Intn(11) + 1
s.Bot.Send(r,
fmt.Sprintf("Результаты первого раунда\nИгрок %s: %d очков.\nИгрок %s: %d очков.",
s.PlayerOne.FName, s.PlayerOne.Counter, s.PlayerTwo.FName, s.PlayerTwo.Counter),
)
s.RoundTwo(r)
}
// Round Two
func (s *DuelController) RoundTwo(r tele.Recipient) {
time.Sleep(time.Second * 3)
s.PlayerOne.Counter += rand.Intn(11) + 1
s.PlayerTwo.Counter += rand.Intn(11) + 1
s.Bot.Send(r,
fmt.Sprintf("Результаты второго раунда\nИгрок %s: %d очков.\nИгрок %s: %d очков.",
s.PlayerOne.FName, s.PlayerOne.Counter, s.PlayerTwo.FName, s.PlayerTwo.Counter),
)
s.RoundThree(r)
}
// Round Three
func (s *DuelController) RoundThree(r tele.Recipient) {
time.Sleep(time.Second * 3)
s.PlayerOne.Counter += rand.Intn(11) + 1
s.PlayerTwo.Counter += rand.Intn(11) + 1
s.Bot.Send(r,
fmt.Sprintf("Результаты третьего раунда\nИгрок %s: %d очков.\nИгрок %s: %d очков.",
s.PlayerOne.FName, s.PlayerOne.Counter, s.PlayerTwo.FName, s.PlayerTwo.Counter),
)
s.CheckWin(r)
}
// Check Win
func (s *DuelController) CheckWin(r tele.Recipient) {
userOne := s.DC.GetUser(s.PlayerOne.ID)
userTwo := s.DC.GetUser(s.PlayerTwo.ID)
time.Sleep(time.Second * 3)
var winner int = 0
if s.PlayerOne.Counter > s.PlayerTwo.Counter {
winner = 1
} else {
winner = 2
}
if winner == 1 {
s.Inited = false
s.Bot.Send(r, fmt.Sprintf("Игрок %s победил!", s.PlayerOne.FName))
s.DC.UpdateDick(userOne.ID, userOne.DickSize+(s.Summ/2))
s.DC.UpdateDick(userTwo.ID, userTwo.DickSize-(s.Summ/2))
} else if winner == 2 {
s.Inited = false
s.Bot.Send(r, fmt.Sprintf("Игрок %s победил!", s.PlayerTwo.FName))
s.DC.UpdateDick(userOne.ID, userOne.DickSize-(s.Summ/2))
s.DC.UpdateDick(userTwo.ID, userTwo.DickSize+(s.Summ/2))
} else {
s.Inited = false
s.Bot.Send(r, "Ничья!")
}
}

53
src/handlers/data.go → src/handlers/logic.go
View File

@@ -2,42 +2,40 @@ package handlers
import (
"bot/src/controllers"
"bot/src/models"
"fmt"
tele "gopkg.in/telebot.v4"
)
type Data struct {
Db *controllers.DataController
type Handlers struct {
DC *controllers.DataController
}
func (d *Data) InitDB() {
d.Db = controllers.NewDB()
go d.Db.CronRun()
func NewHandlers(db *controllers.DataController) *Handlers {
return &Handlers{DC: db}
}
func (d *Data) Dick(c tele.Context) error {
var user models.User
func (d *Handlers) Dick(c tele.Context) error {
var retMessage string
id := c.Sender().ID
lName := c.Sender().LastName
fName := c.Sender().FirstName
d.Db.DB.First(&user, id)
user := d.DC.GetUser(id)
if user.ID == 0 {
d.DC.CreateUser(id, lName, fName)
user = d.DC.GetUser(id)
}
newRange := controllers.RandRange()
newSize := user.DickSize + newRange
if user.IsBlocked {
if user.Blocked {
retMessage = fmt.Sprintf("%s %s, ты уже отращивал свой писюн, следющая попытка будет после 22:00 по МСК.", fName, lName)
} else {
if user.ID != id {
d.Db.DB.Create(&models.User{ID: id, FirstName: fName, LastName: lName, DickSize: newSize, IsBlocked: false, IsAdmin: false})
} else {
d.Db.DB.Model(&user).Update("dick_size", newSize)
}
d.DC.UpdateDick(id, newSize)
if newRange == 0 {
retMessage = fmt.Sprintf("%s %s, твой писюн сегодня не подрос. Не переживай, просто он, видимо, решил взять выходной — пусть лучше отдохнет перед завтрашними подвигами!", fName, lName)
@@ -46,17 +44,14 @@ func (d *Data) Dick(c tele.Context) error {
} else if newRange < 0 {
retMessage = fmt.Sprintf("%s %s, твой писюн сегодня сократился на %d см. Сейчас он равен %d см.", fName, lName, -newRange, newSize)
}
d.Db.DB.Model(&user).Update("is_blocked", true)
}
return c.Send(retMessage)
}
func (d *Data) TopDick(c tele.Context) error {
var users []models.User
func (d *Handlers) TopDick(c tele.Context) error {
var retMessage string
d.Db.DB.Order("dick_size desc").Limit(10).Find(&users)
users := d.DC.GetTopUsers()
for iter, user := range users {
retMessage += fmt.Sprintf("%d. %s %s, %d см\n", iter+1, user.FirstName, user.LastName, user.DickSize)
@@ -65,14 +60,14 @@ func (d *Data) TopDick(c tele.Context) error {
return c.Send(retMessage)
}
func (d *Data) ClearStatistics(c tele.Context) error {
var user models.User
d.Db.DB.First(&user, c.Sender().ID)
// func (d *Handlers) ClearStatistics(c tele.Context) error {
// var user models.User
// d.DC.DC.First(&user, c.Sender().ID)
if user.IsAdmin {
d.Db.DB.Model(&models.User{}).Update("dick_size", 0)
return c.Send("Статистика очищена!")
}
// if user.IsAdmin {
// d.DC.DB.Model(&models.User{}).Update("dick_size", 0)
// return c.Send("Статистика очищена!")
// }
return c.Send("Ты не админ!")
}
// return c.Send("Ты не админ!")
// }

13
src/models/user.go
View File

@@ -1,10 +1,11 @@
package models
type User struct {
ID int64 `json:"id"`
FirstName string `json:"first_name"`
LastName string `json:"last_name"`
DickSize int `json:"dick_size"`
IsBlocked bool `json:"is_blocked"`
IsAdmin bool `json:"is_admin"`
ID int64 `sql:"id"`
FirstName string `sql:"fname"`
LastName string `sql:"lname"`
DickSize int `sql:"dick_size"`
Blocked bool `sql:"blocked"`
Admin bool `sql:"admin"`
Trick int `sql:"trick"`
}