from flask import request, jsonify
from marshmallow import ValidationError
from app.api.enums.loan_status import LoanStatus
from app.api.helpers.response_helper import ResponseHelper
from app.models import Repayment
from app.models.customer import Customer
from app.models.loan import Loan
from app.utils.logger import logger
from app.api.schemas.repayment import RepaymentSchema
from app.api.services.base_service import BaseService
from app.api.enums import TransactionType 
from threading import Thread
from app.extensions import db

class RepaymentService(BaseService):
    TRANSACTION_TYPE = TransactionType.REPAYMENT 

    @staticmethod
    def process_request(data):
        """
        Process the Repayment request.

        Args:
            data (dict): The request data.

        Returns:
            dict: A standardized response.
        """
        try:
            with db.session.begin():
                validated_data = RepaymentService.validate_data(data, RepaymentSchema())

                customer_id =  validated_data.get('customerId')
                request_id =  validated_data.get('requestId')
                loan_id = validated_data.get('debtId')
                account_id = validated_data.get('accountId')
                loan_ref = validated_data.get('loanRef')
                # customer = Customer.get_customer_with_loan_list(customer_id)
                transaction_id  = validated_data.get('transactionId')
                initiated_by = validated_data.get('initiatedBy')
                logger.error(f"HERE 0002a **** ")
                if(RepaymentService.validate_account_ownership(account_id = account_id, customer_id = customer_id)):
                    logger.error(f"HERE 0001a **** ")
                    # Check loan exists
                    loan = Loan.get_customer_loan(loan_id = loan_id, customer_id = customer_id)

                    # Save the repayment details  
                    repayment = Repayment.create_repayment(
                        customer_id = customer_id,
                        loan = loan,
                        transaction_id = transaction_id
                    )

                    if not repayment:
                        logger.error(f"Failed to save repayment details")
                        return ResponseHelper.error(result_description="Failed to save repayment details.")

                    #Update Loan status
                    Loan.update_status(loan_id = loan_id, status = LoanStatus.START_REPAY) # repay started bu user
                    transaction = RepaymentService.log_transaction(validated_data = validated_data)

                    if not transaction:
                        logger.error(f"Failed to log transaction")
                        return ResponseHelper.error(result_description="Failed to log transaction.") 
                else:
                    logger.error(f"Invalid Customer or AccountID {account_id} to CustomerID{customer_id} ")
                    return ResponseHelper.error(result_description="Invalid Customer or Account")

                # Simulated processing logic
                # TODO start using repayment_id instead if id or Id
                response_data = {
                    "Id": repayment.id,
                    "repayment_id": repayment.id,
                    "initiated_by": repayment.initiated_by,
                    "transactionId": transaction_id,
                    "customerId": customer_id,
                    "productId": loan.product_id,
                    "loanRef": loan_ref,
                    "debtId": loan_id
                }

                # Call Kafka in a background thread
                thread = Thread(target=RepaymentService.async_send_to_kafka, args=(response_data, request_id, "LOAN_REPAYMENT"))
                thread.start()

                db.session.commit()
                return ResponseHelper.success(data=response_data)

        except ValidationError as err:

            logger.error(f"Validation Error: {getattr(err, 'messages', str(err))}")
            db.session.rollback()
            return ResponseHelper.unprocessable_entity(result_description="Validation exception")
        
        except  ValueError as err:
            logger.error(f"{getattr(err, 'messages', str(err))}")
            db.session.rollback()
            return ResponseHelper.error(result_description=str(err))

        except Exception as e:
            logger.error(f"An error occurred: {str(e)}", exc_info=True)
            db.session.rollback()
            return ResponseHelper.internal_server_error()