from datetime import datetime, timezone

from app.config import get_settings
from app.core.storage import get_minio_service
from .celery_app import celery_app
from packages.adapters.sqlalchemy_impl.session import SessionLocal, build_session_factory
from packages.adapters.sqlalchemy_impl.generation_task_repository import SQLAlchemyGenerationTaskRepository
from packages.adapters.sqlalchemy_impl.generated_video_repository import SQLAlchemyGeneratedVideoRepository
from packages.domain import GeneratedVideo, GenerationTaskStatus

settings = get_settings()
if SessionLocal is None:
    build_session_factory(settings.database_url)


@celery_app.task(name="worker.generate_video")
def generate_video(task_id: str) -> dict:
    session = SessionLocal()
    try:
        task_repo = SQLAlchemyGenerationTaskRepository(session)
        video_repo = SQLAlchemyGeneratedVideoRepository(session)
        storage_service = get_minio_service()

        task = task_repo.get(task_id)
        if task is None:
            return {"ok": False, "error": f"generation task {task_id} not found"}

        task.status = GenerationTaskStatus.RUNNING
        task.progress = 10.0
        task.started_at = task.started_at or datetime.now(timezone.utc)
        task_repo.update(task)

        file_name = f"{task.id}.mp4"
        storage_key = f"workspaces/{task.workspace_id}/projects/{task.project_id}/generated/{task.id}/{file_name}"
        file_url = storage_service.get_url(storage_key)

        video = GeneratedVideo.create(
            workspace_id=task.workspace_id,
            project_id=task.project_id,
            generation_task_id=task.id,
            name=file_name,
            file_url=file_url,
            file_size=1024,
            duration=10.0,
            width=1920,
            height=1080,
            fps=25.0,
        )
        video_repo.create(video)

        task.status = GenerationTaskStatus.COMPLETED
        task.progress = 100.0
        task.result_count = 1
        task.completed_at = datetime.now(timezone.utc)
        task_repo.update(task)

        return {"ok": True, "task_id": task.id, "video_id": video.id, "file_url": file_url}
    except Exception as error:
        try:
            task_repo = SQLAlchemyGenerationTaskRepository(session)
            task = task_repo.get(task_id)
            if task is not None:
                task.status = GenerationTaskStatus.FAILED
                task.error_message = str(error)
                task.completed_at = datetime.now(timezone.utc)
                task_repo.update(task)
        except Exception:
            pass
        return {"ok": False, "task_id": task_id, "error": str(error)}
    finally:
        session.close()