roa2web-service-auto/backend/modules/telegram/bot_main.py

"""
Main entry point for ROA2WEB Telegram Bot

This bot provides access to the ROA2WEB ERP system through Telegram
using direct command handlers for financial data queries.
"""

import asyncio
import glob
import logging
import os
from pathlib import Path
from dotenv import load_dotenv
# Note: uvicorn and threading removed - internal API now served via main.py

# ============================================================================
# LOAD ENVIRONMENT VARIABLES FIRST - BEFORE ANY APP IMPORTS
# ============================================================================
# This ensures all modules can access environment variables at import time
env_path = Path(__file__).parent.parent / '.env'
load_dotenv(env_path)

# Telegram imports
from telegram.ext import (
    Application,
    CommandHandler,
    CallbackQueryHandler,
    MessageHandler,
    filters
)

# Import database initialization
from backend.modules.telegram.db import (
    init_database,
    cleanup_expired_codes,
    cleanup_expired_sessions,
    cleanup_expired_email_codes
)

# Import bot handlers
from backend.modules.telegram.bot.handlers import (
    start_command,
    help_command,
    clear_command,
    companies_command,
    unlink_command,
    selectcompany_command,
    dashboard_command,
    sold_command,
    facturi_command,
    trezorerie_command,
    # FAZA 3: New command handlers with button interface
    menu_command,
    trezorerie_casa_command,
    trezorerie_banca_command,
    clienti_command,
    furnizori_command,
    evolutie_command,
    # FAZA 6: Cache management commands
    clearcache_command,
    togglecache_command,
    # Text message handlers
    handle_text_message,
    # FAZA 4: Callback and error handlers
    button_callback,
    error_handler
)

# Import email authentication handler
from backend.modules.telegram.bot.email_handlers import email_login_handler

# Import receipt handlers (US-005: PDF/JPG OCR fiscal receipt flow)
from backend.modules.telegram.handlers.receipt_handlers import (
    handle_document_message,
    handle_photo_message,
    handle_receipt_callback,
)

# Note: internal_api import removed - now served via main.py at /api/telegram/internal/*

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

# Environment variables (already loaded above)
TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN')
BACKEND_URL = os.getenv('BACKEND_URL', 'http://localhost:8000')
# Note: INTERNAL_API_PORT removed - internal API now served via main.py


# ============================================================================
# TELEGRAM BOT SETUP
# ============================================================================

def create_telegram_application() -> Application:
    """
    Create and configure the Telegram bot application.

    Returns:
        Application: Configured Telegram application
    """
    logger.info("Creating Telegram application...")

    # Create application with concurrent_updates so multiple users can use the bot
    # in parallel (e.g. two users uploading receipts simultaneously).
    application = (
        Application.builder()
        .token(TELEGRAM_BOT_TOKEN)
        .concurrent_updates(True)
        .build()
    )

    # Register email authentication conversation handler (must be before other handlers)
    application.add_handler(email_login_handler)

    # Register essential command handlers
    application.add_handler(CommandHandler("start", start_command))
    application.add_handler(CommandHandler("menu", menu_command))
    application.add_handler(CommandHandler("help", help_command))
    application.add_handler(CommandHandler("unlink", unlink_command))

    # =========================================================================
    # LEGACY COMMAND HANDLERS (kept for backwards compatibility, hidden from help)
    # =========================================================================
    # NOTE: These commands are redundant with the button interface.
    # They're kept for users who already know them, but we push buttons in help.
    # Consider removing completely if migration is successful.

    application.add_handler(CommandHandler("clear", clear_command))
    application.add_handler(CommandHandler("companies", companies_command))
    application.add_handler(CommandHandler("selectcompany", selectcompany_command))
    application.add_handler(CommandHandler("dashboard", dashboard_command))
    application.add_handler(CommandHandler("sold", sold_command))
    application.add_handler(CommandHandler("facturi", facturi_command))
    application.add_handler(CommandHandler("trezorerie", trezorerie_command))
    application.add_handler(CommandHandler("trezorerie_casa", trezorerie_casa_command))
    application.add_handler(CommandHandler("trezorerie_banca", trezorerie_banca_command))
    application.add_handler(CommandHandler("clienti", clienti_command))
    application.add_handler(CommandHandler("furnizori", furnizori_command))
    application.add_handler(CommandHandler("evolutie", evolutie_command))

    # FAZA 6: Cache management commands
    application.add_handler(CommandHandler("clearcache", clearcache_command))
    application.add_handler(CommandHandler("togglecache", togglecache_command))

    # Text message handler (for direct code input and future NLP)
    # IMPORTANT: This must be registered BEFORE CallbackQueryHandler
    # filters.TEXT & ~filters.COMMAND ensures we only process non-command text messages
    application.add_handler(MessageHandler(
        filters.TEXT & ~filters.COMMAND,
        handle_text_message
    ))

    # US-006: Receipt handlers (PDF/JPG fiscal receipt OCR flow)
    # IMPORTANT: receipt CallbackQueryHandler must be registered BEFORE the
    # catch-all button_callback so `receipt:*` callbacks are routed correctly.
    application.add_handler(MessageHandler(
        filters.Document.PDF | filters.Document.IMAGE,
        handle_document_message
    ))
    application.add_handler(MessageHandler(filters.PHOTO, handle_photo_message))
    application.add_handler(CallbackQueryHandler(
        handle_receipt_callback,
        pattern=r'^receipt:'
    ))

    # FAZA 4: Register callback query handler (for inline buttons)
    application.add_handler(CallbackQueryHandler(button_callback))

    # Register error handler
    application.add_error_handler(error_handler)

    logger.info("Telegram application configured with all handlers")

    return application


