plane/web/components/views/modal.tsx

import React from "react";

import { useRouter } from "next/router";

import { mutate } from "swr";

// headless ui
import { Dialog, Transition } from "@headlessui/react";
// services
import viewsService from "services/views.service";
import workspaceService from "services/workspace.service";
// hooks
import useToast from "hooks/use-toast";
// components
import { ViewForm } from "components/views";
// types
import { ICurrentUserResponse, IView } from "types";
// fetch-keys
import { VIEWS_LIST, WORKSPACE_VIEWS_LIST } from "constants/fetch-keys";

type Props = {
  isOpen: boolean;
  viewType: "project" | "workspace";
  handleClose: () => void;
  data?: IView | null;
  preLoadedData?: Partial<IView> | null;
  user: ICurrentUserResponse | undefined;
};

export const CreateUpdateViewModal: React.FC<Props> = ({
  isOpen,
  viewType,
  handleClose,
  data,
  preLoadedData,
  user,
}) => {
  const router = useRouter();
  const { workspaceSlug, projectId } = router.query;

  const { setToastAlert } = useToast();

  const onClose = () => {
    handleClose();
  };

  const createView = async (payload: IView) => {
    payload = {
      ...payload,
      query_data: payload.query,
    };

    if (viewType === "project") {
      await viewsService
        .createView(workspaceSlug as string, projectId as string, payload, user)
        .then(() => {
          mutate(VIEWS_LIST(projectId as string));
          handleClose();

          setToastAlert({
            type: "success",
            title: "Success!",
            message: "View created successfully.",
          });
        })
        .catch(() => {
          setToastAlert({
            type: "error",
            title: "Error!",
            message: "View could not be created. Please try again.",
          });
        });
    } else {
      await workspaceService
        .createView(workspaceSlug as string, payload)
        .then(() => {
          mutate(WORKSPACE_VIEWS_LIST(workspaceSlug as string));
          handleClose();

          setToastAlert({
            type: "success",
            title: "Success!",
            message: "View created successfully.",
          });
        })
        .catch(() => {
          setToastAlert({
            type: "error",
            title: "Error!",
            message: "View could not be created. Please try again.",
          });
        });
    }
  };

  const updateView = async (payload: IView) => {
    const payloadData = {
      ...payload,
      query_data: payload.query,
    };
    if (viewType === "project") {
      await viewsService
        .updateView(workspaceSlug as string, projectId as string, data?.id ?? "", payloadData, user)
        .then((res) => {
          mutate<IView[]>(
            VIEWS_LIST(projectId as string),
            (prevData) =>
              prevData?.map((p) => {
                if (p.id === res.id) return { ...p, ...payloadData };

                return p;
              }),
            false
          );
          onClose();

          setToastAlert({
            type: "success",
            title: "Success!",
            message: "View updated successfully.",
          });
        })
        .catch(() => {
          setToastAlert({
            type: "error",
            title: "Error!",
            message: "View could not be updated. Please try again.",
          });
        });
    } else {
      await workspaceService
        .updateView(workspaceSlug as string, data?.id ?? "", payloadData)
        .then((res) => {
          mutate<IView[]>(
            WORKSPACE_VIEWS_LIST(workspaceSlug as string),
            (prevData) =>
              prevData?.map((p) => {
                if (p.id === res.id) return { ...p, ...payloadData };

                return p;
              }),
            false
          );
          onClose();

          setToastAlert({
            type: "success",
            title: "Success!",
            message: "View updated successfully.",
          });
        })
        .catch(() => {
          setToastAlert({
            type: "error",
            title: "Error!",
            message: "View could not be updated. Please try again.",
          });
        });
    }
  };

  const handleFormSubmit = async (formData: IView) => {
    if (viewType === "project") {
      if (!workspaceSlug || !projectId) return;

      if (!data) await createView(formData);
      else await updateView(formData);
    } else {
      if (!workspaceSlug) return;

      if (!data) await createView(formData);
      else await updateView(formData);
    }
  };

  return (
    <Transition.Root show={isOpen} as={React.Fragment}>
      <Dialog as="div" className="relative z-20" onClose={handleClose}>
        <Transition.Child
          as={React.Fragment}
          enter="ease-out duration-300"
          enterFrom="opacity-0"
          enterTo="opacity-100"
          leave="ease-in duration-200"
          leaveFrom="opacity-100"
          leaveTo="opacity-0"
        >
          <div className="fixed inset-0 bg-custom-backdrop bg-opacity-50 transition-opacity" />
        </Transition.Child>

        <div className="fixed inset-0 z-20 overflow-y-auto">
          <div className="flex min-h-full items-center justify-center p-4 text-center sm:p-0">
            <Transition.Child
              as={React.Fragment}
              enter="ease-out duration-300"
              enterFrom="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
              enterTo="opacity-100 translate-y-0 sm:scale-100"
              leave="ease-in duration-200"
              leaveFrom="opacity-100 translate-y-0 sm:scale-100"
              leaveTo="opacity-0 translate-y-4 sm:translate-y-0 sm:scale-95"
            >
              <Dialog.Panel className="relative transform rounded-lg border border-custom-border-200 bg-custom-background-100 px-5 py-8 text-left shadow-xl transition-all sm:my-8 sm:w-full sm:max-w-2xl sm:p-6">
                <ViewForm
                  handleFormSubmit={handleFormSubmit}
                  handleClose={handleClose}
                  status={data ? true : false}
                  data={data}
                  viewType={viewType}
                  preLoadedData={preLoadedData}
                />
              </Dialog.Panel>
            </Transition.Child>
          </div>
        </div>
      </Dialog>
    </Transition.Root>
  );
};