fix(civitai): support civitai.red URLs (#897)

tests/frontend/components/modelContextMenuMixin.test.js

@@ -0,0 +1,21 @@
+import { describe, expect, it } from 'vitest';
+
+import { ModelContextMenuMixin } from '../../../static/js/components/ContextMenu/ModelContextMenuMixin.js';
+
+describe('ModelContextMenuMixin.extractModelVersionId', () => {
+  it('accepts civitai.red model URLs', () => {
+    expect(
+      ModelContextMenuMixin.extractModelVersionId(
+        'https://civitai.red/models/65423/nijimecha-artstyle?modelVersionId=777'
+      )
+    ).toEqual({ modelId: '65423', modelVersionId: '777' });
+  });
+
+  it('rejects model-like URLs from unsupported hosts', () => {
+    expect(
+      ModelContextMenuMixin.extractModelVersionId(
+        'https://example.com/models/65423?modelVersionId=777'
+      )
+    ).toEqual({ modelId: null, modelVersionId: null });
+  });
+});

tests/frontend/managers/downloadManager.history.test.js

@@ -136,4 +136,14 @@ describe('DownloadManager version history badges', () => {
     expect(items[1].querySelector('.local-path')?.textContent).toContain('/models/still-local.safetensors');
     expect(items[1].querySelector('.downloaded-badge')).toBeNull();
   });
+
+  it('extracts model and version ids from civitai.red URLs', () => {
+    const manager = new DownloadManager();
+
+    expect(
+      manager.extractModelId('https://civitai.red/models/65423/nijimecha-artstyle?modelVersionId=777')
+    ).toBe('65423');
+    expect(manager.modelVersionId).toBe('777');
+    expect(manager.source).toBeNull();
+  });
 });

tests/frontend/utils/civitaiUtils.test.js

@@ -4,7 +4,10 @@ import {
     getOptimizedUrl,
     getShowcaseUrl,
     getThumbnailUrl,
+    extractCivitaiImageId,
+    extractCivitaiModelUrlParts,
     isCivitaiUrl,
+    isSupportedCivitaiPageHost,
     OptimizationMode
 } from '../../../static/js/utils/civitaiUtils.js';
 
@@ -217,4 +220,43 @@ describe('civitaiUtils', () => {
             expect(isCivitaiUrl('not-a-url')).toBe(false);
         });
     });
+
+    describe('isSupportedCivitaiPageHost', () => {
+        it('accepts civitai.com and civitai.red page hosts', () => {
+            expect(isSupportedCivitaiPageHost('civitai.com')).toBe(true);
+            expect(isSupportedCivitaiPageHost('civitai.red')).toBe(true);
+        });
+
+        it('rejects unrelated hosts', () => {
+            expect(isSupportedCivitaiPageHost('www.civitai.com')).toBe(false);
+            expect(isSupportedCivitaiPageHost('www.civitai.red')).toBe(false);
+            expect(isSupportedCivitaiPageHost('example.com')).toBe(false);
+            expect(isSupportedCivitaiPageHost('')).toBe(false);
+            expect(isSupportedCivitaiPageHost(null)).toBe(false);
+        });
+    });
+
+    describe('extractCivitaiModelUrlParts', () => {
+        it('extracts model and version ids from civitai.red model URLs', () => {
+            expect(
+                extractCivitaiModelUrlParts('https://civitai.red/models/65423/name?modelVersionId=98765')
+            ).toEqual({ modelId: '65423', modelVersionId: '98765' });
+        });
+
+        it('rejects model-like URLs from unsupported hosts', () => {
+            expect(
+                extractCivitaiModelUrlParts('https://example.com/models/65423?modelVersionId=98765')
+            ).toEqual({ modelId: null, modelVersionId: null });
+        });
+    });
+
+    describe('extractCivitaiImageId', () => {
+        it('extracts image ids from civitai.red image URLs', () => {
+            expect(extractCivitaiImageId('https://civitai.red/images/126920345')).toBe('126920345');
+        });
+
+        it('rejects image-like URLs from unsupported hosts', () => {
+            expect(extractCivitaiImageId('https://example.com/images/126920345')).toBe(null);
+        });
+    });
 });