import logging
import logging.config
from flask import Flask
import os

from common.utils.celery_utils import make_celery, init_celery
from common.extensions import db, cache_manager, minio_client
from config.logging_config import configure_logging
from config.config import get_config


def create_app(config_file=None):
    app = Flask(__name__)

    environment = os.getenv('FLASK_ENV', 'development')

    match environment:
        case 'development':
            app.config.from_object(get_config('dev'))
        case 'production':
            app.config.from_object(get_config('prod'))
        case _:
            app.config.from_object(get_config('dev'))

    configure_logging()

    app.logger.info('Starting up eveai_chat_workers...')
    register_extensions(app)

    from . import specialists, retrievers

    celery = make_celery(app.name, app.config)
    init_celery(celery, app)

    register_cache_handlers(app)

    from eveai_chat_workers import tasks
    print(tasks.tasks_ping())

    return app, celery


def register_extensions(app):
    db.init_app(app)
    cache_manager.init_app(app)
    minio_client.init_app(app)


def register_cache_handlers(app):
    from common.utils.cache.config_cache import register_config_cache_handlers
    register_config_cache_handlers(cache_manager)
    from common.utils.cache.crewai_processed_config_cache import register_specialist_cache_handlers
    register_specialist_cache_handlers(cache_manager)
    from eveai_chat_workers.chat_session_cache import register_chat_session_cache_handlers
    register_chat_session_cache_handlers(cache_manager)
    from common.utils.cache.translation_cache import register_translation_cache_handlers
    register_translation_cache_handlers(cache_manager)
    

app, celery = create_app()