"""
Video Background Tasks

영상 생성 관련 백그라운드 태스크를 정의합니다.
"""

from datetime import date
from pathlib import Path

import aiofiles
import httpx
from sqlalchemy import select

from app.database.session import AsyncSessionLocal
from app.video.models import Video
from app.utils.common import generate_task_id
from config import prj_settings


async def download_and_save_video(
    task_id: str,
    video_url: str,
    store_name: str,
) -> None:
    """백그라운드에서 영상을 다운로드하고 Video 테이블을 업데이트합니다.

    Args:
        task_id: 프로젝트 task_id
        video_url: 다운로드할 영상 URL
        store_name: 저장할 파일명에 사용할 업체명
    """
    print(f"[download_and_save_video] START - task_id: {task_id}, store_name: {store_name}")
    try:
        # 저장 경로 생성: media/{날짜}/{uuid7}/{store_name}.mp4
        today = date.today().isoformat()
        unique_id = await generate_task_id()
        # 파일명에 사용할 수 없는 문자 제거
        safe_store_name = "".join(
            c for c in store_name if c.isalnum() or c in (" ", "_", "-")
        ).strip()
        safe_store_name = safe_store_name or "video"
        file_name = f"{safe_store_name}.mp4"

        # 절대 경로 생성
        media_dir = Path("media") / today / unique_id
        media_dir.mkdir(parents=True, exist_ok=True)
        file_path = media_dir / file_name
        print(f"[download_and_save_video] Directory created - path: {file_path}")

        # 영상 파일 다운로드
        print(f"[download_and_save_video] Downloading video - task_id: {task_id}, url: {video_url}")
        async with httpx.AsyncClient() as client:
            response = await client.get(video_url, timeout=120.0)  # 영상은 더 큰 timeout
            response.raise_for_status()

            async with aiofiles.open(str(file_path), "wb") as f:
                await f.write(response.content)
        print(f"[download_and_save_video] File saved - task_id: {task_id}, path: {file_path}")

        # 프론트엔드에서 접근 가능한 URL 생성
        relative_path = f"/media/{today}/{unique_id}/{file_name}"
        base_url = f"http://{prj_settings.PROJECT_DOMAIN}"
        file_url = f"{base_url}{relative_path}"
        print(f"[download_and_save_video] URL generated - task_id: {task_id}, url: {file_url}")

        # Video 테이블 업데이트 (새 세션 사용)
        async with AsyncSessionLocal() as session:
            # 여러 개 있을 경우 가장 최근 것 선택
            result = await session.execute(
                select(Video)
                .where(Video.task_id == task_id)
                .order_by(Video.created_at.desc())
                .limit(1)
            )
            video = result.scalar_one_or_none()

            if video:
                video.status = "completed"
                video.result_movie_url = file_url
                await session.commit()
                print(f"[download_and_save_video] SUCCESS - task_id: {task_id}, status: completed")
            else:
                print(f"[download_and_save_video] Video NOT FOUND in DB - task_id: {task_id}")

    except Exception as e:
        print(f"[download_and_save_video] EXCEPTION - task_id: {task_id}, error: {e}")
        # 실패 시 Video 테이블 업데이트
        async with AsyncSessionLocal() as session:
            # 여러 개 있을 경우 가장 최근 것 선택
            result = await session.execute(
                select(Video)
                .where(Video.task_id == task_id)
                .order_by(Video.created_at.desc())
                .limit(1)
            )
            video = result.scalar_one_or_none()

            if video:
                video.status = "failed"
                await session.commit()
                print(f"[download_and_save_video] FAILED - task_id: {task_id}, status updated to failed")