plane/apps/app/components/project/cycles/create-update-cycle-modal.tsx

import React, { useEffect } from "react";
// next
import { useRouter } from "next/router";
// swr
import { mutate } from "swr";
// react hook form
import { Controller, useForm } from "react-hook-form";
// headless
import { Dialog, Transition } from "@headlessui/react";
// services
import cycleService from "services/cycles.service";
// hooks
import useToast from "hooks/use-toast";
// ui
import { Button, Input, TextArea, CustomSelect } from "components/ui";
// common
import { renderDateFormat } from "helpers/date-time.helper";
// types
import type { ICycle } from "types";
// fetch keys
import { CYCLE_LIST } from "constants/fetch-keys";

type Props = {
  isOpen: boolean;
  setIsOpen: React.Dispatch<React.SetStateAction<boolean>>;
  projectId: string;
  data?: ICycle;
};

const defaultValues: Partial<ICycle> = {
  name: "",
  description: "",
  status: "draft",
  start_date: new Date().toString(),
  end_date: new Date().toString(),
};

const CreateUpdateCycleModal: React.FC<Props> = ({ isOpen, setIsOpen, data, projectId }) => {
  const router = useRouter();
  const { workspaceSlug } = router.query;

  const { setToastAlert } = useToast();

  const {
    register,
    formState: { errors, isSubmitting },
    handleSubmit,
    control,
    reset,
    setError,
  } = useForm<ICycle>({
    defaultValues,
  });

  useEffect(() => {
    if (data) {
      setIsOpen(true);
      reset(data);
    } else {
      reset(defaultValues);
    }
  }, [data, setIsOpen, reset]);

  const onSubmit = async (formData: ICycle) => {
    if (!workspaceSlug) return;
    const payload = {
      ...formData,
      start_date: formData.start_date ? renderDateFormat(formData.start_date) : null,
      end_date: formData.end_date ? renderDateFormat(formData.end_date) : null,
    };
    if (!data) {
      await cycleService
        .createCycle(workspaceSlug as string, projectId, payload)
        .then((res) => {
          mutate<ICycle[]>(CYCLE_LIST(projectId), (prevData) => [res, ...(prevData ?? [])], false);

          handleClose();
          setToastAlert({
            title: "Success",
            type: "success",
            message: "Cycle created successfully",
          });
        })
        .catch((err) => {
          Object.keys(err).map((key) => {
            setError(key as keyof typeof defaultValues, {
              message: err[key].join(", "),
            });
          });
        });
    } else {
      await cycleService
        .updateCycle(workspaceSlug as string, projectId, data.id, payload)
        .then((res) => {
          mutate(CYCLE_LIST(projectId));
          handleClose();

          setToastAlert({
            title: "Success",
            type: "success",
            message: "Cycle updated successfully",
          });
        })
        .catch((err) => {
          Object.keys(err).map((key) => {
            setError(key as keyof typeof defaultValues, {
              message: err[key].join(", "),
            });
          });
        });
    }
  };

  const handleClose = () => {
    setIsOpen(false);
    reset(defaultValues);
  };

  return (
    <Transition.Root show={isOpen} as={React.Fragment}>
      <Dialog as="div" className="relative z-30" 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-gray-500 bg-opacity-75 transition-opacity" />
        </Transition.Child>

        <div className="fixed inset-0 z-10 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 overflow-hidden rounded-lg bg-white px-5 py-8 text-left shadow-xl transition-all sm:my-8 sm:w-full sm:max-w-2xl sm:p-6">
                <form onSubmit={handleSubmit(onSubmit)}>
                  <div className="space-y-5">
                    <Dialog.Title as="h3" className="text-lg font-medium leading-6 text-gray-900">
                      {data ? "Update" : "Create"} Cycle
                    </Dialog.Title>
                    <div className="space-y-3">
                      <div>
                        <Input
                          id="name"
                          label="Name"
                          name="name"
                          type="name"
                          placeholder="Enter name"
                          autoComplete="off"
                          error={errors.name}
                          register={register}
                          validations={{
                            required: "Name is required",
                          }}
                        />
                      </div>
                      <div>
                        <TextArea
                          id="description"
                          name="description"
                          label="Description"
                          placeholder="Enter description"
                          error={errors.description}
                          register={register}
                        />
                      </div>
                      <div>
                        <h6 className="text-gray-500">Status</h6>
                        <Controller
                          name="status"
                          control={control}
                          render={({ field }) => (
                            <CustomSelect
                              {...field}
                              label={
                                <span className="capitalize">{field.value ?? "Select Status"}</span>
                              }
                              input
                            >
                              {[
                                { label: "Draft", value: "draft" },
                                { label: "Started", value: "started" },
                                { label: "Completed", value: "completed" },
                              ].map((item) => (
                                <CustomSelect.Option key={item.value} value={item.value}>
                                  {item.label}
                                </CustomSelect.Option>
                              ))}
                            </CustomSelect>
                          )}
                        />
                      </div>
                      <div className="flex gap-x-2">
                        <div className="w-full">
                          <Input
                            id="start_date"
                            label="Start Date"
                            name="start_date"
                            type="date"
                            placeholder="Enter start date"
                            error={errors.start_date}
                            register={register}
                            validations={{
                              required: "Start date is required",
                            }}
                          />
                        </div>
                        <div className="w-full">
                          <Input
                            id="end_date"
                            label="End Date"
                            name="end_date"
                            type="date"
                            placeholder="Enter end date"
                            error={errors.end_date}
                            register={register}
                            validations={{
                              required: "End date is required",
                            }}
                          />
                        </div>
                      </div>
                    </div>
                  </div>
                  <div className="mt-5 flex justify-end gap-2">
                    <Button theme="secondary" onClick={handleClose}>
                      Cancel
                    </Button>
                    <Button type="submit" disabled={isSubmitting}>
                      {data
                        ? isSubmitting
                          ? "Updating Cycle..."
                          : "Update Cycle"
                        : isSubmitting
                        ? "Creating Cycle..."
                        : "Create Cycle"}
                    </Button>
                  </div>
                </form>
              </Dialog.Panel>
            </Transition.Child>
          </div>
        </div>
      </Dialog>
    </Transition.Root>
  );
};

export default CreateUpdateCycleModal;