feat(misc): add VAE and Upscaler model management page

2026-03-24 22:52:12 -03:00 · 2026-01-31 07:28:10 +08:00
parent b86bd44c65
commit 0a340d397c
37 changed files with 1164 additions and 38 deletions
--- a/static/js/components/ContextMenu/MiscContextMenu.js
+++ b/static/js/components/ContextMenu/MiscContextMenu.js
@@ -0,0 +1,85 @@
+import { BaseContextMenu } from './BaseContextMenu.js';
+import { ModelContextMenuMixin } from './ModelContextMenuMixin.js';
+import { getModelApiClient, resetAndReload } from '../../api/modelApiFactory.js';
+import { showDeleteModal, showExcludeModal } from '../../utils/modalUtils.js';
+import { moveManager } from '../../managers/MoveManager.js';
+import { i18n } from '../../i18n/index.js';
+
+export class MiscContextMenu extends BaseContextMenu {
+    constructor() {
+        super('miscContextMenu', '.model-card');
+        this.nsfwSelector = document.getElementById('nsfwLevelSelector');
+        this.modelType = 'misc';
+        this.resetAndReload = resetAndReload;
+
+        this.initNSFWSelector();
+    }
+
+    // Implementation needed by the mixin
+    async saveModelMetadata(filePath, data) {
+        return getModelApiClient().saveModelMetadata(filePath, data);
+    }
+
+    showMenu(x, y, card) {
+        super.showMenu(x, y, card);
+
+        // Update the "Move to other root" label based on current model type
+        const moveOtherItem = this.menu.querySelector('[data-action="move-other"]');
+        if (moveOtherItem) {
+            const currentType = card.dataset.sub_type || 'vae';
+            const otherType = currentType === 'vae' ? 'upscaler' : 'vae';
+            const typeLabel = i18n.t(`misc.modelTypes.${otherType}`);
+            moveOtherItem.innerHTML = `<i class="fas fa-exchange-alt"></i> ${i18n.t('misc.contextMenu.moveToOtherTypeFolder', { otherType: typeLabel })}`;
+        }
+    }
+
+    handleMenuAction(action) {
+        // First try to handle with common actions
+        if (ModelContextMenuMixin.handleCommonMenuActions.call(this, action)) {
+            return;
+        }
+
+        const apiClient = getModelApiClient();
+
+        // Otherwise handle misc-specific actions
+        switch (action) {
+            case 'details':
+                // Show misc details
+                this.currentCard.click();
+                break;
+            case 'replace-preview':
+                // Add new action for replacing preview images
+                apiClient.replaceModelPreview(this.currentCard.dataset.filepath);
+                break;
+            case 'delete':
+                showDeleteModal(this.currentCard.dataset.filepath);
+                break;
+            case 'copyname':
+                // Copy misc model name
+                if (this.currentCard.querySelector('.fa-copy')) {
+                    this.currentCard.querySelector('.fa-copy').click();
+                }
+                break;
+            case 'refresh-metadata':
+                // Refresh metadata from CivitAI
+                apiClient.refreshSingleModelMetadata(this.currentCard.dataset.filepath);
+                break;
+            case 'move':
+                moveManager.showMoveModal(this.currentCard.dataset.filepath, this.currentCard.dataset.sub_type);
+                break;
+            case 'move-other':
+                {
+                    const currentType = this.currentCard.dataset.sub_type || 'vae';
+                    const otherType = currentType === 'vae' ? 'upscaler' : 'vae';
+                    moveManager.showMoveModal(this.currentCard.dataset.filepath, otherType);
+                }
+                break;
+            case 'exclude':
+                showExcludeModal(this.currentCard.dataset.filepath);
+                break;
+        }
+    }
+}
+
+// Mix in shared methods
+Object.assign(MiscContextMenu.prototype, ModelContextMenuMixin);