valera/internal/states/pause_bot_state/pause_bot_state.go

package pause_bot_state

import (
	"fmt"
	"log"
	"net/http"
	"valera/internal/db"
	"valera/internal/states"

	tgbot "github.com/go-telegram-bot-api/telegram-bot-api"
	"golang.org/x/exp/slices"
)

var names = []string{
	"/pause",
}

type pauseBotState struct {
	bot      *tgbot.BotAPI
	dataBase *db.DB
}

func NewPauseBotState(bot *tgbot.BotAPI, dataBase *db.DB) states.BotState {
	return &pauseBotState{
		bot:      bot,
		dataBase: dataBase,
	}
}

func (s *pauseBotState) Do(text string, chatInfo *db.ChatInfo) error {
	if chatInfo.Status == string(db.UserStatePause) {
		duration, err := getDuration(text)
		if err != nil {
			log.Println(err)
			return nil
		}
		if err := s.dataBase.SetPause(chatInfo.ChatID, duration); err != nil {
			return nil
		}
		msg := tgbot.NewMessage(chatInfo.ChatID, fmt.Sprintf("Поставил паузу %v, отдыхай", duration))
		msg.ReplyMarkup = tgbot.NewRemoveKeyboard(false)
		_, _ = s.bot.Send(msg)
		return s.dataBase.SetStatusToChat(chatInfo.ChatID, db.UserStateNone)
	}

	if !slices.Contains(names, text) {
		return nil
	}

	msg := tgbot.NewMessage(chatInfo.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д"),
		),
	}...)
	_, _ = s.bot.Send(msg)
	return s.dataBase.SetStatusToChat(chatInfo.ChatID, db.UserStatePause)
}

func (s *pauseBotState) GetHandler() (string, func(http.ResponseWriter, *http.Request)) {
	return names[0], func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusInternalServerError)
	}
}