from loguru import logger from fastapi import APIRouter, HTTPException from chain_service.database.models.chain import Chain from chain_service.dependencies.chain_repository import ChainRepositoryDependency from chain_service.dependencies.file_uploader_service import ( FileUploaderServiceDependency, ) router = APIRouter(prefix="/chain") @router.post("/") async def chain_upsert_controller( chain: Chain, chain_repository: ChainRepositoryDependency, file_uploader_service: FileUploaderServiceDependency, ): try: upserted_chain = await chain_repository.upsert(chain) await file_uploader_service.upload_from_chain(upserted_chain) return upserted_chain except Exception: logger.exception(f"Error during chain upsert {chain.model_dump_json()}") return HTTPException(status_code=500, detail="Error during chain upsert") @router.get("/list") async def chain_list_controller(chain_repository: ChainRepositoryDependency): try: chains = await chain_repository.get_list() return chains except Exception: logger.exception("Error during chain list") return HTTPException(status_code=500, detail="Error during chain list") @router.get("/{chain_id}") async def chain_get_controller( chain_id: str, chain_repository: ChainRepositoryDependency ): try: assert (chain := await chain_repository.get_by_id(chain_id)) return chain except AssertionError: logger.warning(f"Chain not found {chain_id}") return HTTPException(status_code=404, detail="Chain not found") except Exception: logger.exception("Error during chain get") return HTTPException(status_code=500, detail="Error during chain get") @router.delete("/delete/{chain_id}") async def chain_delete_controller( chain_id: str, chain_repository: ChainRepositoryDependency ): try: await chain_repository.delete_by_id(chain_id) return {} except Exception: logger.exception("Error during chain deletion") return HTTPException(status_code=500, detail="Error during chain deletion")