import logging

import convert_obj_to_md
from redis_client import redis_client
from rq import Worker

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def process_task(task):
    """
    태스크 정보를 기반으로 파일 확장자에 맞는 변환 함수를 호출합니다.
    """
    file_ext = task.get("extension")
    input_path = task.get("input_path")
    output_path = task.get("output_path")

    logger.info(f"작업 처리 중: {task}")

    try:
        if file_ext == "hwp":
            convert_obj_to_md.convert_hwp_to_md(input_path, output_path)
        else:
            convert_obj_to_md.convert_to_md(input_path, output_path)
        logger.info(f"변환 완료: {task}")
        return {"status": "success", "output_path": output_path}

    except Exception as e:
        logger.error(f"작업 처리 중 에러: {e}", exc_info=True)
        return {"status": "error", "message": str(e)}


if __name__ == "__main__":
    listen = ["task_queue1"]

    worker = Worker(listen, connection=redis_client)
    worker.work()