plane/web/components/labels/create-update-label-inline.tsx

import React, { forwardRef, useEffect } from "react";

import { useRouter } from "next/router";

import { mutate } from "swr";

// react-hook-form
import { Controller, SubmitHandler, useForm } from "react-hook-form";
// hooks
import useUserAuth from "hooks/use-user-auth";
// react-color
import { TwitterPicker } from "react-color";
// headless ui
import { Popover, Transition } from "@headlessui/react";
// services
import issuesService from "services/issues.service";
// ui
import { Input, PrimaryButton, SecondaryButton } from "components/ui";
// types
import { IIssueLabels } from "types";
// fetch-keys
import { PROJECT_ISSUE_LABELS } from "constants/fetch-keys";
import { getRandomLabelColor, LABEL_COLOR_OPTIONS } from "constants/label";

type Props = {
  labelForm: boolean;
  setLabelForm: React.Dispatch<React.SetStateAction<boolean>>;
  isUpdating: boolean;
  labelToUpdate: IIssueLabels | null;
  onClose?: () => void;
};

const defaultValues: Partial<IIssueLabels> = {
  name: "",
  color: "rgb(var(--color-text-200))",
};

export const CreateUpdateLabelInline = forwardRef<HTMLDivElement, Props>(
  function CreateUpdateLabelInline(props, ref) {
    const { labelForm, setLabelForm, isUpdating, labelToUpdate, onClose } = props;

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

    const { user } = useUserAuth();

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

    const handleClose = () => {
      setLabelForm(false);
      reset(defaultValues);
      if (onClose) onClose();
    };

    const handleLabelCreate: SubmitHandler<IIssueLabels> = async (formData) => {
      if (!workspaceSlug || !projectId || isSubmitting) return;

      await issuesService
        .createIssueLabel(workspaceSlug as string, projectId as string, formData, user)
        .then((res) => {
          mutate<IIssueLabels[]>(
            PROJECT_ISSUE_LABELS(projectId as string),
            (prevData) => [res, ...(prevData ?? [])],
            false
          );
          handleClose();
        });
    };

    const handleLabelUpdate: SubmitHandler<IIssueLabels> = async (formData) => {
      if (!workspaceSlug || !projectId || isSubmitting) return;

      await issuesService
        .patchIssueLabel(
          workspaceSlug as string,
          projectId as string,
          labelToUpdate?.id ?? "",
          formData,
          user
        )
        .then(() => {
          reset(defaultValues);
          mutate<IIssueLabels[]>(
            PROJECT_ISSUE_LABELS(projectId as string),
            (prevData) =>
              prevData?.map((p) => (p.id === labelToUpdate?.id ? { ...p, ...formData } : p)),
            false
          );
          handleClose();
        });
    };

    useEffect(() => {
      if (!labelForm && isUpdating) return;

      reset();
    }, [labelForm, isUpdating, reset]);

    useEffect(() => {
      if (!labelToUpdate) return;

      setValue(
        "color",
        labelToUpdate.color && labelToUpdate.color !== "" ? labelToUpdate.color : "#000"
      );
      setValue("name", labelToUpdate.name);
    }, [labelToUpdate, setValue]);

    useEffect(() => {
      if (labelToUpdate) {
        setValue(
          "color",
          labelToUpdate.color && labelToUpdate.color !== "" ? labelToUpdate.color : "#000"
        );
        return;
      }

      setValue("color", getRandomLabelColor());
    }, [labelToUpdate, setValue]);

    return (
      <div
        className={`flex scroll-m-8 items-center gap-2 rounded border border-custom-border-200 bg-custom-background-100 px-3.5 py-2 ${
          labelForm ? "" : "hidden"
        }`}
        ref={ref}
      >
        <div className="flex-shrink-0">
          <Popover className="relative z-10 flex h-full w-full items-center justify-center">
            {({ open }) => (
              <>
                <Popover.Button
                  className={`group inline-flex items-center text-base font-medium focus:outline-none ${
                    open ? "text-custom-text-100" : "text-custom-text-200"
                  }`}
                >
                  <span
                    className="h-4 w-4 rounded-full"
                    style={{
                      backgroundColor: watch("color"),
                    }}
                  />
                </Popover.Button>

                <Transition
                  as={React.Fragment}
                  enter="transition ease-out duration-200"
                  enterFrom="opacity-0 translate-y-1"
                  enterTo="opacity-100 translate-y-0"
                  leave="transition ease-in duration-150"
                  leaveFrom="opacity-100 translate-y-0"
                  leaveTo="opacity-0 translate-y-1"
                >
                  <Popover.Panel className="absolute top-full left-0 z-20 mt-3 w-screen max-w-xs px-2 sm:px-0">
                    <Controller
                      name="color"
                      control={control}
                      render={({ field: { value, onChange } }) => (
                        <TwitterPicker
                          colors={LABEL_COLOR_OPTIONS}
                          color={value}
                          onChange={(value) => onChange(value.hex)}
                        />
                      )}
                    />
                  </Popover.Panel>
                </Transition>
              </>
            )}
          </Popover>
        </div>
        <div className="flex flex-1 flex-col justify-center">
          <Input
            type="text"
            id="labelName"
            name="name"
            register={register}
            placeholder="Label title"
            validations={{
              required: "Label title is required",
            }}
            error={errors.name}
          />
        </div>
        <SecondaryButton onClick={() => handleClose()}>Cancel</SecondaryButton>
        {isUpdating ? (
          <PrimaryButton onClick={handleSubmit(handleLabelUpdate)} loading={isSubmitting}>
            {isSubmitting ? "Updating" : "Update"}
          </PrimaryButton>
        ) : (
          <PrimaryButton onClick={handleSubmit(handleLabelCreate)} loading={isSubmitting}>
            {isSubmitting ? "Adding" : "Add"}
          </PrimaryButton>
        )}
      </div>
    );
  }
);