plane/apps/app/components/onboarding/join-workspaces.tsx

import React, { useState } from "react";

import useSWR, { mutate } from "swr";

// services
import workspaceService from "services/workspace.service";
import userService from "services/user.service";
// hooks
import useUser from "hooks/use-user";
// ui
import { PrimaryButton, SecondaryButton } from "components/ui";
// icons
import { CheckCircleIcon } from "@heroicons/react/24/outline";
// helpers
import { truncateText } from "helpers/string.helper";
// types
import { ICurrentUserResponse, IUser, IWorkspaceMemberInvitation, OnboardingSteps } from "types";
// fetch-keys
import { CURRENT_USER, USER_WORKSPACE_INVITATIONS } from "constants/fetch-keys";
// constants
import { ROLE } from "constants/workspace";

type Props = {
  stepChange: (steps: Partial<OnboardingSteps>) => Promise<void>;
};

export const JoinWorkspaces: React.FC<Props> = ({ stepChange }) => {
  const [isJoiningWorkspaces, setIsJoiningWorkspaces] = useState(false);
  const [invitationsRespond, setInvitationsRespond] = useState<string[]>([]);

  const { user } = useUser();

  const { data: invitations, mutate: mutateInvitations } = useSWR(USER_WORKSPACE_INVITATIONS, () =>
    workspaceService.userWorkspaceInvitations()
  );

  const handleInvitation = (
    workspace_invitation: IWorkspaceMemberInvitation,
    action: "accepted" | "withdraw"
  ) => {
    if (action === "accepted") {
      setInvitationsRespond((prevData) => [...prevData, workspace_invitation.id]);
    } else if (action === "withdraw") {
      setInvitationsRespond((prevData) =>
        prevData.filter((item: string) => item !== workspace_invitation.id)
      );
    }
  };

  // complete onboarding
  const finishOnboarding = async () => {
    if (!user) return;

    mutate<ICurrentUserResponse>(
      CURRENT_USER,
      (prevData) => {
        if (!prevData) return prevData;

        return {
          ...prevData,
          is_onboarded: true,
        };
      },
      false
    );

    await userService.updateUserOnBoard({ userRole: user.role }, user);
    await stepChange({ workspace_join: true });
  };

  const submitInvitations = async () => {
    if (invitationsRespond.length <= 0) return;

    setIsJoiningWorkspaces(true);

    await workspaceService
      .joinWorkspaces({ invitations: invitationsRespond })
      .then(async () => {
        await mutateInvitations();
        await finishOnboarding();

        setIsJoiningWorkspaces(false);
      })
      .catch(() => setIsJoiningWorkspaces(false));
  };

  return (
    <div className="w-full space-y-7 sm:space-y-10">
      <h5 className="sm:text-lg">We see that someone has invited you to</h5>
      <h4 className="text-xl sm:text-2xl font-semibold">Join a workspace</h4>
      <div className="md:w-3/5 space-y-4">
        {invitations &&
          invitations.map((invitation) => {
            const isSelected = invitationsRespond.includes(invitation.id);

            return (
              <div
                key={invitation.id}
                className={`flex cursor-pointer items-center gap-2 border py-5 px-3.5 rounded ${
                  isSelected
                    ? "border-custom-primary-100"
                    : "border-custom-border-300 hover:bg-custom-background-80"
                }`}
                onClick={() => handleInvitation(invitation, isSelected ? "withdraw" : "accepted")}
              >
                <div className="flex-shrink-0">
                  <div className="grid place-items-center h-9 w-9 rounded">
                    {invitation.workspace.logo && invitation.workspace.logo !== "" ? (
                      <img
                        src={invitation.workspace.logo}
                        height="100%"
                        width="100%"
                        className="rounded"
                        alt={invitation.workspace.name}
                      />
                    ) : (
                      <span className="grid place-items-center h-9 w-9 py-1.5 px-3 rounded bg-gray-700 uppercase text-white">
                        {invitation.workspace.name[0]}
                      </span>
                    )}
                  </div>
                </div>
                <div className="min-w-0 flex-1">
                  <div className="text-sm font-medium">
                    {truncateText(invitation.workspace.name, 30)}
                  </div>
                  <p className="text-xs text-custom-text-200">{ROLE[invitation.role]}</p>
                </div>
                <span
                  className={`flex-shrink-0 ${
                    isSelected ? "text-custom-primary-100" : "text-custom-text-200"
                  }`}
                >
                  <CheckCircleIcon className="h-5 w-5" />
                </span>
              </div>
            );
          })}
      </div>
      <div className="flex items-center gap-3">
        <PrimaryButton
          type="submit"
          size="md"
          onClick={submitInvitations}
          disabled={isJoiningWorkspaces || invitationsRespond.length === 0}
        >
          Accept & Join
        </PrimaryButton>
        <SecondaryButton size="md" onClick={finishOnboarding} outline>
          Skip for now
        </SecondaryButton>
      </div>
    </div>
  );
};