mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
b72cf7ba98
- Add CivitAI URL utility with optimization strategies for showcase and thumbnail modes - Replace /original=true with /optimized=true for showcase videos to reduce bandwidth - Remove redundant crossorigin and referrerpolicy attributes from video elements - Use media type detection to apply appropriate optimization (image vs video) - Integrate URL optimization into showcase rendering for improved loading times
120 lines
3.7 KiB
JavaScript
120 lines
3.7 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 image domain
|
|
if (url.hostname.toLowerCase() !== 'image.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);
|
|
return parsed.hostname.toLowerCase() === 'image.civitai.com';
|
|
} catch (e) {
|
|
return false;
|
|
}
|
|
}
|