add

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()
+			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
 }