__main__.py

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import logging
import os
import sys
from pathlib import Path
from uniborg import Uniborg
from uniborg.storage import Storage
from alchemysession import AlchemySessionContainer
from telethon import events, TelegramClient


logging.basicConfig(level=logging.INFO)

from kopp import Config

if Config.DB_URI is None:
    logging.warning("No DB_URI Found!")


if len(Config.SUDO_USERS) == 0:
    logging.warning(
        "'SUDO_USERS' should have atleast one value"
    )
    sys.exit(1)


if Config.HU_STRING_SESSION:
    # for Running on Heroku
    session_id_s = Config.HU_STRING_SESSION.split()
    container = AlchemySessionContainer(
        engine=Config.DB_URI
    )
    borg = None
    for session_id in session_id_s:
        session = container.new_session(session_id)
        borg = Uniborg(
            session,
            n_plugin_path="stdplugins/",
            db_plugin_path="dbplugins/",
            api_config=Config,
            api_id=Config.APP_ID,
            api_hash=Config.API_HASH,
            load_tgbot=True if borg is None else False
        )
    borg.run_until_disconnected()

elif Config.TG_BOT_TOKEN_BF_HER:
    # user defined 'TG_BOT_TOKEN_BF_HER'
    # but did not define, 'HU_STRING_SESSION'
    logging.info(
        "[] did not provide / generate "
        "'HU_STRING_SESSION', trying to work-around"
    )
    temp_borg = TelegramClient(
        "temp_bot_session",
        api_id=Config.APP_ID,
        api_hash=Config.API_HASH
    ).start(bot_token=Config.TG_BOT_TOKEN_BF_HER)
    @temp_borg.on(events.NewMessage(chats=Config.SUDO_USERS))
    async def on_new_message(event):
        from .helper_sign_in import bleck_megick
        await bleck_megick(event, Config, None)
    logging.info(
        f"please send /start to your '@{Config.TG_BOT_USER_NAME_BF_HER}'"
    )
    temp_borg.run_until_disconnected()

else:
    # throw error
    logging.error(
        "USAGE EXAMPLE:\n"
        "python3 -m kopp"
        "\n 👆👆 Please follow the above format to run your userbot."
        "\n Bot quitting."
    )