# ============================================================================
# STARTUP/SHUTDOWN
# ============================================================================
# Note: Internal API server removed - now served via main.py at /api/telegram/internal/*

def startup_cleanup() -> int:
    """
    Remove orphan receipt temp files left over from a previous bot crash.

    Receipt OCR flow writes downloaded files to `/tmp/receipt_*.*` and unlinks
    them after confirm/cancel. If the bot died between download and cleanup,
    those files remain on disk; we clean them on startup. Returns the count
    of unlinked files (for logging).
    """
    count = 0
    for path_str in glob.glob('/tmp/receipt_*.*'):
        try:
            Path(path_str).unlink(missing_ok=True)
            count += 1
        except OSError as e:
            logger.warning(f"Failed to unlink orphan receipt file {path_str}: {e}")
    return count


async def startup():
    """
    Initialize the bot application on startup.
    """
    logger.info("🚀 ROA2WEB Telegram Bot - Starting up...")

    # US-006: Sweep orphan receipt temp files from previous crashes
    try:
        orphans = startup_cleanup()
        if orphans:
            logger.info(f"🧹 Cleaned up {orphans} orphan receipt temp files")
    except Exception as e:
        logger.warning(f"⚠️ Receipt orphan cleanup failed (non-critical): {e}")

    # Initialize database
    try:
        logger.info("Initializing SQLite database...")
        await init_database()
        logger.info("✅ Database initialized successfully")
    except Exception as e:
        logger.error(f"❌ Failed to initialize database: {e}")
        raise

    # Cleanup expired data
    try:
        logger.info("Cleaning up expired data...")
        expired_codes = await cleanup_expired_codes()
        expired_sessions = await cleanup_expired_sessions()
        expired_email_codes = await cleanup_expired_email_codes()
        logger.info(f"✅ Cleanup complete: {expired_codes} codes, {expired_sessions} sessions, {expired_email_codes} email codes removed")
    except Exception as e:
        logger.warning(f"⚠️ Cleanup failed (non-critical): {e}")

    logger.info("✅ Startup complete")


async def shutdown():
    """
    Clean up resources on shutdown.
    """
    logger.info("👋 ROA2WEB Telegram Bot - Shutting down...")
    logger.info("✅ Shutdown complete")


async def scheduled_cleanup():
    """
    Background task to periodically clean up expired data.
    Runs every hour to remove expired auth codes, sessions, and email codes.
    """
    while True:
        try:
            await asyncio.sleep(3600)  # Sleep for 1 hour
            logger.info("🧹 Running scheduled cleanup...")
            expired_codes = await cleanup_expired_codes()
            expired_sessions = await cleanup_expired_sessions()
            expired_email_codes = await cleanup_expired_email_codes()
            logger.info(f"✅ Scheduled cleanup: {expired_codes} codes, {expired_sessions} sessions, {expired_email_codes} email codes removed")
        except Exception as e:
            logger.error(f"❌ Error in scheduled cleanup: {e}")


# ============================================================================
# MAIN APPLICATION
# ============================================================================

async def main():
    """
    Main application entry point.

    Runs both the Telegram bot and internal API server concurrently.
    """
    try:
        # Run startup
        await startup()

        # Create Telegram application
        telegram_app = create_telegram_application()

        # Note: Internal API server removed - now served via main.py

        # Start scheduled cleanup task in background
        cleanup_task = asyncio.create_task(scheduled_cleanup())
        logger.info("✅ Scheduled cleanup task started")

        # Initialize and start Telegram bot
        logger.info("🤖 Starting Telegram bot polling...")
        await telegram_app.initialize()
        await telegram_app.start()
        await telegram_app.updater.start_polling(
            drop_pending_updates=True,
            poll_interval=0,        # No delay between polls
            timeout=30              # Long poll timeout 30 seconds (reduces requests from ~6/min to ~2/min)
        )

        logger.info("✅ Telegram bot is now running and polling for updates")
        logger.info(f"📱 Bot ready to receive messages at @{(await telegram_app.bot.get_me()).username}")
        logger.info("🎯 Bot is operational with direct command handlers!")

        # Keep running until interrupted
        await asyncio.Event().wait()

    except KeyboardInterrupt:
        logger.info("⚠️ Received interrupt signal")
    except Exception as e:
        logger.error(f"❌ Fatal error: {e}", exc_info=True)
        raise
    finally:
        # Stop Telegram bot gracefully
        try:
            if 'telegram_app' in locals():
                logger.info("Stopping Telegram bot...")
                await telegram_app.updater.stop()
                await telegram_app.stop()
                await telegram_app.shutdown()
                logger.info("✅ Telegram bot stopped")
        except Exception as e:
            logger.error(f"Error stopping Telegram bot: {e}")

        await shutdown()


# ============================================================================
# ENTRY POINT
# ============================================================================

if __name__ == "__main__":
    # Check required environment variables
    if not os.getenv('TELEGRAM_BOT_TOKEN'):
        logger.error("❌ TELEGRAM_BOT_TOKEN is required")
        logger.error("Please set it in .env file")
        exit(1)

    # Display startup banner
    logger.info("=" * 60)
    logger.info("  ROA2WEB TELEGRAM BOT")
    logger.info("  Financial ERP Assistant with Direct Commands")
    logger.info("=" * 60)

    # Run the main application
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logger.info("👋 Application stopped by user")
    except Exception as e:
        logger.error(f"❌ Application failed: {e}", exc_info=True)
        exit(1)