72 lines
1.9 KiB
TypeScript
72 lines
1.9 KiB
TypeScript
import { isMediaInputFailure } from "@/lib/ai/runtime/request-failure-classification";
|
|
|
|
function parseUrl(value: string) {
|
|
try {
|
|
return new URL(String(value || "").trim());
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export function isUrlWithinPublicBase(mediaUrl: string, publicBaseUrl: string) {
|
|
const media = parseUrl(mediaUrl);
|
|
const base = parseUrl(publicBaseUrl);
|
|
if (!media || !base || media.host.toLowerCase() !== base.host.toLowerCase()) {
|
|
return false;
|
|
}
|
|
|
|
const basePath = base.pathname.replace(/\/+$/, "");
|
|
if (!basePath) return true;
|
|
return media.pathname === basePath || media.pathname.startsWith(`${basePath}/`);
|
|
}
|
|
|
|
export function shouldStageVip98kReferenceUrl(params: {
|
|
mediaUrl: string;
|
|
sourcePublicUrl: string;
|
|
targetPublicUrl?: string;
|
|
}) {
|
|
if (
|
|
params.targetPublicUrl &&
|
|
isUrlWithinPublicBase(params.mediaUrl, params.targetPublicUrl)
|
|
) {
|
|
return false;
|
|
}
|
|
return isUrlWithinPublicBase(params.mediaUrl, params.sourcePublicUrl);
|
|
}
|
|
|
|
export function getStorageObjectKeyFromPublicUrl(
|
|
mediaUrl: string,
|
|
publicBaseUrl: string,
|
|
) {
|
|
if (!isUrlWithinPublicBase(mediaUrl, publicBaseUrl)) return null;
|
|
const media = parseUrl(mediaUrl);
|
|
const base = parseUrl(publicBaseUrl);
|
|
if (!media || !base) return null;
|
|
const basePath = base.pathname.replace(/\/+$/, "");
|
|
const encodedKey = media.pathname.slice(basePath.length).replace(/^\/+/, "");
|
|
if (!encodedKey) return null;
|
|
try {
|
|
return decodeURIComponent(encodedKey);
|
|
} catch {
|
|
return encodedKey;
|
|
}
|
|
}
|
|
|
|
export function getVip98kStagedMediaRetryDelay(params: {
|
|
status: number;
|
|
message: string;
|
|
attempt: number;
|
|
maxAttempts: number;
|
|
stagedReferenceCount: number;
|
|
}) {
|
|
if (
|
|
params.status !== 400 ||
|
|
params.stagedReferenceCount < 1 ||
|
|
params.attempt >= params.maxAttempts ||
|
|
!isMediaInputFailure(params.message)
|
|
) {
|
|
return null;
|
|
}
|
|
return params.attempt * 20_000;
|
|
}
|