import {
  DeleteDisciplinaryActionPayload,
  DeleteDisciplinaryActionResponse,
  DisciplinaryAction,
  UpdateDisciplinaryActionPayload,
} from '@/interface/DisciplinaryAction';
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 { DisciplinaryActionActions } from './actionTypes';

interface DisciplinaryActionState {
  disciplinaryActions: DisciplinaryAction[];
  loading: boolean;
  deleting: boolean;
  error: string | null;
}

const initialState: DisciplinaryActionState = {
  disciplinaryActions: [],
  loading: false,
  deleting: false,
  error: null,
};

export const fetchDisciplinaryActions = createAsyncThunk(
  DisciplinaryActionActions.FETCH_DISCIPLINARY_ACTIONS_REQUEST,
  async (payload: string) => {
    const response = (await apiGet(
      `/hr/employees/${payload}/disciplinary-actions`,
    )) as AxiosResponse<DisciplinaryAction[]>;
    return response.data;
  },
);

export const createDisciplinaryAction = createAsyncThunk(
  DisciplinaryActionActions.CREATE_DISCIPLINARY_ACTION_REQUEST,
  async (payload: FormData, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        `/hr/employees/${payload.get('employee_id')}/disciplinary-actions`,
        payload,
        {
          headers: {
            'Content-Type': 'multipart/form-data',
          },
        },
      )) as AxiosResponse<DisciplinaryAction>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateDisciplinaryAction = createAsyncThunk(
  DisciplinaryActionActions.UPDATE_DISCIPLINARY_ACTION_REQUEST,
  async (payload: UpdateDisciplinaryActionPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPut(
        `/hr/employees/${payload.employee_id}/disciplinary-actions/${payload.id}`,
        payload,
      )) as AxiosResponse<DisciplinaryAction>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteDisciplinaryAction = createAsyncThunk(
  DisciplinaryActionActions.DELETE_DISCIPLINARY_ACTION_REQUEST,
  async (payload: DeleteDisciplinaryActionPayload) => {
    const response = (await apiDelete(
      `/hr/employees/${payload.employee_id}/disciplinary-actions/${payload.disciplinary_action_id}`,
    )) as AxiosResponse<DeleteDisciplinaryActionResponse>;

    const responseData = response.data;
    responseData.id = payload.disciplinary_action_id;
    return responseData;
  },
);

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

    builder
      .addCase(createDisciplinaryAction.pending, (state: Draft<DisciplinaryActionState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createDisciplinaryAction.fulfilled,
        (state: Draft<DisciplinaryActionState>, action: PayloadAction<DisciplinaryAction>) => {
          state.loading = false;
          state.error = null;
          state.disciplinaryActions.unshift(action.payload);
        },
      )
      .addCase(
        createDisciplinaryAction.rejected,
        (state: Draft<DisciplinaryActionState>, action: RejectedActionPayload) => {
          state.loading = false;
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateDisciplinaryAction.pending, (state: Draft<DisciplinaryActionState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateDisciplinaryAction.fulfilled,
        (state: Draft<DisciplinaryActionState>, action: PayloadAction<DisciplinaryAction>) => {
          state.loading = false;
          state.error = null;

          const index = state.disciplinaryActions.findIndex(
            (item) => item.id === action.payload.id,
          );
          if (index > -1) {
            state.disciplinaryActions[index] = action.payload;
          }

          message.success('Successfully updated Disciplinary Action', 8);
        },
      )
      .addCase(
        updateDisciplinaryAction.rejected,
        (state: Draft<DisciplinaryActionState>, action: RejectedActionPayload) => {
          state.loading = false;
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteDisciplinaryAction.pending, (state: Draft<DisciplinaryActionState>) => {
        state.deleting = true;
        state.error = null;
      })
      .addCase(
        deleteDisciplinaryAction.fulfilled,
        (
          state: Draft<DisciplinaryActionState>,
          action: PayloadAction<DeleteDisciplinaryActionResponse>,
        ) => {
          state.deleting = false;
          state.error = null;
          state.disciplinaryActions = state.disciplinaryActions.filter(
            (item) => item.id !== action.payload.id,
          );
          message.success(action.payload.message, 10);
        },
      )
      .addCase(deleteDisciplinaryAction.rejected, (state: Draft<DisciplinaryActionState>) => {
        state.deleting = false;
        message.error(
          'Failed to delete Disciplinary Action. Please try again or contact support',
          10,
        );
      });
  },
});

export default DisciplinaryActionSlice;
export type { DisciplinaryActionState };