plane/web/components/gantt-chart/contexts/index.tsx

import React, { createContext, useState } from "react";
// types
import { ChartContextData, ChartContextActionPayload, ChartContextReducer } from "../types";
// data
import { allViewsWithData, currentViewDataWithView } from "../data";

export const ChartContext = createContext<ChartContextReducer | undefined>(undefined);

const chartReducer = (state: ChartContextData, action: ChartContextActionPayload): ChartContextData => {
  switch (action.type) {
    case "CURRENT_VIEW":
      return { ...state, currentView: action.payload };
    case "CURRENT_VIEW_DATA":
      return { ...state, currentViewData: action.payload };
    case "RENDER_VIEW":
      return { ...state, currentViewData: action.payload };
    case "PARTIAL_UPDATE":
      return { ...state, ...action.payload };
    default:
      return state;
  }
};

const initialView = "month";

export const ChartContextProvider: React.FC<{ children: React.ReactNode }> = ({ children }) => {
  // states;
  const [state, dispatch] = useState<ChartContextData>({
    currentView: initialView,
    currentViewData: currentViewDataWithView(initialView),
    renderView: [],
    allViews: allViewsWithData,
    activeBlock: null,
  });
  const [scrollLeft, setScrollLeft] = useState(0);
  const [scrollTop, setScrollTop] = useState(0);

  const handleDispatch = (action: ChartContextActionPayload): ChartContextData => {
    const newState = chartReducer(state, action);
    dispatch(() => newState);
    return newState;
  };

  const updateScrollLeft = (scrollLeft: number) => setScrollLeft(scrollLeft);

  const updateScrollTop = (scrollTop: number) => setScrollTop(scrollTop);

  return (
    <ChartContext.Provider
      value={{
        ...state,
        scrollLeft,
        updateScrollLeft,
        scrollTop,
        updateScrollTop,
        dispatch: handleDispatch,
      }}
    >
      {children}
    </ChartContext.Provider>
  );
};