import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  TrainingProvider,
  CreateTrainingProviderPayload,
  UpdateTrainingProviderPayload,
  DeleteTrainingProviderPayload,
} from '@/interface/Training';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { TrainingProvidersActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface TrainingProviderState {
  providers: TrainingProvider[];
  loading: boolean;
  error: string | null;
}

const initialState: TrainingProviderState = {
  providers: [],
  loading: false,
  error: null,
};

export const fetchTrainingProviders = createAsyncThunk(
  TrainingProvidersActions.FETCH_TRAINING_PROVIDERS_REQUEST,
  async (params?: { active?: boolean; search?: string }) => {
    const queryParams = new URLSearchParams();
    if (params?.active !== undefined) queryParams.append('active', String(params.active));
    if (params?.search) queryParams.append('search', params.search);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/training/providers${query}`)) as AxiosResponse<
      TrainingProvider[]
    >;
    return response.data;
  },
);

export const createTrainingProvider = createAsyncThunk(
  TrainingProvidersActions.CREATE_TRAINING_PROVIDER_REQUEST,
  async (payload: CreateTrainingProviderPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/training/providers',
        payload,
      )) as AxiosResponse<{ data: TrainingProvider }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateTrainingProvider = createAsyncThunk(
  TrainingProvidersActions.UPDATE_TRAINING_PROVIDER_REQUEST,
  async (payload: UpdateTrainingProviderPayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/training/providers/${id}`,
        data,
      )) as AxiosResponse<{ data: TrainingProvider }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteTrainingProvider = createAsyncThunk(
  TrainingProvidersActions.DELETE_TRAINING_PROVIDER_REQUEST,
  async (id: string, { rejectWithValue }) => {
    try {
      const response = (await apiDelete(
        `/training/providers/${id}`,
      )) as AxiosResponse<DeleteTrainingProviderPayload>;
      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 TrainingProvidersSlice = createSlice({
  name: 'trainingProviders',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchTrainingProviders.pending, (state: Draft<TrainingProviderState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchTrainingProviders.fulfilled,
        (state: Draft<TrainingProviderState>, action: PayloadAction<TrainingProvider[]>) => {
          state.providers = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchTrainingProviders.rejected, (state: Draft<TrainingProviderState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load training providers';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createTrainingProvider.pending, (state: Draft<TrainingProviderState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createTrainingProvider.fulfilled,
        (state: Draft<TrainingProviderState>, action: PayloadAction<TrainingProvider>) => {
          state.providers.unshift(action.payload);
          state.error = null;
          state.loading = false;
          message.success('Training provider created successfully');
        },
      )
      .addCase(
        createTrainingProvider.rejected,
        (state: Draft<TrainingProviderState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateTrainingProvider.pending, (state: Draft<TrainingProviderState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateTrainingProvider.fulfilled,
        (state: Draft<TrainingProviderState>, action: PayloadAction<TrainingProvider>) => {
          const index = state.providers.findIndex((p) => p.id === action.payload.id);
          if (index !== -1) {
            state.providers[index] = action.payload;
          }
          state.loading = false;
          state.error = null;
          message.success('Training provider updated successfully');
        },
      )
      .addCase(
        updateTrainingProvider.rejected,
        (state: Draft<TrainingProviderState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteTrainingProvider.pending, (state: Draft<TrainingProviderState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        deleteTrainingProvider.fulfilled,
        (
          state: Draft<TrainingProviderState>,
          action: PayloadAction<DeleteTrainingProviderPayload>,
        ) => {
          state.providers = state.providers.filter((p) => p.id !== action.payload.id);
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(
        deleteTrainingProvider.rejected,
        (state: Draft<TrainingProviderState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );
  },
});

export default TrainingProvidersSlice;
export type { TrainingProviderState };