from typing import List
from app.api.recovery_of_promotional import schema
from app.models.main.recovery_of_promotional import RecoveryOfPromotionalBase, TblRecoveryOfPromotional
from app.utils.schemas_utils import CustomResponse
from app.locale.messages import Messages
from fastapi import HTTPException
from app.dependency.authantication import JWTPayloadSchema
from sqlalchemy.orm import Session

class RecoveryOfPromotionalService:
    def __init__(self, db:Session, token:JWTPayloadSchema):
        self.db = db
        self.token = token
        
    async def create_recovery(self, request:schema.RecoveryOfPromotionalCreate):
        created_recovery = RecoveryOfPromotionalBase.model_validate(request.model_dump())
        TblRecoveryOfPromotional.create_recovery(created_recovery, self.db)
        self.db.commit()
        return CustomResponse(status="1", message=Messages.RECOVERY_OF_PROMOTIONAL)
    
    async def get_recovery(self, group_id:int):
        new_get_recovery = TblRecoveryOfPromotional.get_recovery(group_id, self.db)
        if not new_get_recovery:
            raise HTTPException(status_code=404, detail="Get group ID not found")
        return [schema.RecoveryOfPromotionalResponse.model_validate(get_group) for get_group in new_get_recovery]
    
    async def update_recovery(self, request:List[schema.RecoveryOfPromotionalupdate]):
        for req in request:
            updated_recovery = RecoveryOfPromotionalBase.model_validate(req.model_dump())
            if updated_recovery.recovery_id is None:
                return CustomResponse(status="-1", message=Messages.RECOVERY_OF_PROMOTIONAL_NOT)
            TblRecoveryOfPromotional.update_recovery(updated_recovery.recovery_id, updated_recovery, self.db)
        self.db.commit()
        return CustomResponse(status="1", message=Messages.RECOVERY_OF_PROMOTIONAL_UPDATE)
    
    async def delete_recovery(self, recovery_id:int):
        deleted_recovery = TblRecoveryOfPromotional.delete_recovery(recovery_id, self.db)
        if not deleted_recovery:
            return CustomResponse(status="-1", message=Messages.RECOVERY_OF_PROMOTIONAL_NOT)
        return CustomResponse(status="1", message=Messages.RECOVERY_OF_PROMOTIONAL_DELETE)