import os
import time
import logging 
from logging import ERROR, getLogger
from logging.handlers import TimedRotatingFileHandler

import pyromod.listen
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.jobstores.mongodb import MongoDBJobStore
from pymongo import MongoClient
from pyrogram import Client

from misskaty.vars import API_HASH, API_ID, BOT_TOKEN, DATABASE_URI, USER_SESSION, TZ

logger = getLogger(__name__)    
logger.setLevel(logging.INFO)

# CONFIGURATION TO ROTATE LOGS EVERYDAY
rotate = TimedRotatingFileHandler("MissKatyLogs.txt", when='D', interval=1, backupCount=0, encoding=None, delay=False, utc=False)
logger.addHandler(rotate)
formater = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
rotate.setFormatter(formater)

getLogger("pyrogram").setLevel(ERROR)

MOD_LOAD = []
MOD_NOLOAD = []
HELPABLE = {}
cleanmode = {}
botStartTime = time.time()

# Pyrogram Bot Client
app = Client(
    "MissKatyBot",
    api_id=API_ID,
    api_hash=API_HASH,
    bot_token=BOT_TOKEN,
)

# Pyrogram UserBot Client
user = Client(
    "YasirUBot",
    session_string=USER_SESSION,
)

pymonclient = MongoClient(DATABASE_URI)

jobstores = {
    'default': MongoDBJobStore(
        client=pymonclient,
        database="MissKatyDB",
        collection='nightmode')}

scheduler = AsyncIOScheduler(
    jobstores=jobstores,
    timezone=TZ)

app.start()
user.start()
bot = app.get_me()
ubot = user.get_me()
BOT_ID = bot.id
BOT_NAME = bot.first_name
BOT_USERNAME = bot.username
UBOT_ID = ubot.id
UBOT_NAME = ubot.first_name
UBOT_USERNAME = ubot.username