import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  CreateTaxBracketPayload,
  TaxBracket,
  DeleteTaxBracketPayload,
  UpdateTaxBracketPayload,
} from '@/interface/TaxBracket';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { TaxBracketsActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface TaxBracketState {
  taxBrackets: TaxBracket[];
  loading: boolean;
  error: string | null;
}

const initialState: TaxBracketState = {
  taxBrackets: [],
  loading: false,
  error: null,
};

export const fetchTaxBrackets = createAsyncThunk(
  TaxBracketsActions.FETCH_TAX_BRACKETS_REQUEST,
  async (params?: { year?: number; country_code?: string }) => {
    const queryParams = new URLSearchParams();
    if (params?.year) queryParams.append('year', String(params.year));
    if (params?.country_code) queryParams.append('country_code', params.country_code);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/payroll/tax-brackets${query}`)) as AxiosResponse<
      TaxBracket[]
    >;
    return response.data;
  },
);

export const createTaxBracket = createAsyncThunk(
  TaxBracketsActions.CREATE_TAX_BRACKET_REQUEST,
  async (payload: CreateTaxBracketPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/payroll/tax-brackets',
        payload,
      )) as AxiosResponse<{ data: TaxBracket }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateTaxBracket = createAsyncThunk(
  TaxBracketsActions.UPDATE_TAX_BRACKET_REQUEST,
  async (payload: UpdateTaxBracketPayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/payroll/tax-brackets/${id}`,
        data,
      )) as AxiosResponse<{ data: TaxBracket }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteTaxBracket = createAsyncThunk(
  TaxBracketsActions.DELETE_TAX_BRACKET_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(
      `/payroll/tax-brackets/${id}`,
    )) as AxiosResponse<DeleteTaxBracketPayload>;
    const responseData = response.data;
    responseData.id = id;
    return responseData;
  },
);

export const TaxBracketsSlice = createSlice({
  name: 'taxBrackets',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchTaxBrackets.pending, (state: Draft<TaxBracketState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchTaxBrackets.fulfilled,
        (state: Draft<TaxBracketState>, action: PayloadAction<TaxBracket[]>) => {
          state.taxBrackets = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchTaxBrackets.rejected, (state: Draft<TaxBracketState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load tax brackets. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createTaxBracket.pending, (state: Draft<TaxBracketState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createTaxBracket.fulfilled,
        (state: Draft<TaxBracketState>, action: PayloadAction<TaxBracket>) => {
          state.taxBrackets.push(action.payload);
          state.taxBrackets.sort((a, b) => a.sort_order - b.sort_order);
          state.error = null;
          state.loading = false;
          message.success('Tax bracket created successfully');
        },
      )
      .addCase(
        createTaxBracket.rejected,
        (state: Draft<TaxBracketState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateTaxBracket.pending, (state: Draft<TaxBracketState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateTaxBracket.fulfilled,
        (state: Draft<TaxBracketState>, action: PayloadAction<TaxBracket>) => {
          const index = state.taxBrackets.findIndex((t) => t.id === action.payload.id);
          if (index !== -1) {
            state.taxBrackets[index] = action.payload;
          }
          state.loading = false;
          state.error = null;
          message.success('Tax bracket updated successfully');
        },
      )
      .addCase(
        updateTaxBracket.rejected,
        (state: Draft<TaxBracketState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteTaxBracket.pending, (state: Draft<TaxBracketState>) => {
        state.loading = true;
      })
      .addCase(
        deleteTaxBracket.fulfilled,
        (state: Draft<TaxBracketState>, action: PayloadAction<DeleteTaxBracketPayload>) => {
          state.taxBrackets = state.taxBrackets.filter((t) => t.id !== action.payload.id);
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(deleteTaxBracket.rejected, (state: Draft<TaxBracketState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to delete tax bracket. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });
  },
});

export default TaxBracketsSlice;
export type { TaxBracketState };