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; let retries = 0;
const maxRetries = 5; const maxRetries = 5;
let isExtract = (mode === RateLimiterMode.Extract || mode === RateLimiterMode.ExtractStatus) let isExtract =
let rpcName = mode === RateLimiterMode.Extract ||
isExtract mode === RateLimiterMode.ExtractStatus;
? "auth_credit_usage_chunk_extract" let rpcName = isExtract
: "auth_credit_usage_chunk_test_22_credit_pack_n_extract"; ? "auth_credit_usage_chunk_extract"
: "auth_credit_usage_chunk_test_22_credit_pack_n_extract";
while (retries < maxRetries) { while (retries < maxRetries) {
({ data, error } = await supabase_service.rpc( ({ data, error } = await supabase_service.rpc(
rpcName, rpcName,
@ -333,7 +334,12 @@ export async function supaAuthenticateUser(
mode === RateLimiterMode.Extract || mode === RateLimiterMode.Extract ||
mode === RateLimiterMode.Search) 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 // check the origin of the request and make sure its from firecrawl.dev
// const origin = req.headers.origin; // const origin = req.headers.origin;
// if (origin && origin.includes("firecrawl.dev")){ // if (origin && origin.includes("firecrawl.dev")){
@ -352,7 +358,7 @@ export async function supaAuthenticateUser(
team_id: teamId ?? undefined, team_id: teamId ?? undefined,
// Now we have a test suite plan // Now we have a test suite plan
plan: "testSuite", 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_MONTHLY:
case process.env.STRIPE_PRICE_ID_ETIER_SCALE_1_YEARLY: case process.env.STRIPE_PRICE_ID_ETIER_SCALE_1_YEARLY:
return "etierscale1"; 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_MONTHLY:
case process.env.STRIPE_PRICE_ID_EXTRACT_STARTER_YEARLY: case process.env.STRIPE_PRICE_ID_EXTRACT_STARTER_YEARLY:
return "extract_starter"; 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_MONTHLY:
case process.env.STRIPE_PRICE_ID_EXTRACT_EXPLORER_YEARLY: case process.env.STRIPE_PRICE_ID_EXTRACT_EXPLORER_YEARLY:
return "extract_explorer"; return "extract_explorer";