import datetime
import zoneinfo
from time import sleep

import croniter
import pytz
from telebot.apihelper import ApiTelegramException

from daemons.bot import bot
from helpers import now
from helpers.keyboards import main_keyboard
from helpers.models import UserSchema
from helpers.mongo import mongo
from helpers.ruz import ruz


def process():
    for user in mongo.users_collection.find({"hse_id": {"$ne": None}, "next_notify_time": {"$lte": now()}}):
        try:
            lessons = mongo.get_today_lessons(UserSchema().load(user))
            if len(lessons) == 0:
                ans = "Сегодня у тебя нет пар."
            else:
                ans = ruz.schedule_builder(lessons)
            bot.send_message(
                user['chat_id'],
                "Напоминаю о занятиях сегодня!\n" + ans,
                reply_markup=main_keyboard()
            )
        except:
            pass
        hours, minutes = user['notify_daily'].split(':')
        cron = croniter.croniter(f"{minutes} {hours} * * *", now())
        next_date = cron.get_next(datetime.datetime)
        next_date = next_date.replace(tzinfo=pytz.timezone('Europe/Moscow'))
        mongo.users_collection.update_one({"chat_id": user['chat_id']}, {"$set": {"next_notify_time": next_date}})
    for user in mongo.users_collection.find({"notify_minutes": {"$ne": None}, "hse_id": {"$ne": None}}):
        for lesson in mongo.lessons_collection.find({
            "hse_user_id": user["hse_id"],
            "begin": {"$lte": now() + datetime.timedelta(minutes=5)},
            "notified": False
        }):
            ans = ""
            ans += f"Аудитория: {lesson['building']}, {lesson['auditorium']}\n"
            ans += f"Начало: {lesson['begin'].strftime('%H:%M')}\n"
            ans += f"Конец: {lesson['end'].strftime('%H:%M')}\n"
            ans += f"Преподаватель: {(lesson['lecturer'] or 'Неизвестно')}\n"
            try:
                bot.send_message(
                    user["chat_id"],
                    "Уведомляю о занятиях!\n" + ans
                )
            except ApiTelegramException:
                pass
            mongo.lessons_collection.update_one({"_id": lesson['_id']}, {"$set": {"notified": True}})


def notify():
    while True:
        print("notify start")
        begin = datetime.datetime.now()
        process()
        end = datetime.datetime.now()
        print('notify finished')
        print("time elapsed", (end - begin).total_seconds())
        sleep(60 * 2)