Will Miao
0ec714c122
Menus are `position: fixed`, so only the window can clip them, but they were placed at the raw cursor position with a naive flip: near the bottom edge the menu opened below the window (nothing to scroll it into view), and a tall menu that flipped up could land above the top edge. The sidebar folder menu and the view-options menu did no viewport math at all, so right-clicking a folder at the bottom of the tree showed no options. Add a shared positionFixedMenu helper that picks the side with room, clamps on both axes, and caps the height (scrolling internally) when the menu is taller than the window. BaseContextMenu uses it and exposes a prepareMenu hook so per-card entries are toggled before the menu is measured; the Recipe/Lora/Checkpoint/Embedding/Other/Workflow menus used to mutate the menu after positioning and so measured the previous content. Submenus get the same treatment via positionSubmenu. Fixes #1167
80 lines
3.0 KiB
JavaScript
80 lines
3.0 KiB
JavaScript
import { BaseContextMenu } from './BaseContextMenu.js';
|
|
import { ModelContextMenuMixin } from './ModelContextMenuMixin.js';
|
|
import { getModelApiClient, resetAndReload } from '../../api/modelApiFactory.js';
|
|
import { moveManager } from '../../managers/MoveManager.js';
|
|
import { showDeleteModal, showExcludeModal } from '../../utils/modalUtils.js';
|
|
import { sendEmbeddingToWorkflow } from '../../utils/uiHelpers.js';
|
|
|
|
export class EmbeddingContextMenu extends BaseContextMenu {
|
|
constructor() {
|
|
super('embeddingContextMenu', '.model-card');
|
|
this.nsfwSelector = document.getElementById('nsfwLevelSelector');
|
|
this.modelType = 'embedding';
|
|
this.resetAndReload = resetAndReload;
|
|
|
|
this.initNSFWSelector();
|
|
}
|
|
|
|
// Implementation needed by the mixin
|
|
async saveModelMetadata(filePath, data) {
|
|
return getModelApiClient().saveModelMetadata(filePath, data);
|
|
}
|
|
|
|
prepareMenu() {
|
|
this.updateExcludeMenuItem();
|
|
}
|
|
|
|
handleMenuAction(action) {
|
|
// First try to handle with common actions
|
|
if (ModelContextMenuMixin.handleCommonMenuActions.call(this, action)) {
|
|
return;
|
|
}
|
|
|
|
const apiClient = getModelApiClient();
|
|
|
|
// Otherwise handle embedding-specific actions
|
|
switch(action) {
|
|
case 'details':
|
|
// Show embedding 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 embedding name
|
|
if (this.currentCard.querySelector('.fa-copy')) {
|
|
this.currentCard.querySelector('.fa-copy').click();
|
|
}
|
|
break;
|
|
case 'sendtoworkflow': {
|
|
const folder = this.currentCard.dataset.folder || '';
|
|
const name = this.currentCard.dataset.file_name || '';
|
|
const embeddingCode = folder ? `embedding:${folder}/${name}` : `embedding:${name}`;
|
|
sendEmbeddingToWorkflow(embeddingCode);
|
|
break;
|
|
}
|
|
case 'refresh-metadata':
|
|
// Refresh metadata from CivitAI
|
|
apiClient.refreshSingleModelMetadata(this.currentCard.dataset.filepath);
|
|
break;
|
|
case 'move':
|
|
moveManager.showMoveModal(this.currentCard.dataset.filepath);
|
|
break;
|
|
case 'exclude':
|
|
showExcludeModal(this.currentCard.dataset.filepath);
|
|
break;
|
|
case 'restore':
|
|
this.restoreExcludedModel(this.currentCard.dataset.filepath);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Mix in shared methods
|
|
Object.assign(EmbeddingContextMenu.prototype, ModelContextMenuMixin);
|