mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-04-10 12:52:15 -03:00
30db8c3d1d
- Update CSP whitelist to use wildcard *.civitai.com for all CDN subdomains - Fix hostname parsing to use parsed.hostname instead of parsed.netloc (handles ports) - Update rewrite_preview_url() to support all CivitAI CDN subdomains - Update rewriteCivitaiUrl() frontend function to support subdomains - Add comprehensive tests for edge cases (ports, subdomains, invalid URLs) - Add security note explaining wildcard CSP design decision Fixes CSP blocking of images from image-b2.civitai.com and other CDN subdomains
122 lines
3.9 KiB
JavaScript
122 lines
3.9 KiB
JavaScript
/**
|
|
* CivitAI URL utilities
|
|
* Functions for working with CivitAI media URLs
|
|
*/
|
|
|
|
/**
|
|
* Optimization strategies for CivitAI URLs
|
|
*/
|
|
export const OptimizationMode = {
|
|
/** Full quality for showcase/display - uses /optimized=true only */
|
|
SHOWCASE: 'showcase',
|
|
/** Thumbnail size for cards - uses /width=450,optimized=true */
|
|
THUMBNAIL: 'thumbnail',
|
|
};
|
|
|
|
/**
|
|
* Rewrite Civitai preview URLs to use optimized renditions.
|
|
* Mirrors the backend's rewrite_preview_url() function from py/utils/civitai_utils.py
|
|
*
|
|
* @param {string|null} sourceUrl - Original preview URL from the Civitai API
|
|
* @param {string|null} mediaType - Optional media type hint ("image" or "video")
|
|
* @param {string} mode - Optimization mode ('showcase' or 'thumbnail')
|
|
* @returns {[string|null, boolean]} - Tuple of [rewritten URL or original, wasRewritten flag]
|
|
*/
|
|
export function rewriteCivitaiUrl(sourceUrl, mediaType = null, mode = OptimizationMode.THUMBNAIL) {
|
|
if (!sourceUrl) {
|
|
return [sourceUrl, false];
|
|
}
|
|
|
|
try {
|
|
const url = new URL(sourceUrl);
|
|
|
|
// Check if it's a CivitAI CDN domain (supports all subdomains like image-b2.civitai.com)
|
|
const hostname = url.hostname.toLowerCase();
|
|
if (hostname === 'civitai.com' || !hostname.endsWith('.civitai.com')) {
|
|
return [sourceUrl, false];
|
|
}
|
|
|
|
// Determine replacement based on mode and media type
|
|
let replacement;
|
|
if (mode === OptimizationMode.SHOWCASE) {
|
|
// Full quality for showcase - no width restriction
|
|
replacement = '/optimized=true';
|
|
} else {
|
|
// Thumbnail mode with width restriction
|
|
replacement = '/width=450,optimized=true';
|
|
if (mediaType && mediaType.toLowerCase() === 'video') {
|
|
replacement = '/transcode=true,width=450,optimized=true';
|
|
}
|
|
}
|
|
|
|
// Replace /original=true with optimized version
|
|
if (!url.pathname.includes('/original=true')) {
|
|
return [sourceUrl, false];
|
|
}
|
|
|
|
const updatedPath = url.pathname.replace('/original=true', replacement, 1);
|
|
|
|
if (updatedPath === url.pathname) {
|
|
return [sourceUrl, false];
|
|
}
|
|
|
|
url.pathname = updatedPath;
|
|
return [url.toString(), true];
|
|
} catch (e) {
|
|
// Invalid URL
|
|
return [sourceUrl, false];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the optimized URL for a media item, falling back to original if not a CivitAI URL
|
|
*
|
|
* @param {string} url - Original URL
|
|
* @param {string} type - Media type ("image" or "video")
|
|
* @param {string} mode - Optimization mode ('showcase' or 'thumbnail')
|
|
* @returns {string} - Optimized URL or original URL
|
|
*/
|
|
export function getOptimizedUrl(url, type = 'image', mode = OptimizationMode.THUMBNAIL) {
|
|
const [optimizedUrl] = rewriteCivitaiUrl(url, type, mode);
|
|
return optimizedUrl || url;
|
|
}
|
|
|
|
/**
|
|
* Get showcase-optimized URL (full quality)
|
|
*
|
|
* @param {string} url - Original URL
|
|
* @param {string} type - Media type ("image" or "video")
|
|
* @returns {string} - Optimized URL for showcase display
|
|
*/
|
|
export function getShowcaseUrl(url, type = 'image') {
|
|
return getOptimizedUrl(url, type, OptimizationMode.SHOWCASE);
|
|
}
|
|
|
|
/**
|
|
* Get thumbnail-optimized URL (width=450)
|
|
*
|
|
* @param {string} url - Original URL
|
|
* @param {string} type - Media type ("image" or "video")
|
|
* @returns {string} - Optimized URL for thumbnail display
|
|
*/
|
|
export function getThumbnailUrl(url, type = 'image') {
|
|
return getOptimizedUrl(url, type, OptimizationMode.THUMBNAIL);
|
|
}
|
|
|
|
/**
|
|
* Check if a URL is from CivitAI
|
|
*
|
|
* @param {string} url - URL to check
|
|
* @returns {boolean} - True if it's a CivitAI URL
|
|
*/
|
|
export function isCivitaiUrl(url) {
|
|
if (!url) return false;
|
|
try {
|
|
const parsed = new URL(url);
|
|
const hostname = parsed.hostname.toLowerCase();
|
|
return hostname.endsWith('.civitai.com') && hostname !== 'civitai.com';
|
|
} catch (e) {
|
|
return false;
|
|
}
|
|
}
|