add framework

cmd/valera/main.go

@@ -4,34 +4,27 @@ import (
 	"fmt"
 	"log"
 	"net/http"
-	"strconv"
 	"valera/internal/calories"
 	"valera/internal/commands"
 	"valera/internal/config"
 	"valera/internal/db"
 	"valera/internal/pause"
+	"valera/internal/states"
+	"valera/internal/states/clear_bot_state"
+	"valera/internal/states/go_bot_state"
+	"valera/internal/states/help_bot_state"
+	"valera/internal/states/ping_bot_state"
+	"valera/internal/states/start_bot_state"
+	"valera/internal/states/stat_bot_state"
 
 	tgbot "github.com/go-telegram-bot-api/telegram-bot-api"
 )
 
-var (
-	workoutTypes = []string{
-		"Отжимания",
-		"Пресс",
-		"Подтягивания",
-		"Приседания",
-	}
-)
-
 const (
-	version = "v1.6.3"
+	version = "v1.7.0"
 )
 
 func main() {
-	run()
-}
-
-func run() {
 	cfg := config.NewAppConfig()
 
 	bot, err := tgbot.NewBotAPI(cfg.TgConfig.Token)
@@ -52,46 +45,19 @@ func run() {
 		panic(err)
 	}
 
+	botStates := []states.BotState{
+		clear_bot_state.NewClearBotState(bot, dataBase),
+		go_bot_state.NewGoBotState(bot, dataBase),
+		stat_bot_state.NewStatBotState(bot, dataBase),
+		start_bot_state.NewStartBotState(bot, dataBase, version),
+		help_bot_state.NewHelpBotState(bot, dataBase),
+		ping_bot_state.NewPingBotState(bot, dataBase),
+	}
+
 	go func() {
-		http.HandleFunc("/go", func(w http.ResponseWriter, _ *http.Request) {
-			chats, err := dataBase.GetAllChats()
-			if err != nil {
-				w.Header().Add("Content-Type", "application/json")
-				w.WriteHeader(http.StatusInternalServerError)
-				_, _ = fmt.Fprintf(w, `{"result":"error"}`)
-				return
-			}
-			for _, chatID := range chats {
-				sendGoToChat(bot, dataBase, chatID)
-			}
-			w.Header().Add("Content-Type", "application/json")
-			w.WriteHeader(http.StatusOK)
-			_, _ = fmt.Fprintf(w, `{"result":"ok"}`)
-		})
-
-		http.HandleFunc("/stat", func(w http.ResponseWriter, _ *http.Request) {
-			chats, err := dataBase.GetAllChats()
-			if err != nil {
-				fmt.Println(err)
-				w.Header().Add("Content-Type", "application/json")
-				w.WriteHeader(http.StatusInternalServerError)
-				_, _ = fmt.Fprintf(w, `{"result":"error"}`)
-				return
-			}
-			for _, chatID := range chats {
-				if err := sendStatToChat(bot, dataBase, chatID, "Напоминаю:\n- Cегодня больше не жрем!\n\n"); err != nil {
-					fmt.Println(err)
-					w.Header().Add("Content-Type", "application/json")
-					w.WriteHeader(http.StatusInternalServerError)
-					_, _ = fmt.Fprintf(w, `{"result":"error"}`)
-					return
-				}
-			}
-			w.Header().Add("Content-Type", "application/json")
-			w.WriteHeader(http.StatusOK)
-			_, _ = fmt.Fprintf(w, `{"result":"ok"}`)
-		})
-
+		for _, s := range botStates {
+			http.HandleFunc(s.GetHandler())
+		}
 		log.Println("Server is start up! port", cfg.Port)
 		log.Fatal(http.ListenAndServe(cfg.Port, nil))
 	}()
@@ -105,60 +71,19 @@ func run() {
 		text := update.Message.Text
 		chatID := update.Message.Chat.ID
 		username := update.Message.From.UserName
-
-		if text == "/c" {
-			msg := tgbot.NewMessage(chatID, "Чистка")
-			msg.ReplyMarkup = tgbot.NewRemoveKeyboard(false)
-			_, _ = bot.Send(msg)
-			if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
-				log.Println(err)
-			}
-			continue
-		}
-
 		userInfoDTO, err := dataBase.GetChatInfo(chatID)
 		if err != nil {
 			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
+		for _, s := range botStates {
+			if err := s.Do(text, userInfoDTO, username); err != nil {
+				log.Println(err)
 			}
 		}
-		if isWorkout {
-			continue
-		}
 
 		switch userInfoDTO.GetStatus() {
-		case db.UserStateGo:
-			if err := dataBase.SetStatusToChat(chatID, db.UserState(text)); err != nil {
-				log.Println(err)
-			}
-			msg := tgbot.NewMessage(chatID, fmt.Sprintf("%s, отпишись сколько раз ты выполнил упражнение", text))
-			msg.ReplyMarkup = tgbot.NewRemoveKeyboard(false)
-			_, _ = bot.Send(msg)
-			continue
 		case db.UserStateEat:
 			count, err := calories.CalcCalories(text)
 			if err != nil {
@@ -198,22 +123,6 @@ func run() {
 
 		command := commands.Command(text)
 		switch command {
-		case commands.Start:
-			_, _ = bot.Send(tgbot.NewMessage(chatID, fmt.Sprintf("Здорова, я Валера (%s), твой тренер (%d).", version, chatID)))
-			if err := dataBase.SetStatusToChat(chatID, db.UserStateNone); err != nil {
-				log.Println(err)
-			}
-		case commands.Help:
-			_, _ = bot.Send(tgbot.NewMessage(chatID, "Вот что я умею:\n\n1) Предлагать размяться\n - несколько видов упражнений 6 раз в день\n2) Показывать статистику за день\n3) Считать калории\n - введи одно число, я запишу калории\n - введи 2 числа через пробел одно количество грамм другое количество калорий в 100 граммах (порядок не важен), я посчитаю\n - введи название того что ты съел, я посчитаю (если знаю)"))
-		case commands.Ping:
-			_, _ = bot.Send(tgbot.NewMessage(chatID, "pong"))
-		case commands.Go:
-			sendGoToChat(bot, dataBase, chatID)
-		case commands.Stat:
-			if err := sendStatToChat(bot, dataBase, chatID, ""); err != nil {
-				fmt.Println(err)
-				continue
-			}
 		case commands.Eat:
 			if _, err := bot.Send(tgbot.NewMessage(chatID, "Вижу ты поел, отпишись сколько калорий было")); err == nil {
 				if err := dataBase.SetStatusToChat(chatID, db.UserStateEat); err != nil {
@@ -241,30 +150,3 @@ func run() {
 		}
 	}
 }
-
-func sendGoToChat(bot *tgbot.BotAPI, dataBase *db.DB, chatID int64) {
-	msg := tgbot.NewMessage(chatID, "Давай немного разомнемся, выбирай:")
-	rows := make([][]tgbot.KeyboardButton, 0, len(workoutTypes))
-	for _, workoutType := range workoutTypes {
-		rows = append(rows, tgbot.NewKeyboardButtonRow(tgbot.NewKeyboardButton(workoutType)))
-	}
-	msg.ReplyMarkup = tgbot.NewReplyKeyboard(rows...)
-	if _, err := bot.Send(msg); err == nil {
-		if err := dataBase.SetStatusToChat(chatID, db.UserStateGo); err != nil {
-			log.Println(err)
-		}
-	}
-}
-
-func sendStatToChat(bot *tgbot.BotAPI, dataBase *db.DB, chatID int64, prefix string) error {
-	stat, err := dataBase.GetStat(chatID)
-	if err != nil {
-		return err
-	}
-	msgText := prefix + "Результаты за сегодня:\n"
-	for k, v := range stat {
-		msgText += fmt.Sprintf("- %s: %d\n", k, v)
-	}
-	_, _ = bot.Send(tgbot.NewMessage(chatID, msgText))
-	return nil
-}