from rest_framework import serializers, status

from api.models import TtSetting, TtActivity, UserConstraintSetting, UserSchedulingPreference, TtActivityWeek, TtWeek
from api.models.audit_trail import AuditTrail
from api.models.audit_trail_details import AuditTrailDetails
from api.validator import BaseValidator
from api.utils import log_critical_error, get_exception_detail, get_ip
from rest_framework.exceptions import ValidationError
from api.translation import __
from api.views.medical.base import MedicalApiBase
from backend.kafka import send_request


class RotationSchedule(MedicalApiBase):
    def validate_request(self,request):
        rules = {
            "schedule": "array",
            "socket_id": "nullable",
            "constraint_breaking_mode": "nullable",
        }

        # 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
        # validation for "schedule" array, must have activity_ids, week_id, slot

    def post(self, request):
        # Validate input
        try:
            self.validate_request(request)
            setting_params = {
                "slot_per_week",
                "slot_per_day",
                "minute_per_slot"
            }
            settings = TtSetting.get_multiple_setting(setting_params)
            slot_per_week = int(settings["slot_per_week"])
            slot_per_day = int(settings["slot_per_day"])
            minute_per_slot = int(settings["minute_per_slot"])
            redis_activity_table = "activity"

            schedule_data = request.data.get("schedule")
            action = "rotation_schedule"
            method = "schedule"

            user = request.user
            kafka_topic = self.kafka_config['REQUEST_TOPIC']

            constraint_breaking_mode = 1 if request.data.get("constraint_breaking_mode") == 1 else 0
            # constraint setting
            user_constraint_setting_list = UserConstraintSetting.get_setting(user.id)
            # scheduling preference
            user_scheduling_preference_list = UserSchedulingPreference.get_setting(user.id)
            # loop schedule data, update week pattern in db and redis, call engine schedule

            all_week_id = [val["week_id"] for val in schedule_data]
            all_activity_id = [act_id for val in schedule_data for act_id in val["activity_ids"]]

            weeks_by_id = TtWeek.objects.in_bulk(all_week_id)
            for val in schedule_data:
                # update remove week_pattern, update week relation and update redis week
                request_slot = val["slot"]
                week_id = val["week_id"]
                week_obj = weeks_by_id.get(week_id)
                week = int(week_obj.week)
                activity_ids = val["activity_ids"]
                redis_data = {
                    "update": {},
                    "delete": {}
                }

                TtActivity.objects.filter(id__in=activity_ids).update(week_pattern_id=None)
                TtActivityWeek.objects.filter(activity_id__in=activity_ids).delete()
                activity_week_to_create = []
                for activity_id in activity_ids:
                    # 2026-10-06 engine not yet done for medical schedule, so will using normal schedule way to handle first, then need update week pattern in redis
                    redis_data["update"].setdefault(redis_activity_table, []).append({
                        "id": activity_id,
                        "week_pattern_id": None,
                        "week_pattern": [week - 1],# redis week start from 0, so need -1
                    })

                    activity_week_to_create.append(TtActivityWeek(
                        activity_id=activity_id,
                        week_id=week_id
                    ))
                TtActivityWeek.objects.bulk_create(activity_week_to_create)
                # 2026-10-06 engine not support scheduled fixed slot with multiple activity, so will 1 by 1 activity call to engine
                kafka_request_data = {
                    "session_id": getattr(user, "username", str(user.id)),
                    "activities": activity_ids,
                    "slot": ((week - 1) * slot_per_week + int(request_slot)),
                    "constraint_breaking_mode": constraint_breaking_mode,
                    "constraint_setting": UserConstraintSetting.kafka_group_settings_data(user_constraint_setting_list),
                    "scheduling_preference": UserSchedulingPreference.kafka_group_settings_data(user_scheduling_preference_list),
                }

                send_request(kafka_topic, kafka_request_data, request.data.get("socket_id"), method)

            skip_audit_trail = True
            if not skip_audit_trail:
                # insert audit trail and details
                # can insert the parent first, then only insert details
                activities = TtActivity.objects.filter(id__in=all_activity_id)
                name = ",".join(activities.values_list("name", flat=True))

                audit_trail = AuditTrail.objects.create(
                    user_id=request.user.id,
                    type=self.audit_type,
                    ip_address=get_ip(request)
                )
                remark_param = {
                    "name":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="",
                )
                
            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)