import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  Certification,
  CreateCertificationPayload,
  UpdateCertificationPayload,
  DeleteCertificationPayload,
} from '@/interface/Training';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { CertificationsActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface CertificationState {
  certifications: Certification[];
  expiringCertifications: any[];
  complianceReport: any | null;
  loading: boolean;
  error: string | null;
}

const initialState: CertificationState = {
  certifications: [],
  expiringCertifications: [],
  complianceReport: null,
  loading: false,
  error: null,
};

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

export const createCertification = createAsyncThunk(
  CertificationsActions.CREATE_CERTIFICATION_REQUEST,
  async (payload: CreateCertificationPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/training/certifications',
        payload,
      )) as AxiosResponse<{ data: Certification }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateCertification = createAsyncThunk(
  CertificationsActions.UPDATE_CERTIFICATION_REQUEST,
  async (payload: UpdateCertificationPayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/training/certifications/${id}`,
        data,
      )) as AxiosResponse<{ data: Certification }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteCertification = createAsyncThunk(
  CertificationsActions.DELETE_CERTIFICATION_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(
      `/training/certifications/${id}`,
    )) as AxiosResponse<DeleteCertificationPayload>;
    const responseData = response.data;
    responseData.id = id;
    return responseData;
  },
);

export const fetchExpiringCertifications = createAsyncThunk(
  CertificationsActions.FETCH_EXPIRING_REQUEST,
  async (params?: { days?: number }) => {
    const queryParams = new URLSearchParams();
    if (params?.days) queryParams.append('days', String(params.days));
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/training/certifications/expiring${query}`)) as AxiosResponse;
    return response.data;
  },
);

export const fetchComplianceReport = createAsyncThunk(
  CertificationsActions.FETCH_COMPLIANCE_REPORT_REQUEST,
  async () => {
    const response = (await apiGet(
      '/training/certifications/compliance-report',
    )) as AxiosResponse;
    return response.data;
  },
);

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

    builder
      .addCase(createCertification.pending, (state: Draft<CertificationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createCertification.fulfilled,
        (
          state: Draft<CertificationState>,
          action: PayloadAction<Certification>,
        ) => {
          state.certifications.unshift(action.payload);
          state.error = null;
          state.loading = false;
          message.success('Certification created successfully');
        },
      )
      .addCase(
        createCertification.rejected,
        (state: Draft<CertificationState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateCertification.pending, (state: Draft<CertificationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateCertification.fulfilled,
        (
          state: Draft<CertificationState>,
          action: PayloadAction<Certification>,
        ) => {
          const index = state.certifications.findIndex((c) => c.id === action.payload.id);
          if (index !== -1) {
            state.certifications[index] = action.payload;
          }
          state.loading = false;
          state.error = null;
          message.success('Certification updated successfully');
        },
      )
      .addCase(
        updateCertification.rejected,
        (state: Draft<CertificationState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteCertification.pending, (state: Draft<CertificationState>) => {
        state.loading = true;
      })
      .addCase(
        deleteCertification.fulfilled,
        (
          state: Draft<CertificationState>,
          action: PayloadAction<DeleteCertificationPayload>,
        ) => {
          state.certifications = state.certifications.filter(
            (c) => c.id !== action.payload.id,
          );
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(deleteCertification.rejected, (state: Draft<CertificationState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to delete certification. Please try again or contact support';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(
        fetchExpiringCertifications.pending,
        (state: Draft<CertificationState>) => {
          state.loading = true;
          state.error = null;
        },
      )
      .addCase(
        fetchExpiringCertifications.fulfilled,
        (state: Draft<CertificationState>, action: PayloadAction<any[]>) => {
          state.expiringCertifications = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(
        fetchExpiringCertifications.rejected,
        (state: Draft<CertificationState>) => {
          state.loading = false;
          state.error =
            'Failed to load expiring certifications. Please try again or contact support';
          message.error(
            'Failed to load expiring certifications. Please try again or contact support',
          );
        },
      );

    builder
      .addCase(fetchComplianceReport.pending, (state: Draft<CertificationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchComplianceReport.fulfilled,
        (state: Draft<CertificationState>, action: PayloadAction<any>) => {
          state.complianceReport = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchComplianceReport.rejected, (state: Draft<CertificationState>) => {
        state.loading = false;
        state.error = 'Failed to load compliance report. Please try again or contact support';
        message.error(
          'Failed to load compliance report. Please try again or contact support',
        );
      });
  },
});

export default CertificationsSlice;
export type { CertificationState };