import os
import shutil
from pathlib import Path
from typing import List

from config import OUTPUT_DIR, UPLOAD_DIR
from fastapi import FastAPI, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from redis_client import redis_client
from rq import Queue
from worker import process_task

# RQ 작업 큐 생성
task_queue = Queue("task_queue1", connection=redis_client)

app = FastAPI()

# CORS 설정
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 모든 오리진 허용
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 업로드 및 출력 디렉토리 생성
os.makedirs(UPLOAD_DIR, exist_ok=True)
os.makedirs(OUTPUT_DIR, exist_ok=True)


@app.post("/upload/")
async def upload_directory(files: List[UploadFile]):
    """사용자가 업로드한 파일들을 UPLOAD_DIR에 저장"""
    uploaded_files = []
    for file in files:
        file_path = Path(UPLOAD_DIR) / file.filename
        with open(file_path, "wb") as buffer:
            shutil.copyfileobj(file.file, buffer)
        uploaded_files.append(file.filename)
    return {"message": "파일 업로드 성공", "files": uploaded_files}


@app.get("/convert/")
async def convert_files():
    """업로드된 파일을 변환 큐(RQ)에 등록하고 Job ID 반환"""
    files = os.listdir(UPLOAD_DIR)
    enqueued_tasks = []

    for file in files:
        file_ext = file.split(".")[-1].lower()
        if file_ext in [
            "txt",
            "html",
            "pdf",
            "hwp",
            "pptx",
            "xlsx",
            "docx",
        ]:
            task = {
                "filename": file,
                "extension": file_ext,
                "input_path": os.path.join(UPLOAD_DIR, file),
                "output_path": os.path.join(OUTPUT_DIR, f"{Path(file).stem}.md"),
            }
            job = task_queue.enqueue(process_task, task)  # RQ에 작업 등록
            enqueued_tasks.append({"task": task, "job_id": job.id})

    return {"message": "작업이 큐에 추가되었습니다.", "tasks": enqueued_tasks}


@app.get("/task/{job_id}")
async def get_task_status(job_id: str):
    """RQ에서 특정 Job ID의 상태 확인"""
    job = task_queue.fetch_job(job_id)
    if not job:
        return {"error": "존재하지 않는 작업입니다."}

    return {
        "job_id": job.id,
        "status": job.get_status(),
        "result": job.result,
        "enqueued_at": str(job.enqueued_at),
        "ended_at": str(job.ended_at) if job.ended_at else None,
    }


@app.get("/download/{filename}")
async def download_file(filename: str):
    """변환된 Markdown 파일을 다운로드합니다."""
    file_path = Path(OUTPUT_DIR) / filename
    if not file_path.exists():
        return {"error": "파일이 존재하지 않습니다."}
    return file_path.read_text(encoding="utf-8")