rgthree-comfy/web/common/model_info_service.js

import { rgthreeApi } from "./rgthree_api.js";
import { api } from "../../scripts/api.js";
class ModelInfoService extends EventTarget {
    constructor() {
        super();
        this.loraToInfo = new Map();
        api.addEventListener("rgthree-refreshed-lora-info", this.handleLoraAsyncUpdate.bind(this));
    }
    setFreshLoraData(file, info) {
        this.loraToInfo.set(file, info);
        this.dispatchEvent(new CustomEvent("rgthree-model-service-lora-details", { detail: { lora: info } }));
    }
    async getLora(file, refresh = false, light = false) {
        if (this.loraToInfo.has(file) && !refresh) {
            return this.loraToInfo.get(file);
        }
        return this.fetchLora(file, refresh, light);
    }
    async fetchLora(file, refresh = false, light = false) {
        let info = null;
        if (!refresh) {
            info = await rgthreeApi.getLorasInfo(file, light);
        }
        else {
            info = await rgthreeApi.refreshLorasInfo(file);
        }
        if (!light) {
            this.loraToInfo.set(file, info);
        }
        return info;
    }
    async refreshLora(file) {
        return this.fetchLora(file, true);
    }
    async clearLoraFetchedData(file) {
        await rgthreeApi.clearLorasInfo(file);
        this.loraToInfo.delete(file);
        return null;
    }
    async saveLoraPartial(file, data) {
        let info = await rgthreeApi.saveLoraInfo(file, data);
        this.loraToInfo.set(file, info);
        return info;
    }
    handleLoraAsyncUpdate(event) {
        var _a;
        const info = (_a = event.detail) === null || _a === void 0 ? void 0 : _a.data;
        if (info === null || info === void 0 ? void 0 : info.file) {
            this.setFreshLoraData(info.file, info);
        }
    }
}
export const SERVICE = new ModelInfoService();