calendar/apps/web/modules/onboarding/hooks/useSubmitOnboarding.ts

import { useSession } from "next-auth/react";
import { useRouter } from "next/navigation";
import { useState } from "react";

import { setShowNewOrgModalFlag } from "@calcom/features/ee/organizations/hooks/useWelcomeModal";
import { useFlagMap } from "@calcom/features/flags/context/provider";
import { CreationSource } from "@calcom/prisma/enums";
import { trpc } from "@calcom/trpc/react";
import { showToast } from "@calcom/ui/components/toast";

import type { OnboardingState } from "../store/onboarding-store";

export const useSubmitOnboarding = () => {
  const router = useRouter();
  const { update: updateSession } = useSession();
  const [isSubmitting, setIsSubmitting] = useState(false);
  const [error, setError] = useState<string | null>(null);
  const flags = useFlagMap();

  const intentToCreateOrg = trpc.viewer.organizations.intentToCreateOrg.useMutation();

  const submitOnboarding = async (
    store: OnboardingState,
    userEmail: string,
    invitesToSubmit: OnboardingState["invites"]
  ) => {
    setIsSubmitting(true);
    setError(null);

    try {
      const { selectedPlan, organizationDetails, organizationBrand, teams, inviteRole, resetOnboarding } =
        store;

      if (selectedPlan !== "organization") {
        throw new Error("Only organization plan is currently supported");
      }

      // Prepare teams data
      const teamsData = teams
        .filter((team) => team.name.trim().length > 0)
        .map((team) => ({
          id: -1, // New team
          name: team.name,
          isBeingMigrated: false,
          slug: null,
        }));

      // Prepare invites data
      const invitedMembersData = invitesToSubmit
        .filter((invite) => invite.email.trim().length > 0)
        .map((invite) => ({
          email: invite.email,
          teamName: invite.team,
          teamId: -1,
          role: inviteRole,
        }));

      const result = await intentToCreateOrg.mutateAsync({
        name: organizationDetails.name,
        slug: organizationDetails.link,
        bio: organizationDetails.bio || null,
        logo: organizationBrand.logo,
        brandColor: organizationBrand.color,
        bannerUrl: organizationBrand.banner,
        orgOwnerEmail: userEmail,
        seats: null,
        pricePerSeat: null,
        isPlatform: false,
        creationSource: CreationSource.WEBAPP,
        teams: teamsData,
        invitedMembers: invitedMembersData,
      });

      // If there's a checkout URL, redirect to Stripe (billing enabled flow)
      if (result.checkoutUrl) {
        window.location.href = result.checkoutUrl;
        return;
      }

      // No checkout URL means billing is disabled (self-hosted flow)
      // Organization has already been created by the backend
      showToast("Organization created successfully!", "success");
      // Refresh session to get updated organization data
      await updateSession();
      // Set flag to show welcome modal after personal onboarding redirect
      setShowNewOrgModalFlag();
      skipToPersonal(resetOnboarding);
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : "Failed to create organization";
      setError(errorMessage);
      showToast(errorMessage, "error");
      console.error("Organization creation error:", err);
    } finally {
      setIsSubmitting(false);
    }
  };

  const skipToPersonal = (resetOnboarding: () => void) => {
    resetOnboarding();
    const gettingStartedPath = flags["onboarding-v3"] ? "/onboarding/personal/settings" : "/getting-started";
    router.push(gettingStartedPath);
  };

  return {
    submitOnboarding,
    skipToPersonal,
    isSubmitting,
    error,
  };
};