import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import {
  CheckIn,
  CheckInResponse,
  PaginatedCheckIns,
  CreateCheckInPayload,
  UpdateCheckInPayload,
  CompleteCheckInPayload,
  RescheduleCheckInPayload,
  CheckInActionItem,
} from '@/interface/CheckIn';
import { apiDelete, apiGet, apiPost, apiPut } from '@/services/api/api';
import Helpers from '@/utilities/Helpers';
import { createAsyncThunk, createSlice, Draft, PayloadAction } from '@reduxjs/toolkit';
import { message } from 'antd';
import axios, { AxiosResponse } from 'axios';
import { CheckInActions } from './actionTypes';
import { PaginationLinks, PaginationMeta } from '@/interface/Paginated';
import { GetParams, toQueryString } from '@/interface/GetParams';

interface CheckInState {
  checkIns: CheckIn[];
  myCheckIns: CheckIn[];
  checkInHistory: CheckIn[];
  overdueActions: CheckInActionItem[];
  currentCheckIn: CheckIn | null;
  links?: PaginationLinks;
  meta?: PaginationMeta;
  loading: boolean;
  error: string | null;
}

const initialState: CheckInState = {
  checkIns: [],
  myCheckIns: [],
  checkInHistory: [],
  overdueActions: [],
  currentCheckIn: null,
  loading: false,
  error: null,
};

export const fetchCheckIns = createAsyncThunk(
  CheckInActions.FETCH_CHECK_INS_REQUEST,
  async (params?: GetParams) => {
    const query = toQueryString(params);
    const response = (await apiGet(`/check-ins${query}`)) as AxiosResponse<PaginatedCheckIns>;
    return response.data;
  },
);

export const fetchCheckIn = createAsyncThunk(
  CheckInActions.FETCH_CHECK_IN_REQUEST,
  async (id: string) => {
    const response = (await apiGet(`/check-ins/${id}`)) as AxiosResponse<CheckInResponse>;
    return response.data.data;
  },
);

export const fetchMyCheckIns = createAsyncThunk(
  CheckInActions.FETCH_MY_CHECK_INS_REQUEST,
  async () => {
    const response = (await apiGet('/check-ins/my-check-ins')) as AxiosResponse<{ data: CheckIn[] }>;
    return response.data.data;
  },
);

export const fetchCheckInHistory = createAsyncThunk(
  CheckInActions.FETCH_CHECK_IN_HISTORY_REQUEST,
  async (employeeId: string) => {
    const response = (await apiGet(`/check-ins/history?employee_id=${employeeId}`)) as AxiosResponse<{ data: CheckIn[] }>;
    return response.data.data;
  },
);

export const fetchOverdueActions = createAsyncThunk(
  CheckInActions.FETCH_OVERDUE_ACTIONS_REQUEST,
  async () => {
    const response = (await apiGet('/check-ins/overdue-actions')) as AxiosResponse<{ data: CheckInActionItem[] }>;
    return response.data.data;
  },
);

