l9_stud_bot/modules/tg/handlers.go

package tg

import (
	"fmt"
	"strconv"
	"strings"

	"git.l9labs.ru/anufriev.g.a/l9_stud_bot/modules/database"
	"git.l9labs.ru/anufriev.g.a/l9_stud_bot/modules/ssau_parser"
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
	"xorm.io/builder"
)

// Приветственное сообщение
func (bot *Bot) Start(user *database.TgUser) error {
	user.PosTag = database.Ready
	_, err := bot.DB.Update(user)
	if err != nil {
		return err
	}
	msg := tgbotapi.NewMessage(
		user.TgId,
		`Привет! У меня можно посмотреть в удобном формате <b>ближайшие пары</b>, расписание <b>по дням</b> и даже <b>по неделям</b>!
Просто напиши мне <b>номер группы</b> или <b>фамилию преподавателя</b>`)
	msg.ParseMode = tgbotapi.ModeHTML
	_, err = bot.TG.Send(msg)
	return err
}

// Поиск расписания по запросу
func (bot *Bot) Find(user *database.TgUser, query string) (tgbotapi.Message, error) {
	// Поиск в БД
	var groups []database.Group
	bot.DB.Where(builder.Like{"GroupName", query}).Find(&groups)

	var teachers []database.Teacher
	bot.DB.Where(builder.Like{"FirstName", query}).Find(&teachers)

	// Поиск на сайте
	list, siteErr := ssau_parser.SearchInRasp(query)

	allGroups := groups
	allTeachers := teachers

	// Добавляем результаты поиска на сайте к результатам из БД
	for _, elem := range list {
		if strings.Contains(elem.Url, "group") {
			exists := false
			for _, group := range groups {
				if elem.Id == group.GroupId {
					exists = true
					break
				}
			}
			if !exists {
				allGroups = append(allGroups, database.Group{GroupId: elem.Id, GroupName: elem.Text})
			}
		}
		if strings.Contains(elem.Url, "staff") {
			exists := false
			for _, teacher := range teachers {
				if elem.Id == teacher.TeacherId {
					exists = true
					break
				}
			}
			if !exists {
				teacher := ssau_parser.ParseTeacherName(elem.Text)
				teacher.TeacherId = elem.Id
				allTeachers = append(allTeachers, teacher)
			}
		}
	}

	// Если получен единственный результат, сразу выдать (подключить) расписание
	if len(allGroups) == 1 || len(allTeachers) == 1 {
		if user.PosTag == database.Add {
			msg := tgbotapi.NewMessage(user.TgId, "Подключено!")
			keyboard := tgbotapi.NewReplyKeyboard(
				[]tgbotapi.KeyboardButton{
					tgbotapi.NewKeyboardButton("Моё расписание"),
				})
			msg.ReplyMarkup = keyboard
			user.PosTag = database.Ready
			bot.DB.Update(user)
			return bot.TG.Send(msg)
		} else {
			var sheduleId int64
			var isGroup bool
			if len(allGroups) == 1 {
				sheduleId = allGroups[0].GroupId
				isGroup = true
			} else {
				sheduleId = allTeachers[0].TeacherId
				isGroup = false
			}
			shedule := database.ShedulesInUser{
				IsTeacher: !isGroup,
				SheduleId: sheduleId,
			}
			msg := tgbotapi.NewMessage(
				user.TgId,
				fmt.Sprintf(
					"Тут должно было быть расписание, но его пока не завезли\n%d",
					shedule.SheduleId,
				),
			)
			return bot.TG.Send(msg)
			/*
				err := bot.GetSummary([]database.ShedulesInUser{shedule}, false)
				if err != nil {
					return err
				}*/
		}

		// Если получено несколько групп
	} else if len(allGroups) != 0 {
		msg := tgbotapi.NewMessage(user.TgId, "Вот что я нашёл\nВыбери нужную группу")
		msg.ReplyMarkup = GenerateKeyboard(GenerateGroupsArray(allGroups, user.PosTag == database.Add))
		return bot.TG.Send(msg)
		// Если получено несколько преподавателей
	} else if len(allTeachers) != 0 {
		msg := tgbotapi.NewMessage(user.TgId, "Вот что я нашёл\nВыбери нужного преподавателя")
		msg.ReplyMarkup = GenerateKeyboard(GenerateTeachersArray(allTeachers, user.PosTag == database.Add))
		return bot.TG.Send(msg)
		// Если ничего не получено
	} else {
		var msg tgbotapi.MessageConfig
		if siteErr != nil {
			msg = tgbotapi.NewMessage(
				user.TgId,
				"К сожалению, у меня ничего не нашлось, а на сайте ssau.ru/rasp произошла какая-то ошибка :(\n"+
					"Повтори попытку позже",
			)
			bot.Debug.Printf("sasau error: %s", siteErr)
		} else {
			msg = tgbotapi.NewMessage(
				user.TgId,
				"К сожалению, я ничего не нашёл ):\nПроверь свой запрос",
			)
		}

		return bot.TG.Send(msg)
	}
}

