import json

from sqlalchemy.orm import Session

from packages.adapters.sqlalchemy_impl.models import AssetModel
from packages.domain import Asset


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

    def list_by_library(self, library_id: str) -> list[Asset]:
        models = self.session.query(AssetModel).filter(AssetModel.library_id == library_id).all()
        return [
            Asset(
                id=model.id,
                workspace_id=model.workspace_id,
                project_id=model.project_id,
                library_id=model.library_id,
                name=model.name,
                storage_key=model.storage_key,
                mime_type=model.mime_type,
                metadata=json.loads(model.metadata_json),
                created_at=model.created_at,
            )
            for model in models
        ]

    def create(self, asset: Asset) -> Asset:
        model = AssetModel(
            id=asset.id,
            workspace_id=asset.workspace_id,
            project_id=asset.project_id,
            library_id=asset.library_id,
            name=asset.name,
            storage_key=asset.storage_key,
            mime_type=asset.mime_type,
            metadata_json=json.dumps(asset.metadata),
            created_at=asset.created_at,
        )
        self.session.add(model)
        self.session.commit()
        return asset