export const createCheckIn = createAsyncThunk(
  CheckInActions.CREATE_CHECK_IN_REQUEST,
  async (payload: CreateCheckInPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/check-ins', payload)) as AxiosResponse<CheckInResponse>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const createRecurringCheckIns = createAsyncThunk(
  CheckInActions.CREATE_RECURRING_CHECK_INS_REQUEST,
  async (payload: CreateCheckInPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/check-ins/recurring', payload)) as AxiosResponse<{ data: CheckIn[] }>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateCheckIn = createAsyncThunk(
  CheckInActions.UPDATE_CHECK_IN_REQUEST,
  async (payload: UpdateCheckInPayload & { id: string }, { rejectWithValue }) => {
    try {
      const response = (await apiPut(`/check-ins/${payload.id}`, payload)) as AxiosResponse<CheckInResponse>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteCheckIn = createAsyncThunk(
  CheckInActions.DELETE_CHECK_IN_REQUEST,
  async (id: string, { rejectWithValue }) => {
    try {
      const response = (await apiDelete(`/check-ins/${id}`)) as AxiosResponse<{ message: string }>;
      return { message: response.data.message, id };
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const completeCheckIn = createAsyncThunk(
  CheckInActions.COMPLETE_CHECK_IN_REQUEST,
  async ({ id, ...payload }: CompleteCheckInPayload & { id: string }, { rejectWithValue }) => {
    try {
      const response = (await apiPost(`/check-ins/${id}/complete`, payload)) as AxiosResponse<CheckInResponse>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const rescheduleCheckIn = createAsyncThunk(
  CheckInActions.RESCHEDULE_CHECK_IN_REQUEST,
  async ({ id, ...payload }: RescheduleCheckInPayload & { id: string }, { rejectWithValue }) => {
    try {
      const response = (await apiPost(`/check-ins/${id}/reschedule`, payload)) as AxiosResponse<CheckInResponse>;
      return response.data.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const CheckInSlice = createSlice({
  name: 'checkIns',
  initialState,
  reducers: {
    clearCurrentCheckIn: (state: Draft<CheckInState>) => {
      state.currentCheckIn = null;
    },
  },
  extraReducers(builder) {
    builder
      .addCase(fetchCheckIns.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(fetchCheckIns.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<PaginatedCheckIns>) => {
        state.checkIns = action.payload.data;
        state.meta = action.payload.meta;
        state.links = action.payload.links;
        state.loading = false;
      })
      .addCase(fetchCheckIns.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchCheckIn.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(fetchCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn>) => {
        state.currentCheckIn = action.payload;
        state.loading = false;
      })
      .addCase(fetchCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

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

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

    builder
      .addCase(fetchOverdueActions.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckInActionItem[]>) => {
        state.overdueActions = action.payload;
      });

    builder
      .addCase(createCheckIn.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(createCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn>) => {
        state.checkIns.unshift(action.payload);
        state.loading = false;
        message.success('Check-in scheduled successfully', 10);
      })
      .addCase(createCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 10);
      });

    builder
      .addCase(createRecurringCheckIns.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(createRecurringCheckIns.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn[]>) => {
        state.checkIns.unshift(...action.payload);
        state.loading = false;
        message.success(`${action.payload.length} recurring check-ins created successfully`, 10);
      })
      .addCase(createRecurringCheckIns.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 10);
      });

    builder
      .addCase(updateCheckIn.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(updateCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn>) => {
        state.currentCheckIn = action.payload;
        const idx = state.checkIns.findIndex((c) => c.id === action.payload.id);
        if (idx !== -1) state.checkIns[idx] = action.payload;
        state.loading = false;
        message.success('Check-in updated successfully', 10);
      })
      .addCase(updateCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 10);
      });

    builder
      .addCase(deleteCheckIn.pending, (state: Draft<CheckInState>) => {})
      .addCase(deleteCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<{ message: string; id: string }>) => {
        state.checkIns = state.checkIns.filter((c) => c.id !== action.payload.id);
        message.success(action.payload.message, 10);
      })
      .addCase(deleteCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(completeCheckIn.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(completeCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn>) => {
        state.currentCheckIn = action.payload;
        const idx = state.checkIns.findIndex((c) => c.id === action.payload.id);
        if (idx !== -1) state.checkIns[idx] = action.payload;
        state.loading = false;
        message.success('Check-in completed successfully', 10);
      })
      .addCase(completeCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 10);
      });

    builder
      .addCase(rescheduleCheckIn.pending, (state: Draft<CheckInState>) => {
        state.loading = true;
      })
      .addCase(rescheduleCheckIn.fulfilled, (state: Draft<CheckInState>, action: PayloadAction<CheckIn>) => {
        state.currentCheckIn = action.payload;
        const idx = state.checkIns.findIndex((c) => c.id === action.payload.id);
        if (idx !== -1) state.checkIns[idx] = action.payload;
        state.loading = false;
        message.success('Check-in rescheduled successfully', 10);
      })
      .addCase(rescheduleCheckIn.rejected, (state: Draft<CheckInState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 10);
      });
  },
});

export const { clearCurrentCheckIn } = CheckInSlice.actions;
export default CheckInSlice;
export type { CheckInState };