// Получить расписание из кнопки
func (bot *Bot) GetShedule(user *database.TgUser, query *tgbotapi.CallbackQuery) error {
	data := strings.Split(query.Data, "_")
	if len(data) != 3 {
		return fmt.Errorf("wrong button format: %s", query.Data)
	}
	isGroup := data[1] == "group"
	isAdd := data[0] == "true"
	groupId, err := strconv.ParseInt(data[2], 0, 64)
	if err != nil {
		return err
	}
	if !isAdd {
		shedule := database.ShedulesInUser{
			IsTeacher: !isGroup,
			SheduleId: groupId,
		}
		msg := tgbotapi.NewEditMessageText(
			query.Message.Chat.ID,
			query.Message.MessageID,
			fmt.Sprintf(
				"Тут должно было быть расписание, но его пока не завезли\n%d",
				shedule.SheduleId,
			),
		)
		bot.TG.Request(msg)
	}
	/*
		err = bot.GetSummary([]database.ShedulesInUser{shedule}, false)
		if err != nil {
			return err
		}*/
	return err
}

/*
	func (bot *Bot) HandleSummary(query *tgbotapi.CallbackQuery) error {
		data := strings.Split(query.Data, "_")
		shedule, dt, err := ParseQuery(data)
		if err != nil {
			return err
		}
		if data[1] == "personal" {
			switch data[0] {
			case "day":
				bot.GetPersonalDaySummary(int(dt), *query.Message)
			case "week":
				bot.GetPersonalWeekSummary(int(dt), *query.Message)
			default:
				bot.GetPersonalSummary(*query.Message)
			}
		} else {
			switch data[0] {
			case "day":
				bot.GetDaySummary(shedule, dt, false, *query.Message)
			case "week":
				bot.GetWeekSummary(shedule, dt, false, *query.Message)
			default:
				bot.GetSummary(shedule, false, *query.Message)
			}
		}
		return nil
	}

	func (bot *Bot) Confirm(query *tgbotapi.CallbackQuery) error {
		isGroup := bot.TG_user.PosTag == "confirm_add_group"
		groupId, err := strconv.ParseInt(query.Data, 0, 64)
		if err != nil {
			return err
		}
		var groups []database.ShedulesInUser
		err = bot.DB.Find(&groups, &database.ShedulesInUser{
			L9Id:      bot.TG_user.L9Id,
			SheduleId: groupId,
			IsTeacher: !isGroup,
		})
		if err != nil {
			return err
		}
		if len(groups) == 0 {
			shInUser := database.ShedulesInUser{
				L9Id:      bot.TG_user.L9Id,
				IsTeacher: !isGroup,
				SheduleId: groupId,
			}
			bot.DB.InsertOne(shInUser)
			bot.DeleteMsg(query)
			msg := tgbotapi.NewMessage(bot.TG_user.TgId, "Подключено!")
			keyboard := tgbotapi.NewReplyKeyboard([]tgbotapi.KeyboardButton{tgbotapi.NewKeyboardButton("Главное меню")})
			msg.ReplyMarkup = keyboard
			bot.TG.Send(msg)

			bot.TG_user.PosTag = "ready"
			err = bot.UpdateUserDB()
			if err != nil {
				return err
			}
		} else {
			var msg string
			if isGroup {
				msg = "Эта группа уже подключена!"
			} else {
				msg = "Этот преподаватель уже подключен!"
			}
			callback := tgbotapi.NewCallback(query.ID, msg)
			bot.TG.Request(callback)
		}
		return nil
	}

	func (bot *Bot) Cancel(query *tgbotapi.CallbackQuery) error {
		bot.TG_user.PosTag = "ready"
		err := bot.UpdateUserDB()
		if err != nil {
			return err
		}
		bot.DeleteMsg(query)

		return nil
	}
*/

func (bot *Bot) Etc(user *database.TgUser) {
	msg := tgbotapi.NewMessage(user.TgId, "Oй!")
	bot.TG.Send(msg)
}