import {
  CreateEmployeeCategoryPayload,
  DeleteEmployeeCategoryResponse,
  EmployeeCategory,
} from '@/interface/EmployeeCategory';
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 { EmployeeCategoryActions } from './actionTypes';

interface EmployeeCategoryState {
  employeeCategories: EmployeeCategory[];
  loading: boolean;
  error: string | null;
}

const initialState: EmployeeCategoryState = {
  employeeCategories: [],
  loading: false,
  error: null,
};

export const fetchEmployeeCategories = createAsyncThunk(
  EmployeeCategoryActions.FETCH_EMPLOYEE_CATEGORY_REQUEST,
  async () => {
    const response = (await apiGet('/hr/employee-categories')) as AxiosResponse<EmployeeCategory[]>;
    return response.data;
  },
);

export const createEmployeeCategory = createAsyncThunk(
  EmployeeCategoryActions.CREATE_EMPLOYEE_CATEGORY_REQUEST,
  async (payload: CreateEmployeeCategoryPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/hr/employee-categories',
        payload,
      )) as AxiosResponse<EmployeeCategory>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateEmployeeCategory = createAsyncThunk(
  EmployeeCategoryActions.UPDATE_EMPLOYEE_CATEGORY_REQUEST,
  async (payload: CreateEmployeeCategoryPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPut(
        `/hr/employee-categories/${payload.id}`,
        payload,
      )) as AxiosResponse<EmployeeCategory>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteEmployeeCategory = createAsyncThunk(
  EmployeeCategoryActions.DELETE_EMPLOYEE_CATEGORY_REQUEST,
  async (payload: string) => {
    const response = (await apiDelete(
      `/hr/employee-categories/${payload}`,
    )) as AxiosResponse<DeleteEmployeeCategoryResponse>;
    const responseData = response.data;
    responseData.id = payload;
    return responseData;
  },
);

export const EmployeeCategorySlice = createSlice({
  name: 'employeeCategories',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchEmployeeCategories.pending, (state: Draft<EmployeeCategoryState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchEmployeeCategories.fulfilled,
        (state: Draft<EmployeeCategoryState>, action: PayloadAction<EmployeeCategory[]>) => {
          state.employeeCategories = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchEmployeeCategories.rejected, (state: Draft<EmployeeCategoryState>) => {
        state.loading = false;
        message.error(
          'Failed to load Staff Categories. Please try again or contact support if the issue persists',
        );
      });

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

    builder
      .addCase(updateEmployeeCategory.pending, (state: Draft<EmployeeCategoryState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateEmployeeCategory.fulfilled,
        (state: Draft<EmployeeCategoryState>, action: PayloadAction<EmployeeCategory>) => {
          state.loading = false;
          state.error = null;
          message.success('Successfully updated Staff Category', 10);
        },
      )
      .addCase(
        updateEmployeeCategory.rejected,
        (state: Draft<EmployeeCategoryState>, action: RejectedActionPayload) => {
          state.loading = false;
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteEmployeeCategory.pending, (state: Draft<EmployeeCategoryState>) => {})
      .addCase(
        deleteEmployeeCategory.fulfilled,
        (
          state: Draft<EmployeeCategoryState>,
          action: PayloadAction<DeleteEmployeeCategoryResponse>,
        ) => {
          state.employeeCategories = state.employeeCategories.filter(
            (category) => category.id !== action.payload.id,
          );
          message.success(action.payload.message, 10);
        },
      )
      .addCase(deleteEmployeeCategory.rejected, (state: Draft<EmployeeCategoryState>, action: RejectedActionPayload) => {
        message.error(Helpers.handleServerError(action.payload), 8);
      });
  },
});

export default EmployeeCategorySlice;
export type { EmployeeCategoryState };