from telethon import TelegramClient
from dotenv import load_dotenv
import os

from utils.logger import logger


for folder in ['sessions']:
    if not os.path.exists(folder):
        os.makedirs(folder)

load_dotenv()
client = TelegramClient(session='utils/resetter', api_hash=os.getenv('API_HASH'), api_id=os.getenv('API_ID')).start(bot_token=os.getenv('BOT_TOKEN'))
client.parse_mode = 'html'


def connect():
    logger.info('Connecting to Telegram...')
    client.start()
    client.run_until_disconnected()


def disconnect():
    logger.info('Disconnecting from Telegram...')
    client.disconnect()