mv to framework

cmd/valera/main.go

@@ -8,11 +8,12 @@ import (
 	"valera/internal/commands"
 	"valera/internal/config"
 	"valera/internal/db"
-	"valera/internal/pause"
 	"valera/internal/states"
 	"valera/internal/states/clear_bot_state"
+	"valera/internal/states/eat_bot_state"
 	"valera/internal/states/go_bot_state"
 	"valera/internal/states/help_bot_state"
+	"valera/internal/states/pause_bot_state"
 	"valera/internal/states/ping_bot_state"
 	"valera/internal/states/start_bot_state"
 	"valera/internal/states/stat_bot_state"
@@ -52,6 +53,8 @@ func main() {
 		start_bot_state.NewStartBotState(bot, dataBase, version),
 		help_bot_state.NewHelpBotState(bot, dataBase),
 		ping_bot_state.NewPingBotState(bot, dataBase),
+		pause_bot_state.NewPauseBotState(bot, dataBase),
+		eat_bot_state.NewEatBotState(bot, dataBase),
 	}
 
 	go func() {
@@ -85,68 +88,7 @@ func main() {
 
 		switch userInfoDTO.GetStatus() {
 		case db.UserStateEat:
-			count, err := calories.CalcCalories(text)
-			if err != nil {
-				log.Println(err)
-				continue
-			}
-			if err := dataBase.AddCalories(chatID, db.NewCalories(count, username)); err != nil {
-				log.Println(err)
-				continue
-			}
-			if count <= 0 {
-				_, _ = bot.Send(tgbot.NewMessage(chatID, "Все фигня, давай по новой"))
-				continue
-			}
-			_, _ = bot.Send(tgbot.NewMessage(chatID, fmt.Sprintf("Калории, фу, %d, записал.", count)))
-			if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
-				log.Println(err)
-			}
-			continue
-		case db.UserStatePause:
-			duration, err := pause.GetDuration(text)
-			if err != nil {
-				log.Println(err)
-				continue
-			}
-			if err := dataBase.SetPause(chatID, duration); err != nil {
-				log.Println(err)
-			}
-			msg := tgbot.NewMessage(chatID, fmt.Sprintf("Поставил паузу %v, отдыхай", duration))
-			msg.ReplyMarkup = tgbot.NewRemoveKeyboard(false)
-			_, _ = bot.Send(msg)
-			if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
-				log.Println(err)
-			}
-			continue
-		}
-
-		command := commands.Command(text)
-		switch command {
-		case commands.Eat:
-			if _, err := bot.Send(tgbot.NewMessage(chatID, "Вижу ты поел, отпишись сколько калорий было")); err == nil {
-				if err := dataBase.SetStatusToChat(chatID, db.UserStateEat); err != nil {
-					log.Println(err)
-				}
-			}
-		case commands.Pause:
-			msg := tgbot.NewMessage(chatID, "Хочешь отдохнуть? Сколько времени тебе нужно?")
-			msg.ReplyMarkup = tgbot.NewReplyKeyboard([][]tgbot.KeyboardButton{
-				tgbot.NewKeyboardButtonRow(
-					tgbot.NewKeyboardButton("1ч"),
-					tgbot.NewKeyboardButton("4ч"),
-				),
-				tgbot.NewKeyboardButtonRow(
-					tgbot.NewKeyboardButton("1д"),
-					tgbot.NewKeyboardButton("2д"),
-					tgbot.NewKeyboardButton("7д"),
-				),
-			}...)
-			if _, err := bot.Send(msg); err == nil {
-				if err := dataBase.SetStatusToChat(chatID, db.UserStatePause); err != nil {
-					log.Println(err)
-				}
-			}
+			
 		}
 	}
 }