import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  AllowanceMatrixResponse,
  CreateEmployeeAllowancePayload,
  DeleteEmployeeAllowancePayload,
  EmployeeAllowance,
  UpdateEmployeeAllowancePayload,
} from '@/interface/EmployeeAllowance';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { EmployeeAllowancesActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface EmployeeAllowanceState {
  employeeAllowances: EmployeeAllowance[];
  matrix: AllowanceMatrixResponse | null;
  loading: boolean;
  matrixLoading: boolean;
  error: string | null;
}

const initialState: EmployeeAllowanceState = {
  employeeAllowances: [],
  matrix: null,
  loading: false,
  matrixLoading: false,
  error: null,
};

export const fetchEmployeeAllowances = createAsyncThunk(
  EmployeeAllowancesActions.FETCH_EMPLOYEE_ALLOWANCES_REQUEST,
  async (
    params?: {
      employee_id?: string;
      allowance_type_id?: string;
      branch_id?: string;
      org_unit_id?: string;
      search?: string;
    },
  ) => {
    const queryParams = new URLSearchParams();
    if (params?.employee_id) queryParams.append('employee_id', params.employee_id);
    if (params?.allowance_type_id)
      queryParams.append('allowance_type_id', params.allowance_type_id);
    if (params?.branch_id) queryParams.append('branch_id', params.branch_id);
    if (params?.org_unit_id) queryParams.append('org_unit_id', params.org_unit_id);
    if (params?.search) queryParams.append('search', params.search);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/payroll/employee-allowances${query}`)) as AxiosResponse<
      EmployeeAllowance[]
    >;
    return response.data;
  },
);

export const fetchAllowanceMatrix = createAsyncThunk(
  EmployeeAllowancesActions.FETCH_ALLOWANCE_MATRIX_REQUEST,
  async (params?: { branch_id?: string; org_unit_id?: string; search?: string; salary_grade_id?: string; salary_notch_id?: string }) => {
    const queryParams = new URLSearchParams();
    if (params?.branch_id) queryParams.append('branch_id', params.branch_id);
    if (params?.org_unit_id) queryParams.append('org_unit_id', params.org_unit_id);
    if (params?.search) queryParams.append('search', params.search);
    if (params?.salary_grade_id) queryParams.append('salary_grade_id', params.salary_grade_id);
    if (params?.salary_notch_id) queryParams.append('salary_notch_id', params.salary_notch_id);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/payroll/allowances/matrix${query}`)) as AxiosResponse<
      AllowanceMatrixResponse
    >;
    return response.data;
  },
);

export const createEmployeeAllowance = createAsyncThunk(
  EmployeeAllowancesActions.CREATE_EMPLOYEE_ALLOWANCE_REQUEST,
  async (payload: CreateEmployeeAllowancePayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/payroll/employee-allowances',
        payload,
      )) as AxiosResponse<{ data: EmployeeAllowance }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateEmployeeAllowance = createAsyncThunk(
  EmployeeAllowancesActions.UPDATE_EMPLOYEE_ALLOWANCE_REQUEST,
  async (payload: UpdateEmployeeAllowancePayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/payroll/employee-allowances/${id}`,
        data,
      )) as AxiosResponse<{ data: EmployeeAllowance }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteEmployeeAllowance = createAsyncThunk(
  EmployeeAllowancesActions.DELETE_EMPLOYEE_ALLOWANCE_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(
      `/payroll/employee-allowances/${id}`,
    )) as AxiosResponse<DeleteEmployeeAllowancePayload>;
    const responseData = response.data;
    responseData.id = id;
    return responseData;
  },
);

export const EmployeeAllowancesSlice = createSlice({
  name: 'employeeAllowances',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchEmployeeAllowances.pending, (state: Draft<EmployeeAllowanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchEmployeeAllowances.fulfilled,
        (
          state: Draft<EmployeeAllowanceState>,
          action: PayloadAction<EmployeeAllowance[]>,
        ) => {
          state.employeeAllowances = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchEmployeeAllowances.rejected, (state: Draft<EmployeeAllowanceState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load employee allowances. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchAllowanceMatrix.pending, (state: Draft<EmployeeAllowanceState>) => {
        state.matrixLoading = true;
        state.error = null;
      })
      .addCase(
        fetchAllowanceMatrix.fulfilled,
        (
          state: Draft<EmployeeAllowanceState>,
          action: PayloadAction<AllowanceMatrixResponse>,
        ) => {
          state.matrix = action.payload;
          state.error = null;
          state.matrixLoading = false;
        },
      )
      .addCase(fetchAllowanceMatrix.rejected, (state: Draft<EmployeeAllowanceState>, action: RejectedActionPayload) => {
        state.matrixLoading = false;
        state.error = 'Failed to load allowance matrix. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createEmployeeAllowance.pending, (state: Draft<EmployeeAllowanceState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createEmployeeAllowance.fulfilled,
        (
          state: Draft<EmployeeAllowanceState>,
          action: PayloadAction<EmployeeAllowance>,
        ) => {
          state.employeeAllowances.unshift(action.payload);
          state.error = null;
          state.loading = false;
          message.success('Employee allowance created successfully');
        },
      )
      .addCase(
        createEmployeeAllowance.rejected,
        (state: Draft<EmployeeAllowanceState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

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

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

export default EmployeeAllowancesSlice;
export type { EmployeeAllowanceState };
