import type { ParsedUrlQuery } from "querystring";
import { stringify } from "querystring";

import logger from "@calcom/lib/logger";
import { safeStringify } from "@calcom/lib/safeStringify";
import type { RedirectType } from "@calcom/prisma/client";

const log = logger.getSubLogger({ prefix: ["lib", "getTemporaryOrgRedirect"] });
export const getTemporaryOrgRedirect = async ({
  slug,
  redirectType,
  eventTypeSlug,
  currentQuery,
}: {
  slug: string;
  redirectType: RedirectType;
  eventTypeSlug: string | null;
  currentQuery: ParsedUrlQuery;
}) => {
  const prisma = (await import("@calcom/prisma")).default;
  log.debug(
    `Looking for redirect for`,
    safeStringify({
      slug,
      redirectType,
      eventTypeSlug,
    })
  );
  const redirect = await prisma.tempOrgRedirect.findUnique({
    where: {
      from_type_fromOrgId: {
        type: redirectType,
        from: slug,
        fromOrgId: 0,
      },
    },
  });

  if (redirect) {
    log.debug(`Redirecting ${slug} to ${redirect.toUrl}`);
    const newDestinationWithoutQuery = eventTypeSlug ? `${redirect.toUrl}/${eventTypeSlug}` : redirect.toUrl;
    const currentQueryString = stringify(currentQuery);
    return {
      redirect: {
        permanent: false,
        destination: `${newDestinationWithoutQuery}${currentQueryString ? `?${currentQueryString}` : ""}`,
      },
    } as const;
  }
  return null;
};