import logging

from telegram import ForceReply, Update
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters

from cm_bot_hal import CM_BOT_HAL

logging.basicConfig(
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
)
logging.getLogger("httpx").setLevel(logging.WARNING)

logger = logging.getLogger(__name__)

async def menu_cmd_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    menu = [
        'MENU',
        '/1 - Get Acc',
        '/2 <link> - Set Security Pin'
    ]
    await update.message.reply_text('\n'.join(menu))

async def get_acc_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    await update.message.reply_text('Start Getting CM Account ...')
    try:
        bot = CM_BOT_HAL()
        user = bot.get_user_api()
        msg = [
            f'Username: {user["username"]}',
            f'Password: {user["password"]}',
            f'Link: {user["link"]}'
        ]
        await update.message.reply_text('\n'.join(msg))
    except Exception as e:
        await update.message.reply_text(f'Error: {e}')

async def set_security_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    if len(context.args) == 0 or len(context.args) > 1:
        await update.message.reply_text('CMD is wrong, please check and retry ...')
        return
    bot = CM_BOT_HAL()
    if bot.is_whatsapp_url(context.args[0]) == False:
        await update.message.reply_text('Link Format Wrong, please check and retry ...')
        return
    await update.message.reply_text('Start Setting Security Pin ...')
    try:
        bot.set_security_pin_api(context.args[0])
        del bot
        await update.message.reply_text('Security Pin Set Done!')
    except Exception as e:
        await update.message.reply_text(f'Error: {e}')

async def insert_to_user_table_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    if len(context.args) == 0 or len(context.args) != 4:
        await update.message.reply_text('CMD is wrong, please check and retry ...')
        return
    bot = CM_BOT_HAL()
    f_username, f_password, t_username, t_password = context.args
    bot.insert_user_to_table_user(
            {
                'f_username': f_username,
                'f_password': f_password,
                't_username': t_username,
                't_password': t_password
            }
        )
    await update.message.reply_text(f'Done insert {f_username} into user table.')


def main() -> None:
    """Start the bot."""
    # application = Application.builder().token("5327571437:AAFlowwnAysTEMx6LtYQNTevGCboKDZoYzY").build()
    application = Application.builder().token("5315819168:AAH31xwNgPdnk123x97XalmTW6fQV5EUCFU").build()
    application.add_handler(CommandHandler("menu", menu_cmd_handler))
    application.add_handler(CommandHandler("1", get_acc_handler))
    application.add_handler(CommandHandler("2", set_security_handler))
    application.add_handler(CommandHandler("3", insert_to_user_table_handler))
    # application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))

    # Start the Telegram bot
    print("Starting Telegram bot...")
    application.run_polling(allowed_updates=Update.ALL_TYPES)

if __name__ == "__main__":
    main()