feat: add model updates check to global context menu

Add a new "Check Model Updates" option to the global context menu that allows users to manually check for model updates. This includes:

- Adding refreshUpdates endpoint to API configuration
- Implementing checkModelUpdates method with proper loading states
- Adding internationalization support for update messages
- Handling success/error states with appropriate user feedback
- Automatically reloading models after update check completes

The feature provides users with manual control over update checks and improves visibility into model update availability.
This commit is contained in:
Will Miao
2025-10-25 21:32:08 +08:00
parent d77b6d78b7
commit 9ca2b9dd56
14 changed files with 216 additions and 4 deletions

View File

@@ -1,11 +1,15 @@
import { BaseContextMenu } from './BaseContextMenu.js';
import { showToast } from '../../utils/uiHelpers.js';
import { state } from '../../state/index.js';
import { translate } from '../../utils/i18nHelpers.js';
import { getCompleteApiConfig, getCurrentModelType } from '../../api/apiConfig.js';
import { resetAndReload } from '../../api/modelApiFactory.js';
export class GlobalContextMenu extends BaseContextMenu {
constructor() {
super('globalContextMenu');
this._cleanupInProgress = false;
this._updateCheckInProgress = false;
}
showMenu(x, y, origin = null) {
@@ -25,6 +29,11 @@ export class GlobalContextMenu extends BaseContextMenu {
console.error('Failed to trigger example images download:', error);
});
break;
case 'check-model-updates':
this.checkModelUpdates(menuItem).catch((error) => {
console.error('Failed to check model updates:', error);
});
break;
default:
console.warn(`Unhandled global context menu action: ${action}`);
break;
@@ -101,4 +110,74 @@ export class GlobalContextMenu extends BaseContextMenu {
menuItem?.classList.remove('disabled');
}
}
async checkModelUpdates(menuItem) {
if (this._updateCheckInProgress) {
return;
}
const modelType = getCurrentModelType();
const apiConfig = getCompleteApiConfig(modelType);
if (!apiConfig?.endpoints?.refreshUpdates) {
console.warn('Refresh updates endpoint not configured for model type:', modelType);
return;
}
this._updateCheckInProgress = true;
menuItem?.classList.add('disabled');
const displayName = apiConfig.config?.displayName ?? 'Model';
const loadingMessage = translate(
'globalContextMenu.checkModelUpdates.loading',
{ type: displayName },
`Checking for ${displayName} updates...`
);
state.loadingManager?.showSimpleLoading?.(loadingMessage);
try {
const response = await fetch(apiConfig.endpoints.refreshUpdates, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ force: false })
});
let payload = {};
try {
payload = await response.json();
} catch {
payload = {};
}
if (!response.ok || payload.success !== true) {
const errorMessage = payload?.error || response.statusText || 'Unknown error';
throw new Error(errorMessage);
}
const records = Array.isArray(payload.records) ? payload.records : [];
if (records.length > 0) {
showToast('globalContextMenu.checkModelUpdates.success', { count: records.length, type: displayName }, 'success');
} else {
showToast('globalContextMenu.checkModelUpdates.none', { type: displayName }, 'info');
}
await resetAndReload(false);
} catch (error) {
console.error('Error checking model updates:', error);
showToast(
'globalContextMenu.checkModelUpdates.error',
{ message: error?.message ?? 'Unknown error', type: displayName },
'error'
);
} finally {
state.loadingManager?.hide?.();
if (typeof state.loadingManager?.restoreProgressBar === 'function') {
state.loadingManager.restoreProgressBar();
}
menuItem?.classList.remove('disabled');
this._updateCheckInProgress = false;
}
}
}