main.go

@@ -15,15 +15,8 @@ import (
 	"valera/db"
 )
 
-var (
-	workoutTypes = []string{
-		"Отжимания",
-		"Пресс",
-	}
-)
-
 const (
-	version = "v1.4.0"
+	version = "v1.3.0"
 )
 
 type Opts struct {
@@ -159,43 +152,27 @@ func run() {
 			log.Println(err)
 			continue
 		}
-
-		isWorkout := false
-		for _, workoutType := range workoutTypes {
-			if userInfoDTO.GetStatus() == db.UserState(workoutType) {
-				count, err := strconv.Atoi(text)
-				if err != nil {
-					log.Println(err)
-					continue
-				}
-				if err := dataBase.AddWorkout(chatID, db.NewWorkout(workoutType, count, username)); err != nil {
-					log.Println(err)
-					continue
-				}
-				msgText := fmt.Sprintf("Отлично, %s, записал.", text)
-				if count <= 0 {
-					msgText = "Плохо, хочешь быть толстым и не красивым?"
-				}
-				_, _ = bot.Send(tgbot.NewMessage(chatID, msgText))
-				if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
-					log.Println(err)
-				}
-				isWorkout = true
-				break
-			}
-		}
-		if isWorkout {
-			continue
-		}
-
 		switch userInfoDTO.GetStatus() {
 		case db.UserStateGo:
-			if err := dataBase.SetStatusToChat(chatID, db.UserState(text)); err != nil {
+			count, err := strconv.Atoi(text)
+			if err != nil {
 				log.Println(err)
+				continue
 			}
-			msg := tgbot.NewMessage(chatID, fmt.Sprintf("%s, отпишись сколько раз ты выполнил упражнение", text))
+			if err := dataBase.AddWorkout(chatID, db.NewWorkout("Отжимания", count, username)); err != nil {
+				log.Println(err)
+				continue
+			}
+			msgText := fmt.Sprintf("Отлично, %s, записал.", text)
+			if count <= 0 {
+				msgText = "Плохо, хочешь быть толстым и не красивым?"
+			}
+			msg := tgbot.NewMessage(chatID, msgText)
 			msg.ReplyMarkup = tgbot.NewRemoveKeyboard(false)
 			_, _ = bot.Send(msg)
+			if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
+				log.Println(err)
+			}
 			continue
 		case db.UserStateEat:
 			count, err := strconv.Atoi(text)
@@ -247,12 +224,16 @@ func run() {
 }
 
 func sendGoToChat(bot *tgbot.BotAPI, dataBase *db.DB, chatID int64) {
-	msg := tgbot.NewMessage(chatID, "Давай немного разомнемся, выбирай:")
+	msg := tgbot.NewMessage(chatID, "Давай немного разомнемся, отжимания, отпишись сколько раз ты выполнил упражнение")
-	row := tgbot.NewKeyboardButtonRow()
+	msg.ReplyMarkup = tgbot.NewReplyKeyboard(
-	for _, workoutType := range workoutTypes {
+		tgbot.NewKeyboardButtonRow(
-		row = append(row, tgbot.NewKeyboardButton(workoutType))
+			tgbot.NewKeyboardButton("1"),
-	}
+			tgbot.NewKeyboardButton("2"),
-	msg.ReplyMarkup = tgbot.NewReplyKeyboard(row)
+			tgbot.NewKeyboardButton("3"),
+			tgbot.NewKeyboardButton("5"),
+			tgbot.NewKeyboardButton("8"),
+		),
+	)
 	if _, err := bot.Send(msg); err == nil {
 		if err := dataBase.SetStatusToChat(chatID, db.UserStateGo); err != nil {
 			log.Println(err)