talia_bot/app/main.py

# app/main.py
# Este es el archivo principal del bot. Aquí se inicia todo y se configuran los comandos.

import logging
from telegram import Update
from telegram.ext import (
    Application,
    CommandHandler,
    CallbackQueryHandler,
    ConversationHandler,
    MessageHandler,
    ContextTypes,
    filters,
)

# Importamos las configuraciones y herramientas que creamos en otros archivos
from config import TELEGRAM_BOT_TOKEN
from permissions import get_user_role
from modules.onboarding import handle_start as onboarding_handle_start
from modules.agenda import get_agenda
from modules.citas import request_appointment
from modules.equipo import (
    propose_activity_start,
    get_description,
    get_duration,
    cancel_proposal,
    view_requests_status,
    DESCRIPTION,
    DURATION,
)
from modules.aprobaciones import view_pending, handle_approval_action
from modules.servicios import get_service_info
from modules.admin import get_system_status
from modules.print import print_handler
from modules.create_tag import create_tag_conv_handler
from scheduler import schedule_daily_summary

# Configuramos el sistema de logs para ver mensajes de estado en la consola
logging.basicConfig(
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
)
logger = logging.getLogger(__name__)

async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """
    Se ejecuta cuando el usuario escribe /start.
    Muestra un mensaje de bienvenida y un menú según el rol del usuario.
    """
    chat_id = update.effective_chat.id
    user_role = get_user_role(chat_id)

    logger.info(f"Usuario {chat_id} inició conversación con el rol: {user_role}")

    # Obtenemos el texto y los botones de bienvenida desde el módulo de onboarding
    response_text, reply_markup = onboarding_handle_start(user_role)

    # Respondemos al usuario
    await update.message.reply_text(response_text, reply_markup=reply_markup)

async def button_dispatcher(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """
    Esta función maneja los clics en los botones del menú.
    Dependiendo de qué botón se presione, ejecuta una acción diferente.
    """
    query = update.callback_query
    await query.answer() # Avisa a Telegram que recibimos el clic
    logger.info(f"El despachador recibió una consulta: {query.data}")

    # Texto por defecto si no encontramos la acción
    response_text = "Acción no reconocida."
    reply_markup = None

    # Diccionario de acciones simples (que solo devuelven texto)
    simple_handlers = {
        'view_agenda': get_agenda,
        'view_requests_status': view_requests_status,
        'schedule_appointment': request_appointment,
        'get_service_info': get_service_info,
        'view_system_status': get_system_status,
        'manage_users': lambda: "Función de gestión de usuarios no implementada.",
    }

    # Diccionario de acciones complejas (que devuelven texto y botones)
    complex_handlers = {
        'view_pending': view_pending,
    }

    # Buscamos qué función ejecutar según el dato del botón (query.data)
    if query.data in simple_handlers:
        response_text = simple_handlers[query.data]()
    elif query.data in complex_handlers:
        response_text, reply_markup = complex_handlers[query.data]()
    elif query.data.startswith(('approve:', 'reject:')):
        # Manejo especial para botones de aprobar o rechazar
        response_text = handle_approval_action(query.data)

    # Editamos el mensaje original con la nueva información
    await query.edit_message_text(text=response_text, reply_markup=reply_markup, parse_mode='Markdown')

def main() -> None:
    """Función principal que arranca el bot."""
    # Verificamos que tengamos el token del bot
    if not TELEGRAM_BOT_TOKEN:
        logger.error("TELEGRAM_BOT_TOKEN no está configurado en las variables de entorno.")
        return

    # Creamos la aplicación del bot
    application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()

    # Programamos el resumen diario
    schedule_daily_summary(application)

    # Configuramos un "manejador de conversación" para proponer actividades
    # Esto permite que el bot haga varias preguntas seguidas (descripción, duración)
    conv_handler = ConversationHandler(
        entry_points=[CallbackQueryHandler(propose_activity_start, pattern='^propose_activity$')],
        states={
            DESCRIPTION: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_description)],
            DURATION: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_duration)],
        },
        fallbacks=[CommandHandler('cancel', cancel_proposal)],
        per_message=False
    )

    # Registramos todos los manejadores de eventos en la aplicación
    application.add_handler(conv_handler)
    application.add_handler(create_tag_conv_handler())
    application.add_handler(CommandHandler("start", start))
    application.add_handler(CommandHandler("print", print_handler))
    application.add_handler(CallbackQueryHandler(button_dispatcher))

    # Iniciamos el bot (se queda escuchando mensajes)
    logger.info("Iniciando Talía Bot...")
    application.run_polling()

# Si este archivo se ejecuta directamente, llamamos a la función main()
if __name__ == "__main__":
    main()