import { Employee } from '@/interface/Employee';
import { RejectedActionPayload, ServerErrorResponse } from '@/interface/Shared';
import { apiGet } 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 { OrgChartActions } from './actionTypes';

interface OrgChartState {
  employees: Employee[];
  loading: boolean;
  error: string | null;
}

const initialState: OrgChartState = {
  employees: [],
  loading: false,
  error: null,
};

export const fetchOrgChart = createAsyncThunk(
  OrgChartActions.FETCH_ORG_CHART,
  async (_, { rejectWithValue }) => {
    try {
      const response = (await apiGet('/hr/org-chart')) as AxiosResponse<Employee[]>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

export const fetchOrgChartSubtree = createAsyncThunk(
  OrgChartActions.FETCH_ORG_CHART_SUBTREE,
  async (employeeId: string, { rejectWithValue }) => {
    try {
      const response = (await apiGet(`/hr/org-chart/${employeeId}`)) as AxiosResponse<Employee[]>;
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error) && error.response) {
        return rejectWithValue(error.response.data as ServerErrorResponse);
      }
      throw error;
    }
  },
);

const OrgChartSlice = createSlice({
  name: 'orgChart',
  initialState,
  reducers: {},
  extraReducers(builder) {
    builder
      .addCase(fetchOrgChart.pending, (state: Draft<OrgChartState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(fetchOrgChart.fulfilled, (state: Draft<OrgChartState>, action: PayloadAction<Employee[]>) => {
        state.loading = false;
        state.error = null;
        state.employees = action.payload;
      })
      .addCase(fetchOrgChart.rejected, (state: Draft<OrgChartState>, action: RejectedActionPayload) => {
        state.loading = false;
        state.error = Helpers.handleServerError(action.payload);
        message.error(state.error, 10);
      });

    builder
      .addCase(fetchOrgChartSubtree.pending, (state: Draft<OrgChartState>) => {
        state.loading = true;
        state.error = null;
      })
      .addCase(
        fetchOrgChartSubtree.fulfilled,
        (state: Draft<OrgChartState>, action: PayloadAction<Employee[]>) => {
          state.loading = false;
          state.error = null;
          state.employees = action.payload;
        },
      )
      .addCase(
        fetchOrgChartSubtree.rejected,
        (state: Draft<OrgChartState>, action: RejectedActionPayload) => {
          state.loading = false;
          state.error = Helpers.handleServerError(action.payload);
          message.error(state.error, 10);
        },
      );
  },
});

export default OrgChartSlice;
export type { OrgChartState };