import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  PayrollPeriod,
  CreatePayrollPeriodPayload,
  DeletePayrollPeriodPayload,
} from '@/interface/PayrollRun';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { PayrollPeriodsActions } from './actionTypes';
import { apiDelete, apiGet, apiPost } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface PayrollPeriodsState {
  periods: PayrollPeriod[];
  selectedPeriod: PayrollPeriod | null;
  loading: boolean;
  error: string | null;
}

const initialState: PayrollPeriodsState = {
  periods: [],
  selectedPeriod: null,
  loading: false,
  error: null,
};

export const fetchPayrollPeriods = createAsyncThunk(
  PayrollPeriodsActions.FETCH_PERIODS_REQUEST,
  async (params?: { year?: number }) => {
    const queryParams = new URLSearchParams();
    if (params?.year) queryParams.append('year', String(params.year));
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/payroll/payroll-periods${query}`)) as AxiosResponse<
      PayrollPeriod[]
    >;
    return response.data;
  },
);

export const fetchPayrollPeriod = createAsyncThunk(
  PayrollPeriodsActions.FETCH_PERIOD_REQUEST,
  async (id: string) => {
    const response = (await apiGet(
      `/payroll/payroll-periods/${id}`,
    )) as AxiosResponse<PayrollPeriod>;
    return response.data;
  },
);

export const createPayrollPeriod = createAsyncThunk(
  PayrollPeriodsActions.CREATE_PERIOD_REQUEST,
  async (payload: CreatePayrollPeriodPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/payroll/payroll-periods', payload)) as AxiosResponse<{
        data: PayrollPeriod;
      }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deletePayrollPeriod = createAsyncThunk(
  PayrollPeriodsActions.DELETE_PERIOD_REQUEST,
  async (id: string, { rejectWithValue }) => {
    try {
      const response = (await apiDelete(
        `/payroll/payroll-periods/${id}`,
      )) as AxiosResponse<DeletePayrollPeriodPayload>;
      const responseData = response.data;
      responseData.id = id;
      return responseData;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const lockPayrollPeriod = createAsyncThunk(
  PayrollPeriodsActions.LOCK_PERIOD_REQUEST,
  async (id: string) => {
    const response = (await apiPost(
      `/payroll/payroll-periods/${id}/lock`,
      {},
    )) as AxiosResponse<{ data: PayrollPeriod }>;
    return response.data.data;
  },
);

export const unlockPayrollPeriod = createAsyncThunk(
  PayrollPeriodsActions.UNLOCK_PERIOD_REQUEST,
  async (id: string) => {
    const response = (await apiPost(
      `/payroll/payroll-periods/${id}/unlock`,
      {},
    )) as AxiosResponse<{ data: PayrollPeriod }>;
    return response.data.data;
  },
);

export const PayrollPeriodsSlice = createSlice({
  name: 'payrollPeriods',
  initialState,
  reducers: {
    clearSelectedPeriod(state) {
      state.selectedPeriod = null;
    },
  },
  extraReducers(builder) {
    builder
      .addCase(fetchPayrollPeriods.pending, (state: Draft<PayrollPeriodsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchPayrollPeriods.fulfilled,
        (state: Draft<PayrollPeriodsState>, action: PayloadAction<PayrollPeriod[]>) => {
          state.periods = action.payload;
          state.loading = false;
          state.error = null;
        },
      )
      .addCase(fetchPayrollPeriods.rejected, (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load payroll periods';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchPayrollPeriod.pending, (state: Draft<PayrollPeriodsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchPayrollPeriod.fulfilled,
        (state: Draft<PayrollPeriodsState>, action: PayloadAction<PayrollPeriod>) => {
          state.selectedPeriod = action.payload;
          state.loading = false;
          state.error = null;
        },
      )
      .addCase(fetchPayrollPeriod.rejected, (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load payroll period';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createPayrollPeriod.pending, (state: Draft<PayrollPeriodsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createPayrollPeriod.fulfilled,
        (state: Draft<PayrollPeriodsState>, action: PayloadAction<PayrollPeriod>) => {
          state.periods.unshift(action.payload);
          state.loading = false;
          state.error = null;
          message.success('Payroll period created successfully');
        },
      )
      .addCase(
        createPayrollPeriod.rejected,
        (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deletePayrollPeriod.pending, (state: Draft<PayrollPeriodsState>) => {
        state.loading = true;
      })
      .addCase(
        deletePayrollPeriod.fulfilled,
        (state: Draft<PayrollPeriodsState>, action: PayloadAction<DeletePayrollPeriodPayload>) => {
          state.periods = state.periods.filter((p) => p.id !== action.payload.id);
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(
        deletePayrollPeriod.rejected,
        (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(lockPayrollPeriod.fulfilled, (state: Draft<PayrollPeriodsState>, action) => {
        const index = state.periods.findIndex((p) => p.id === action.payload.id);
        if (index !== -1) state.periods[index] = action.payload;
        message.success('Payroll period locked');
      })
      .addCase(lockPayrollPeriod.rejected, (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
        state.error = 'Failed to lock payroll period';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(unlockPayrollPeriod.fulfilled, (state: Draft<PayrollPeriodsState>, action) => {
        const index = state.periods.findIndex((p) => p.id === action.payload.id);
        if (index !== -1) state.periods[index] = action.payload;
        message.success('Payroll period unlocked');
      })
      .addCase(unlockPayrollPeriod.rejected, (state: Draft<PayrollPeriodsState>, action: RejectedActionPayload) => {
        state.error = 'Failed to unlock payroll period';
        message.error(Helpers.handleServerError(action.payload), 8);
      });
  },
});

export const { clearSelectedPeriod } = PayrollPeriodsSlice.actions;
export default PayrollPeriodsSlice;
export type { PayrollPeriodsState };