Update auth.ts

This commit is contained in:
Nicolas 2025-02-14 11:12:02 -03:00
parent ce9d3d32f6
commit 7ac2b99210

View File

@ -94,11 +94,12 @@ export async function getACUC(
let retries = 0;
const maxRetries = 5;
let isExtract = (mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus)
let rpcName =
isExtract
? "auth_credit_usage_chunk_extract"
: "auth_credit_usage_chunk_test_22_credit_pack_n_extract";
let isExtract =
mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.ExtractStatus;
let rpcName = isExtract
? "auth_credit_usage_chunk_extract"
: "auth_credit_usage_chunk_test_22_credit_pack_n_extract";
while (retries < maxRetries) {
({ data, error } = await supabase_service.rpc(
rpcName,
@ -333,7 +334,12 @@ export async function supaAuthenticateUser(
mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.Search)
) {
return { success: true, team_id: `preview_${iptoken}`, chunk: null, plan: "free" };
return {
success: true,
team_id: `preview_${iptoken}`,
chunk: null,
plan: "free",
};
// check the origin of the request and make sure its from firecrawl.dev
// const origin = req.headers.origin;
// if (origin && origin.includes("firecrawl.dev")){
@ -352,7 +358,7 @@ export async function supaAuthenticateUser(
team_id: teamId ?? undefined,
// Now we have a test suite plan
plan: "testSuite",
chunk
chunk,
};
}
@ -390,12 +396,11 @@ function getPlanByPriceId(price_id: string | null): PlanType {
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_1_MONTHLY:
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_1_YEARLY:
return "etierscale1";
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_2_YEARLY:
return "etierscale2";
case process.env.STRIPE_PRICE_ID_EXTRACT_STARTER_MONTHLY:
case process.env.STRIPE_PRICE_ID_EXTRACT_STARTER_YEARLY:
return "extract_starter";
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_2_MONTHLY:
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_2_YEARLY:
return "etierscale2";
case process.env.STRIPE_PRICE_ID_EXTRACT_EXPLORER_MONTHLY:
case process.env.STRIPE_PRICE_ID_EXTRACT_EXPLORER_YEARLY:
return "extract_explorer";