import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  TrainingAttendance,
  CreateTrainingAttendancePayload,
  MarkBulkAttendancePayload,
  DeleteTrainingAttendancePayload,
} from '@/interface/Training';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { TrainingAttendanceActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface TrainingAttendanceState {
  attendance: TrainingAttendance[];
  loading: boolean;
  error: string | null;
}

const initialState: TrainingAttendanceState = {
  attendance: [],
  loading: false,
  error: null,
};

export const fetchAttendance = createAsyncThunk(
  TrainingAttendanceActions.FETCH_ATTENDANCE_REQUEST,
  async (params?: { session_id?: string; date?: string; employee_id?: string }) => {
    const queryParams = new URLSearchParams();
    if (params?.session_id) queryParams.append('session_id', params.session_id);
    if (params?.date) queryParams.append('date', params.date);
    if (params?.employee_id) queryParams.append('employee_id', params.employee_id);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/training/attendance${query}`)) as AxiosResponse<
      TrainingAttendance[]
    >;
    return response.data;
  },
);

export const createAttendance = createAsyncThunk(
  TrainingAttendanceActions.CREATE_ATTENDANCE_REQUEST,
  async (payload: CreateTrainingAttendancePayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/training/attendance',
        payload,
      )) as AxiosResponse<{ data: TrainingAttendance }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateAttendance = createAsyncThunk(
  TrainingAttendanceActions.UPDATE_ATTENDANCE_REQUEST,
  async (
    payload: {
      id: string;
      check_in_time?: string;
      check_out_time?: string;
      status?: string;
      hours_attended?: number;
      notes?: string;
    },
    { rejectWithValue },
  ) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/training/attendance/${id}`,
        data,
      )) as AxiosResponse<{ data: TrainingAttendance }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteAttendance = createAsyncThunk(
  TrainingAttendanceActions.DELETE_ATTENDANCE_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(
      `/training/attendance/${id}`,
    )) as AxiosResponse<DeleteTrainingAttendancePayload>;
    const responseData = response.data;
    responseData.id = id;
    return responseData;
  },
);

export const markBulkAttendance = createAsyncThunk(
  TrainingAttendanceActions.MARK_BULK_ATTENDANCE_REQUEST,
  async (payload: MarkBulkAttendancePayload, { rejectWithValue }) => {
    try {
      const { training_session_id, ...data } = payload;
      const response = (await apiPost(
        `/training/attendance/sessions/${training_session_id}/bulk`,
        data,
      )) as AxiosResponse<{ message: string; data: TrainingAttendance[] }>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const TrainingAttendanceSlice = createSlice({
  name: 'trainingAttendance',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchAttendance.pending, (state: Draft<TrainingAttendanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchAttendance.fulfilled,
        (
          state: Draft<TrainingAttendanceState>,
          action: PayloadAction<TrainingAttendance[]>,
        ) => {
          state.attendance = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchAttendance.rejected, (state: Draft<TrainingAttendanceState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load attendance records. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createAttendance.pending, (state: Draft<TrainingAttendanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createAttendance.fulfilled,
        (
          state: Draft<TrainingAttendanceState>,
          action: PayloadAction<TrainingAttendance>,
        ) => {
          state.attendance.unshift(action.payload);
          state.error = null;
          state.loading = false;
          message.success('Attendance recorded successfully');
        },
      )
      .addCase(
        createAttendance.rejected,
        (state: Draft<TrainingAttendanceState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateAttendance.pending, (state: Draft<TrainingAttendanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateAttendance.fulfilled,
        (
          state: Draft<TrainingAttendanceState>,
          action: PayloadAction<TrainingAttendance>,
        ) => {
          const index = state.attendance.findIndex((a) => a.id === action.payload.id);
          if (index !== -1) {
            state.attendance[index] = action.payload;
          }
          state.loading = false;
          state.error = null;
          message.success('Attendance updated successfully');
        },
      )
      .addCase(
        updateAttendance.rejected,
        (state: Draft<TrainingAttendanceState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteAttendance.pending, (state: Draft<TrainingAttendanceState>) => {
        state.loading = true;
      })
      .addCase(
        deleteAttendance.fulfilled,
        (
          state: Draft<TrainingAttendanceState>,
          action: PayloadAction<DeleteTrainingAttendancePayload>,
        ) => {
          state.attendance = state.attendance.filter((a) => a.id !== action.payload.id);
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(deleteAttendance.rejected, (state: Draft<TrainingAttendanceState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to delete attendance record. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(markBulkAttendance.pending, (state: Draft<TrainingAttendanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        markBulkAttendance.fulfilled,
        (
          state: Draft<TrainingAttendanceState>,
          action: PayloadAction<{ message: string; data: TrainingAttendance[] }>,
        ) => {
          for (const record of action.payload.data) {
            const index = state.attendance.findIndex((a) => a.id === record.id);
            if (index !== -1) {
              state.attendance[index] = record;
            } else {
              state.attendance.push(record);
            }
          }
          state.loading = false;
          state.error = null;
          message.success(action.payload.message);
        },
      )
      .addCase(
        markBulkAttendance.rejected,
        (state: Draft<TrainingAttendanceState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );
  },
});

export default TrainingAttendanceSlice;
export type { TrainingAttendanceState };