import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  QuizAssignment,
  QuizAssignmentDetail,
  CreateAssignmentPayload,
  PaginationMeta,
  Assignee,
} from '@/interface/Assessment';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import { AssignmentsActions } from './actionTypes';
import { apiDelete, apiGet, apiPost } from '@/services/api/api';
import axios, { AxiosResponse } from 'axios';

interface AssignmentsState {
  assignments: QuizAssignment[];
  selectedAssignment: QuizAssignmentDetail | null;
  assignees: Assignee[];
  meta: PaginationMeta | null;
  loading: boolean;
  error: string | null;
}

const initialState: AssignmentsState = {
  assignments: [],
  selectedAssignment: null,
  assignees: [],
  meta: null,
  loading: false,
  error: null,
};

export const fetchAssignments = createAsyncThunk(
  AssignmentsActions.FETCH_ASSIGNMENTS_REQUEST,
  async (params?: { quiz_id?: string; page?: number }) => {
    const queryParams = new URLSearchParams();
    if (params?.quiz_id) queryParams.append('quiz_id', params.quiz_id);
    if (params?.page) queryParams.append('page', params.page.toString());
    const query = queryParams.toString() ? `?${queryParams.toString()}` : '';
    const response = (await apiGet(`/assessment/assignments${query}`)) as AxiosResponse<{
      data: QuizAssignment[];
      meta: PaginationMeta;
    }>;
    return response.data;
  },
);

export const fetchAssignment = createAsyncThunk(
  AssignmentsActions.FETCH_ASSIGNMENT_REQUEST,
  async (id: string) => {
    const response = (await apiGet(
      `/assessment/assignments/${id}`,
    )) as AxiosResponse<QuizAssignmentDetail>;
    return response.data;
  },
);

export const createAssignment = createAsyncThunk(
  AssignmentsActions.CREATE_ASSIGNMENT_REQUEST,
  async (payload: CreateAssignmentPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/assessment/assignments', payload)) as AxiosResponse<{
        data: QuizAssignment;
      }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteAssignment = createAsyncThunk(
  AssignmentsActions.DELETE_ASSIGNMENT_REQUEST,
  async (id: string, { rejectWithValue }) => {
    try {
      await apiDelete(`/assessment/assignments/${id}`);
      return { id };
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const fetchAssignees = createAsyncThunk(
  AssignmentsActions.FETCH_ASSIGNEES_REQUEST,
  async (assignmentId: string) => {
    const response = (await apiGet(
      `/assessment/assignments/${assignmentId}/assignees`,
    )) as AxiosResponse<{ data: Assignee[] }>;
    return response.data.data;
  },
);

export const AssignmentsSlice = createSlice({
  name: 'assessmentAssignments',
  initialState,
  reducers: {
    clearSelectedAssignment: (state: Draft<AssignmentsState>) => {
      state.selectedAssignment = null;
      state.assignees = [];
    },
  },
  extraReducers(builder) {
    builder
      .addCase(fetchAssignments.pending, (state: Draft<AssignmentsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchAssignments.fulfilled,
        (
          state: Draft<AssignmentsState>,
          action: PayloadAction<{ data: QuizAssignment[]; meta: PaginationMeta }>,
        ) => {
          state.assignments = action.payload.data;
          state.meta = action.payload.meta;
          state.loading = false;
        },
      )
      .addCase(fetchAssignments.rejected, (state: Draft<AssignmentsState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = 'Failed to load assignments';
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchAssignment.pending, (state: Draft<AssignmentsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchAssignment.fulfilled,
        (state: Draft<AssignmentsState>, action: PayloadAction<QuizAssignmentDetail>) => {
          state.selectedAssignment = action.payload;
          state.loading = false;
        },
      )
      .addCase(fetchAssignment.rejected, (state: Draft<AssignmentsState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createAssignment.pending, (state: Draft<AssignmentsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        createAssignment.fulfilled,
        (state: Draft<AssignmentsState>, action: PayloadAction<QuizAssignment>) => {
          state.assignments.unshift(action.payload);
          state.loading = false;
          message.success('Quiz assigned successfully');
        },
      )
      .addCase(
        createAssignment.rejected,
        (state: Draft<AssignmentsState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(deleteAssignment.pending, (state: Draft<AssignmentsState>) => {
        state.loading = true;
      })
      .addCase(
        deleteAssignment.fulfilled,
        (state: Draft<AssignmentsState>, action: PayloadAction<{ id: string }>) => {
          state.assignments = state.assignments.filter((a) => a.id !== action.payload.id);
          state.loading = false;
          message.success('Assignment deleted');
        },
      )
      .addCase(
        deleteAssignment.rejected,
        (state: Draft<AssignmentsState>, action: RejectedActionPayload) => {
          state.loading = false;
          message.error(Helpers.handleServerError(action.payload), 10);
        },
      );

    builder
      .addCase(fetchAssignees.pending, (state: Draft<AssignmentsState>) => {
        state.loading = true;
      })
      .addCase(
        fetchAssignees.fulfilled,
        (state: Draft<AssignmentsState>, action: PayloadAction<Assignee[]>) => {
          state.assignees = action.payload;
          state.loading = false;
        },
      )
      .addCase(fetchAssignees.rejected, (state: Draft<AssignmentsState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });
  },
});

export const { clearSelectedAssignment } = AssignmentsSlice.actions;
export default AssignmentsSlice;
export type { AssignmentsState };