sourcebot/packages/web/src/app/invite/actions.ts

"use server";

import { withAuth } from "@/actions";
import { isServiceError } from "@/lib/utils";
import { orgNotFound, ServiceError } from "@/lib/serviceError";
import { sew } from "@/actions";
import { addUserToOrganization } from "@/lib/authUtils";
import { prisma } from "@/prisma";
import { StatusCodes } from "http-status-codes";
import { ErrorCode } from "@/lib/errorCodes";

export const joinOrganization = async (orgId: number, inviteLinkId?: string) => sew(async () =>
    withAuth(async (userId) => {
        const org = await prisma.org.findUnique({
            where: {
                id: orgId,
            },
        });
        
        if (!org) {
            return orgNotFound();
        }

        // If member approval is required we must be using a valid invite link
        if (org.memberApprovalRequired) {
            if (!org.inviteLinkEnabled) {
                return {
                    statusCode: StatusCodes.BAD_REQUEST,
                    errorCode: ErrorCode.INVITE_LINK_NOT_ENABLED,
                    message: "Invite link is not enabled.",
                } satisfies ServiceError;
            }

            if (org.inviteLinkId !== inviteLinkId) {
                return {
                    statusCode: StatusCodes.BAD_REQUEST,
                    errorCode: ErrorCode.INVALID_INVITE_LINK,
                    message: "Invalid invite link.",
                } satisfies ServiceError;
            }
        }

        const addUserToOrgRes = await addUserToOrganization(userId, org.id);
        if (isServiceError(addUserToOrgRes)) {
            return addUserToOrgRes;
        }

        return {
            success: true,
        }
    })
)