from rest_framework import serializers, status
from api.models import TtLocation, TtSuitability
from api.models.audit_trail import AuditTrail
from api.models.audit_trail_details import AuditTrailDetails
from api.views.admin.base import AdminApiBase
from api.validator import BaseValidator
from api.utils import decrypt_aes_128_cbc, encrypt_aes_128_cbc, generate_kafka_data, log_critical_error, get_exception_detail, \
    get_add_new_data, get_ip, bulk_sync_to_redis, get_next_available_code
from rest_framework.exceptions import ValidationError
from api.translation import __
from backend.kafka import send_request
from backend.redis_client import redis_client


class SuitabilityCreate(AdminApiBase):
    def validate_request(self,request):
        rules = {
            "code": "nullable|not_exists:api.TtSuitability,code",
            "name": "required",
            "desc": "nullable",
            "department_id": "nullable|exists:api.TtDepartment,id",
            "staff_id": "nullable|array|exists:api.TtStaff,id",
            "location_id": "nullable|array|exists:api.TtLocation,id",
            "tag": "nullable|array|exists:api.TtTag,id",
        }

        # if need rename field, can put here
        attribute = {
            # "email": __("attr.email"),
        }

        validator = BaseValidator(request.data,rules,attribute)
        error = validator.validate()
        if error:
            raise ValidationError(error)

        # custom error after basic validation

    def post(self, request):
        # Validate input
        try:
            self.validate_request(request)

            old_new_data = {
                "old_data": None,
                "new_data": None,
            }

            action="suitability_create" # this is for audit log only

            idata = {}
            # not insert to suitability table
            ignore_field = ['staff_id','location_id']
            valid_fields = {f.column for f in TtSuitability._meta.fields}

            for key, val in request.data.items():
                if key not in ignore_field:
                    match key:
                        case _:
                            if key in valid_fields:#for insert, key can check in default part, cause if got special handle, i can check the way to use key also
                                # default the key will same as db column name
                                idata[key] = val

            # if FE no pass in code or is null or empty, will used default code
            code = request.data.get("code")
            if not code:
                idata["code"] = get_next_available_code(TtSuitability, "code", "MV", 10)
            idata['status'] = TtSuitability.STATUS_TO_CODE['active']
            idata['created_by'] = request.user.id

            suitability = TtSuitability.objects.create(**idata)

            old_new_data['new_data']=get_add_new_data(suitability)

            new_ids = request.data.get("staff_id", [])
            # the field name is passing the name in tt_suitability table, so the key not same as param
            old_new_data = self.update_m2m_field(suitability,'staff',None,new_ids,old_new_data)

            new_ids = request.data.get("location_id", [])
            # the field name is passing the name in tt_suitability table, so the key not same as param
            old_new_data = self.update_m2m_field(suitability,'location',None,new_ids,old_new_data)

            new_ids = request.data.get("tag", [])
            old_new_data = self.update_m2m_field(suitability,'tag',None,new_ids,old_new_data,False,True)

            redis_suitability_table = "suitability"
            # from here make it same name as db first, got different, only do extra handle
            redis_suitability_column = [
                "id","code","name","department_id","staff_id","location_id"
            ]
            redis_data = {
                "insert": {}
            }
            redis_suitability_data = {}
            for column in redis_suitability_column:
                match column:
                    case "staff_id" | "location_id":
                        redis_suitability_data[column] = request.data.get(column) or []
                    case _:
                        redis_suitability_data[column] = getattr(suitability, column)
            redis_data["insert"].setdefault(redis_suitability_table, []).append(redis_suitability_data)
            bulk_sync_to_redis(redis_client, redis_data)

            # kafka push
            method = "suitability_create"
            kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC']
            new_data = old_new_data.get("new_data")
            kafka_suitability = []
            if new_data and kafka_topic:
                relation_fields = [
                    "staff","location","tag"
                ]
                kafka_suitability = generate_kafka_data(new_data, relation_fields)
                kafka_request_data = {
                    "session_id": request.user.name,
                    "suitability": kafka_suitability,
                }
                send_request(kafka_topic, kafka_request_data, None, method)

            # insert audit trail and details
            # can insert the parent first, then only insert details
            audit_trail = AuditTrail.objects.create(
                user_id=request.user.id,
                type=self.audit_type,
                ip_address=get_ip(request)
            )
            remark_param = {
                "name":suitability.name,
            }

            # call to a function for insert details, so in future if want change to used worker, can modify in function only
            AuditTrailDetails.custom_insert(
                audit_trail=audit_trail,
                action=action,
                remark_param=remark_param,
                new_data=old_new_data['new_data'],
            )
            response = {

            }

            return self.api_response(data=response)
        except ValidationError as e:
            first_message = e.detail['error']
            errors = e.detail['errors']
            return self.api_response(error=first_message,errors=errors,code=status.HTTP_400_BAD_REQUEST)
        except Exception as e:
            e_details = get_exception_detail(e)
            log_critical_error(user_id=None,descr=e_details['descr'],url=e_details['url'],trace=e_details['trace'])
            return self.api_response(error=__("message.internal_server_error"),code=status.HTTP_500_INTERNAL_SERVER_ERROR)