rgthree-comfy/web/common/rgthree_api.js

class RgthreeApi {
    constructor(baseUrl) {
        this.getCheckpointsPromise = null;
        this.getSamplersPromise = null;
        this.getSchedulersPromise = null;
        this.getLorasPromise = null;
        this.getWorkflowsPromise = null;
        this.baseUrl = baseUrl || "./rgthree/api";
    }
    apiURL(route) {
        return `${this.baseUrl}${route}`;
    }
    fetchApi(route, options) {
        return fetch(this.apiURL(route), options);
    }
    async fetchJson(route, options) {
        const r = await this.fetchApi(route, options);
        return await r.json();
    }
    async postJson(route, json) {
        const body = new FormData();
        body.append("json", JSON.stringify(json));
        return await rgthreeApi.fetchJson(route, { method: "POST", body });
    }
    getLoras(force = false) {
        if (!this.getLorasPromise || force) {
            this.getLorasPromise = this.fetchJson("/loras", { cache: "no-store" });
        }
        return this.getLorasPromise;
    }
    async fetchApiJsonOrNull(route, options) {
        const response = await this.fetchJson(route, options);
        if (response.status === 200 && response.data) {
            return response.data || null;
        }
        return null;
    }
    async getLorasInfo(...args) {
        const params = new URLSearchParams();
        const isSingleLora = typeof args[0] == 'string';
        if (isSingleLora) {
            params.set("file", args[0]);
        }
        params.set("light", (isSingleLora ? args[1] : args[0]) === false ? '0' : '1');
        const path = `/loras/info?` + params.toString();
        return await this.fetchApiJsonOrNull(path);
    }
    async refreshLorasInfo(file) {
        const path = `/loras/info/refresh` + (file ? `?file=${encodeURIComponent(file)}` : '');
        const infos = await this.fetchApiJsonOrNull(path);
        return infos;
    }
    async clearLorasInfo(file) {
        const path = `/loras/info/clear` + (file ? `?file=${encodeURIComponent(file)}` : '');
        await this.fetchApiJsonOrNull(path);
        return;
    }
    async saveLoraInfo(lora, data) {
        const body = new FormData();
        body.append("json", JSON.stringify(data));
        return await this.fetchApiJsonOrNull(`/loras/info?file=${encodeURIComponent(lora)}`, { cache: "no-store", method: "POST", body });
    }
}
export const rgthreeApi = new RgthreeApi();