valera/internal/states/weight_bot_state/weight_bot_state.go

package weight_bot_state

import (
	"fmt"
	"log"
	"net/http"
	"strconv"
	"strings"

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

	"valera/internal/db"
	"valera/internal/modules/times"
	"valera/internal/states"
)

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

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

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

func (s *weightBotState) Do(text string, chatInfo *db.ChatInfo) error {
	if chatInfo.Status == string(db.UserStateWeight) {
		weight, err := getWeight(text)
		if err != nil {
			log.Println(err)
			return nil
		}
		if weight <= 0 {
			_, _ = s.bot.Send(tgbot.NewMessage(chatInfo.ChatID, "Все фигня, давай по новой"))
			return nil
		}
		if err := s.dataBase.AddWeight(chatInfo.ChatID, db.NewWeight(weight, chatInfo.Username)); err != nil {
			log.Println(err)
			return nil
		}
		stat, err := s.dataBase.GetStatBetween(chatInfo.ChatID, times.GetStartDayMinus(1), times.GetStartDay())
		if err != nil {
			log.Println(err)
			return nil
		}
		dWeight := 0.0
		if stat["Вес кг"] != 0 {
			dWeight = weight - stat["Вес кг"]
		}
		_, _ = s.bot.Send(tgbot.NewMessage(chatInfo.ChatID, fmt.Sprintf("%vкг, записал.\nИзменение веса: %vкг", weight, dWeight)))
		return s.dataBase.SetStatusToChat(chatInfo.ChatID, db.UserStateNone)
	}

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

	s.sendWeightToChat(chatInfo.ChatID)
	return nil
}

func (s *weightBotState) GetHandler() (string, func(http.ResponseWriter, *http.Request)) {
	return names[0], func(w http.ResponseWriter, r *http.Request) {
		chats, err := s.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 {
			s.sendWeightToChat(chatID)
		}
		w.Header().Add("Content-Type", "application/json")
		w.WriteHeader(http.StatusOK)
		_, _ = fmt.Fprintf(w, `{"result":"ok"}`)
	}
}

func (s *weightBotState) sendWeightToChat(chatID int64) {
	msg := tgbot.NewMessage(chatID, "Давай взвешивайся, отпишись сколько кг ты весишь")
	if _, err := s.bot.Send(msg); err == nil {
		if err := s.dataBase.SetStatusToChat(chatID, db.UserStateWeight); err != nil {
			log.Println(err)
		}
	}
}

func getWeight(text string) (float64, error) {
	number := strings.ReplaceAll(text, ",", ".")
	return strconv.ParseFloat(number, 64)
}