from sqlalchemy.orm import Session

from packages.adapters.sqlalchemy_impl.models import GeneratedVideoModel
from packages.domain import GeneratedVideo


class SQLAlchemyGeneratedVideoRepository:
    def __init__(self, session: Session):
        self.session = session

    def create(self, video: GeneratedVideo) -> GeneratedVideo:
        model = GeneratedVideoModel(
            id=video.id,
            workspace_id=video.workspace_id,
            project_id=video.project_id,
            generation_task_id=video.generation_task_id,
            name=video.name,
            file_url=video.file_url,
            file_size=video.file_size,
            duration=video.duration,
            thumbnail_url=video.thumbnail_url,
            width=video.width,
            height=video.height,
            fps=video.fps,
            generated_at=video.generated_at,
            created_at=video.created_at,
        )
        self.session.add(model)
        self.session.commit()
        return video

    def get(self, video_id: str) -> GeneratedVideo | None:
        model = self.session.query(GeneratedVideoModel).filter(GeneratedVideoModel.id == video_id).first()
        if model is None:
            return None
        return GeneratedVideo(
            id=model.id,
            workspace_id=model.workspace_id,
            project_id=model.project_id,
            generation_task_id=model.generation_task_id,
            name=model.name,
            file_url=model.file_url,
            file_size=int(model.file_size or 0),
            duration=model.duration,
            thumbnail_url=model.thumbnail_url,
            width=int(model.width or 0),
            height=int(model.height or 0),
            fps=model.fps,
            generated_at=model.generated_at,
            created_at=model.created_at,
        )

    def list_by_project(self, project_id: str) -> list[GeneratedVideo]:
        models = self.session.query(GeneratedVideoModel).filter(GeneratedVideoModel.project_id == project_id).all()
        return [self.get(model.id) for model in models if self.get(model.id) is not None]

    def list_by_generation_task(self, generation_task_id: str) -> list[GeneratedVideo]:
        models = (
            self.session.query(GeneratedVideoModel)
            .filter(GeneratedVideoModel.generation_task_id == generation_task_id)
            .all()
        )
        return [self.get(model.id) for model in models if self.get(model.id) is not None]