plane/web/hooks/use-calendar-issues-view.tsx

import { useContext, useState } from "react";

import { useRouter } from "next/router";

import useSWR from "swr";

// contexts
import { issueViewContext } from "contexts/issue-view.context";
// services
import issuesService from "services/issues.service";
import cyclesService from "services/cycles.service";
import modulesService from "services/modules.service";
// helpers
import { renderDateFormat } from "helpers/date-time.helper";
// types
import { IIssue } from "types";
// fetch-keys
import {
  CYCLE_ISSUES_WITH_PARAMS,
  MODULE_ISSUES_WITH_PARAMS,
  PROJECT_ISSUES_LIST_WITH_PARAMS,
  VIEW_ISSUES,
} from "constants/fetch-keys";

const useCalendarIssuesView = () => {
  const {
    display_filters: displayFilters,
    filters,
    setFilters,
    resetFilterToDefault,
    setNewFilterDefaultView,
  } = useContext(issueViewContext);

  const [activeMonthDate, setActiveMonthDate] = useState(new Date());

  // previous month's first date
  const previousMonthYear =
    activeMonthDate.getMonth() === 0
      ? activeMonthDate.getFullYear() - 1
      : activeMonthDate.getFullYear();
  const previousMonthMonth = activeMonthDate.getMonth() === 0 ? 11 : activeMonthDate.getMonth() - 1;

  const previousMonthFirstDate = new Date(previousMonthYear, previousMonthMonth, 1);

  // next month's last date
  const nextMonthYear =
    activeMonthDate.getMonth() === 11
      ? activeMonthDate.getFullYear() + 1
      : activeMonthDate.getFullYear();
  const nextMonthMonth = (activeMonthDate.getMonth() + 1) % 12;
  const nextMonthFirstDate = new Date(nextMonthYear, nextMonthMonth, 1);

  const nextMonthLastDate = new Date(
    nextMonthFirstDate.getFullYear(),
    nextMonthFirstDate.getMonth() + 1,
    0
  );

  const router = useRouter();
  const { workspaceSlug, projectId, cycleId, moduleId, viewId } = router.query;

  const params: any = {
    assignees: filters?.assignees ? filters?.assignees.join(",") : undefined,
    state: filters?.state ? filters?.state.join(",") : undefined,
    priority: filters?.priority ? filters?.priority.join(",") : undefined,
    type: displayFilters?.type ? displayFilters?.type : undefined,
    labels: filters?.labels ? filters?.labels.join(",") : undefined,
    created_by: filters?.created_by ? filters?.created_by.join(",") : undefined,
    start_date: filters?.start_date ? filters?.start_date.join(",") : undefined,
    target_date: `${renderDateFormat(previousMonthFirstDate)};after,${renderDateFormat(
      nextMonthLastDate
    )};before`,
  };

  const { data: projectCalendarIssues, mutate: mutateProjectCalendarIssues } = useSWR(
    workspaceSlug && projectId
      ? PROJECT_ISSUES_LIST_WITH_PARAMS(projectId.toString(), params)
      : null,
    workspaceSlug && projectId
      ? () =>
          issuesService.getIssuesWithParams(workspaceSlug.toString(), projectId.toString(), params)
      : null
  );

  const { data: cycleCalendarIssues, mutate: mutateCycleCalendarIssues } = useSWR(
    workspaceSlug && projectId && cycleId
      ? CYCLE_ISSUES_WITH_PARAMS(cycleId.toString(), params)
      : null,
    workspaceSlug && projectId && cycleId
      ? () =>
          cyclesService.getCycleIssuesWithParams(
            workspaceSlug.toString(),
            projectId.toString(),
            cycleId.toString(),
            params
          )
      : null
  );

  const { data: moduleCalendarIssues, mutate: mutateModuleCalendarIssues } = useSWR(
    workspaceSlug && projectId && moduleId
      ? MODULE_ISSUES_WITH_PARAMS(moduleId.toString(), params)
      : null,
    workspaceSlug && projectId && moduleId
      ? () =>
          modulesService.getModuleIssuesWithParams(
            workspaceSlug.toString(),
            projectId.toString(),
            moduleId.toString(),
            params
          )
      : null
  );

  const { data: viewCalendarIssues, mutate: mutateViewCalendarIssues } = useSWR(
    workspaceSlug && projectId && viewId && params ? VIEW_ISSUES(viewId.toString(), params) : null,
    workspaceSlug && projectId && viewId && params
      ? () =>
          issuesService.getIssuesWithParams(workspaceSlug.toString(), projectId.toString(), params)
      : null
  );

  const calendarIssues = cycleId
    ? (cycleCalendarIssues as IIssue[])
    : moduleId
    ? (moduleCalendarIssues as IIssue[])
    : viewId
    ? (viewCalendarIssues as IIssue[])
    : (projectCalendarIssues as IIssue[]);

  return {
    activeMonthDate,
    setActiveMonthDate,
    calendarIssues: calendarIssues ?? [],
    mutateIssues: cycleId
      ? mutateCycleCalendarIssues
      : moduleId
      ? mutateModuleCalendarIssues
      : viewId
      ? mutateViewCalendarIssues
      : mutateProjectCalendarIssues,
    filters,
    setFilters,
    params,
    resetFilterToDefault,
    setNewFilterDefaultView,
  } as const;
};

export default useCalendarIssuesView;