import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  TrainingEvaluation,
  CreateTrainingEvaluationPayload,
  UpdateTrainingEvaluationPayload,
  DeleteTrainingEvaluationPayload,
  SessionEvaluationReport,
} from '@/interface/Training';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { TrainingEvaluationsActions } from './actionTypes';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface TrainingEvaluationState {
  evaluations: TrainingEvaluation[];
  pendingEvaluations: TrainingEvaluation[];
  sessionReport: SessionEvaluationReport | null;
  programSummary: Record<string, unknown> | null;
  loading: boolean;
  error: string | null;
}

const initialState: TrainingEvaluationState = {
  evaluations: [],
  pendingEvaluations: [],
  sessionReport: null,
  programSummary: null,
  loading: false,
  error: null,
};

export const fetchTrainingEvaluations = createAsyncThunk(
  TrainingEvaluationsActions.FETCH_REQUEST,
  async (params?: { session_id?: string; employee_id?: string }) => {
    const queryParams = new URLSearchParams();
    if (params?.session_id) queryParams.append('session_id', params.session_id);
    if (params?.employee_id) queryParams.append('employee_id', params.employee_id);
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/training/training-evaluations${query}`)) as AxiosResponse<
      TrainingEvaluation[]
    >;
    return response.data;
  },
);

export const fetchPendingEvaluations = createAsyncThunk(
  TrainingEvaluationsActions.FETCH_PENDING_REQUEST,
  async () => {
    const response = (await apiGet(
      '/training/training-evaluations/pending',
    )) as AxiosResponse<TrainingEvaluation[]>;
    return response.data;
  },
);

export const fetchSessionEvaluationReport = createAsyncThunk(
  TrainingEvaluationsActions.FETCH_SESSION_REPORT_REQUEST,
  async (sessionId: string) => {
    const response = (await apiGet(
      `/training/training-sessions/${sessionId}/evaluation-report`,
    )) as AxiosResponse<SessionEvaluationReport>;
    return response.data;
  },
);

export const fetchProgramEvaluationSummary = createAsyncThunk(
  TrainingEvaluationsActions.FETCH_PROGRAM_SUMMARY_REQUEST,
  async (programId: string) => {
    const response = (await apiGet(
      `/training/training-programs/${programId}/evaluation-summary`,
    )) as AxiosResponse<Record<string, unknown>>;
    return response.data;
  },
);

export const createTrainingEvaluation = createAsyncThunk(
  TrainingEvaluationsActions.CREATE_REQUEST,
  async (payload: CreateTrainingEvaluationPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost(
        '/training/training-evaluations',
        payload,
      )) as AxiosResponse<{ data: TrainingEvaluation }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateTrainingEvaluation = createAsyncThunk(
  TrainingEvaluationsActions.UPDATE_REQUEST,
  async (payload: UpdateTrainingEvaluationPayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(
        `/training/training-evaluations/${id}`,
        data,
      )) as AxiosResponse<{ data: TrainingEvaluation }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteTrainingEvaluation = createAsyncThunk(
  TrainingEvaluationsActions.DELETE_REQUEST,
  async (id: string, { rejectWithValue }) => {
    try {
      const response = (await apiDelete(
        `/training/training-evaluations/${id}`,
      )) as AxiosResponse<DeleteTrainingEvaluationPayload>;
      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 TrainingEvaluationsSlice = createSlice({
  name: 'trainingEvaluations',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchTrainingEvaluations.pending, (state: Draft<TrainingEvaluationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchTrainingEvaluations.fulfilled,
        (state: Draft<TrainingEvaluationState>, action: PayloadAction<TrainingEvaluation[]>) => {
          state.evaluations = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchTrainingEvaluations.rejected, (state: Draft<TrainingEvaluationState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load training evaluations';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchPendingEvaluations.pending, (state: Draft<TrainingEvaluationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchPendingEvaluations.fulfilled,
        (state: Draft<TrainingEvaluationState>, action: PayloadAction<TrainingEvaluation[]>) => {
          state.pendingEvaluations = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(fetchPendingEvaluations.rejected, (state: Draft<TrainingEvaluationState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load pending evaluations';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(
        fetchSessionEvaluationReport.pending,
        (state: Draft<TrainingEvaluationState>) => {
          state.loading = true;
          state.error = null;
        },
      )
      .addCase(
        fetchSessionEvaluationReport.fulfilled,
        (
          state: Draft<TrainingEvaluationState>,
          action: PayloadAction<SessionEvaluationReport>,
        ) => {
          state.sessionReport = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(
        fetchSessionEvaluationReport.rejected,
        (state: Draft<TrainingEvaluationState>) => {
          state.loading = false;
          state.error = 'Failed to load session evaluation report';
          message.error(Helpers.handleServerError(action.payload), 8);
        },
      );

    builder
      .addCase(
        fetchProgramEvaluationSummary.pending,
        (state: Draft<TrainingEvaluationState>) => {
          state.loading = true;
          state.error = null;
        },
      )
      .addCase(
        fetchProgramEvaluationSummary.fulfilled,
        (
          state: Draft<TrainingEvaluationState>,
          action: PayloadAction<Record<string, unknown>>,
        ) => {
          state.programSummary = action.payload;
          state.error = null;
          state.loading = false;
        },
      )
      .addCase(
        fetchProgramEvaluationSummary.rejected,
        (state: Draft<TrainingEvaluationState>) => {
          state.loading = false;
          state.error = 'Failed to load program evaluation summary';
          message.error(Helpers.handleServerError(action.payload), 8);
        },
      );

    builder
      .addCase(createTrainingEvaluation.pending, (state: Draft<TrainingEvaluationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createTrainingEvaluation.fulfilled,
        (state: Draft<TrainingEvaluationState>, action: PayloadAction<TrainingEvaluation>) => {
          state.evaluations.unshift(action.payload);
          state.error = null;
          state.loading = false;
          message.success('Training evaluation submitted successfully');
        },
      )
      .addCase(
        createTrainingEvaluation.rejected,
        (state: Draft<TrainingEvaluationState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(updateTrainingEvaluation.pending, (state: Draft<TrainingEvaluationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        updateTrainingEvaluation.fulfilled,
        (state: Draft<TrainingEvaluationState>, action: PayloadAction<TrainingEvaluation>) => {
          const index = state.evaluations.findIndex((e) => e.id === action.payload.id);
          if (index !== -1) {
            state.evaluations[index] = action.payload;
          }
          state.loading = false;
          state.error = null;
          message.success('Training evaluation updated successfully');
        },
      )
      .addCase(
        updateTrainingEvaluation.rejected,
        (state: Draft<TrainingEvaluationState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteTrainingEvaluation.pending, (state: Draft<TrainingEvaluationState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        deleteTrainingEvaluation.fulfilled,
        (
          state: Draft<TrainingEvaluationState>,
          action: PayloadAction<DeleteTrainingEvaluationPayload>,
        ) => {
          state.evaluations = state.evaluations.filter((e) => e.id !== action.payload.id);
          state.loading = false;
          message.success(action.payload.message);
        },
      )
      .addCase(
        deleteTrainingEvaluation.rejected,
        (state: Draft<TrainingEvaluationState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );
  },
});

export default TrainingEvaluationsSlice;
export type { TrainingEvaluationState };