import {
  AssignWorkflowPayload,
  DeleteLeaveWorkflowResponse,
  LeaveWorkflow,
  UpdateLeaveWorkflowPayload,
  WorkflowActionResponse,
  WorkflowAssignment,
  WorkflowAssignmentActionResponse,
  CreateLeaveWorkflowPayload,
} from '@/interface/LeaveWorkflow';
import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
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 { LeaveWorkflowsActions } from './actionTypes';

interface LeaveWorkflowsState {
  workflows: LeaveWorkflow[];
  assignments: WorkflowAssignment[];
  loading: boolean;
  actionLoading: boolean;
  error: string | null;
}

const initialState: LeaveWorkflowsState = {
  workflows: [],
  assignments: [],
  loading: false,
  actionLoading: false,
  error: null,
};

const replaceWorkflow = (workflows: LeaveWorkflow[], updated: LeaveWorkflow): LeaveWorkflow[] => {
  const index = workflows.findIndex((item) => item.id === updated.id);

  if (index === -1) {
    return workflows;
  }

  const next = [...workflows];
  next[index] = updated;
  return next;
};

const replaceAssignment = (
  assignments: WorkflowAssignment[],
  updated: WorkflowAssignment,
): WorkflowAssignment[] => {
  const index = assignments.findIndex((item) => item.id === updated.id);

  if (index === -1) {
    return [updated, ...assignments];
  }

  const next = [...assignments];
  next[index] = updated;
  return next;
};

export const fetchLeaveWorkflows = createAsyncThunk(
  LeaveWorkflowsActions.FETCH_LEAVE_WORKFLOWS_REQUEST,
  async () => {
    const response = (await apiGet('/leave/workflows')) as AxiosResponse<LeaveWorkflow[] | { data: LeaveWorkflow[] }>;
    return Array.isArray(response.data) ? response.data : response.data.data;
  },
);

export const createLeaveWorkflow = createAsyncThunk(
  LeaveWorkflowsActions.CREATE_LEAVE_WORKFLOW_REQUEST,
  async (payload: CreateLeaveWorkflowPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/leave/workflows', payload)) as AxiosResponse<WorkflowActionResponse>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const updateLeaveWorkflow = createAsyncThunk(
  LeaveWorkflowsActions.UPDATE_LEAVE_WORKFLOW_REQUEST,
  async (payload: UpdateLeaveWorkflowPayload, { rejectWithValue }) => {
    try {
      const { id, ...data } = payload;
      const response = (await apiPut(`/leave/workflows/${id}`, data)) as AxiosResponse<WorkflowActionResponse>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const deleteLeaveWorkflow = createAsyncThunk(
  LeaveWorkflowsActions.DELETE_LEAVE_WORKFLOW_REQUEST,
  async (id: string) => {
    const response = (await apiDelete(`/leave/workflows/${id}`)) as AxiosResponse<DeleteLeaveWorkflowResponse>;
    const data = response.data;
    data.id = id;
    return data;
  },
);

export const fetchWorkflowAssignments = createAsyncThunk(
  LeaveWorkflowsActions.FETCH_WORKFLOW_ASSIGNMENTS_REQUEST,
  async () => {
    const response = (await apiGet('/leave/workflows-assignments')) as AxiosResponse<WorkflowAssignment[]>;
    return response.data;
  },
);

export const assignWorkflow = createAsyncThunk(
  LeaveWorkflowsActions.ASSIGN_WORKFLOW_REQUEST,
  async (payload: AssignWorkflowPayload, { rejectWithValue }) => {
    try {
      const response = (await apiPost('/leave/workflows-assignments', payload)) as AxiosResponse<WorkflowAssignmentActionResponse>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const LeaveWorkflowsSlice = createSlice({
  name: 'leaveWorkflows',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchLeaveWorkflows.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchLeaveWorkflows.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<LeaveWorkflow[]>) => {
          state.workflows = action.payload;
          state.loading = false;
          state.error = null;
        },
      )
      .addCase(fetchLeaveWorkflows.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(createLeaveWorkflow.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.actionLoading = true;
        state.error = null;
      })
      .addCase(
        createLeaveWorkflow.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<WorkflowActionResponse>) => {
          state.actionLoading = false;
          state.error = null;
          state.workflows.unshift(action.payload.data);
          message.success(action.payload.message || 'Leave workflow created successfully.');
        },
      )
      .addCase(createLeaveWorkflow.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.actionLoading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });

    builder
      .addCase(updateLeaveWorkflow.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.actionLoading = true;
        state.error = null;
      })
      .addCase(
        updateLeaveWorkflow.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<WorkflowActionResponse>) => {
          state.actionLoading = false;
          state.error = null;
          state.workflows = replaceWorkflow(state.workflows, action.payload.data);
          message.success(action.payload.message || 'Leave workflow updated successfully.');
        },
      )
      .addCase(updateLeaveWorkflow.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.actionLoading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });

    builder
      .addCase(deleteLeaveWorkflow.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.actionLoading = true;
        state.error = null;
      })
      .addCase(
        deleteLeaveWorkflow.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<DeleteLeaveWorkflowResponse>) => {
          state.actionLoading = false;
          state.error = null;
          state.workflows = state.workflows.filter((item) => item.id !== action.payload.id);
          message.success(action.payload.message || 'Leave workflow deleted successfully.');
        },
      )
      .addCase(deleteLeaveWorkflow.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.actionLoading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(fetchWorkflowAssignments.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchWorkflowAssignments.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<WorkflowAssignment[]>) => {
          state.assignments = action.payload;
          state.loading = false;
          state.error = null;
        },
      )
      .addCase(fetchWorkflowAssignments.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.loading = false;
        message.error(Helpers.handleServerError(action.payload), 8);
      });

    builder
      .addCase(assignWorkflow.pending, (state: Draft<LeaveWorkflowsState>) => {
        state.actionLoading = true;
        state.error = null;
      })
      .addCase(
        assignWorkflow.fulfilled,
        (state: Draft<LeaveWorkflowsState>, action: PayloadAction<WorkflowAssignmentActionResponse>) => {
          state.actionLoading = false;
          state.error = null;
          state.assignments = replaceAssignment(state.assignments, action.payload.data);
          message.success(action.payload.message || 'Workflow assignment saved successfully.');
        },
      )
      .addCase(assignWorkflow.rejected, (state: Draft<LeaveWorkflowsState>, action: RejectedActionPayload) => {
        state.actionLoading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });
  },
});

export default LeaveWorkflowsSlice;
export type { LeaveWorkflowsState };