Class ModelManager

apps/node/src/modules/model-manager/model-manager.ts:16–204

let updatingMetaFromCivitAI = false;
class ModelManager {
    constructor() {
        channelService.on(CHANNELS.MAIN, CHANNEL_EVENTS.UPDATE_MODEL_META, (ev) => {
            this.updateModelMeta(ev.payload);
        });
    }

    getMetaFilePath = () => {
        const appDir = getAppDataDir();
        return path.resolve(appDir, 'models.meta.json');
    }

    tryUpdateMetaFromCivitAI = async (dispatch: TaskEventDispatcher) => {
        if (updatingMetaFromCivitAI) {
            return;
        }
        updatingMetaFromCivitAI = true;
        const metas = this.getModelMetas();
        const metaPath = this.getMetaFilePath();
        dispatch({
            message: "Updating model meta from CivitAI, you can config your api token key in settings pannel"
        });
        try {
            const modelFiles = await this.getAllInstalledModels();
            for (const modelType in modelFiles) {
                const models = modelFiles[modelType];
                for (const model of models) {
                    const modelKey = this.getModelMetaKey(modelType, model.name)
                    if (!metas[modelKey]) {
                        await fetchFileInfo(modelKey, model.path);
                    } else {
                        const meta = metas[modelKey];
                        if (meta.failed) {
                            const lastCheck = new Date(metas[modelKey].last_check);
                            if (new Date().getTime() - lastCheck.getTime() > 1000 * 60 * 60 * 24 * 3) {
                                await fetchFileInfo(modelKey, model.path);
                            }
                        }
                    }
                }
            }
        } catch (error) {
            console.error('Failed to update model meta from CivitAI', error);
        }

        updatingMetaFromCivitAI = false;
        fs.writeFileSync(metaPath, JSON.stringify(metas, null, 2));

        channelService.emit(CHANNELS.MAIN, {
            type: CHANNEL_EVENTS.MODEL_META_UPDATED,
            payload: {
                message: "Model meta updated from CivitAI successrm "
            }
        });

        async function fetchFileInfo(modelKey: string, file: string) {
            try {
                dispatch({