mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-10-05 01:25:33 -03:00
Will Miao
e42d649df0
The owner could not tell from the UI what "Buzz Price Tracking" enabled, what the "Price alert threshold" number meant, or what "Price Alerts" was alerting about. That was not a copy problem: the implementation exposed our mechanism (a page scrape) and our SQL predicates as the user's concepts. Two concrete defects came from the same root: - the alert population included versions the user already owns (neither the event generator nor the query filtered on is_in_library; in the owner's library 28 of 52 gated versions were already downloaded, so most "alerts" were about files already on disk, which cannot become cheaper *for them*); - a threshold-filtered state list lived in a notification surface, so an empty panel had three indistinguishable causes and read as a broken feature. The information model is now the version plus ownership: cost is shown only where a decision exists. Owned -> nothing. Not owned and free -> nothing. Not owned and gated -> the price when it is known, `Paid` without a number when it is not, and early access keeps its countdown because "free on <date>" decides between waiting and paying. The numeric threshold has no place in that model: every decision is categorical (wait / pay / skip), so the setting, the comparison and the whole alert-state machine are gone. - both alert-state columns are removed from the schema rather than left dead; a database created by an unreleased build has them dropped on open (native ALTER TABLE ... DROP COLUMN, guarded), which is a no-op for everyone else - gate events are emitted only for versions the user does not have, and the price-drop event goes with the threshold it belonged to - both alert endpoints, PriceAlertsHandler and the service-registry adapter field it needed are removed: events already reach the UI through the refresh response - the bell tab, panel, CSS, both entry points, the unread watermark and their locales are removed; the setting keeps only the enable flag and the refresh interval and is framed as plumbing - "Price unavailable" is replaced by `Paid`: the gate is certain from the public API, only the number is best-effort, and that is our plumbing, not the user's problem Verified against a copy of the owner's real database: 52 gated versions -> 28 owned (now silent) + 24 the feature is actually about; the drop migration ran and both removed endpoints 404.
400 lines
14 KiB
JavaScript
400 lines
14 KiB
JavaScript
// Create the new hierarchical state structure
|
|
import { getStorageItem, getMapFromStorage } from '../utils/storageHelpers.js';
|
|
import { MODEL_TYPES } from '../api/apiConfig.js';
|
|
import { DEFAULT_PATH_TEMPLATES, DEFAULT_FILENAME_TEMPLATES, DEFAULT_PRIORITY_TAG_CONFIG } from '../utils/constants.js';
|
|
|
|
const DEFAULT_SETTINGS_BASE = Object.freeze({
|
|
civitai_api_key: '',
|
|
civitai_api_key_set: false,
|
|
huggingface_api_key: '',
|
|
huggingface_api_key_set: false,
|
|
civitai_host: 'civitai.com',
|
|
download_backend: 'python',
|
|
aria2c_path: '',
|
|
use_portable_settings: false,
|
|
language: 'en',
|
|
show_only_sfw: false,
|
|
enable_metadata_archive_db: false,
|
|
enable_civarchive_api: true,
|
|
enable_openmodeldb_api: true,
|
|
metadata_provider_order: 'civitai_archive_sqlite',
|
|
proxy_enabled: false,
|
|
proxy_type: 'http',
|
|
proxy_host: '',
|
|
proxy_port: '',
|
|
proxy_username: '',
|
|
proxy_password: '',
|
|
default_lora_root: '',
|
|
default_checkpoint_root: '',
|
|
default_embedding_root: '',
|
|
default_other_roots: {},
|
|
enable_other_models: false,
|
|
enabled_other_sub_types: ['vae', 'upscaler', 'text_encoder'],
|
|
recipes_path: '',
|
|
base_model_path_mappings: {},
|
|
download_path_templates: {},
|
|
download_filename_templates: {},
|
|
example_images_path: '',
|
|
example_images_open_mode: 'system',
|
|
example_images_local_root: '',
|
|
example_images_open_uri_template: '',
|
|
optimize_example_images: true,
|
|
auto_download_example_images: false,
|
|
blur_mature_content: true,
|
|
mature_blur_level: 'R',
|
|
card_blur_amount: 8,
|
|
autoplay_on_hover: false,
|
|
display_density: 'default',
|
|
recipes_layout: 'grid',
|
|
card_info_display: 'always',
|
|
model_name_display: 'model_name',
|
|
lora_syntax_format: 'legacy',
|
|
model_card_footer_action: 'example_images',
|
|
show_version_on_card: true,
|
|
include_trigger_words: false,
|
|
compact_mode: false,
|
|
priority_tags: { ...DEFAULT_PRIORITY_TAG_CONFIG },
|
|
version_grouping: 'same_base',
|
|
hide_early_access_updates: false,
|
|
hide_paid_updates: false,
|
|
price_tracking_enabled: false,
|
|
price_check_ttl_hours: 24,
|
|
auto_organize_exclusions: [],
|
|
metadata_refresh_skip_paths: [],
|
|
skip_previously_downloaded_model_versions: false,
|
|
download_skip_base_models: [],
|
|
unknown_base_model_routing: 'diffusion_model',
|
|
backup_auto_enabled: true,
|
|
backup_retention_count: 5,
|
|
sidecar_storage_mode: 'alongside',
|
|
sidecar_storage_path: '',
|
|
strip_lora_on_copy: false,
|
|
use_new_license_icons: true,
|
|
group_by_model: false,
|
|
sticky_controls: false,
|
|
llm_provider: 'openai',
|
|
llm_api_key: '',
|
|
llm_api_base: '',
|
|
llm_model: '',
|
|
});
|
|
|
|
export function createDefaultSettings() {
|
|
return {
|
|
...DEFAULT_SETTINGS_BASE,
|
|
base_model_path_mappings: {},
|
|
download_path_templates: { ...DEFAULT_PATH_TEMPLATES },
|
|
download_filename_templates: { ...DEFAULT_FILENAME_TEMPLATES },
|
|
priority_tags: { ...DEFAULT_PRIORITY_TAG_CONFIG },
|
|
default_other_roots: {},
|
|
enabled_other_sub_types: ['vae', 'upscaler', 'text_encoder'],
|
|
// Standalone-only fields populated by GET /api/lm/settings; in plugin
|
|
// mode the backend omits folder_paths/folder_path_schema and these
|
|
// defaults apply.
|
|
standalone_mode: false,
|
|
folder_paths: {},
|
|
folder_path_schema: [],
|
|
};
|
|
}
|
|
|
|
// Load preview versions from localStorage for each model type
|
|
const loraPreviewVersions = getMapFromStorage('loras_preview_versions');
|
|
const checkpointPreviewVersions = getMapFromStorage('checkpoints_preview_versions');
|
|
const embeddingPreviewVersions = getMapFromStorage('embeddings_preview_versions');
|
|
const otherPreviewVersions = getMapFromStorage('other_preview_versions');
|
|
|
|
export const state = {
|
|
// Global state
|
|
global: {
|
|
settings: createDefaultSettings(),
|
|
loadingManager: null,
|
|
observer: null,
|
|
},
|
|
|
|
// Page-specific states
|
|
pages: {
|
|
[MODEL_TYPES.LORA]: {
|
|
currentPage: 1,
|
|
isLoading: false,
|
|
hasMore: true,
|
|
sortBy: 'name',
|
|
activeFolder: getStorageItem(`${MODEL_TYPES.LORA}_activeFolder`),
|
|
activeLetterFilter: null,
|
|
previewVersions: loraPreviewVersions,
|
|
searchManager: null,
|
|
searchOptions: {
|
|
filename: true,
|
|
modelname: true,
|
|
tags: false,
|
|
creator: false,
|
|
recursive: getStorageItem(`${MODEL_TYPES.LORA}_recursiveSearch`, true),
|
|
},
|
|
filters: {
|
|
baseModel: [],
|
|
tags: {},
|
|
license: {},
|
|
modelTypes: [],
|
|
search: '',
|
|
tagLogic: 'any',
|
|
},
|
|
bulkMode: false,
|
|
selectedLoras: new Set(),
|
|
loraMetadataCache: new Map(),
|
|
showFavoritesOnly: false,
|
|
showUpdateAvailableOnly: false,
|
|
duplicatesMode: false,
|
|
viewMode: 'active',
|
|
excludedViewState: {
|
|
sortBy: 'name:asc',
|
|
search: '',
|
|
},
|
|
activeViewSnapshot: null,
|
|
},
|
|
|
|
recipes: {
|
|
currentPage: 1,
|
|
isLoading: false,
|
|
hasMore: true,
|
|
sortBy: 'date:desc',
|
|
activeFolder: getStorageItem('recipes_activeFolder'),
|
|
searchManager: null,
|
|
searchOptions: {
|
|
title: true,
|
|
tags: true,
|
|
loraName: true,
|
|
loraModel: true,
|
|
prompt: true,
|
|
recursive: getStorageItem('recipes_recursiveSearch', true),
|
|
},
|
|
filters: {
|
|
baseModel: [],
|
|
tags: {},
|
|
license: {},
|
|
modelTypes: [],
|
|
loraAvailability: [],
|
|
search: ''
|
|
},
|
|
pageSize: 20,
|
|
showFavoritesOnly: false,
|
|
duplicatesMode: false,
|
|
bulkMode: false,
|
|
selectedModels: new Set(),
|
|
},
|
|
|
|
[MODEL_TYPES.CHECKPOINT]: {
|
|
currentPage: 1,
|
|
isLoading: false,
|
|
hasMore: true,
|
|
sortBy: 'name',
|
|
activeFolder: getStorageItem(`${MODEL_TYPES.CHECKPOINT}_activeFolder`),
|
|
previewVersions: checkpointPreviewVersions,
|
|
searchManager: null,
|
|
searchOptions: {
|
|
filename: true,
|
|
modelname: true,
|
|
creator: false,
|
|
recursive: getStorageItem(`${MODEL_TYPES.CHECKPOINT}_recursiveSearch`, true),
|
|
},
|
|
filters: {
|
|
baseModel: [],
|
|
tags: {},
|
|
license: {},
|
|
modelTypes: [],
|
|
search: '',
|
|
tagLogic: 'any',
|
|
},
|
|
modelType: 'checkpoint', // 'checkpoint' or 'diffusion_model'
|
|
bulkMode: false,
|
|
selectedModels: new Set(),
|
|
metadataCache: new Map(),
|
|
showFavoritesOnly: false,
|
|
showUpdateAvailableOnly: false,
|
|
duplicatesMode: false,
|
|
viewMode: 'active',
|
|
excludedViewState: {
|
|
sortBy: 'name:asc',
|
|
search: '',
|
|
},
|
|
activeViewSnapshot: null,
|
|
},
|
|
|
|
[MODEL_TYPES.EMBEDDING]: {
|
|
currentPage: 1,
|
|
isLoading: false,
|
|
hasMore: true,
|
|
sortBy: 'name',
|
|
activeFolder: getStorageItem(`${MODEL_TYPES.EMBEDDING}_activeFolder`),
|
|
activeLetterFilter: null,
|
|
previewVersions: embeddingPreviewVersions,
|
|
searchManager: null,
|
|
searchOptions: {
|
|
filename: true,
|
|
modelname: true,
|
|
tags: false,
|
|
creator: false,
|
|
recursive: getStorageItem(`${MODEL_TYPES.EMBEDDING}_recursiveSearch`, true),
|
|
},
|
|
filters: {
|
|
baseModel: [],
|
|
tags: {},
|
|
license: {},
|
|
modelTypes: [],
|
|
search: '',
|
|
tagLogic: 'any',
|
|
},
|
|
bulkMode: false,
|
|
selectedModels: new Set(),
|
|
metadataCache: new Map(),
|
|
showFavoritesOnly: false,
|
|
showUpdateAvailableOnly: false,
|
|
duplicatesMode: false,
|
|
viewMode: 'active',
|
|
excludedViewState: {
|
|
sortBy: 'name:asc',
|
|
search: '',
|
|
},
|
|
activeViewSnapshot: null,
|
|
},
|
|
|
|
[MODEL_TYPES.OTHER]: {
|
|
currentPage: 1,
|
|
isLoading: false,
|
|
hasMore: true,
|
|
sortBy: 'name',
|
|
activeFolder: getStorageItem(`${MODEL_TYPES.OTHER}_activeFolder`),
|
|
previewVersions: otherPreviewVersions,
|
|
searchManager: null,
|
|
searchOptions: {
|
|
filename: true,
|
|
modelname: true,
|
|
tags: false,
|
|
creator: false,
|
|
recursive: getStorageItem(`${MODEL_TYPES.OTHER}_recursiveSearch`, true),
|
|
},
|
|
filters: {
|
|
baseModel: [],
|
|
tags: {},
|
|
license: {},
|
|
modelTypes: [],
|
|
search: '',
|
|
tagLogic: 'any',
|
|
},
|
|
bulkMode: false,
|
|
selectedModels: new Set(),
|
|
metadataCache: new Map(),
|
|
showFavoritesOnly: false,
|
|
showUpdateAvailableOnly: false,
|
|
duplicatesMode: false,
|
|
viewMode: 'active',
|
|
excludedViewState: {
|
|
sortBy: 'name:asc',
|
|
search: '',
|
|
},
|
|
activeViewSnapshot: null,
|
|
}
|
|
},
|
|
|
|
// Current active page - use MODEL_TYPES constants
|
|
currentPageType: MODEL_TYPES.LORA,
|
|
|
|
// Backward compatibility - proxy properties
|
|
get currentPage() { return this.pages[this.currentPageType].currentPage; },
|
|
set currentPage(value) { this.pages[this.currentPageType].currentPage = value; },
|
|
|
|
get isLoading() { return this.pages[this.currentPageType].isLoading; },
|
|
set isLoading(value) { this.pages[this.currentPageType].isLoading = value; },
|
|
|
|
get hasMore() { return this.pages[this.currentPageType].hasMore; },
|
|
set hasMore(value) { this.pages[this.currentPageType].hasMore = value; },
|
|
|
|
get sortBy() { return this.pages[this.currentPageType].sortBy; },
|
|
set sortBy(value) { this.pages[this.currentPageType].sortBy = value; },
|
|
|
|
get activeFolder() { return this.pages[this.currentPageType].activeFolder; },
|
|
set activeFolder(value) { this.pages[this.currentPageType].activeFolder = value; },
|
|
|
|
get loadingManager() { return this.global.loadingManager; },
|
|
set loadingManager(value) { this.global.loadingManager = value; },
|
|
|
|
get observer() { return this.global.observer; },
|
|
set observer(value) { this.global.observer = value; },
|
|
|
|
get previewVersions() { return this.pages.loras.previewVersions; },
|
|
set previewVersions(value) { this.pages.loras.previewVersions = value; },
|
|
|
|
get searchManager() { return this.pages[this.currentPageType].searchManager; },
|
|
set searchManager(value) { this.pages[this.currentPageType].searchManager = value; },
|
|
|
|
get searchOptions() { return this.pages[this.currentPageType].searchOptions; },
|
|
set searchOptions(value) { this.pages[this.currentPageType].searchOptions = value; },
|
|
|
|
get filters() { return this.pages[this.currentPageType].filters; },
|
|
set filters(value) { this.pages[this.currentPageType].filters = value; },
|
|
|
|
get bulkMode() {
|
|
const currentType = this.currentPageType;
|
|
if (currentType === MODEL_TYPES.LORA) {
|
|
return this.pages.loras.bulkMode;
|
|
} else {
|
|
return this.pages[currentType].bulkMode;
|
|
}
|
|
},
|
|
set bulkMode(value) {
|
|
const currentType = this.currentPageType;
|
|
if (currentType === MODEL_TYPES.LORA) {
|
|
this.pages.loras.bulkMode = value;
|
|
} else {
|
|
this.pages[currentType].bulkMode = value;
|
|
}
|
|
},
|
|
|
|
get selectedLoras() { return this.pages.loras.selectedLoras; },
|
|
set selectedLoras(value) { this.pages.loras.selectedLoras = value; },
|
|
|
|
get selectedModels() {
|
|
const currentType = this.currentPageType;
|
|
if (currentType === MODEL_TYPES.LORA) {
|
|
return this.pages.loras.selectedLoras;
|
|
} else {
|
|
return this.pages[currentType].selectedModels;
|
|
}
|
|
},
|
|
set selectedModels(value) {
|
|
const currentType = this.currentPageType;
|
|
if (currentType === MODEL_TYPES.LORA) {
|
|
this.pages.loras.selectedLoras = value;
|
|
} else {
|
|
this.pages[currentType].selectedModels = value;
|
|
}
|
|
},
|
|
|
|
get loraMetadataCache() { return this.pages.loras.loraMetadataCache; },
|
|
set loraMetadataCache(value) { this.pages.loras.loraMetadataCache = value; },
|
|
|
|
get settings() { return this.global.settings; },
|
|
set settings(value) { this.global.settings = value; }
|
|
};
|
|
|
|
// Get the current page state
|
|
export function getCurrentPageState() {
|
|
return state.pages[state.currentPageType];
|
|
}
|
|
|
|
// Set the current page type
|
|
export function setCurrentPageType(pageType) {
|
|
if (state.pages[pageType]) {
|
|
state.currentPageType = pageType;
|
|
return true;
|
|
}
|
|
console.warn(`Unknown page type: ${pageType}`);
|
|
return false;
|
|
}
|
|
|
|
// Initialize page state when a page loads
|
|
export function initPageState(pageType) {
|
|
if (setCurrentPageType(pageType)) {
|
|
console.log(`Initialized state for page: ${pageType}`);
|
|
return getCurrentPageState();
|
|
}
|
|
return null;
|
|
}
|