diff --git a/src/controllers/duelController.go b/src/controllers/duelController.go index f7e2bec..e7261bd 100644 --- a/src/controllers/duelController.go +++ b/src/controllers/duelController.go @@ -38,17 +38,6 @@ func NewDuel(bot *tele.Bot, dc *DataController) *DuelController { } } -func (s *DuelController) ClearStats() { - s.Inited = false - s.PlayerOne.Counter = 0 - s.PlayerTwo.Counter = 0 - s.PlayerOne.FName = "" - s.PlayerTwo.FName = "" - s.Summ = 0 - s.PlayerOne.ID = 0 - s.PlayerTwo.ID = 0 -} - // Start Match func (s *DuelController) StartMatch(c tele.Context) error { if s.Inited { @@ -56,46 +45,40 @@ func (s *DuelController) StartMatch(c tele.Context) error { } var text string + user := s.DC.GetUser(s.PlayerOne.ID) if strings.Contains(c.Text(), " ") { text = strings.Split(c.Text(), " ")[1] - fmt.Println("Ставка текст", text) + fmt.Println("Ставка", text) } else { - s.ClearStats() return c.Send("Некоректная команда, используйте /duel <сумма>") } summ, err := strconv.Atoi(text) if err != nil { - s.ClearStats() return c.Send("Некорректная команда, используйте /duel <сумма>") } else { - s.Summ = summ - } + if user.DickSize < summ { + return c.Send("У вас недостаточно длинный пенис для состязания!") + } - if summ <= 0 { - s.ClearStats() - return c.Send("Некорректная сумма, используйте положительное значение!") + if summ <= 0 { + s.Summ = summ + } else { + return c.Send("Некорректная сумма, используйте положительное значение!") + } } s.Inited = true s.PlayerOne.ID = c.Sender().ID s.PlayerOne.FName = c.Sender().FirstName - user := s.DC.GetUser(s.PlayerOne.ID) - - if user.DickSize < summ { - s.ClearStats() - 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 { - s.ClearStats() return c.Send("Дуэль еще не началась! Дождитесь старта матча или начните новую дуель с помощью /duel <сумма>") } @@ -105,12 +88,10 @@ func (s *DuelController) AcceptMatch(c tele.Context) error { user := s.DC.GetUser(s.PlayerTwo.ID) if user.DickSize < s.Summ { - s.ClearStats() return c.Send("У вас недостаточно длинный пенис для состязания!") } if s.PlayerTwo.ID == s.PlayerOne.ID { - s.ClearStats() return c.Send("Вы не можете принять участие в своей же дуэли!") } @@ -181,5 +162,8 @@ func (s *DuelController) CheckWin(r tele.Recipient) { s.Bot.Send(r, "Ничья!") } - s.ClearStats() + s.PlayerOne.Counter = 0 + s.PlayerTwo.Counter = 0 + s.Summ = 0 + s.Inited = false }