import { CreateLeaveTypePayload, DeleteLeaveTypePayload, LeaveType, UpdateLeaveTypePayload } from '@/interface/LeaveType';
import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import axios, { AxiosResponse } from 'axios';
import { LeaveTypesActions } from './actionTypes';

interface LeaveTypesState {
  leaveTypes: LeaveType[];
  loading: boolean;
  error: string | null;
}

const initialState: LeaveTypesState = {
  leaveTypes: [],
  loading: false,
  error: null,
};

const extractCollectionData = (payload: LeaveType[] | { data: LeaveType[] }): LeaveType[] => {
  return Array.isArray(payload) ? payload : payload.data;
};

export const fetchLeaveTypes = createAsyncThunk(LeaveTypesActions.FETCH_LEAVE_TYPES_REQUEST, async () => {
  const response = (await apiGet('/leave/leave-types')) as AxiosResponse<LeaveType[] | { data: LeaveType[] }>;
  return extractCollectionData(response.data);
});

export const createLeaveType = createAsyncThunk(
  LeaveTypesActions.CREATE_LEAVE_TYPE_REQUEST,
  async (payload: CreateLeaveTypePayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/leave/leave-types', payload)) as AxiosResponse<{ data: LeaveType }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateLeaveType = createAsyncThunk(
  LeaveTypesActions.UPDATE_LEAVE_TYPE_REQUEST,
  async (payload: UpdateLeaveTypePayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(`/leave/leave-types/${id}`, data)) as AxiosResponse<{ data: LeaveType }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteLeaveType = createAsyncThunk(
  LeaveTypesActions.DELETE_LEAVE_TYPE_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(`/leave/leave-types/${id}`)) as AxiosResponse<DeleteLeaveTypePayload>;
    const responseData = response.data;
    responseData.id = id;
    return responseData;
  },
);

export const LeaveTypesSlice = createSlice({
  name: 'leaveTypes',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchLeaveTypes.pending, (state: Draft<LeaveTypesState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(fetchLeaveTypes.fulfilled, (state: Draft<LeaveTypesState>, action: PayloadAction<LeaveType[]>) => {
        state.leaveTypes = action.payload;
        state.loading = false;
        state.error = null;
      })
      .addCase(fetchLeaveTypes.rejected, (state: Draft<LeaveTypesState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createLeaveType.pending, (state: Draft<LeaveTypesState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(createLeaveType.fulfilled, (state: Draft<LeaveTypesState>, action: PayloadAction<LeaveType>) => {
        state.leaveTypes.unshift(action.payload);
        state.loading = false;
        state.error = null;
        message.success('Leave type created successfully');
      })
      .addCase(createLeaveType.rejected, (state: Draft<LeaveTypesState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });

    builder
      .addCase(updateLeaveType.pending, (state: Draft<LeaveTypesState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(updateLeaveType.fulfilled, (state: Draft<LeaveTypesState>, action: PayloadAction<LeaveType>) => {
        const index = state.leaveTypes.findIndex((item) => item.id === action.payload.id);
        if (index !== -1) {
          state.leaveTypes[index] = action.payload;
        }
        state.loading = false;
        state.error = null;
        message.success('Leave type updated successfully');
      })
      .addCase(updateLeaveType.rejected, (state: Draft<LeaveTypesState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });

    builder
      .addCase(deleteLeaveType.pending, (state: Draft<LeaveTypesState>) => {
        state.loading = true;
      })
      .addCase(
        deleteLeaveType.fulfilled,
        (state: Draft<LeaveTypesState>, action: PayloadAction<DeleteLeaveTypePayload>) => {
          state.leaveTypes = state.leaveTypes.filter((item) => item.id !== action.payload.id);
          state.loading = false;
          state.error = null;
          message.success(action.payload.message);
        },
      )
      .addCase(deleteLeaveType.rejected, (state: Draft<LeaveTypesState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });
  },
});

export default LeaveTypesSlice;
export type { LeaveTypesState };