plane/web/components/onboarding/join-workspaces.tsx

import React, { useState } from "react";

import useSWR, { mutate } from "swr";

// services
import { WorkspaceService } from "services/workspace.service";
// hooks
import useUser from "hooks/use-user";
// ui
import { Button } from "@plane/ui";
// icons
import { CheckCircle } from "lucide-react";
// helpers
import { truncateText } from "helpers/string.helper";
// types
import { IWorkspaceMemberInvitation, TOnboardingSteps } from "types";
// fetch-keys
import { USER_WORKSPACES, USER_WORKSPACE_INVITATIONS } from "constants/fetch-keys";
// constants
import { ROLE } from "constants/workspace";
import { trackEvent } from "helpers/event-tracker.helper";

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

// services
const workspaceService = new WorkspaceService();

export const JoinWorkspaces: React.FC<Props> = ({ finishOnboarding, stepChange, updateLastWorkspace }) => {
  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));
    }
  };

  const handleNextStep = async () => {
    if (!user) return;

    await stepChange({ workspace_join: true });

    if (user.onboarding_step.workspace_create && user.onboarding_step.workspace_invite) await finishOnboarding();
  };

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

    setIsJoiningWorkspaces(true);

    await workspaceService
      .joinWorkspaces({ invitations: invitationsRespond })
      .then(async (res) => {
        trackEvent(
          'WORKSPACE_USER_INVITE_ACCEPT',
          res
        )
        await mutateInvitations();
        await mutate(USER_WORKSPACES);
        await updateLastWorkspace();

        await handleNextStep();
      })
      .finally(() => 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="max-h-[37vh] overflow-y-auto 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-200 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"}`}>
                  <CheckCircle className="h-5 w-5" />
                </span>
              </div>
            );
          })}
      </div>
      <div className="flex items-center gap-3">
        <Button
          variant="primary"
          type="submit"
          size="md"
          onClick={submitInvitations}
          disabled={invitationsRespond.length === 0}
          loading={isJoiningWorkspaces}
        >
          Accept & Join
        </Button>
        <Button variant="neutral-primary" size="md" onClick={handleNextStep}>
          Skip for now
        </Button>
      </div>
    </div>
  );
};