import { PensionCountryTemplate, PensionSetupStatus } from '@/interface/PensionSetup';
import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import { apiGet, apiPost } 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 { PensionSetupActions } from './actionTypes';

interface PensionSetupState {
  countries: PensionCountryTemplate[];
  status: PensionSetupStatus | null;
  loading: boolean;
  applying: boolean;
  error: string | null;
}

const initialState: PensionSetupState = {
  countries: [],
  status: null,
  loading: false,
  applying: false,
  error: null,
};

export const fetchPensionSetupCountries = createAsyncThunk(
  PensionSetupActions.FETCH_COUNTRIES_REQUEST,
  async () => {
    const response = (await apiGet('/payroll/pension-setup/countries')) as AxiosResponse<
      PensionCountryTemplate[]
    >;
    return response.data;
  },
);

export const fetchPensionSetupStatus = createAsyncThunk(
  PensionSetupActions.FETCH_STATUS_REQUEST,
  async () => {
    const response = (await apiGet(
      '/payroll/pension-setup/status',
    )) as AxiosResponse<PensionSetupStatus>;
    return response.data;
  },
);

export const applyPensionTemplate = createAsyncThunk(
  PensionSetupActions.APPLY_TEMPLATE_REQUEST,
  async (countryCode: string, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/payroll/pension-setup/apply', {
        country_code: countryCode,
      })) as AxiosResponse<{ message: string; country_code: string; country_name: string }>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const PensionSetupSlice = createSlice({
  name: 'pensionSetup',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchPensionSetupCountries.pending, (state: Draft<PensionSetupState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchPensionSetupCountries.fulfilled,
        (state: Draft<PensionSetupState>, action: PayloadAction<PensionCountryTemplate[]>) => {
          state.countries = action.payload;
          state.loading = false;
        },
      )
      .addCase(fetchPensionSetupCountries.rejected, (state: Draft<PensionSetupState>) => {
        state.loading = false;
        state.error = 'Failed to load available countries';
      });

    builder
      .addCase(fetchPensionSetupStatus.pending, (state: Draft<PensionSetupState>) => {
        state.loading = true;
      })
      .addCase(
        fetchPensionSetupStatus.fulfilled,
        (state: Draft<PensionSetupState>, action: PayloadAction<PensionSetupStatus>) => {
          state.status = action.payload;
          state.loading = false;
        },
      )
      .addCase(fetchPensionSetupStatus.rejected, (state: Draft<PensionSetupState>) => {
        state.loading = false;
      });

    builder
      .addCase(applyPensionTemplate.pending, (state: Draft<PensionSetupState>) => {
        state.applying = true;
        state.error = null;
      })
      .addCase(applyPensionTemplate.fulfilled, (state: Draft<PensionSetupState>) => {
        state.applying = false;
        message.success('Pension structure applied successfully');
      })
      .addCase(
        applyPensionTemplate.rejected,
        (state: Draft<PensionSetupState>, action: RejectedActionPayload) => {
          state.applying = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );
  },
});

export default PensionSetupSlice;