import telebot
import os
import time
from dotenv import load_dotenv
from Kanobu_Parse import Kanobu_Parser
from Animag_Parse import Animag_Parser
from Anibest_Parse import Anibest_Parser


class Bot:
    def __init__(self) -> None:

        self.checker = {
            'kanobu': '',
            'goha': '',
            'animag': '',
            'anibest': ''
        }

        load_dotenv()

        self.bot = telebot.TeleBot(os.getenv("TOKEN"))

        while True:
            print("Parsing...")
            self.bot_workplace()
            time.sleep(300)

    def bot_workplace(self) -> None:
        try:
            kpp = Kanobu_Parser.Parse()
            app = Animag_Parser.Parse()
            abp = Anibest_Parser.Parse()
            chat_id = '-1002027564410'
            kpp_ramaining_text = None
            app_ramaining_text = None
            abp_ramaining_text = None

            kpp_image, kpp_headline, kpp_text, kpp_video, kpp_url = kpp.split('++')
            kpp_caption = (kpp_headline + "\n\n" + kpp_text)[:1024]
            kpp_splited_text = kpp_caption.split('. ')
            kpp_caption = '. '.join([i for i in kpp_splited_text[0:len(kpp_splited_text)-1]])
            kpp_caption += '.'
            if len(kpp_caption + "\n\n" + kpp_video + "\n" + kpp_url) < 1024 and len(kpp_splited_text[-1] + (kpp_headline + "\n" + kpp_text)[1024:]) == 0:
                kpp_caption += "\n\n" + kpp_video + "\n" + kpp_url
            else:
                kpp_ramaining_text = kpp_splited_text[-1] + (kpp_headline + "\n" + kpp_text)[1024:] + "\n\n" + kpp_video + "\n" + kpp_url

            app_image, app_headline, app_text, app_video, app_url = app.split('++')
            app_caption = (app_headline + "\n\n" + app_text)[:1024]
            app_splited_text = app_caption.split('. ')
            app_caption = '. '.join([i for i in app_splited_text[0:len(app_splited_text)-1]])
            app_caption += '.'
            if len(app_caption + "\n\n" + app_video + "\n" + app_url) < 1024 and len(app_splited_text[-1] + (app_headline + "\n" + app_text)[1024:]) == 0:
                app_caption += "\n\n" + app_video + "\n" + app_url
            else:
                app_ramaining_text = app_splited_text[-1] + (app_headline + "\n" + app_text)[1024:] + "\n\n" + app_video + "\n" + app_url

            abp_image, abp_headline, abp_text, abp_video, abp_url = abp.split('++')
            abp_caption = (abp_headline + "\n\n" + abp_text)[:1024]
            abp_splited_text = abp_caption.split('. ')
            abp_caption = '. '.join([i for i in abp_splited_text[0:len(abp_splited_text)-1]])
            abp_caption += '.'
            if len(abp_caption + "\n\n" + abp_video + "\n" + abp_url) < 1024 and len(abp_splited_text[-1] + (abp_headline + "\n" + abp_text)[1024:]) == 0:
                abp_caption += "\n\n" + abp_video + "\n" + abp_url
            else:
                abp_ramaining_text = abp_splited_text[-1] + (abp_headline + "\n" + abp_text)[1024:] + "\n\n" + abp_video + "\n" + abp_url

            if self.checker['kanobu'] != kpp_caption:
                self.bot.send_photo(chat_id, photo=kpp_image, caption=kpp_caption, parse_mode='Markdown')
                if kpp_ramaining_text is not None:
                    self.bot.send_message(chat_id, kpp_ramaining_text, parse_mode='Markdown', disable_web_page_preview=True)
                self.checker['kanobu'] = kpp_caption

            if self.checker['animag'] != app_caption:
                self.bot.send_photo(chat_id, photo=app_image, caption=app_caption, parse_mode='Markdown')
                if app_ramaining_text is not None:
                    self.bot.send_message(chat_id, app_ramaining_text, parse_mode='Markdown', disable_web_page_preview=True)
                self.checker['animag'] = app_caption

            if self.checker['anibest'] != abp_caption:
                self.bot.send_photo(chat_id, photo=abp_image, caption=abp_caption, parse_mode='Markdown')
                if abp_ramaining_text is not None:
                    self.bot.send_message(chat_id, abp_ramaining_text, parse_mode='Markdown', disable_web_page_preview=True)
                self.checker['anibest'] = abp_caption

        except Exception as e:
            print(e)


if __name__ == "__main__":
    bot = Bot()
    bot.bot_workplace()